gitlab-org--gitlab-foss/app/services/google_cloud/service_accounts_service.rb

# frozen_string_literal: true

module GoogleCloud
  ##
  # GCP keys used to store Google Cloud Service Accounts
  GCP_KEYS = %w[GCP_PROJECT_ID GCP_SERVICE_ACCOUNT GCP_SERVICE_ACCOUNT_KEY].freeze

  ##
  # This service deals with GCP Service Accounts in GitLab

  class ServiceAccountsService < ::BaseService
    ##
    # Find GCP Service Accounts in a GitLab project
    #
    # This method looks up GitLab project's CI vars
    # and returns Google Cloud Service Accounts combinations
    # aligning GitLab project and environment to GCP projects

    def find_for_project
      group_vars_by_environment.map do |environment_scope, value|
        {
          environment: environment_scope,
          gcp_project: value['GCP_PROJECT_ID'],
          service_account_exists: value['GCP_SERVICE_ACCOUNT'].present?,
          service_account_key_exists: value['GCP_SERVICE_ACCOUNT_KEY'].present?
        }
      end
    end

    def add_for_project(environment, gcp_project_id, service_account, service_account_key)
      project_var_create_or_replace(
        environment,
        'GCP_PROJECT_ID',
        gcp_project_id
      )
      project_var_create_or_replace(
        environment,
        'GCP_SERVICE_ACCOUNT',
        service_account
      )
      project_var_create_or_replace(
        environment,
        'GCP_SERVICE_ACCOUNT_KEY',
        service_account_key
      )
    end

    private

    def group_vars_by_environment
      filtered_vars = @project.variables.filter { |variable| GCP_KEYS.include? variable.key }
      filtered_vars.each_with_object({}) do |variable, grouped|
        grouped[variable.environment_scope] ||= {}
        grouped[variable.environment_scope][variable.key] = variable.value
      end
    end

    def project_var_create_or_replace(environment_scope, key, value)
      params = { key: key, filter: { environment_scope: environment_scope } }
      existing_variable = ::Ci::VariablesFinder.new(@project, params).execute.first
      existing_variable.destroy if existing_variable
      @project.variables.create!(key: key, value: value, environment_scope: environment_scope, protected: true)
    end
  end
end