module Tako
  class Repository
    class << self
      def proxy_configs
        @proxy_configs
      end

      def proxy_connections
        @proxy_connections
      end

      def add(shard_name, conf)
        @proxy_configs ||= {}
        @proxy_connections ||= {}

        shard_name = shard_name.to_sym
        return if @proxy_configs[shard_name]

        temporary_class = Class.new(ActiveRecord::Base)
        const_set("TAKO_AR_CLASS_#{shard_name.upcase}", temporary_class)
        temporary_class.establish_connection(conf)

        @proxy_connections[shard_name] = temporary_class.connection
        @proxy_configs[shard_name] = conf
      end

      def shard(shard_name, base = nil)
        Proxy.new(shard_name, @proxy_connections[shard_name.to_sym], base)
      end
    end
  end
end