mailcatcher/lib/mail_catcher/mail.rb

require 'mail'
require 'sqlite3'
require 'eventmachine'

module MailCatcher::Mail
  class << 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])
      result = @@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 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"]
      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 ['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|
        part = 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
  end
end