2017-06-02 07:41:30 -04:00
|
|
|
def save(settings, topic)
|
2016-10-02 18:30:44 -04:00
|
|
|
if settings.save
|
2017-06-02 07:41:30 -04:00
|
|
|
puts "Saved #{topic}".color(:green)
|
2016-10-02 18:30:44 -04:00
|
|
|
else
|
2017-06-02 07:41:30 -04:00
|
|
|
puts "Could not save #{topic}".color(:red)
|
2016-10-02 18:30:44 -04:00
|
|
|
puts
|
|
|
|
settings.errors.full_messages.map do |message|
|
|
|
|
puts "--> #{message}".color(:red)
|
|
|
|
end
|
|
|
|
puts
|
2017-06-02 07:41:30 -04:00
|
|
|
exit(1)
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
envs = %w{ GITLAB_PROMETHEUS_METRICS_ENABLED GITLAB_SHARED_RUNNERS_REGISTRATION_TOKEN }
|
|
|
|
|
|
|
|
if envs.any? {|env_name| ENV[env_name].present? }
|
|
|
|
settings = ApplicationSetting.current || ApplicationSetting.create_from_defaults
|
|
|
|
if ENV['GITLAB_SHARED_RUNNERS_REGISTRATION_TOKEN'].present?
|
|
|
|
settings.set_runners_registration_token(ENV['GITLAB_SHARED_RUNNERS_REGISTRATION_TOKEN'])
|
|
|
|
save(settings, 'Runner Registration Token')
|
|
|
|
end
|
|
|
|
|
|
|
|
if ENV['GITLAB_PROMETHEUS_METRICS_ENABLED'].present?
|
|
|
|
value = Gitlab::Utils.to_boolean(ENV['GITLAB_PROMETHEUS_METRICS_ENABLED'])
|
|
|
|
settings.prometheus_metrics_enabled = value
|
2017-06-02 08:21:58 -04:00
|
|
|
save(settings, 'Prometheus metrics enabled flag')
|
2016-10-02 18:30:44 -04:00
|
|
|
end
|
|
|
|
end
|