debian-mirror-gitlab/app/services/google_cloud/service_accounts_service.rb

78 lines
2.4 KiB
Ruby
Raw Normal View History

2021-12-11 22:18:48 +05:30
# 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
2022-05-07 20:08:51 +05:30
# aligning GitLab project and ref to GCP projects
2021-12-11 22:18:48 +05:30
def find_for_project
2022-05-07 20:08:51 +05:30
group_vars_by_ref.map do |environment_scope, value|
2021-12-11 22:18:48 +05:30
{
2022-05-07 20:08:51 +05:30
ref: environment_scope,
2021-12-11 22:18:48 +05:30
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
2022-05-07 20:08:51 +05:30
def add_for_project(ref, gcp_project_id, service_account, service_account_key, is_protected)
2022-01-26 12:08:38 +05:30
project_var_create_or_replace(
2022-05-07 20:08:51 +05:30
ref,
2022-01-26 12:08:38 +05:30
'GCP_PROJECT_ID',
2022-03-02 08:16:31 +05:30
gcp_project_id,
is_protected
2022-01-26 12:08:38 +05:30
)
project_var_create_or_replace(
2022-05-07 20:08:51 +05:30
ref,
2022-01-26 12:08:38 +05:30
'GCP_SERVICE_ACCOUNT',
2022-03-02 08:16:31 +05:30
service_account,
is_protected
2022-01-26 12:08:38 +05:30
)
project_var_create_or_replace(
2022-05-07 20:08:51 +05:30
ref,
2022-01-26 12:08:38 +05:30
'GCP_SERVICE_ACCOUNT_KEY',
2022-03-02 08:16:31 +05:30
service_account_key,
is_protected
2022-01-26 12:08:38 +05:30
)
end
2021-12-11 22:18:48 +05:30
private
2022-05-07 20:08:51 +05:30
def group_vars_by_ref
2022-03-02 08:16:31 +05:30
filtered_vars = project.variables.filter { |variable| GCP_KEYS.include? variable.key }
2021-12-11 22:18:48 +05:30
filtered_vars.each_with_object({}) do |variable, grouped|
grouped[variable.environment_scope] ||= {}
grouped[variable.environment_scope][variable.key] = variable.value
end
end
2022-01-26 12:08:38 +05:30
2022-03-02 08:16:31 +05:30
def project_var_create_or_replace(environment_scope, key, value, is_protected)
2022-05-07 20:08:51 +05:30
change_params = { variable_params: { key: key, value: value, environment_scope: environment_scope, protected: is_protected } }
filter_params = { key: key, filter: { environment_scope: environment_scope } }
existing_variable = ::Ci::VariablesFinder.new(project, filter_params).execute.first
if existing_variable
change_params[:action] = :update
change_params[:variable] = existing_variable
else
change_params[:action] = :create
end
::Ci::ChangeVariableService.new(container: project, current_user: current_user, params: change_params).execute
2022-01-26 12:08:38 +05:30
end
2021-12-11 22:18:48 +05:30
end
end