Extract ActionMailbox::PostfixRelayer

2 files changed, 68 insertions, 29 deletions

diff --git a/lib/action_mailbox/postfix_relayer.rb b/lib/action_mailbox/postfix_relayer.rb
new file mode 100644
index 0000000000..de4fdc06d0
--- /dev/null
+++ b/lib/action_mailbox/postfix_relayer.rb
@@ -0,0 +1,60 @@
+# frozen_string_literal: true
+
+require "net/http"
+require "uri"
+require "openssl"
+
+module ActionMailbox
+  class PostfixRelayer
+    class Result < Struct.new(:output)
+      def success?
+        !failure?
+      end
+
+      def failure?
+        output.match?(/\A[45]\.\d\.\d /)
+      end
+    end
+
+    attr_reader :uri, :username, :password, :user_agent
+
+    def initialize(url:, username: "actionmailbox", password:, user_agent: nil)
+      @uri, @username, @password, @user_agent = URI(url), username, password, user_agent || "Postfix"
+    end
+
+    def relay(source)
+      case response = post(source)
+      when Net::HTTPSuccess
+        Result.new "2.0.0 Successfully relayed message to Postfix ingress"
+      when Net::HTTPUnauthorized
+        Result.new "4.7.0 Invalid credentials for Postfix ingress"
+      else
+        Result.new "4.0.0 HTTP #{response.code}"
+      end
+    rescue IOError, SocketError, SystemCallError => error
+      Result.new "4.4.2 Network error relaying to Postfix ingress: #{error.message}"
+    rescue Timeout::Error
+      Result.new "4.4.2 Timed out relaying to Postfix ingress"
+    rescue => error
+      Result.new "4.0.0 Error relaying to Postfix ingress: #{error.message}"
+    end
+
+    private
+      def post(source)
+        client.post uri.path, source,
+          "Content-Type"  => "message/rfc822",
+          "User-Agent"    => user_agent,
+          "Authorization" => "Basic #{Base64.strict_encode64(username + ":" + password)}"
+      end
+
+      def client
+        @client ||= Net::HTTP.new(uri.host, uri.port).tap do |connection|
+          connection.use_ssl      = uri.scheme == "https"
+          connection.verify_mode  = OpenSSL::SSL::VERIFY_PEER
+
+          connection.open_timeout = 1
+          connection.read_timeout = 10
+        end
+      end
+  end
+end
diff --git a/lib/tasks/ingress.rake b/lib/tasks/ingress.rake
index 6620e1fe43..6d1c0797ff 100644
--- a/lib/tasks/ingress.rake
+++ b/lib/tasks/ingress.rake
@@ -2,45 +2,24 @@
 
 namespace :action_mailbox do
   namespace :ingress do
-    desc "Pipe an inbound email from STDIN to the Postfix ingress at the given URL"
+    desc "Pipe an inbound email from STDIN to the Postfix ingress (URL and INGRESS_PASSWORD required)"
     task :postfix do
       require "active_support"
       require "active_support/core_ext/object/blank"
-      require "http"
+      require "action_mailbox/ingresses/postfix/relayer"
 
-      url, password = ENV.values_at("URL", "INGRESS_PASSWORD")
+      url, password, user_agent = ENV.values_at("URL", "INGRESS_PASSWORD", "USER_AGENT")
 
       if url.blank? || password.blank?
-        puts "4.3.5 URL and INGRESS_PASSWORD are required"
+        echo "4.3.5 URL and INGRESS_PASSWORD are required"
         exit 1
       end
 
-      begin
-        response = HTTP.basic_auth(user: "actionmailbox", pass: password)
-          .timeout(connect: 1, write: 10, read: 10)
-          .post(url, body: STDIN.read,
-            headers: { "Content-Type" => "message/rfc822", "User-Agent" => ENV.fetch("USER_AGENT", "Postfix") })
-
-        case
-        when response.status.success?
-          puts "2.0.0 HTTP #{response.status}"
-        when response.status.unauthorized?
-          puts "4.7.0 HTTP #{response.status}"
-          exit 1
-        when response.status.unsupported_media_type?
-          puts "5.6.1 HTTP #{response.status}"
-          exit 1
-        else
-          puts "4.0.0 HTTP #{response.status}"
-          exit 1
+      ActionMailbox::PostfixRelayer.new(url: url, password: password, user_agent: user_agent)
+        .relay(STDIN.read).tap do |result|
+          echo result.output
+          exit result.success? ? 0 : 1
         end
-      rescue HTTP::ConnectionError => error
-        puts "4.4.2 Error connecting to the Postfix ingress: #{error.message}"
-        exit 1
-      rescue HTTP::TimeoutError
-        puts "4.4.7 Timed out piping to the Postfix ingress"
-        exit 1
-      end
     end
   end
 end