module SidekiqScheduler::Client

def delayed_push(queue=nil, timestamp, item)
  raise(ArgumentError, "Message must be a Hash of the form: { 'class' => SomeClass, 'args' => ['bob', 1, :foo => 'bar'] }") unless item.is_a?(Hash)
  raise(ArgumentError, "Message must include a class and set of arguments: #{item.inspect}") if !item['class'] || !item['args']
  timestamp = timestamp.to_i
  item['queue'] = queue.to_s if queue
  item['class'] = item['class'].to_s if !item['class'].is_a?(String)
  # Add item to the list for this timestamp
  Sidekiq.redis { |r| r.rpush("delayed:#{timestamp}", MultiJson.encode(item)) }
  # Add timestamp to zset. Score and value are based on the timestamp
  # as querying will be based on that
  Sidekiq.redis { |r| r.zadd('delayed_queue_schedule', timestamp, timestamp) }
end

def remove_all_delayed(klass, *args)
  remove_all_delayed_from_queue(nil, klass, *args)
end

def remove_all_delayed_from_queue(queue, klass, *args)
  count = 0
  item = {'class' => klass.to_s, 'args' => args}
  item['queue'] = queue.to_s if queue
  search = MultiJson.encode(item)
  Array(Sidekiq.redis { |r| r.keys("delayed:*") }).each do |key|
    count += Sidekiq.redis { |r| r.lrem(key, 0, search) }
  end
  count
end

def remove_delayed(timestamp, klass, *args)
  remove_delayed_from_queue(nil, timestamp, klass, *args)
end

def remove_delayed_from_queue(queue, timestamp, klass, *args)
  timestamp = timestamp.to_i
  item = {'class' => klass.to_s, 'args' => args}
  item['queue'] = queue.to_s if queue
  search = MultiJson.encode(item)
  count = Sidekiq.redis { |r| r.lrem("delayed:#{timestamp}", 0, search) }
  remove_scheduler_queue(timestamp)
  count
end

def remove_scheduler_queue(timestamp)
  key = "delayed:#{timestamp}"
  if 0 == Sidekiq.redis { |r| r.llen(key) }
    Sidekiq.redis { |r| r.del(key) }
    Sidekiq.redis { |r| r.zrem('delayed_queue_schedule', timestamp) }
  end
end