require "sapience"
require "sapience/extensions/action_cable/tagged_logger_proxy"
require "sapience/extensions/action_controller/live"
require "sapience/extensions/action_controller/log_subscriber"
require "sapience/extensions/action_dispatch/debug_exceptions"
require "sapience/extensions/action_view/streaming_template_renderer"
require "sapience/extensions/active_job/logging"
require "sapience/extensions/active_model_serializers/logging"
require "sapience/extensions/active_record/log_subscriber"
require "sapience/extensions/rails/rack/logger"
require "sapience/extensions/rails/rack/logger_info_as_debug"
require "sapience/extensions/action_view/log_subscriber"

module Sapience
  class Rails < ::Rails::Engine

    # Replace Rails logger initializer
    ::Rails::Application::Bootstrap.initializers.delete_if { |i| i.name == :initialize_logger }

    initializer :initialize_logger, group: :all do
      Sapience.configure
      ::Rails.logger = Sapience[::Rails]
      [:active_record, :action_controller, :action_mailer, :action_view].each do |name|
        ActiveSupport.on_load(name) { include Sapience::Loggable }
      end
      ActiveSupport.on_load(:action_cable) { self.logger = Sapience["ActionCable"] }
    end

    # Before any initializers run, but after the gems have been loaded
    config.before_initialize do
      # Replace the Mongoid Logger
      Mongoid.logger              = Sapience[Mongoid] if defined?(Mongoid)
      Moped.logger                = Sapience[Moped] if defined?(Moped)

      # Replace the Resque Logger
      Resque.logger               = Sapience[Resque] if defined?(Resque) && Resque.respond_to?(:logger)

      # Replace the Sidekiq logger
      Sidekiq::Logging.logger     = Sapience[Sidekiq] if defined?(Sidekiq)

      # Replace the Sidetiq logger
      Sidetiq.logger              = Sapience[Sidetiq] if defined?(Sidetiq)

      # Replace the Raven logger
      # Raven::Configuration.logger = Sapience[Raven::Configuration] if defined?(Raven::Configuration)
      Raven.send(:include) { Sapience::Loggable }

      # Replace the Sneakers logger
      Sneakers.configure(log: Sapience[Sneakers]) if defined?(Sneakers)

      # Replace the Bugsnag logger
      Bugsnag.configure { |config| config.logger = Sapience[Bugsnag] } if defined?(Bugsnag)

      # Set the logger for concurrent-ruby
      Concurrent.global_logger = Sapience[Concurrent] if defined?(Concurrent)
    end

    # Before any initializers run, but after the gems have been loaded
    config.after_initialize do
      # Replace the Bugsnag logger
      Bugsnag.configure { |config| config.logger = Sapience[Bugsnag] } if defined?(Bugsnag)
      Sapience::Extensions::ActionController::LogSubscriber.attach_to :action_controller
      # Sapience::Extensions::ActiveSupport::MailerLogSubscriber.attach_to :action_mailer
      Sapience::Extensions::ActiveRecord::LogSubscriber.attach_to :active_record
      Sapience::Extensions::ActionView::LogSubscriber.attach_to :action_view
      # Sapience::Extensions::ActiveJob::LogSubscriber.attach_to :active_job
    end
  end
end