lib/howitzer/mailtrap_api/client.rb in howitzer-2.4.0 vs lib/howitzer/mailtrap_api/client.rb in howitzer-2.5.0

@@ -30,9 +30,27 @@
 
       def find_attachments(message)
         JSON.parse(RestClient.get("#{BASE_URL}/messages/#{message['id']}/attachments", 'Api-Token' => @api_token))
       end
 
+      def get_html_body(message)
+        RestClient.get("#{BASE_URL}/messages/#{message['id']}/body.html", 'Api-Token' => @api_token).body
+      rescue => e
+        raise Howitzer::CommunicationError, e.message
+      end
+
+      def get_txt_body(message)
+        RestClient.get("#{BASE_URL}/messages/#{message['id']}/body.txt", 'Api-Token' => @api_token).body
+      rescue => e
+        raise Howitzer::CommunicationError, e.message
+      end
+
+      def get_raw_body(message)
+        RestClient.get("#{BASE_URL}/messages/#{message['id']}/body.raw", 'Api-Token' => @api_token).body
+      rescue => e
+        raise Howitzer::CommunicationError, e.message
+      end
+
       private
 
       def messages(recipient)
         JSON.parse(RestClient.get("#{BASE_URL}/messages?search=#{recipient}", 'Api-Token' => @api_token))
       end