module Redis::Commands::Sets

def sadd(key, *members)
  members.flatten!(1)
  send_command([:sadd, key].concat(members))
end

def sadd?(key, *members)
  members.flatten!(1)
  send_command([:sadd, key].concat(members), &Boolify)
end

def scard(key)
  send_command([:scard, key])
end

def sdiff(*keys)
  keys.flatten!(1)
  send_command([:sdiff].concat(keys))
end

def sdiffstore(destination, *keys)
  keys.flatten!(1)
  send_command([:sdiffstore, destination].concat(keys))
end

def sinter(*keys)
  keys.flatten!(1)
  send_command([:sinter].concat(keys))
end

def sinterstore(destination, *keys)
  keys.flatten!(1)
  send_command([:sinterstore, destination].concat(keys))
end

def sismember(key, member)
  send_command([:sismember, key, member], &Boolify)
end

def smembers(key)
  send_command([:smembers, key])
end

def smismember(key, *members)
  members.flatten!(1)
  send_command([:smismember, key].concat(members)) do |reply|
    reply.map(&Boolify)
  end
end

def smove(source, destination, member)
  send_command([:smove, source, destination, member], &Boolify)
end

def spop(key, count = nil)
  if count.nil?
    send_command([:spop, key])
  else
    send_command([:spop, key, Integer(count)])
  end
end

def srandmember(key, count = nil)
  if count.nil?
    send_command([:srandmember, key])
  else
    send_command([:srandmember, key, count])
  end
end

def srem(key, *members)
  members.flatten!(1)
  send_command([:srem, key].concat(members))
end

def srem?(key, *members)
  members.flatten!(1)
  send_command([:srem, key].concat(members), &Boolify)
end

def sscan(key, cursor, **options)
  _scan(:sscan, cursor, [key], **options)
end

def sscan_each(key, **options, &block)
  return to_enum(:sscan_each, key, **options) unless block_given?
  cursor = 0
  loop do
    cursor, keys = sscan(key, cursor, **options)
    keys.each(&block)
    break if cursor == "0"
  end
end

def sunion(*keys)
  keys.flatten!(1)
  send_command([:sunion].concat(keys))
end

def sunionstore(destination, *keys)
  keys.flatten!(1)
  send_command([:sunionstore, destination].concat(keys))
end