internetee-registry/app/models/concerns/domain/registry_lockable.rb

module Domain::RegistryLockable
  extend ActiveSupport::Concern

  LOCK_STATUSES = if Feature.obj_and_extensions_statuses_enabled?
                    [DomainStatus::SERVER_OBJ_UPDATE_PROHIBITED,
                     DomainStatus::SERVER_DELETE_PROHIBITED,
                     DomainStatus::SERVER_TRANSFER_PROHIBITED].freeze
                  else
                    [DomainStatus::SERVER_UPDATE_PROHIBITED,
                     DomainStatus::SERVER_DELETE_PROHIBITED,
                     DomainStatus::SERVER_TRANSFER_PROHIBITED].freeze
                  end

  def apply_registry_lock
    return unless registry_lockable?
    return if locked_by_registrant?

    transaction do
      self.admin_store_statuses_history = self.statuses
      self.statuses |= LOCK_STATUSES
      self.locked_by_registrant_at = Time.zone.now
      alert_registrar_lock_changes!(lock: true)

      save!
    end
  end

  def registry_lockable?
    (statuses & [DomainStatus::PENDING_DELETE_CONFIRMATION,
                 DomainStatus::PENDING_CREATE, DomainStatus::PENDING_UPDATE,
                 DomainStatus::PENDING_DELETE, DomainStatus::PENDING_RENEW,
                 DomainStatus::PENDING_TRANSFER]).empty?
  end

  def locked_by_registrant?
    return false unless locked_by_registrant_at

    (statuses & LOCK_STATUSES).count == 3
  end

  def remove_registry_lock
    return unless locked_by_registrant?

    transaction do
      LOCK_STATUSES.each do |domain_status|
        statuses.delete([domain_status])
      end
      self.locked_by_registrant_at = nil
      self.statuses = admin_store_statuses_history || []
      alert_registrar_lock_changes!(lock: false)

      save!
    end
  end

  def alert_registrar_lock_changes!(lock: true)
    translation = lock ? 'locked' : 'unlocked'
    registrar.notifications.create!(
      text: I18n.t("notifications.texts.registrar_#{translation}",
                   domain_name: name),
      attached_obj_id: name,
      attached_obj_type: self.class.name
    )
  end
end