Expand router with real routing object and 4-way address options

1 files changed, 26 insertions, 0 deletions

diff --git a/lib/action_mailroom/router/route.rb b/lib/action_mailroom/router/route.rb
new file mode 100644
index 0000000000..316c77d711
--- /dev/null
+++ b/lib/action_mailroom/router/route.rb
@@ -0,0 +1,26 @@
+class ActionMailroom::Router::Route
+  class InvalidAddressError < StandardError; end
+
+  attr_reader :address, :mailbox_name
+
+  def initialize(address, to:)
+    @address, @mailbox_name = address, to
+  end
+
+  def match?(inbound_email)
+    case address
+    when String
+      inbound_email.mail.to.include?(address)
+    when Regexp
+      inbound_email.mail.to.detect { |recipient| address.match?(recipient) }
+    when Proc
+      address.call(inbound_email)
+    else
+      address.try(:match?, inbound_email) || raise(InvalidAddressError)
+    end
+  end
+
+  def mailbox_class
+    "#{mailbox_name.to_s.capitalize}Mailbox".constantize
+  end
+end