Contents

# frozen_string_literal: true

class CodeSuggestionsSetup
  class << self
    def configure!
      puts 'Enabling code_suggestions_tokens_api feature flag...'
      Feature.enable(:code_suggestions_tokens_api)

      activate_cloud_license

      # Due to the various async Sidekiq processes involved, we wait to verify
      # that the code suggestions access token has been generated before proceeding
      verify_code_suggestions_access_token

      puts 'Enabling application setting instance_level_code_suggestions_enabled...'
      ApplicationSetting.last.update(instance_level_code_suggestions_enabled: true)
    end

    private

    def activate_cloud_license
      puts 'Activating cloud license...'
      activation_result = ::GitlabSubscriptions::ActivateService.new.execute(ENV.fetch('QA_EE_ACTIVATION_CODE', nil))

      if activation_result[:success]
        puts 'Cloud license activation successful'
      else
        puts 'Cloud license activation failed!'
        puts Array(result[:errors]).join(' ')
        exit 1
      end
    end

    def verify_code_suggestions_access_token
      puts 'Waiting for code suggestions access token to be available...'

      max_attempts = 3
      attempts = 0

      until (tokens = ::Ai::ServiceAccessToken.active.count)&.positive? || attempts == max_attempts
        puts 'Attempting to verify access token exists...'
        attempts += 1
        sleep 30
      end

      return if tokens&.positive?

      puts "Failed to create code suggestions access token after #{max_attempts} attempts"
      exit 1
    end
  end
end

CodeSuggestionsSetup.configure!

Version data entries

3 entries across 3 versions & 1 rubygems

Version	Path
gitlab-qa-14.0.0	support/setup/code_suggestions_setup.rb
gitlab-qa-13.2.1	support/setup/code_suggestions_setup.rb
gitlab-qa-13.2.0	support/setup/code_suggestions_setup.rb