module DbCharmer
  module ActiveRecord
    module ClassAttributes
      @@db_charmer_opts = {}
      def db_charmer_opts=(opts)
        @@db_charmer_opts[self.name] = opts
      end

      def db_charmer_opts
        @@db_charmer_opts[self.name] || {}
      end

      #-----------------------------------------------------------------------------
      @@db_charmer_connection_proxies = {}
      def db_charmer_connection_proxy=(proxy)
        @@db_charmer_connection_proxies[self.name] = proxy
      end

      def db_charmer_connection_proxy
        @@db_charmer_connection_proxies[self.name]
      end

      #-----------------------------------------------------------------------------
      @@db_charmer_default_connections = {}
      def db_charmer_default_connection=(conn)
        @@db_charmer_default_connections[self.name] = conn
      end

      def db_charmer_default_connection
        @@db_charmer_default_connections[self.name]
      end

      #-----------------------------------------------------------------------------
      @@db_charmer_slaves = {}
      def db_charmer_slaves=(slaves)
        @@db_charmer_slaves[self.name] = slaves
      end

      def db_charmer_slaves
        @@db_charmer_slaves[self.name] || []
      end

      def db_charmer_random_slave
        return nil unless db_charmer_slaves.any?
        db_charmer_slaves[rand(db_charmer_slaves.size)]
      end

      #-----------------------------------------------------------------------------
      @@db_charmer_force_slave_reads = {}
      def db_charmer_force_slave_reads=(force)
        @@db_charmer_force_slave_reads[self.name] = force
      end

      def db_charmer_force_slave_reads
        @@db_charmer_force_slave_reads[self.name]
      end

      # Slave reads are used in two cases:
      #  - per-model slave reads are enabled (see db_magic method for more details)
      #  - global slave reads enforcing is enabled (in a controller action)
      def db_charmer_force_slave_reads?
        db_charmer_force_slave_reads || DbCharmer.force_slave_reads?
      end

      #-----------------------------------------------------------------------------
      @@db_charmer_connection_levels = Hash.new(0)
      def db_charmer_connection_level=(level)
        @@db_charmer_connection_levels[self.name] = level
      end

      def db_charmer_connection_level
        @@db_charmer_connection_levels[self.name] || 0
      end

      def db_charmer_top_level_connection?
        db_charmer_connection_level.zero?
      end

      #-----------------------------------------------------------------------------
      @@db_charmer_database_remappings = Hash.new
      def db_charmer_remapped_connection
        return nil if (db_charmer_connection_level || 0) > 0
        name = :master
        proxy = db_charmer_connection_proxy
        name = proxy.db_charmer_connection_name.to_sym if proxy

        remapped = @@db_charmer_database_remappings[name]
        remapped ? DbCharmer::ConnectionFactory.connect(remapped, true) : nil
      end

      def db_charmer_database_remappings
        @@db_charmer_database_remappings
      end

      def db_charmer_database_remappings=(mappings)
        raise "Mappings must be nil or respond to []" if mappings && (! mappings.respond_to?(:[]))
        @@db_charmer_database_remappings = mappings || { }
      end

      #-----------------------------------------------------------------------------
      @@db_charmer_allocated_shard_connection = nil
      def db_charmer_allocated_shard_connection
        @@db_charmer_allocated_shard_connection
      end

      def db_charmer_allocated_shard_connection=(conn)
        @@db_charmer_allocated_shard_connection = conn
      end

      # For sharded models, return the table name without the schema prefix
      def table_name_without_schema
        if self.respond_to?(:orig_table_name)
          self.orig_table_name
        else
          self.table_name
        end
      end
    end
  end
end