]> nmode's Git Repositories - signal-cli/blob - src/main/java/org/asamk/signal/commands/GetUserStatusCommand.java
97e4af340a1c091cb6dbaab87fa454161c7c34b8
[signal-cli] / src / main / java / org / asamk / signal / commands / GetUserStatusCommand.java
1 package org.asamk.signal.commands;
2
3 import net.sourceforge.argparse4j.inf.Namespace;
4 import net.sourceforge.argparse4j.inf.Subparser;
5
6 import org.asamk.signal.commands.exceptions.CommandException;
7 import org.asamk.signal.commands.exceptions.IOErrorException;
8 import org.asamk.signal.manager.Manager;
9 import org.asamk.signal.manager.api.Pair;
10 import org.asamk.signal.output.JsonWriter;
11 import org.asamk.signal.output.OutputWriter;
12 import org.asamk.signal.output.PlainTextWriter;
13 import org.slf4j.Logger;
14 import org.slf4j.LoggerFactory;
15
16 import java.io.IOException;
17 import java.util.HashSet;
18 import java.util.Map;
19 import java.util.UUID;
20 import java.util.stream.Collectors;
21
22 public class GetUserStatusCommand implements JsonRpcLocalCommand {
23
24 private final static Logger logger = LoggerFactory.getLogger(GetUserStatusCommand.class);
25
26 @Override
27 public String getName() {
28 return "getUserStatus";
29 }
30
31 @Override
32 public void attachToSubparser(final Subparser subparser) {
33 subparser.help("Check if the specified phone number/s have been registered");
34 subparser.addArgument("recipient").help("Phone number").nargs("+");
35 }
36
37 @Override
38 public void handleCommand(
39 final Namespace ns, final Manager m, final OutputWriter outputWriter
40 ) throws CommandException {
41 // Get a map of registration statuses
42 Map<String, Pair<String, UUID>> registered;
43 try {
44 registered = m.areUsersRegistered(new HashSet<>(ns.getList("recipient")));
45 } catch (IOException e) {
46 throw new IOErrorException("Unable to check if users are registered", e);
47 }
48
49 // Output
50 if (outputWriter instanceof JsonWriter jsonWriter) {
51
52 var jsonUserStatuses = registered.entrySet().stream().map(entry -> {
53 final var number = entry.getValue().first();
54 final var uuid = entry.getValue().second();
55 return new JsonUserStatus(entry.getKey(), number, uuid == null ? null : uuid.toString(), uuid != null);
56 }).collect(Collectors.toList());
57
58 jsonWriter.write(jsonUserStatuses);
59 } else {
60 final var writer = (PlainTextWriter) outputWriter;
61
62 for (var entry : registered.entrySet()) {
63 final var uuid = entry.getValue().second();
64 writer.println("{}: {}", entry.getKey(), uuid != null);
65 }
66 }
67 }
68
69 private record JsonUserStatus(String recipient, String number, String uuid, boolean isRegistered) {}
70 }