internetee-registry/app/controllers/api/v1/registrant/registry_locks_controller.rb

module Api
  module V1
    module Registrant
      class RegistryLocksController < BaseController
        before_action :set_domain

        def create
          if @domain.apply_registry_lock
            render json: @domain
          else
            render json: { errors: [{ base: ['Domain cannot be locked'] }] },
                   status: :unprocessable_entity
          end
        end

        def destroy
          if @domain.remove_registry_lock
            render json: @domain
          else
            render json: { errors: [{ base: ['Domain not locked'] }] },
                   status: :unprocessable_entity
          end
        end

        private

        def set_domain
          domain_pool = associated_domains(current_user)
          @domain = domain_pool.find_by(uuid: params[:domain_uuid])

          return if @domain
          render json: { errors: [{ base: ['Domain not found'] }] },
                 status: :not_found and return
        end
      end
    end
  end
end