require "worker_roulette/version"
require 'oj'
require 'redis'
require 'hiredis'
require 'em-hiredis'
require 'connection_pool'
require "digest/sha1"

Dir[File.join(File.dirname(__FILE__),'worker_roulette','**','*.rb')].sort.each { |file| require file.gsub(".rb", "")}

module WorkerRoulette
  JOB_BOARD = "job_board"
  JOB_NOTIFICATIONS = "new_job_ready"

  def self.start(config = {})
    @redis_config               = {host: 'localhost', port: 6379, db: 14, driver: :hiredis, timeout: 5, evented: false, pool_size: 10}.merge(config)
    @pool_config                = Hash[size: @redis_config.delete(:pool_size), timeout: @redis_config.delete(:timeout)]
    @evented                    = @redis_config.delete(:evented)

    @foreman_connection_pool    = ConnectionPool.new(@pool_config) {new_redis}
    @tradesman_connection_pool  = ConnectionPool.new(@pool_config) {new_redis}
    @pubsub_connection_pool     = ConnectionPool.new(@pool_config) {new_redis_pubsub}
  end

  def self.foreman(sender, channel = nil)
    raise "WorkerRoulette not Started" unless @foreman_connection_pool
    Foreman.new(sender, @foreman_connection_pool, channel)
  end

  def self.tradesman(channel = nil)
    raise "WorkerRoulette not Started" unless @tradesman_connection_pool
    Tradesman.new(@tradesman_connection_pool, @pubsub_connection_pool, channel)
  end

  def self.a_foreman(sender, channel = nil)
    foreman(sender, channel)
  end

  def self.a_tradesman(channel = nil)
    raise "WorkerRoulette not Started" unless @tradesman_connection_pool
    ATradesman.new(@tradesman_connection_pool, @pubsub_connection_pool, channel)
  end

  def self.tradesman_connection_pool
    @tradesman_connection_pool
  end

  def self.pubsub_connection_pool
    @pubsub_connection_pool
  end

  def self.pool_size
    (@pool_config ||= {})[:size]
  end

  def self.redis_config
    (@redis_config ||= {}).dup
  end

  def self.dump(obj)
    Oj.dump(obj)
  rescue Oj::ParseError => e
    {'error' => e, 'unparsable_string' => obj}
  end

  def self.load(json)
    Oj.load(json)
  rescue Oj::ParseError => e
    {'error' => e, 'unparsable_string' => obj}
  end

  def self.job_board_key(namespace = nil)
    "#{namespace + ':' if namespace}#{WorkerRoulette::JOB_BOARD}"
  end

  def self.sender_key(sender, namespace = nil)
    "#{namespace + ':' if namespace}#{sender}"
  end

  def self.counter_key(sender, namespace = nil)
    "#{namespace + ':' if namespace}counter_key"
  end
private
  def self.new_redis
    if @evented
      require 'eventmachine'
      redis = EM::Hiredis::Client.new(@redis_config[:host], @redis_config[:port], @redis_config[:password], @redis_config[:db])
      redis.connect
    else
      Redis.new(@redis_config)
    end
  end

  def self.new_redis_pubsub
    if @evented
     new_redis.pubsub
    else
      new_redis
    end
  end
end