68 lines
2.1 KiB
Ruby
68 lines
2.1 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module Gitlab
|
|
module GithubImport
|
|
module Clients
|
|
class Proxy
|
|
attr_reader :client
|
|
|
|
delegate :each_object, :user, :octokit, to: :client
|
|
|
|
REPOS_COUNT_CACHE_KEY = 'github-importer/provider-repo-count/%{type}/%{user_id}'
|
|
|
|
def initialize(access_token, client_options)
|
|
@client = pick_client(access_token, client_options)
|
|
end
|
|
|
|
def repos(search_text, options)
|
|
return { repos: filtered(client.repos, search_text) } if use_legacy?
|
|
|
|
fetch_repos_via_graphql(search_text, options)
|
|
end
|
|
|
|
def count_repos_by(relation_type, user_id)
|
|
return if use_legacy?
|
|
|
|
key = format(REPOS_COUNT_CACHE_KEY, type: relation_type, user_id: user_id)
|
|
|
|
::Gitlab::Cache::Import::Caching.read_integer(key, timeout: 5.minutes) ||
|
|
fetch_and_cache_repos_count_via_graphql(relation_type, key)
|
|
end
|
|
|
|
private
|
|
|
|
def fetch_repos_via_graphql(search_text, options)
|
|
response = client.search_repos_by_name_graphql(search_text, options)
|
|
{
|
|
repos: response.dig(:data, :search, :nodes),
|
|
page_info: response.dig(:data, :search, :pageInfo),
|
|
count: response.dig(:data, :search, :repositoryCount)
|
|
}
|
|
end
|
|
|
|
def pick_client(access_token, client_options)
|
|
return Gitlab::GithubImport::Client.new(access_token) unless use_legacy?
|
|
|
|
Gitlab::LegacyGithubImport::Client.new(access_token, **client_options)
|
|
end
|
|
|
|
def filtered(collection, search_text)
|
|
return collection if search_text.blank?
|
|
|
|
collection.select { |item| item[:name].to_s.downcase.include?(search_text) }
|
|
end
|
|
|
|
def use_legacy?
|
|
Feature.disabled?(:remove_legacy_github_client)
|
|
end
|
|
|
|
def fetch_and_cache_repos_count_via_graphql(relation_type, key)
|
|
response = client.count_repos_by_relation_type_graphql(relation_type: relation_type)
|
|
count = response.dig(:data, :search, :repositoryCount)
|
|
|
|
::Gitlab::Cache::Import::Caching.write(key, count, timeout: 5.minutes)
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|