class Remnant
  module ClassMethods
    def disable!
      @enabled = false
    end

    def enable!
      @enabled = true
    end

    def enabled?
      @enabled
    end

    def color(default = false, heading = false)
      return "\033[0m" if default
      return "\033[0;01;33m" if heading

      @current_color ||= "\033[0;01;36m"
      @next_color ||= "\033[0;01;35m"

      @current_color, @next_color = @next_color, @current_color
      @current_color
    end

    def configure(&block)
      configuration.instance_eval(&block)
    end

    def configuration
      @configuration ||= Remnant::Configuration.new.defaults!
    end

    def handler
      @handler ||= Statsd.new(Remnant.configuration.hostname, Remnant.configuration.port_number)
    end

    #
    # Log everything to Rails logger for output
    #
    def output
      # always log in development mode
      Rails.logger.info "#{color(false, true)}--------------Remnants Discovered--------------#{color(true)}"

      Remnant::Discover.results.map do |remnant_key, ms|
        key = [
          extra_remnant_key,
          remnant_key
        ].compact.join('.')

        Rails.logger.info "#{Remnant.color}#{ms.to_i}ms#{Remnant.color(true)}\t#{key}"
      end
      Rails.logger.info "#{Remnant.color}#{Remnant::GC.time.to_i}ms (#{Remnant::GC.collections} collections)#{Remnant.color(true)}\tGC"

      # filters
      Rails.logger.info ""
      Rails.logger.info("#{color(false, true)}----- Filters (%.2fms) -----#{color(true)}" % Remnant::Filters.total_time)
      Remnant::Filters.filters.map do |filter|
        Rails.logger.info("#{color}%.2fms#{color(true)}\t#{filter[:name]} (#{filter[:type]})" % filter[:ms])
      end

      # template captures
      if Remnant::Template.enabled?
        Rails.logger.info ""
        Rails.logger.info "#{color(false, true)}----- Templates -----#{color(true)}"
        Remnant::Template.trace.root.children.map do |rendering|
          Remnant::Template.trace.log(Rails.logger, rendering)
        end
      end

      # sql captures
      Rails.logger.info ""
      Rails.logger.info("#{color(false, true)}---- Database (%.2fms) -----#{color(true)}" % Remnant::Database.total_time)
      if Remnant::Database.suppress?
        Rails.logger.info "queries suppressed in development mode"
      else
        Remnant::Database.queries.map do |query|
          Rails.logger.info("#{color}%.2fms#{color(true)}\t#{query.sql}" % (query.time * 1000))
        end
      end

      Rails.logger.info "#{color(false, true)}-----------------------------------------------#{color(true)}"
    end

    #
    # Collect data, report statsd timings, run hook
    #
    def collect
      @sample_counter ||= 0

      extra_remnant_key = Remnant::Discover.results.delete(:extra_remnant_key)

      # don't send timings in development or test environments
      unless ::Rails.env.development? || ::Rails.env.test?
        # only log if above sample rate
        if @sample_counter > configuration.sample_rate
          key_prefix = [
                        Remnant.configuration.tag,
                        Remnant.configuration.env,
                        extra_remnant_key
                       ].compact.join('.')

          Remnant::Discover.results.map do |remnant_key, ms|
            Remnant.handler.timing("#{key_prefix}.#{remnant_key}", ms.to_i)
          end

          Remnant.handler.timing("#{key_prefix}.gc", Remnant::GC.time.to_i)
          Remnant.handler.timing("#{key_prefix}.db", Remnant::Database.total_time.to_i)
          Remnant.handler.timing("#{key_prefix}.filters", Remnant::Filters.total_time.to_i)

          @sample_counter = 0
        else
          @sample_counter += 1
        end
      end

      # run hook if given for all environments
      unless Remnant.configuration.custom_hook.nil?
        begin
          Remnant.configuration.custom_hook.call(Remnant::Discover.results)
        rescue => e
          ::Rails.logger.error "Failed to run hook: #{e.message}"
        end
      end

      Remnant::Database.reset
      Remnant::Template.reset
      Remnant::Filters.reset
      Remnant::Discover.results.clear
    end
  end
  extend ClassMethods
end