import org.asamk.signal.manager.Manager;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
+import org.whispersystems.libsignal.util.Pair;
import java.io.IOException;
import java.util.HashSet;
import java.util.Map;
+import java.util.UUID;
import java.util.stream.Collectors;
public class GetUserStatusCommand implements JsonRpcLocalCommand {
@Override
public void attachToSubparser(final Subparser subparser) {
subparser.help("Check if the specified phone number/s have been registered");
- subparser.addArgument("number").help("Phone number").nargs("+");
+ subparser.addArgument("recipient").help("Phone number").nargs("+");
}
@Override
final Namespace ns, final Manager m, final OutputWriter outputWriter
) throws CommandException {
// Get a map of registration statuses
- Map<String, Boolean> registered;
+ Map<String, Pair<String, UUID>> registered;
try {
- registered = m.areUsersRegistered(new HashSet<>(ns.getList("number")));
+ registered = m.areUsersRegistered(new HashSet<>(ns.getList("recipient")));
} catch (IOException e) {
- logger.debug("Failed to check registered users", e);
- throw new IOErrorException("Unable to check if users are registered");
+ throw new IOErrorException("Unable to check if users are registered", e);
}
// Output
- if (outputWriter instanceof JsonWriter) {
- final var jsonWriter = (JsonWriter) outputWriter;
+ if (outputWriter instanceof JsonWriter jsonWriter) {
- var jsonUserStatuses = registered.entrySet()
- .stream()
- .map(entry -> new JsonUserStatus(entry.getKey(), entry.getValue()))
- .collect(Collectors.toList());
+ var jsonUserStatuses = registered.entrySet().stream().map(entry -> {
+ final var number = entry.getValue().first();
+ final var uuid = entry.getValue().second();
+ return new JsonUserStatus(entry.getKey(), number, uuid == null ? null : uuid.toString(), uuid != null);
+ }).collect(Collectors.toList());
jsonWriter.write(jsonUserStatuses);
} else {
final var writer = (PlainTextWriter) outputWriter;
for (var entry : registered.entrySet()) {
- writer.println("{}: {}", entry.getKey(), entry.getValue());
+ final var uuid = entry.getValue().second();
+ writer.println("{}: {}", entry.getKey(), uuid != null);
}
}
}
private static final class JsonUserStatus {
- public String name;
+ public final String recipient;
- public boolean isRegistered;
+ public final String number;
- public JsonUserStatus(String name, boolean isRegistered) {
- this.name = name;
+ public final String uuid;
+
+ public final boolean isRegistered;
+
+ public JsonUserStatus(String recipient, String number, String uuid, boolean isRegistered) {
+ this.recipient = recipient;
+ this.number = number;
+ this.uuid = uuid;
this.isRegistered = isRegistered;
}
}