mailcatcher/lib/mail_catcher/web.rb

141 lines
3.7 KiB
Ruby
Raw Normal View History

require 'sinatra'
require 'pathname'
2011-10-07 10:16:17 -04:00
require 'net/http'
require 'uri'
require 'skinny'
class Sinatra::Request
include Skinny::Helpers
end
2011-10-09 03:45:23 -04:00
class MailCatcher::Web < Sinatra::Base
set :root, Pathname.new(__FILE__).dirname.parent.parent
set :haml, :format => :html5
2011-06-09 23:21:39 -04:00
2011-10-09 03:45:23 -04:00
get '/' do
haml :index
end
2011-06-09 23:21:39 -04:00
2011-10-09 03:45:23 -04:00
delete '/' do
MailCatcher.quit!
status 204
end
2011-06-09 23:21:39 -04:00
2011-10-09 03:45:23 -04:00
get '/messages' do
if request.websocket?
puts "Gots a 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
2011-10-09 03:45:23 -04:00
end
2011-06-09 23:21:39 -04:00
2011-10-09 03:45:23 -04:00
delete '/messages' do
MailCatcher::Mail.delete!
status 204
end
2011-06-09 23:21:39 -04:00
2011-10-09 03:45:23 -04:00
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),
].compact,
"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
2011-10-09 03:45:23 -04:00
end
2011-06-09 23:21:39 -04:00
2011-10-09 03:45:23 -04:00
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")
2011-06-09 23:21:39 -04:00
2011-10-09 03:45:23 -04:00
body = part["body"]
2011-06-09 23:21:39 -04:00
2011-10-09 03:45:23 -04:00
# Rewrite body to link to embedded attachments served by cid
body.gsub! /cid:([^'"> ]+)/, "#{id}/parts/\\1"
2011-06-09 23:21:39 -04:00
2011-10-09 03:45:23 -04:00
# Rewrite body to open links in a new window
body.gsub! /<a\s+/, '<a target="_blank" '
2011-06-09 23:21:39 -04:00
2011-10-09 03:45:23 -04:00
body
else
not_found
end
2011-10-09 03:45:23 -04:00
end
2011-06-09 23:21:39 -04:00
2011-10-09 03:45:23 -04:00
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
2011-10-09 03:45:23 -04:00
end
2011-06-09 23:21:39 -04:00
2011-10-09 03:45:23 -04:00
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
2011-10-09 03:45:23 -04:00
end
2011-06-09 23:21:39 -04:00
2011-10-09 03:45:23 -04:00
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
2011-05-27 09:40:16 -04:00
end
2011-10-09 03:45:23 -04:00
end
2011-06-09 23:21:39 -04:00
2011-10-09 03:45:23 -04:00
get "/messages/:id/parts/: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
2011-10-09 03:45:23 -04:00
end
2011-06-09 23:21:39 -04:00
2011-10-09 03:45:23 -04:00
get "/messages/:id/analysis.?:format?" do
id = params[:id].to_i
if part = MailCatcher::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"])
2011-10-09 03:45:23 -04:00
content_type ".#{params[:format]}" if params[:format].present?
body response.body
else
not_found
2011-10-07 10:16:17 -04:00
end
2011-10-09 03:45:23 -04:00
end
2011-10-07 10:16:17 -04:00
2011-10-09 03:45:23 -04:00
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
2011-06-09 23:21:39 -04:00
end