require 'slavery/version'
require 'slavery/railtie'
require 'active_record'
require 'active_support/core_ext/module/attribute_accessors'

module Slavery
  extend ActiveSupport::Concern

  included do
    class << self
      alias_method_chain :connection, :slavery
    end
  end

  class Error < StandardError; end

  mattr_accessor :disabled

  module ModuleFunctions
    def on_slave(&block)
      ActiveRecord::Base.on_slave(&block)
    end

    def on_master(&block)
      ActiveRecord::Base.on_master(&block)
    end

    def env
      self.env = defined?(Rails) ? Rails.env.to_s : 'development' unless @env
      @env
    end

    def env=(string)
      @env = ActiveSupport::StringInquirer.new(string)
    end
  end
  extend ModuleFunctions

  module ClassMethods
    def on_slave(&block)
      with_slavery true, &block
    end

    def on_master(&block)
      with_slavery false, &block
    end

    def with_slavery(new_value)
      old_value = Thread.current[:on_slave] # Save for recursive nested calls
      Thread.current[:on_slave] = new_value
      yield
    ensure
      Thread.current[:on_slave] = old_value
    end

    def connection_with_slavery
      if Thread.current[:on_slave] and slaveryable?
        slave_connection
      else
        master_connection
      end
    end

    def slaveryable?
      base_transaction_depth = defined?(ActiveSupport::TestCase) &&
        ActiveSupport::TestCase.respond_to?(:use_transactional_fixtures) &&
        ActiveSupport::TestCase.try(:use_transactional_fixtures) ? 1 : 0
      inside_transaction = master_connection.open_transactions > base_transaction_depth
      raise Error.new('on_slave cannot be used inside transaction block!') if inside_transaction

      !Slavery.disabled
    end

    def master_connection
      connection_without_slavery
    end

    def slave_connection
      slave_connection_holder.connection_without_slavery
    end

    # Create an anonymous AR class to hold slave connection
    def slave_connection_holder
      @slave_connection_holder ||= Class.new(ActiveRecord::Base) {
        self.abstract_class = true

        spec = ["#{Slavery.env}_slave", Slavery.env].find do |spec|
          ActiveRecord::Base.configurations[spec]
        end or raise Error.new("#{Slavery.env}_slave or #{Slavery.env} must exist!")

        establish_connection spec
      }
    end
  end
end