Extract Mail-bound methods into mail_ext for future upstream work

5 files changed, 17 insertions, 6 deletions

diff --git a/lib/action_mailbox.rb b/lib/action_mailbox.rb
index fbc8122d9d..417a1a7431 100644
--- a/lib/action_mailbox.rb
+++ b/lib/action_mailbox.rb
@@ -1,4 +1,5 @@
 require "action_mailbox/engine"
+require "action_mailbox/mail_ext"
 
 module ActionMailbox
   extend ActiveSupport::Autoload
diff --git a/lib/action_mailbox/mail_ext.rb b/lib/action_mailbox/mail_ext.rb
new file mode 100644
index 0000000000..abdaae4f9e
--- /dev/null
+++ b/lib/action_mailbox/mail_ext.rb
@@ -0,0 +1,4 @@
+require "mail"
+
+# The hope is to upstream most of these basic additions to the Mail gem's Mail object. But until then, here they lay!
+Dir["#{File.expand_path(File.dirname(__FILE__))}/mail_ext/*"].each { |path| require "action_mailbox/mail_ext/#{File.basename(path)}" }
diff --git a/lib/action_mailbox/mail_ext/from_source.rb b/lib/action_mailbox/mail_ext/from_source.rb
new file mode 100644
index 0000000000..ddc3cbe385
--- /dev/null
+++ b/lib/action_mailbox/mail_ext/from_source.rb
@@ -0,0 +1,5 @@
+module Mail
+  def self.from_source(source)
+    Mail.new Mail::Utilities.binary_unsafe_to_crlf(source.to_s)
+  end
+end
diff --git a/lib/action_mailbox/mail_ext/recipients.rb b/lib/action_mailbox/mail_ext/recipients.rb
new file mode 100644
index 0000000000..58561c7f01
--- /dev/null
+++ b/lib/action_mailbox/mail_ext/recipients.rb
@@ -0,0 +1,5 @@
+class Mail::Message
+  def recipients
+    Array(to) + Array(cc) + Array(bcc)
+  end
+end
diff --git a/lib/action_mailbox/router/route.rb b/lib/action_mailbox/router/route.rb
index 34cc684381..24be8d4804 100644
--- a/lib/action_mailbox/router/route.rb
+++ b/lib/action_mailbox/router/route.rb
@@ -10,9 +10,9 @@ class ActionMailbox::Router::Route
   def match?(inbound_email)
     case address
     when String
-      recipients_from(inbound_email.mail).any? { |recipient| address.casecmp?(recipient) }
+      inbound_email.mail.recipients.any? { |recipient| address.casecmp?(recipient) }
     when Regexp
-      recipients_from(inbound_email.mail).any? { |recipient| address.match?(recipient) }
+      inbound_email.mail.recipients.any? { |recipient| address.match?(recipient) }
     when Proc
       address.call(inbound_email)
     else
@@ -30,8 +30,4 @@ class ActionMailbox::Router::Route
         raise ArgumentError, "Expected a String, Regexp, Proc, or matchable, got #{address.inspect}"
       end
     end
-
-    def recipients_from(mail)
-      Array(mail.to) + Array(mail.cc) + Array(mail.bcc)
-    end
 end