# frozen_string_literal: true

module CanMoveRepositoryStorage
  extend ActiveSupport::Concern

  RepositoryReadOnlyError = Class.new(StandardError)

  # Tries to set repository as read_only, checking for existing Git transfers in
  # progress beforehand. Setting a repository read-only will fail if it is
  # already in that state.
  #
  # @return nil. Failures will raise an exception
  def set_repository_read_only!(skip_git_transfer_check: false)
    with_lock do
      raise RepositoryReadOnlyError, _('Git transfer in progress') if
        !skip_git_transfer_check && git_transfer_in_progress?

      raise RepositoryReadOnlyError, _('Repository already read-only') if
        _safe_read_repository_read_only_column

      raise ActiveRecord::RecordNotSaved, _('Database update failed') unless
        _update_repository_read_only_column(true)

      nil
    end
  end

  # Set repository as writable again. Unlike setting it read-only, this will
  # succeed if the repository is already writable.
  def set_repository_writable!
    with_lock do
      raise ActiveRecord::RecordNotSaved, _('Database update failed') unless
        _update_repository_read_only_column(false)

      nil
    end
  end

  def git_transfer_in_progress?
    reference_counter(type: repository.repo_type).value > 0
  end

  def reference_counter(type:)
    Gitlab::ReferenceCounter.new(type.identifier_for_container(self))
  end

  private

  # Not all resources that can move repositories have the `repository_read_only`
  # in their table, for example groups. We need these methods to override the
  # behavior in those classes in order to access the column.
  def _safe_read_repository_read_only_column
    # This was added originally this way because of
    # https://gitlab.com/gitlab-org/gitlab/-/commit/43f9b98302d3985312c9f8b66018e2835d8293d2
    self.class.where(id: id).pick(:repository_read_only)
  end

  def _update_repository_read_only_column(value)
    update_column(:repository_read_only, value)
  end
end