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

encrypt_ci_trigger_token.rb « background_migration « gitlab « lib - gitlab.com/gitlab-org/gitlab-foss.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
blob: 237c655a48ae716d14348cc1a4d1829538b7979d (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
# frozen_string_literal: true

module Gitlab
  module BackgroundMigration
    # Migration to make sure that all the prevously saved tokens have their encrypted values in the db.
    class EncryptCiTriggerToken < Gitlab::BackgroundMigration::BatchedMigrationJob
      feature_category :continuous_integration
      scope_to ->(relation) { relation.where(encrypted_token: nil) }
      operation_name :update
      # Class that is imitating Ci::Trigger
      class CiTrigger < ::Ci::ApplicationRecord
        ALGORITHM = 'aes-256-gcm'

        self.table_name = 'ci_triggers'

        attr_encrypted :encrypted_token_tmp,
          attribute: :encrypted_token,
          mode: :per_attribute_iv,
          algorithm: 'aes-256-gcm',
          key: Settings.attr_encrypted_db_key_base_32,
          encode: false

        before_save :copy_token_to_encrypted_token

        def copy_token_to_encrypted_token
          self.encrypted_token_tmp = token
        end
      end

      def perform
        each_sub_batch do |sub_batch|
          sub_batch.each do |trigger|
            Gitlab::BackgroundMigration::EncryptCiTriggerToken::CiTrigger.find(trigger.id).save!
          end
        end
      end
    end
  end
end