require 'sinatra/base'
require 'erb'
require 'letter_opener/web/message'
module LetterOpener
  module Web
    class App < Sinatra::Base

      set :root,          File.expand_path(File.dirname(__FILE__))
      set :public_folder, Proc.new { "#{root}/assets" }
      set :views,         Proc.new { "#{root}/views" }

      enable :method_override

      if defined?(Padrino) && Padrino.version < '0.12.0'
        def self.dependencies; []; end
        def self.setup_application!; end
        def self.reload!; end
        set :erb, :engine_class => Padrino::Erubis::SafeBufferTemplate
      end

      def root_path
        "#{env['SCRIPT_NAME']}/"
      end

      get '/' do
        @messages = Message.load_all
        erb :index
      end

      get '/:id.?:format?' do
        @message = Message.find(params[:id])
        halt 404 unless @message
        begin
          @message.render(params[:format]).to_s.
            gsub(/"plain\.html"/, %Q("#{root_path + @message.id}")).
            gsub(/"rich\.html"/,  %Q("#{root_path + @message.id}.html")).
            gsub(/"attachments\//, %Q("#{root_path + @message.id}/attachments/))
        rescue ArgumentError
          halt 404
        end
      end

      get '/:id/attachments/:file' do
        @message = Message.find(params[:id])
        halt 404 unless @message
        halt 404 unless attachment = @message.attachments[params[:file]]
        send_file attachment, :disposition => 'attachment', :type => 'application/octet-stream'
      end

      delete '/' do
        Message.bulk_delete(params[:message_ids].split(',').map(&:strip)) if params[:message_ids]
        redirect root_path
      end

      error 404 do
        'Not Found'
      end

    end
  end
end