module Mongobile
  class App < Sinatra::Base
    include Mongobile::MongoHelper

    helpers do
      include Rack::Utils
      alias_method :h, :escape_html
    end

    set :public, File.expand_path("../../../public", __FILE__)
    set :views, File.expand_path("../../../lib/mongobile/views", __FILE__)

    before do
    end

    get "/" do
      @databases = connection.database_names
      haml :index
    end

    get "/status" do
      haml :status
    end

    get "/ops.json" do
      content_type "application/json"
      status 200
      ops_per_second.to_json
    end

    get "/database/:id" do
      @database = db(params[:id])
      @collections = @database.collections
      haml :"database/show"
    end

    get "/database/:id/:collection" do
      @database = db(params[:id])
      @collection = @database.collection(params[:collection])
      @documents = document_list(@collection)
      haml :"database/collection"
    end

    put "/database/:id/:collection" do
      db.create_collection(params[:collection])
      status 201
    end

    post "/database/:id/:collection/filter" do
      @database = db(params[:id])
      @collection = @database.collection(params[:collection])

      query = nil
      begin
        query = JSON.parse(params[:q])
        params[:q] = query.to_json
      rescue => e
        Timeout.timeout(2) do
          begin
            Thread.start do
              $SAFE=4
              query = Object.module_eval(params[:q])
            end.join
          rescue Exception => e
            puts e.message
            query = {}
          ensure
            params[:q] = query.inspect
          end
        end
      ensure
        query ||= {}
      end

      puts "Processing query: #{query.inspect}"

      @documents = document_list(@collection, query)

      haml :"database/collection"
    end

    post "/database/:id/:collection/delete" do
      @database = db(params[:id])
      @database.drop_collection(params[:collection])

      redirect "/database/#{params[:id]}"
    end

    post "/database/:id/delete" do
      connection.drop_database(params[:id])
      redirect "/"
    end

    post "/database/:id/repair" do
      @database = db(params[:id])
      Thread.start do
        @database.command({:repairDatabase => 1})
      end
      redirect "/database/#{params[:id]}"
    end

    def humanize(v, quote = true)
      if v.nil? && quote
        "null"
      elsif v.kind_of?(Hash)
        JSON.pretty_generate(v)
      elsif v.kind_of?(Array)
        JSON.pretty_generate(v)
      elsif v.kind_of?(Time)
        v.strftime("%d %B %Y %H:%M:%S").inspect
      elsif quote
        v.inspect
      else
        v
      end
    end

    private
    def error_not_found
      status 404
    end
  end
end