danxexe-mailcatcher 0.6.2

data/lib/mail_catcher/mail.rb

@@ -0,0 +1,161 @@
+require "active_support/json"
+require "eventmachine"
+require "mail"
+require "sqlite3"
+
+module MailCatcher::Mail extend self
+  def db
+    @__db ||= begin
+      SQLite3::Database.new(":memory:", :type_translation => true).tap do |db|
+        db.execute(<<-SQL)
+          CREATE TABLE message (
+            id INTEGER PRIMARY KEY ASC,
+            sender TEXT,
+            recipients TEXT,
+            subject TEXT,
+            source BLOB,
+            size TEXT,
+            type TEXT,
+            created_at DATETIME DEFAULT CURRENT_DATETIME
+          )
+        SQL
+        db.execute(<<-SQL)
+          CREATE TABLE message_part (
+            id INTEGER PRIMARY KEY ASC,
+            message_id INTEGER NOT NULL,
+            cid TEXT,
+            type TEXT,
+            is_attachment INTEGER,
+            filename TEXT,
+            charset TEXT,
+            body BLOB,
+            size INTEGER,
+            created_at DATETIME DEFAULT CURRENT_DATETIME
+          )
+        SQL
+      end
+    end
+  end
+
+  def add_message(message)
+    @add_message_query ||= db.prepare("INSERT INTO message (sender, recipients, subject, source, type, size, created_at) VALUES (?, ?, ?, ?, ?, ?, datetime('now'))")
+
+    mail = Mail.new(message[:source])
+    @add_message_query.execute(message[:sender], message[:recipients].to_json, mail.subject, message[:source], mail.mime_type || "text/plain", message[:source].length)
+    message_id = db.last_insert_row_id
+    parts = mail.all_parts
+    parts = [mail] if parts.empty?
+    parts.each do |part|
+      body = part.body.to_s
+      # Only parts have CIDs, not mail
+      cid = part.cid if part.respond_to? :cid
+      add_message_part(message_id, cid, part.mime_type || "text/plain", part.attachment? ? 1 : 0, part.filename, part.charset, body, body.length)
+    end
+
+    EventMachine.next_tick do
+      message = MailCatcher::Mail.message message_id
+      MailCatcher::Events::MessageAdded.push message
+    end
+  end
+
+  def add_message_part(*args)
+    @add_message_part_query ||= db.prepare "INSERT INTO message_part (message_id, cid, type, is_attachment, filename, charset, body, size, created_at) VALUES (?, ?, ?, ?, ?, ?, ?, ?, datetime('now'))"
+    @add_message_part_query.execute(*args)
+  end
+
+  def latest_created_at
+    @latest_created_at_query ||= db.prepare "SELECT created_at FROM message ORDER BY created_at DESC LIMIT 1"
+    @latest_created_at_query.execute.next
+  end
+
+  def messages
+    @messages_query ||= db.prepare "SELECT id, sender, recipients, subject, size, created_at FROM message ORDER BY created_at, id ASC"
+    @messages_query.execute.map do |row|
+      Hash[row.fields.zip(row)].tap do |message|
+        message["recipients"] &&= ActiveSupport::JSON.decode message["recipients"]
+      end
+    end
+  end
+
+  def message(id)
+    @message_query ||= db.prepare "SELECT * FROM message WHERE id = ? LIMIT 1"
+    row = @message_query.execute(id).next
+    row && Hash[row.fields.zip(row)].tap do |message|
+      message["recipients"] &&= ActiveSupport::JSON.decode message["recipients"]
+      message["source"].force_encoding('UTF-8')
+    end
+  end
+
+  def message_has_html?(id)
+    @message_has_html_query ||= db.prepare "SELECT 1 FROM message_part WHERE message_id = ? AND is_attachment = 0 AND type IN ('application/xhtml+xml', 'text/html') LIMIT 1"
+    (!!@message_has_html_query.execute(id).next) || ["text/html", "application/xhtml+xml"].include?(message(id)["type"])
+  end
+
+  def message_has_plain?(id)
+    @message_has_plain_query ||= db.prepare "SELECT 1 FROM message_part WHERE message_id = ? AND is_attachment = 0 AND type = 'text/plain' LIMIT 1"
+    (!!@message_has_plain_query.execute(id).next) || message(id)["type"] == "text/plain"
+  end
+
+  def message_parts(id)
+    @message_parts_query ||= db.prepare "SELECT cid, type, filename, size FROM message_part WHERE message_id = ? ORDER BY filename ASC"
+    @message_parts_query.execute(id).map do |row|
+      Hash[row.fields.zip(row)]
+    end
+  end
+
+  def message_attachments(id)
+    @message_parts_query ||= db.prepare "SELECT cid, type, filename, size FROM message_part WHERE message_id = ? AND is_attachment = 1 ORDER BY filename ASC"
+    @message_parts_query.execute(id).map do |row|
+      Hash[row.fields.zip(row)]
+    end
+  end
+
+  def message_part(message_id, part_id)
+    @message_part_query ||= db.prepare "SELECT * FROM message_part WHERE message_id = ? AND id = ? LIMIT 1"
+    row = @message_part_query.execute(message_id, part_id).next
+    row && Hash[row.fields.zip(row)]
+  end
+
+  def message_part_type(message_id, part_type)
+    @message_part_type_query ||= db.prepare "SELECT * FROM message_part WHERE message_id = ? AND type = ? AND is_attachment = 0 LIMIT 1"
+    row = @message_part_type_query.execute(message_id, part_type).next
+    row && Hash[row.fields.zip(row)]
+  end
+
+  def message_part_html(message_id)
+    part = message_part_type(message_id, "text/html")
+    part ||= message_part_type(message_id, "application/xhtml+xml")
+    part ||= begin
+      message = message(message_id)
+      message if message.present? and ["text/html", "application/xhtml+xml"].include? message["type"]
+    end
+  end
+
+  def message_part_plain(message_id)
+    message_part_type message_id, "text/plain"
+  end
+
+  def message_part_cid(message_id, cid)
+    @message_part_cid_query ||= db.prepare "SELECT * FROM message_part WHERE message_id = ?"
+    @message_part_cid_query.execute(message_id).map do |row|
+      Hash[row.fields.zip(row)]
+    end.find do |part|
+      part["cid"] == cid
+    end
+  end
+
+  def delete!
+    @delete_messages_query ||= db.prepare "DELETE FROM message"
+    @delete_message_parts_query ||= db.prepare "DELETE FROM message_part"
+
+    @delete_messages_query.execute and
+    @delete_message_parts_query.execute
+  end
+
+  def delete_message!(message_id)
+    @delete_messages_query ||= db.prepare "DELETE FROM message WHERE id = ?"
+    @delete_message_parts_query ||= db.prepare "DELETE FROM message_part WHERE message_id = ?"
+    @delete_messages_query.execute(message_id) and
+    @delete_message_parts_query.execute(message_id)
+  end
+end

