Add whitelist to EPP

app/api/repp/api.rb

@@ -9,7 +9,7 @@ module Repp
 
     before do
       unless Rails.env.development?
-        error! 'IP is not whitelisted', 401 unless @current_user.registrar.repp_ip_white?(request.ip)
+        error! I18n.t('ip_is_not_whitelisted'), 401 unless @current_user.registrar.repp_ip_white?(request.ip)
       end
 
       next if Rails.env.test? || Rails.env.development?

app/controllers/epp_controller.rb

@@ -80,7 +80,11 @@ class EppController < ApplicationController
   end
 
   # VALIDATION
+  # rubocop: disable Metrics/PerceivedComplexity
+  # rubocop: disable Metrics/CyclomaticComplexity
   def validate_request
+    handle_errors and return unless ip_white?
+
     validation_method = "validate_#{params[:action]}"
     return unless respond_to?(validation_method, true)
     send(validation_method)
@@ -93,6 +97,22 @@ class EppController < ApplicationController
 
     handle_errors and return if epp_errors.any?
   end
+  # rubocop: enable Metrics/PerceivedComplexity
+  # rubocop: enable Metrics/CyclomaticComplexity
+
+  def ip_white?
+    if current_user
+      unless current_user.registrar.epp_ip_white?(request.ip)
+        epp_errors << {
+          msg: t('ip_is_not_whitelisted'),
+          code: '2201'
+        }
+        return false
+      end
+    end
+
+    true
+  end
 
   # let's follow grape's validations: https://github.com/intridea/grape/#parameter-validation-and-coercion
 

app/controllers/registrar/domains_controller.rb

@@ -24,7 +24,7 @@ class Registrar::DomainsController < Registrar::DeppController # EPP controller
     if response_ok?
       render 'info'
     else
-      flash[:alert] = t(:domain_not_found)
+      flash[:alert] = @data.css('msg').text
       redirect_to registrar_domains_url and return
     end
   end

app/models/depp/user.rb

@@ -32,6 +32,8 @@ module Depp
 
     def request(xml)
       Nokogiri::XML(server.request(xml)).remove_namespaces!
+      rescue EppErrorResponse => e
+        Nokogiri::XML(e.response_xml.to_s).remove_namespaces!
     end
 
     def repp_request(path, params = {})

config/locales/en.yml

@@ -795,3 +795,4 @@ en:
   domain_registrant_change_rejected_body: 'You have rejected domain owner change.'
   registrant_domain_verification_rejected: 'Domain owner change has been rejected successfully.'
   registrant_domain_verification_rejected_failed: 'Something went wrong'
+  ip_is_not_whitelisted: 'IP is not whitelisted'

spec/fabricators/registrar_fabricator.rb

@@ -10,7 +10,7 @@ Fabricator(:registrar) do
   code { sequence(:code) { |i| "REGISTRAR#{i}" } }
   reference_no { sequence(:reference_no) { |i| "RF#{i}" } }
   accounts(count: 1)
-  white_ips { [Fabricate(:white_ip_repp, ipv4: '127.0.0.1')] }
+  white_ips { [Fabricate(:white_ip_repp, ipv4: '127.0.0.1'), Fabricate(:white_ip, ipv4: '127.0.0.1')] }
 end
 
 Fabricator(:registrar_with_no_account_activities, from: :registrar) do