require_relative './base_controller.rb'

module RailsPerformance
  class RailsPerformanceController < RailsPerformance::BaseController
    include RailsPerformance::ApplicationHelper

    protect_from_forgery except: :recent

    if RailsPerformance.enabled
      def index
        @datasource                = RailsPerformance::DataSource.new(**prepare_query, type: :requests)
        db                         = @datasource.db

        @throughput_report_data    = RailsPerformance::Reports::ThroughputReport.new(db).data
        @response_time_report_data = RailsPerformance::Reports::ResponseTimeReport.new(db).data
      end

      def summary
        @datasource                = RailsPerformance::DataSource.new(**prepare_query, type: :requests)
        db                         = @datasource.db

        @throughput_report_data    = RailsPerformance::Reports::ThroughputReport.new(db).data
        @response_time_report_data = RailsPerformance::Reports::ResponseTimeReport.new(db).data
        @data                      = RailsPerformance::Reports::BreakdownReport.new(db, title: "Requests").data

        respond_to do |format|
          format.js {}
          format.any { render plain: "Doesn't open in new window. Wait until full page load." }
        end
      end

      def trace
        @record = RailsPerformance::Models::RequestRecord.find_by(request_id: params[:id])
        @data   = RailsPerformance::Reports::TraceReport.new(request_id: params[:id]).data
        respond_to do |format|
          format.js {}
          format.any { render plain: "Doesn't open in new window. Wait until full page load." }
        end
      end

      def crashes
        @datasource   = RailsPerformance::DataSource.new(**prepare_query({status_eq: 500}), type: :requests)
        db            = @datasource.db
        @data         = RailsPerformance::Reports::CrashReport.new(db).data
      end

      def requests
        @datasource = RailsPerformance::DataSource.new(**prepare_query, type: :requests)
        db          = @datasource.db
        @data       = RailsPerformance::Reports::RequestsReport.new(db, group: :controller_action_format, sort: :count).data
      end

      def recent
        @datasource = RailsPerformance::DataSource.new(**prepare_query, type: :requests)
        db          = @datasource.db
        @data       = RailsPerformance::Reports::RecentRequestsReport.new(db).data(params[:from_timei])

        # example
        # :controller=>"HomeController",
        # :action=>"index",
        # :format=>"html",
        # :status=>"200",
        # :method=>"GET",
        # :path=>"/",
        # :request_id=>"9c9bff5f792a5b3f77cb07fa325f8ddf",
        # :datetime=>2023-06-24 21:22:46 +0300,
        # :datetimei=>1687630966,
        # :duration=>207.225830078125,
        # :db_runtime=>2.055999994277954,
        # :view_runtime=>67.8370000096038,
        # :exception=>nil,
        # :backtrace=>nil,
        # :http_referer=>nil,
        # "email"=>nil,
        # "user_agent"=>"Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/114.0.0.0 Safari/537.36"}]

        respond_to do |page|
          page.html
          page.js
        end
      end

      def slow
        @datasource = RailsPerformance::DataSource.new(**prepare_query, type: :requests)
        db          = @datasource.db
        @data       = RailsPerformance::Reports::SlowRequestsReport.new(db).data
      end

      def sidekiq
        @datasource                = RailsPerformance::DataSource.new(**prepare_query, type: :sidekiq)
        db                         = @datasource.db
        @throughput_report_data    = RailsPerformance::Reports::ThroughputReport.new(db).data
        @response_time_report_data = RailsPerformance::Reports::ResponseTimeReport.new(db).data
        @recent_report_data        = RailsPerformance::Reports::RecentRequestsReport.new(db).data
      end

      def delayed_job
        @datasource                = RailsPerformance::DataSource.new(**prepare_query, type: :delayed_job)
        db                         = @datasource.db
        @throughput_report_data    = RailsPerformance::Reports::ThroughputReport.new(db).data
        @response_time_report_data = RailsPerformance::Reports::ResponseTimeReport.new(db).data
        @recent_report_data        = RailsPerformance::Reports::RecentRequestsReport.new(db).data
      end

      def custom
        @datasource                = RailsPerformance::DataSource.new(**prepare_query, type: :custom)
        db                         = @datasource.db
        @throughput_report_data    = RailsPerformance::Reports::ThroughputReport.new(db).data
        @response_time_report_data = RailsPerformance::Reports::ResponseTimeReport.new(db).data
        @recent_report_data        = RailsPerformance::Reports::RecentRequestsReport.new(db).data
      end

      def grape
        @datasource                = RailsPerformance::DataSource.new(**prepare_query, type: :grape)
        db                         = @datasource.db
        @throughput_report_data    = RailsPerformance::Reports::ThroughputReport.new(db).data
        @recent_report_data        = RailsPerformance::Reports::RecentRequestsReport.new(db).data
      end

      def rake
        @datasource                = RailsPerformance::DataSource.new(**prepare_query, type: :rake)
        db                         = @datasource.db
        @throughput_report_data    = RailsPerformance::Reports::ThroughputReport.new(db).data
        @recent_report_data        = RailsPerformance::Reports::RecentRequestsReport.new(db).data
      end

      private

      def prepare_query(query = params)
        RailsPerformance::Rails::QueryBuilder.compose_from(query)
      end
    end

  end
end