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

module Concerns
  module Domain
    module RegistryLockable
      extend ActiveSupport::Concern

      def apply_registry_lock
        return unless registry_lockable?
        return if locked_by_registrant?

        transaction do
          statuses << DomainStatus::SERVER_UPDATE_PROHIBITED
          statuses << DomainStatus::SERVER_DELETE_PROHIBITED
          statuses << DomainStatus::SERVER_TRANSFER_PROHIBITED
          self.locked_by_registrant_at = Time.zone.now
          alert_registrar_lock_changes!

          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, DomainStatus::FORCE_DELETE]).empty?
      end

      def locked_by_registrant?
        return false unless locked_by_registrant_at

        lock_statuses = [DomainStatus::SERVER_UPDATE_PROHIBITED,
                         DomainStatus::SERVER_DELETE_PROHIBITED,
                         DomainStatus::SERVER_TRANSFER_PROHIBITED]

        (statuses & lock_statuses).count == 3
      end

      def remove_registry_lock
        return unless locked_by_registrant?

        transaction do
          statuses.delete(DomainStatus::SERVER_UPDATE_PROHIBITED)
          statuses.delete(DomainStatus::SERVER_DELETE_PROHIBITED)
          statuses.delete(DomainStatus::SERVER_TRANSFER_PROHIBITED)
          self.locked_by_registrant_at = nil
          alert_registrar_lock_changes!

          save!
        end
      end

      def alert_registrar_lock_changes!
        translation = locked_by_registrant? ? '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
  end
end