require "rails"
require "dryad"

module Dryad
  module Consul
    class Railtie < Rails::Railtie
      initializer "dryad_consul.set_consul" do
        ActiveSupport.on_load(:dryad_consul) do
          Dryad::Consul.configure_consul(Dryad.configuration)
        end
      end

      initializer "dryad_consul.update_active_record_connection" do
        config.after_initialize do
          db_path = "#{Dryad.configuration.namespace}/#{Dryad.configuration.group}/database.yml"
          db_config = Dryad::Consul::ConfigProvider.load(db_path)
          ActiveRecord::Base.configurations = YAML.load(ERB.new(db_config.payload).result)
          ActiveRecord::Base.establish_connection(Rails.env.to_sym)
        rescue Dryad::Core::ConfigurationNotFound => e
          raise e
        rescue Exception => e
          Rails.logger.warn e.message
        end
      end
    end
  end
  ActiveSupport.run_load_hooks(:dryad_consul, Consul)
end