262 lines
8.8 KiB
Ruby
262 lines
8.8 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module Gitlab
|
|
module DatabaseImporters
|
|
module SelfMonitoring
|
|
module Project
|
|
class CreateService < ::BaseService
|
|
include Stepable
|
|
|
|
STEPS_ALLOWED_TO_FAIL = [
|
|
:validate_application_settings, :validate_project_created, :validate_admins
|
|
].freeze
|
|
|
|
VISIBILITY_LEVEL = Gitlab::VisibilityLevel::INTERNAL
|
|
PROJECT_NAME = 'GitLab Instance Administration'
|
|
|
|
steps :validate_application_settings,
|
|
:validate_project_created,
|
|
:validate_admins,
|
|
:create_group,
|
|
:create_project,
|
|
:save_project_id,
|
|
:add_group_members,
|
|
:add_to_whitelist,
|
|
:add_prometheus_manual_configuration
|
|
|
|
def initialize
|
|
super(nil)
|
|
end
|
|
|
|
def execute!
|
|
result = execute_steps
|
|
|
|
if result[:status] == :success
|
|
result
|
|
elsif STEPS_ALLOWED_TO_FAIL.include?(result[:failed_step])
|
|
success
|
|
else
|
|
raise StandardError, result[:message]
|
|
end
|
|
end
|
|
|
|
private
|
|
|
|
def validate_application_settings
|
|
return success if application_settings
|
|
|
|
log_error('No application_settings found')
|
|
error(_('No application_settings found'))
|
|
end
|
|
|
|
def validate_project_created
|
|
return success unless project_created?
|
|
|
|
log_error('Project already created')
|
|
error(_('Project already created'))
|
|
end
|
|
|
|
def validate_admins
|
|
unless instance_admins.any?
|
|
log_error('No active admin user found')
|
|
return error(_('No active admin user found'))
|
|
end
|
|
|
|
success
|
|
end
|
|
|
|
def create_group
|
|
if project_created?
|
|
log_info(_('Instance administrators group already exists'))
|
|
@group = application_settings.instance_administration_project.owner
|
|
return success(group: @group)
|
|
end
|
|
|
|
@group = ::Groups::CreateService.new(group_owner, create_group_params).execute
|
|
|
|
if @group.persisted?
|
|
success(group: @group)
|
|
else
|
|
error(_('Could not create group'))
|
|
end
|
|
end
|
|
|
|
def create_project
|
|
if project_created?
|
|
log_info('Instance administration project already exists')
|
|
@project = application_settings.instance_administration_project
|
|
return success(project: project)
|
|
end
|
|
|
|
@project = ::Projects::CreateService.new(group_owner, create_project_params).execute
|
|
|
|
if project.persisted?
|
|
success(project: project)
|
|
else
|
|
log_error("Could not create instance administration project. Errors: %{errors}" % { errors: project.errors.full_messages })
|
|
error(_('Could not create project'))
|
|
end
|
|
end
|
|
|
|
def save_project_id
|
|
return success if project_created?
|
|
|
|
result = application_settings.update(instance_administration_project_id: @project.id)
|
|
|
|
if result
|
|
success
|
|
else
|
|
log_error("Could not save instance administration project ID, errors: %{errors}" % { errors: application_settings.errors.full_messages })
|
|
error(_('Could not save project ID'))
|
|
end
|
|
end
|
|
|
|
def add_group_members
|
|
members = @group.add_users(members_to_add, Gitlab::Access::MAINTAINER)
|
|
errors = members.flat_map { |member| member.errors.full_messages }
|
|
|
|
if errors.any?
|
|
log_error('Could not add admins as members to self-monitoring project. Errors: %{errors}' % { errors: errors })
|
|
error(_('Could not add admins as members'))
|
|
else
|
|
success
|
|
end
|
|
end
|
|
|
|
def add_to_whitelist
|
|
return success unless prometheus_enabled?
|
|
return success unless prometheus_listen_address.present?
|
|
|
|
uri = parse_url(internal_prometheus_listen_address_uri)
|
|
return error(_('Prometheus listen_address in config/gitlab.yml is not a valid URI')) unless uri
|
|
|
|
application_settings.add_to_outbound_local_requests_whitelist([uri.normalized_host])
|
|
result = application_settings.save
|
|
|
|
if result
|
|
# Expire the Gitlab::CurrentSettings cache after updating the whitelist.
|
|
# This happens automatically in an after_commit hook, but in migrations,
|
|
# the after_commit hook only runs at the end of the migration.
|
|
Gitlab::CurrentSettings.expire_current_application_settings
|
|
success
|
|
else
|
|
log_error("Could not add prometheus URL to whitelist, errors: %{errors}" % { errors: application_settings.errors.full_messages })
|
|
error(_('Could not add prometheus URL to whitelist'))
|
|
end
|
|
end
|
|
|
|
def add_prometheus_manual_configuration
|
|
return success unless prometheus_enabled?
|
|
return success unless prometheus_listen_address.present?
|
|
|
|
service = project.find_or_initialize_service('prometheus')
|
|
|
|
unless service.update(prometheus_service_attributes)
|
|
log_error('Could not save prometheus manual configuration for self-monitoring project. Errors: %{errors}' % { errors: service.errors.full_messages })
|
|
return error(_('Could not save prometheus manual configuration'))
|
|
end
|
|
|
|
success
|
|
end
|
|
|
|
def application_settings
|
|
@application_settings ||= ApplicationSetting.current_without_cache
|
|
end
|
|
|
|
def project_created?
|
|
application_settings.instance_administration_project.present?
|
|
end
|
|
|
|
def parse_url(uri_string)
|
|
Addressable::URI.parse(uri_string)
|
|
rescue Addressable::URI::InvalidURIError, TypeError
|
|
end
|
|
|
|
def prometheus_enabled?
|
|
Gitlab.config.prometheus.enable if Gitlab.config.prometheus
|
|
rescue Settingslogic::MissingSetting
|
|
log_error('prometheus.enable is not present in config/gitlab.yml')
|
|
|
|
false
|
|
end
|
|
|
|
def prometheus_listen_address
|
|
Gitlab.config.prometheus.listen_address.to_s if Gitlab.config.prometheus
|
|
rescue Settingslogic::MissingSetting
|
|
log_error('Prometheus listen_address is not present in config/gitlab.yml')
|
|
|
|
nil
|
|
end
|
|
|
|
def instance_admins
|
|
@instance_admins ||= User.admins.active
|
|
end
|
|
|
|
def group_owner
|
|
instance_admins.first
|
|
end
|
|
|
|
def members_to_add
|
|
# Exclude admins who are already members of group because
|
|
# `@group.add_users(users)` returns an error if the users parameter contains
|
|
# users who are already members of the group.
|
|
instance_admins - @group.members.collect(&:user)
|
|
end
|
|
|
|
def create_group_params
|
|
{
|
|
name: 'GitLab Instance Administrators',
|
|
path: "gitlab-instance-administrators-#{SecureRandom.hex(4)}",
|
|
visibility_level: VISIBILITY_LEVEL
|
|
}
|
|
end
|
|
|
|
def docs_path
|
|
Rails.application.routes.url_helpers.help_page_path(
|
|
'administration/monitoring/gitlab_instance_administration_project/index'
|
|
)
|
|
end
|
|
|
|
def create_project_params
|
|
{
|
|
initialize_with_readme: true,
|
|
visibility_level: VISIBILITY_LEVEL,
|
|
name: PROJECT_NAME,
|
|
description: "This project is automatically generated and will be used to help monitor this GitLab instance. [More information](#{docs_path})",
|
|
namespace_id: @group.id
|
|
}
|
|
end
|
|
|
|
def internal_prometheus_listen_address_uri
|
|
if prometheus_listen_address.starts_with?('0.0.0.0:')
|
|
# 0.0.0.0:9090
|
|
port = ':' + prometheus_listen_address.split(':').second
|
|
'http://localhost' + port
|
|
|
|
elsif prometheus_listen_address.starts_with?(':')
|
|
# :9090
|
|
'http://localhost' + prometheus_listen_address
|
|
|
|
elsif prometheus_listen_address.starts_with?('http')
|
|
# https://localhost:9090
|
|
prometheus_listen_address
|
|
|
|
else
|
|
# localhost:9090
|
|
'http://' + prometheus_listen_address
|
|
end
|
|
end
|
|
|
|
def prometheus_service_attributes
|
|
{
|
|
api_url: internal_prometheus_listen_address_uri,
|
|
manual_configuration: true,
|
|
active: true
|
|
}
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|