data/lib/mail_catcher/smtp.rb

@@ -0,0 +1,61 @@
+require "eventmachine"
+
+require "mail_catcher/mail"
+
+class MailCatcher::Smtp < EventMachine::Protocols::SmtpServer
+  # We override EM's mail from processing to allow multiple mail-from commands
+  # per [RFC 2821](http://tools.ietf.org/html/rfc2821#section-4.1.1.2)
+  def process_mail_from sender
+    if @state.include? :mail_from
+      @state -= [:mail_from, :rcpt, :data]
+      receive_reset
+    end
+
+    super
+  end
+
+  def current_message
+    @current_message ||= {}
+  end
+
+  def receive_reset
+    @current_message = nil
+    true
+  end
+
+  def receive_sender(sender)
+    current_message[:sender] = sender
+    true
+  end
+
+  def receive_recipient(recipient)
+    current_message[:recipients] ||= []
+    current_message[:recipients] << recipient
+    true
+  end
+
+  def receive_data_chunk(lines)
+    current_message[:source] ||= ""
+    lines.each do |line|
+      current_message[:source] << line << "\r\n"
+    end
+    true
+  end
+
+  def receive_message
+    MailCatcher::Mail.add_message current_message
+    puts "==> SMTP: Received message from '#{current_message[:sender]}' (#{current_message[:source].length} bytes)"
+    true
+  rescue
+    puts "*** Error receiving message: #{current_message.inspect}"
+    puts "    Exception: #{$!}"
+    puts "    Backtrace:"
+    $!.backtrace.each do |line|
+      puts "       #{line}"
+    end
+    puts "    Please submit this as an issue at http://github.com/sj26/mailcatcher/issues"
+    false
+  ensure
+    @current_message = nil
+  end
+end

data/lib/mail_catcher/version.rb

@@ -0,0 +1,3 @@
+module MailCatcher
+  VERSION = "0.6.2"
+end

data/lib/mail_catcher/web.rb

@@ -0,0 +1,21 @@
+require "active_support/core_ext/module/delegation"
+require "rack/builder"
+
+require "mail_catcher/web/application"
+
+module MailCatcher
+  module Web extend self
+    def app
+      @@app ||= Rack::Builder.new do
+        if ENV["MAILCATCHER_ENV"] == "development"
+          require "mail_catcher/web/assets"
+          map("/assets") { run Assets }
+        end
+
+        map("/") { run Application }
+      end
+    end
+
+    delegate :call, :to => :app
+  end
+end

data/lib/mail_catcher/web/application.rb

