gitlab-org--gitlab-foss/app/models/terraform/state.rb

57 lines
1.6 KiB
Ruby

# frozen_string_literal: true
module Terraform
class State < ApplicationRecord
include UsageStatistics
include FileStoreMounter
HEX_REGEXP = %r{\A\h+\z}.freeze
UUID_LENGTH = 32
belongs_to :project
belongs_to :locked_by_user, class_name: 'User'
has_many :versions, class_name: 'Terraform::StateVersion', foreign_key: :terraform_state_id
has_one :latest_version, -> { ordered_by_version_desc }, class_name: 'Terraform::StateVersion', foreign_key: :terraform_state_id
scope :versioning_not_enabled, -> { where(versioning_enabled: false) }
validates :project_id, presence: true
validates :uuid, presence: true, uniqueness: true, length: { is: UUID_LENGTH },
format: { with: HEX_REGEXP, message: 'only allows hex characters' }
default_value_for(:uuid, allows_nil: false) { SecureRandom.hex(UUID_LENGTH / 2) }
default_value_for(:versioning_enabled, true)
mount_file_store_uploader StateUploader
def file_store
super || StateUploader.default_store
end
def latest_file
versioning_enabled ? latest_version&.file : file
end
def local?
file_store == ObjectStorage::Store::LOCAL
end
def locked?
self.lock_xid.present?
end
def update_file!(data, version:)
if versioning_enabled?
new_version = versions.build(version: version)
new_version.assign_attributes(created_by_user: locked_by_user, file: data)
new_version.save!
else
self.file = data
save!
end
end
end
end
Terraform::State.prepend_if_ee('EE::Terraform::State')