debian-mirror-gitlab/lib/gitlab/github_import/clients/search_repos.rb

# frozen_string_literal: true

module Gitlab
  module GithubImport
    module Clients
      module SearchRepos
        def search_repos_by_name_graphql(name, options = {})
          graphql_request(graphql_search_repos_body(name, options))
        end

        def count_repos_by_relation_type_graphql(options)
          graphql_request(count_by_relation_type_query(options))
        end

        private

        def graphql_request(query)
          with_retry do
            octokit.post(
              '/graphql',
              { query: query }.to_json
            ).to_h
          end
        end

        def graphql_search_repos_body(name, options)
          query = search_repos_query(name, options)
          query = "query: \"#{query}\""
          first = options[:first].present? ? ", first: #{options[:first]}" : ''
          after = options[:after].present? ? ", after: \"#{options[:after]}\"" : ''
          <<-TEXT
          {
              search(type: REPOSITORY, #{query}#{first}#{after}) {
                  nodes {
                      __typename
                      ... on Repository {
                          id: databaseId
                          name
                          full_name: nameWithOwner
                          owner { login }
                      }
                  }
                  pageInfo {
                      startCursor
                      endCursor
                      hasNextPage
                      hasPreviousPage
                  },
                  repositoryCount
              }
          }
          TEXT
        end

        def search_repos_query(string, options = {})
          base = "#{string} in:name is:public,private fork:true"

          case options[:relation_type]
          when 'organization' then organization_repos_query(base, options)
          when 'collaborated' then collaborated_repos_query(base)
          when 'owned' then owned_repos_query(base)
          # TODO: remove after https://gitlab.com/gitlab-org/gitlab/-/issues/385113 get done
          else legacy_all_repos_query(base)
          end
        end

        def organization_repos_query(search_string, options)
          if options[:organization_login].present?
            "#{search_string} org:#{options[:organization_login]}"
          else
            organizations_subquery
          end
        end

        def collaborated_repos_query(search_string)
          "#{search_string} #{collaborations_subquery}"
        end

        def owned_repos_query(search_string)
          "#{search_string} user:#{octokit.user.to_h[:login]}"
        end

        def legacy_all_repos_query(search_string)
          [
            search_string,
            "user:#{octokit.user.to_h[:login]}",
            collaborations_subquery,
            organizations_subquery
          ].join(' ')
        end

        def collaborations_subquery
          each_object(:repos, nil, { affiliation: 'collaborator' })
            .map { |repo| "repo:#{repo[:full_name]}" }
            .join(' ')
        end

        def organizations_subquery
          each_object(:organizations)
            .map { |org| "org:#{org[:login]}" }
            .join(' ')
        end

        def count_by_relation_type_query(options)
          query = search_repos_query(nil, options)
          query = "query: \"#{query}\""
          <<-TEXT
          {
            search(type: REPOSITORY, #{query}) {
              repositoryCount
            }
          }
          TEXT
        end
      end
    end
  end
end