internetee-registry/app/controllers/epp/polls_controller.rb

module Epp
  class PollsController < BaseController
    skip_authorization_check # TODO: move authorization under ability

    def poll
      req_poll if params[:parsed_frame].css('poll').first['op'] == 'req'
      ack_poll if params[:parsed_frame].css('poll').first['op'] == 'ack'
    end

    private

    def req_poll
      @notification = current_user.unread_notifications.order('created_at DESC').take

      render_epp_response 'epp/poll/poll_no_messages' and return unless @notification
      if @notification.attached_obj_type && @notification.attached_obj_id
        begin
          @object = Object.const_get(@notification.attached_obj_type).find(@notification.attached_obj_id)
        rescue => problem
          # the data model might be inconsistent; or ...
          # this could happen if the registrar does not dequeue messages, and then the domain was deleted

          # SELECT messages.id, domains.name, messages.body FROM messages LEFT OUTER
          # JOIN domains ON attached_obj_id::INTEGER = domains.id
          # WHERE attached_obj_type = 'Epp::Domain' AND name IS NULL;

          Rails.logger.error 'orphan message, error ignored: ' + problem.to_s
          # now we should dequeue or delete the messages avoid duplicate log alarms
        end
      end

      if @notification.attached_obj_type == 'Keyrelay'
        render_epp_response 'epp/poll/poll_keyrelay'
      else
        render_epp_response 'epp/poll/poll_req'
      end
    end

    def ack_poll
      @notification = current_user.unread_notifications.find_by(id: params[:parsed_frame].css('poll').first['msgID'])

      unless @notification
        epp_errors << {
          code: '2303',
          msg: I18n.t('message_was_not_found'),
          value: { obj: 'msgID', val: params[:parsed_frame].css('poll').first['msgID'] }
        }
        handle_errors and return
      end

      handle_errors(@notification) and return unless @notification.mark_as_read
      render_epp_response 'epp/poll/poll_ack'
    end

    def validate_poll
      requires_attribute 'poll', 'op', values: %(ack req), allow_blank: true
    end

    def resource
      @notification
    end
  end
end