74 lines
1.6 KiB
Ruby
74 lines
1.6 KiB
Ruby
|
# frozen_string_literal: true
|
||
|
|
||
|
class SetSelfMonitoringProjectAlertingToken < ActiveRecord::Migration[5.2]
|
||
|
DOWNTIME = false
|
||
|
|
||
|
module Migratable
|
||
|
module Alerting
|
||
|
class ProjectAlertingSetting < ApplicationRecord
|
||
|
self.table_name = 'project_alerting_settings'
|
||
|
|
||
|
belongs_to :project
|
||
|
|
||
|
validates :token, presence: true
|
||
|
|
||
|
attr_encrypted :token,
|
||
|
mode: :per_attribute_iv,
|
||
|
key: Settings.attr_encrypted_db_key_base_truncated,
|
||
|
algorithm: 'aes-256-gcm'
|
||
|
|
||
|
before_validation :ensure_token
|
||
|
|
||
|
private
|
||
|
|
||
|
def ensure_token
|
||
|
self.token ||= generate_token
|
||
|
end
|
||
|
|
||
|
def generate_token
|
||
|
SecureRandom.hex
|
||
|
end
|
||
|
end
|
||
|
end
|
||
|
|
||
|
class Project < ApplicationRecord
|
||
|
has_one :alerting_setting, inverse_of: :project, class_name: 'Alerting::ProjectAlertingSetting'
|
||
|
end
|
||
|
|
||
|
class ApplicationSetting < ApplicationRecord
|
||
|
self.table_name = 'application_settings'
|
||
|
|
||
|
belongs_to :instance_administration_project, class_name: 'Project'
|
||
|
|
||
|
def self.current_without_cache
|
||
|
last
|
||
|
end
|
||
|
end
|
||
|
end
|
||
|
|
||
|
def setup_alertmanager_token(project)
|
||
|
return unless License.feature_available?(:prometheus_alerts)
|
||
|
|
||
|
project.create_alerting_setting!
|
||
|
end
|
||
|
|
||
|
def up
|
||
|
Gitlab.ee do
|
||
|
project = Migratable::ApplicationSetting.current_without_cache&.instance_administration_project
|
||
|
|
||
|
if project
|
||
|
setup_alertmanager_token(project)
|
||
|
end
|
||
|
end
|
||
|
end
|
||
|
|
||
|
def down
|
||
|
Gitlab.ee do
|
||
|
Migratable::ApplicationSetting.current_without_cache
|
||
|
&.instance_administration_project
|
||
|
&.alerting_setting
|
||
|
&.destroy!
|
||
|
end
|
||
|
end
|
||
|
end
|