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