Basic poll implementation

app/controllers/epp/commands_controller.rb

@@ -9,6 +9,14 @@ class Epp::CommandsController < ApplicationController
 
   private
 
+  def poll
+    @message = current_epp_user.registrar.messages.last
+    if @message.attached_obj_type && @message.attached_obj_id
+      @object = Object.const_get(@message.attached_obj_type).find(@message.attached_obj_id)
+    end
+    render 'epp/poll'
+  end
+
   def create
     send("create_#{OBJECT_TYPES[params_hash['epp']['xmlns:ns2']]}")
   end

app/helpers/epp/domains_helper.rb

@@ -80,7 +80,9 @@ module Epp::DomainsHelper
     @domain = find_domain(secure: false)
 
     handle_errors(@domain) and return unless @domain
-    handle_errors(@domain) and return unless @domain.transfer(domain_transfer_params)
+
+    @domain_transfer = @domain.transfer(domain_transfer_params)
+    handle_errors(@domain) and return unless @domain_transfer
 
     render '/epp/domains/transfer'
   end

app/models/domain_transfer.rb

@@ -13,6 +13,8 @@ class DomainTransfer < ActiveRecord::Base
 
   before_create :set_wait_until
 
+  delegate :name, :valid_to, to: :domain, prefix: true
+
   def set_wait_until
     wait_time = Setting.transfer_wait_time
     return if wait_time == 0

app/models/epp/epp_domain.rb

@@ -285,7 +285,12 @@ class Epp::EppDomain < Domain
 
     pt = pending_transfer
     if pt && params[:action] == 'approve'
-      return approve_pending_transfer(params[:current_user])
+      if approve_pending_transfer(params[:current_user])
+        return pt.reload
+      else
+        return false
+      end
+
     end
 
     if !pt && params[:action] != 'query'
@@ -297,17 +302,24 @@ class Epp::EppDomain < Domain
       return false unless can_be_transferred_to?(params[:current_user].registrar)
     end
 
-    return true if pt
+    return pt if pt
 
     if Setting.transfer_wait_time > 0
-      domain_transfers.create(
+      dt = domain_transfers.create(
         status: DomainTransfer::PENDING,
         transfer_requested_at: Time.zone.now,
         transfer_to: params[:current_user].registrar,
         transfer_from: registrar
       )
+
+      registrar.messages.create(
+        body: I18n.t('transfer_requested'),
+        attached_obj_id: dt.id,
+        attached_obj_type: dt.class.to_s
+      )
+
     else
-      domain_transfers.create(
+      dt = domain_transfers.create(
         status: DomainTransfer::SERVER_APPROVED,
         transfer_requested_at: Time.zone.now,
         transferred_at: Time.zone.now,
@@ -320,6 +332,8 @@ class Epp::EppDomain < Domain
       self.registrar = params[:current_user].registrar
       save(validate: false)
     end
+
+    dt
   end
   # rubocop: enable Metrics/PerceivedComplexity
   # rubocop: enable Metrics/MethodLength

app/models/epp_user.rb

@@ -16,5 +16,9 @@ class EppUser < ActiveRecord::Base
   def to_s
     username
   end
+
+  def queued_messages
+    registrar.messages.queued
+  end
 end
 # rubocop: enable Metrics/ClassLength

app/models/message.rb

@@ -1,3 +1,7 @@
 class Message < ActiveRecord::Base
   belongs_to :registrar
+
+  before_create -> { self.queued = true }
+
+  scope :queued, -> { where(queued: true) }
 end

app/views/epp/domains/partials/_transfer.xml.builder

@@ -0,0 +1,9 @@
+builder.tag!('domain:trnData', 'xmlns:domain' => 'urn:ietf:params:xml:ns:domain-1.0') do
+  builder.tag!('domain:name', dt.domain_name)
+  builder.tag!('domain:trStatus', dt.status)
+  builder.tag!('domain:reID', dt.transfer_to.reg_no)
+  builder.tag!('domain:reDate', dt.transfer_requested_at)
+  builder.tag!('domain:acID', dt.transfer_from.reg_no)
+  builder.tag!('domain:acDate', dt.transferred_at || dt.wait_until)
+  builder.tag!('domain:exDate', dt.domain_valid_to)
+end

app/views/epp/domains/transfer.xml.builder

@@ -5,16 +5,7 @@ xml.epp_head do
     end
 
     xml.resData do
-      xml.tag!('domain:trnData', 'xmlns:domain' => 'urn:ietf:params:xml:ns:domain-1.0') do
-        xml.tag!('domain:name', @domain.name)
-        ldt = @domain.domain_transfers.last
-        xml.tag!('domain:trStatus', ldt.status)
-        xml.tag!('domain:reID', ldt.transfer_to.reg_no)
-        xml.tag!('domain:reDate', ldt.transfer_requested_at)
-        xml.tag!('domain:acID', ldt.transfer_from.reg_no)
-        xml.tag!('domain:acDate', ldt.transferred_at || ldt.wait_until)
-        xml.tag!('domain:exDate', @domain.valid_to)
-      end
+      xml << render('epp/domains/partials/transfer', builder: xml, dt: @domain_transfer)
     end
   end
 

app/views/epp/poll.xml.builder

@@ -0,0 +1,20 @@
+xml.epp_head do
+  xml.response do
+    xml.result('code' => '1301') do
+      xml.msg 'Command completed successfully; ack to dequeue'
+    end
+
+    xml.tag!('msgQ', 'count' => current_epp_user.queued_messages.count, 'id' => @message.id) do
+      xml.qDate @message.created_at
+      xml.msg @message.body
+    end
+
+    xml.resData do
+      if @message.attached_obj_type == 'DomainTransfer'
+        xml << render('epp/domains/partials/transfer', builder: xml, dt: @object)
+      end
+    end if @object
+
+    xml << render('/epp/shared/trID')
+  end
+end