require 'set'
require 'open-uri'

module RedisFailover
  # Redis failover-aware client.
  class Client
    include Util

    REDIS_ERRORS = Errno.constants.map { |c| Errno.const_get(c) }.freeze
    REDIS_READ_OPS = Set[
      :dbsize,
      :echo,
      :exists,
      :get,
      :getbit,
      :getrange,
      :hexists,
      :hget,
      :hgetall,
      :hkeys,
      :hlen,
      :hmget,
      :hvals,
      :info,
      :keys,
      :lastsave,
      :lindex,
      :llen,
      :lrange,
      :mapped_hmget,
      :mapped_mget,
      :mget,
      :ping,
      :scard,
      :sdiff,
      :select,
      :sinter,
      :sismember,
      :smembers,
      :srandmember,
      :strlen,
      :sunion,
      :ttl,
      :type,
      :zcard,
      :zcount,
      :zrange,
      :zrangebyscore,
      :zrank,
      :zrevrange,
      :zrevrangebyscore,
      :zrevrank,
      :zscore
    ].freeze

    # Performance optimization: to avoid unnecessary method_missing calls,
    # we proactively define methods that dispatch to the underlying redis
    # calls.
    Redis.public_instance_methods(false).each do |method|
      define_method(method) do |*args, &block|
        dispatch(method, *args, &block)
      end
    end

    # Creates a new failover redis client.
    #
    # Options:
    #
    #   :host - redis failover server host (required)
    #   :port - redis failover server port (required)
    #   :password - optional password for redis nodes
    #   :namespace - optional namespace for redis nodes
    #   :logger - optional logger override
    #   :retry_failure - indicate if failures should be retried (default true)
    #   :max_retries - max retries for a failure (default 5)
    #
    def initialize(options = {})
      unless options.values_at(:host, :port).all?
        raise ArgumentError, ':host and :port options required'
      end

      Util.logger = options[:logger] if options[:logger]
      @namespace = options[:namespace]
      @password = options[:password]
      @retry = options[:retry_failure] || true
      @max_retries = options[:max_retries] || 3
      @registry_url = "http://#{options[:host]}:#{options[:port]}/redis_servers"
      @redis_servers = nil
      @master = nil
      @slaves = []
      @lock = Mutex.new
      build_clients
    end

    def method_missing(method, *args, &block)
      if redis_operation?(method)
        dispatch(method, *args, &block)
      else
        super
      end
    end

    def respond_to?(method)
      redis_operation?(method) || super
    end

    def inspect
      "#<RedisFailover::Client - master: #{master_info}, slaves: #{slaves_info})>"
    end
    alias_method :to_s, :inspect

    private

    def redis_operation?(method)
      Redis.public_instance_methods(false).include?(method)
    end

    def dispatch(method, *args, &block)
      tries = 0
      begin
        if REDIS_READ_OPS.include?(method)
          # send read operations to a slave
          slave.send(method, *args, &block)
        else
          # direct everything else to master
          master.send(method, *args, &block)
        end
      rescue NoMasterError, *REDIS_ERRORS
        logger.error("No suitable node available for operation `#{method}.`")
        sleep(3)
        build_clients

        if @retry && tries < @max_retries
          tries += 1
          retry
        end

        raise
      end
    end

    def master
      @master or raise NoMasterError
    end

    def slave
      # pick a slave, if none available fallback to master
      @slaves.sample || master
    end

    def build_clients
      @lock.synchronize do
        begin
          logger.info('Attempting to fetch nodes and build redis clients.')
          servers = fetch_redis_servers
          master = new_clients_for(servers[:master]).first if servers[:master]
          slaves = new_clients_for(*servers[:slaves])

          # once clients are successfully created, swap the references
          @master = master
          @slaves = slaves
        rescue => ex
          logger.error("Failed to fetch servers from #{@registry_url} - #{ex.message}")
          logger.error(ex.backtrace.join("\n"))
        end
      end
    end

    def fetch_redis_servers
      open(@registry_url) do |io|
        servers = symbolize_keys(MultiJson.decode(io))
        logger.info("Fetched servers: #{servers}")
        servers
      end
    end

    def new_clients_for(*nodes)
      nodes.map do |node|
        host, port = node.split(':')
        client = Redis.new(:host => host, :port => port, :password => @password)
        if @namespace
          client = Redis::Namespace.new(@namespace, :redis => client)
        end
        client
      end
    end

    def master_info
      return "none" unless @master
      "#{@master.client.host}:#{@master.client.port}"
    end

    def slaves_info
      return "none" if @slaves.empty?
      @slaves.map { |s| "#{s.client.host}:#{s.client.port}" }.join(', ')
    end
  end
end