@@ -0,0 +1,188 @@
+require "pathname"
+require "net/http"
+require "uri"
+
+require "sinatra"
+require "skinny"
+
+require "mail_catcher/events"
+require "mail_catcher/mail"
+
+class Sinatra::Request
+  include Skinny::Helpers
+end
+
+module MailCatcher
+  module Web
+    class Application < Sinatra::Base
+      set :development, ENV["MAILCATCHER_ENV"] == "development"
+      set :root, File.expand_path("#{__FILE__}/../../../..")
+
+      if development?
+        require "sprockets-helpers"
+
+        configure do
+          require "mail_catcher/web/assets"
+          Sprockets::Helpers.configure do |config|
+            config.environment = Assets
+            config.prefix      = "/assets"
+            config.digest      = false
+            config.public_path = public_folder
+            config.debug       = true
+          end
+        end
+
+        helpers do
+          include Sprockets::Helpers
+        end
+      else
+        helpers do
+          def javascript_tag(name)
+            %{<script src="/assets/#{name}.js"></script>}
+          end
+
+          def stylesheet_tag(name)
+            %{<link rel="stylesheet" href="/assets/#{name}.css">}
+          end
+        end
+      end
+
+      get "/" do
+        erb :index
+      end
+
+      delete "/" do
+        if MailCatcher.quittable?
+          MailCatcher.quit!
+          status 204
+        else
+          status 403
+        end
+      end
+
+      get "/messages" do
+        if request.websocket?
+          request.websocket!(
+            :on_start => proc do |websocket|
+              subscription = Events::MessageAdded.subscribe { |message| websocket.send_message message.to_json }
+              websocket.on_close do |websocket|
+                Events::MessageAdded.unsubscribe subscription
+              end
+            end)
+        else
+          content_type :json
+          Mail.messages.to_json
+        end
+      end
+
+      delete "/messages" do
+        Mail.delete!
+        status 204
+      end
+
+      get "/messages/:id.json" do
+        id = params[:id].to_i
+        if message = Mail.message(id)
+          content_type :json
+          message.merge({
+            "formats" => [
+              "source",
+              ("html" if Mail.message_has_html? id),
+              ("plain" if Mail.message_has_plain? id)
+            ].compact,
+            "attachments" => Mail.message_attachments(id).map do |attachment|
+              attachment.merge({"href" => "/messages/#{escape(id)}/parts/#{escape(attachment["cid"])}"})
+            end,
+          }).to_json
+        else
+          not_found
+        end
+      end
+
+      get "/messages/:id.html" do
+        id = params[:id].to_i
+        if part = Mail.message_part_html(id)
+          content_type part["type"], :charset => (part["charset"] || "utf8")
+
+          body = part["body"]
+
+          # Rewrite body to link to embedded attachments served by cid
+          body.gsub! /cid:([^'"> ]+)/, "#{id}/parts/\\1"
+
+          content_type :html
+          body
+        else
+          not_found
+        end
+      end
+
+      get "/messages/:id.plain" do
+        id = params[:id].to_i
+        if part = Mail.message_part_plain(id)
+          content_type part["type"], :charset => (part["charset"] || "utf8")
+          part["body"]
+        else
+          not_found
+        end
+      end
+
+      get "/messages/:id.source" do
+        id = params[:id].to_i
+        if message = Mail.message(id)
+          content_type "text/plain"
+          message["source"]
+        else
+          not_found
+        end
+      end
+
+      get "/messages/:id.eml" do
+        id = params[:id].to_i
+        if message = Mail.message(id)
+          content_type "message/rfc822"
+          message["source"]
+        else
+          not_found
+        end
+      end
+
+      get "/messages/:id/parts/:cid" do
+        id = params[:id].to_i
+        if part = Mail.message_part_cid(id, params[:cid])
+          content_type part["type"], :charset => (part["charset"] || "utf8")
+          attachment part["filename"] if part["is_attachment"] == 1
+          body part["body"].to_s
+        else
+          not_found
+        end
+      end
+
+      get "/messages/:id/analysis.?:format?" do
+        id = params[:id].to_i
+        if part = Mail.message_part_html(id)
+          # TODO: Server-side cache? Make the browser cache based on message create time? Hmm.
+          uri = URI.parse("http://api.getfractal.com/api/v2/validate#{"/format/#{params[:format]}" if params[:format].present?}")
+          response = Net::HTTP.post_form(uri, :api_key => "5c463877265251386f516f7428", :html => part["body"])
+          content_type ".#{params[:format]}" if params[:format].present?
+          body response.body
+        else
+          not_found
+        end
+      end
+
+      delete "/messages/:id" do
+        id = params[:id].to_i
+        if message = Mail.message(id)
+          Mail.delete_message!(id)
+          status 204
+        else
+          not_found
+        end
+      end
+
+      not_found do
+        erb :"404"
+      end
+    end
+  end
+end

data/lib/mailcatcher.rb

@@ -0,0 +1,3 @@
+require "mail_catcher"
+
+Mailcatcher = MailCatcher