Ability to send params to action.

Allows methods of MailView subclasses to access params, as in a typical
Rails controller.

Author: lukeasrodgers

lib/mail_view.rb

@@ -23,6 +23,7 @@ def call(env)
 
   def call(env)
     request = Rack::Request.new(env)
+    @params = request.params || {}
 
     if request.path_info == "" || request.path_info == "/"
       links = self.actions.map do |action|
@@ -110,7 +111,12 @@ def find_preferred_part(mail, formats)
     end
 
     def part_body_url(part)
-      '?part=%s' % Rack::Utils.escape([part.main_type, part.sub_type].compact.join('/'))
+      part_query_string = '?part=%s' % Rack::Utils.escape([part.main_type, part.sub_type].compact.join('/'))
+      if @params.any?
+        part_query_string + '&' + Rack::Utils.build_query(@params)
+      else
+        part_query_string
+      end
     end
 
     def find_part(mail, matching_content_type)
@@ -126,4 +132,9 @@ def find_part(mail, matching_content_type)
         mail
       end
     end
+
+    def params
+      @params
+    end
+
 end

test/test_mail_view.rb

@@ -107,6 +107,34 @@ def nested_multipart_message
       mail.add_part container
       mail
     end
+
+    def message_with_params
+      greeting = params['greeting']
+      email = params['email']
+      Mail.new do
+        to "User <#{email}>"
+        from 'Test Peek <[email protected]>'
+        body "#{greeting}"
+      end
+    end
+
+    def multipart_alternative_with_params
+      greeting = params['greeting']
+      Mail.new do
+        to '[email protected]'
+
+        yield self if block_given?
+
+        text_part do
+          body "Plain greeting #{greeting}"
+        end
+
+        html_part do
+          content_type 'text/html; charset=UTF-8'
+          body "<h1>Html greeting #{greeting}</h1>"
+        end
+      end
+    end
   end
 
   class ISayHelloAndYouSayGoodbyeInterceptor
@@ -131,8 +159,9 @@ def app
     Preview
   end
 
-  def iframe_src_match(content_type)
-    /<iframe[^>]* src="\?part=#{Regexp.escape(Rack::Utils.escape(content_type))}"[^>]*><\/iframe>/
+  def iframe_src_match(content_type, params = '')
+    params = '&' + Rack::Utils.build_query(Rack::Utils.parse_query(params)) if params != ''
+    /<iframe[^>]* src="\?part=#{Regexp.escape(Rack::Utils.escape(content_type)) + params}"[^>]*><\/iframe>/
   end
 
   def unescaped_body
@@ -306,6 +335,31 @@ def test_interceptors
     assert_equal 'Goodbye', last_response.body
   end
 
+  def test_message_with_params
+    get '/message_with_params?greeting=Bonjour&[email protected]'
+    assert last_response.ok?
+    assert_match '[email protected]', unescaped_body, "Email should be in body"
+    assert_match 'Bonjour', unescaped_body, "Greeting should be in body"
+  end
+
+  def test_message_with_params_no_part_url
+    get '/message_with_params?greeting=Bonjour&[email protected]'
+    assert last_response.ok?
+    assert_match iframe_src_match('', '&greeting=Bonjour&[email protected]'), last_response.body, "Iframe src should include params"
+  end
+
+  def test_message_multipart_alternative_plain_with_params
+    get '/multipart_alternative_with_params?part=text%2Fplain&greeting=Bonjour'
+    assert last_response.ok?
+    assert_match 'Plain greeting Bonjour', unescaped_body, "Email should be in body"
+  end
+
+  def test_message_multipart_alternative_html_with_params
+    get '/multipart_alternative_with_params?part=text%2Fhtml&greeting=Bonjour'
+    assert last_response.ok?
+    assert_match 'Html greeting Bonjour', unescaped_body, "Email should be in body"
+  end
+
   unless RUBY_VERSION >= '1.9'
     def test_tmail_html_message
       get '/tmail_html_message'