# https://github.com/mperham/sidekiq/wiki/Monitoring
require 'erb'
require 'yaml'
require 'sinatra/base'
require 'json'

require 'mongo_profiler/web_helpers'

module MongoProfiler
  class Web < Sinatra::Base

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

    helpers WebHelpers


    get '/' do
      @profiles = MongoProfiler.collection.
        group({ key:     %i[method file application_name group_id],
                reduce:  'function(curr, result) { result.total_time += curr.total_time; result.total += 1 }',
                initial: { total: 0, total_time: 0 } })

        # group profilers by group_id and sort by created_at DESC
        @grouped_profiles = @profiles.group_by { |profile| profile['group_id'] }.to_a.reverse

        erb :index
    end

    post '/profiler/enable' do
      MongoProfiler.enable!

      redirect to('/')
    end

    post '/profiler/disable' do
      MongoProfiler.disable!

      redirect to('/')
    end

    get '/profiler/groups/:group_id' do
      @group_id = params[:group_id]
      @profiles = MongoProfiler.collection.find(group_id: @group_id).to_a

      @sample_profile = @profiles.first

      @profiles_count = @profiles.count
      @profiles_total_time = @profiles.reduce(0) { |sum, p| sum + p['total_time'] }

      @grouped_profiles = @profiles.group_by { |profile| profile['method'] }

      erb :group_id
    end

    get '/profiler/:_id' do
      @profile_id = BSON::ObjectId(params[:_id])
      @profile = MongoProfiler.collection.find_one(_id: @profile_id)
      @instrument_payload = JSON.parse(@profile['instrument_payload'])

      @collection_name = @instrument_payload['collection']
      @selector        = @instrument_payload['selector']

      begin
        @selector.each_pair do |key, value|
          if value.is_a? Hash
            if value.first[0] == '$oid'
              @selector[key] = BSON::ObjectId(value.first[1])
            end
          end
        end
        @explain = MongoProfiler.database[@collection_name].find(@selector).explain
      rescue => e
        @explain = { error: "Unable to generate explain: #{e.message}" }
      end

      # http://docs.mongodb.org/manual/core/capped-collections/
      # You can update documents in a collection after inserting them. However, these updates cannot cause the documents to grow. If the update operation causes the document to grow beyond their original size, the update operation will fail.
      # If you plan to update documents in a capped collection, create an index so that these update operations do not require a table scan.
      # MongoRubyProfiler.collection.update({ _id: BSON::ObjectId(params[:_id]) }, '$set' => { explain: explain } )
      erb :show
    end
  end
end