mirror of
https://github.com/moparisthebest/mailcatcher
synced 2024-11-14 05:05:01 -05:00
117 lines
3.1 KiB
Ruby
117 lines
3.1 KiB
Ruby
require 'sinatra'
|
|
require 'json'
|
|
require 'pathname'
|
|
|
|
require 'skinny'
|
|
|
|
class Sinatra::Request
|
|
include Skinny::Helpers
|
|
end
|
|
|
|
module MailCatcher
|
|
class Web < Sinatra::Base
|
|
set :root, Pathname.new(__FILE__).dirname.parent.parent
|
|
set :haml, :format => :html5
|
|
|
|
get '/' do
|
|
haml :index
|
|
end
|
|
|
|
get '/messages' do
|
|
if request.websocket?
|
|
request.websocket!(
|
|
:protocol => "MailCatcher 0.2 Message Push",
|
|
:on_start => proc do |websocket|
|
|
subscription = MailCatcher::Events::MessageAdded.subscribe { |message| websocket.send_message message.to_json }
|
|
websocket.on_close do |websocket|
|
|
MailCatcher::Events::MessageAdded.unsubscribe subscription
|
|
end
|
|
end)
|
|
else
|
|
MailCatcher::Mail.messages.to_json
|
|
end
|
|
end
|
|
|
|
get '/messages/:id.json' do
|
|
id = params[:id].to_i
|
|
if message = MailCatcher::Mail.message(id)
|
|
message.merge({
|
|
"formats" => [
|
|
"source",
|
|
("html" if MailCatcher::Mail.message_has_html? id),
|
|
("plain" if MailCatcher::Mail.message_has_plain? id),
|
|
].flatten,
|
|
"attachments" => MailCatcher::Mail.message_attachments(id).map do |attachment|
|
|
attachment.merge({"href" => "/messages/#{escape(id)}/#{escape(attachment['cid'])}"})
|
|
end,
|
|
}).to_json
|
|
else
|
|
not_found
|
|
end
|
|
end
|
|
|
|
get '/messages/:id.html' do
|
|
id = params[:id].to_i
|
|
if part = MailCatcher::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}/\\1"
|
|
|
|
# Rewrite body to open links in a new window
|
|
body.gsub! /<a\s+/, '<a target="_blank" '
|
|
|
|
body
|
|
else
|
|
not_found
|
|
end
|
|
end
|
|
|
|
get "/messages/:id.plain" do
|
|
id = params[:id].to_i
|
|
if part = MailCatcher::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 = MailCatcher::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 = MailCatcher::Mail.message(id)
|
|
content_type "message/rfc822"
|
|
message["source"]
|
|
else
|
|
not_found
|
|
end
|
|
end
|
|
|
|
get "/messages/:id/:cid" do
|
|
id = params[:id].to_i
|
|
if part = MailCatcher::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
|
|
|
|
not_found do
|
|
"<html><body><h1>No Dice</h1><p>The message you were looking for does not exist, or doesn't have content of this type.</p></body></html>"
|
|
end
|
|
end
|
|
end |