class Dependabot::Clients::GitlabWithRetries

def self.for_gitlab_dot_com(credentials:)
  access_token =
    credentials
    .select { |cred| cred["type"] == "git_source" }
    .select { |cred| cred["password"] }
    .find { |cred| cred["host"] == "gitlab.com" }
    &.fetch("password")
  new(
    endpoint: "https://gitlab.com/api/v4",
    private_token: access_token || ""
  )
end

def self.for_source(source:, credentials:)
  access_token =
    credentials
    .select { |cred| cred["type"] == "git_source" }
    .select { |cred| cred["password"] }
    .find { |cred| cred["host"] == source.hostname }
    &.fetch("password")
  new(
    endpoint: source.api_endpoint,
    private_token: access_token || ""
  )
end

def create_commit(repo, branch_name, commit_message, files, **options)
  @client.create_commit(
    repo,
    branch_name,
    commit_message,
    file_actions(files),
    options
  )
end

def fetch_commit(repo, branch)
  T.unsafe(self).branch(repo, branch).commit.id
end

def fetch_default_branch(repo)
  T.unsafe(self).project(repo).default_branch
end

def file_action(file)
  if file.operation == Dependabot::DependencyFile::Operation::DELETE
    "delete"
  elsif file.operation == Dependabot::DependencyFile::Operation::CREATE
    "create"
  else
    "update"
  end
end

def file_actions(files)
  files.map do |file|
    {
      action: file_action(file),
      encoding: file_encoding(file),
      file_path: file.type == "symlink" ? file.symlink_target : file.path,
      content: file.content
    }
  end
end

def file_encoding(file)
  return ContentEncoding::BASE64 if file.content_encoding == Dependabot::DependencyFile::ContentEncoding::BASE64
  ContentEncoding::TEXT
end

def initialize(max_retries: 3, **args)
  @max_retries = T.let(max_retries || 3, Integer)
  @client = T.let(::Gitlab::Client.new(args), ::Gitlab::Client)
end

def method_missing(method_name, *args, &block)
  retry_connection_failures do
    if @client.respond_to?(method_name)
      mutatable_args = args.map(&:dup)
      T.unsafe(@client).public_send(method_name, *mutatable_args, &block)
    else
      super
    end
  end
end

def respond_to_missing?(method_name, include_private = false)
  @client.respond_to?(method_name) || super
end

def retry_connection_failures(&_blk)
  retry_attempt = 0
  begin
    yield
  rescue *RETRYABLE_ERRORS
    retry_attempt += 1
    retry_attempt <= @max_retries ? retry : raise
  end
end