Welcome to mirror list, hosted at ThFree Co, Russian Federation.

grafana_integration.rb « models « app - gitlab.com/gitlab-org/gitlab-foss.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
blob: 5cd5aa1b08538861a8a8e9cff808ffc636fb7d77 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
# frozen_string_literal: true

class GrafanaIntegration < ApplicationRecord
  belongs_to :project

  attr_encrypted :token,
    mode: :per_attribute_iv,
    algorithm: 'aes-256-gcm',
    key: Settings.attr_encrypted_db_key_base_32

  before_validation :check_token_changes

  validates :grafana_url,
            length: { maximum: 1024 },
            addressable_url: { enforce_sanitization: true, ascii_only: true }

  validates :encrypted_token, :project, presence: true

  validates :enabled, inclusion: { in: [true, false] }

  before_validation :reset_token

  scope :enabled, -> { where(enabled: true) }

  def client
    return unless enabled?

    @client ||= ::Grafana::Client.new(api_url: grafana_url.chomp('/'), token: token)
  end

  def masked_token
    mask(encrypted_token)
  end

  def masked_token_was
    mask(encrypted_token_was)
  end

  private

  def reset_token
    if grafana_url_changed? && !encrypted_token_changed?
      self.token = nil
    end
  end

  def token
    decrypt(:token, encrypted_token)
  end

  def check_token_changes
    return unless [encrypted_token_was, masked_token_was].include?(token)

    clear_attribute_changes [:token, :encrypted_token, :encrypted_token_iv]
  end

  def mask(token)
    token&.squish&.gsub(/./, '*')
  end
end