nmode's Git Repositories - signal-cli/blob - 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: "
  46                     + e.getMessage()
  47                     + " ("
  48                     + e.getClass().getSimpleName()
  49                     + ")", e);
  50         }
  51
  52         // Output
  53         if (outputWriter instanceof JsonWriter jsonWriter) {
  54
  55             var jsonUserStatuses = registered.entrySet().stream().map(entry -> {
  56                 final var number = entry.getValue().first();
  57                 final var uuid = entry.getValue().second();
  58                 return new JsonUserStatus(entry.getKey(), number, uuid == null ? null : uuid.toString(), uuid != null);
  59             }).toList();
  60
  61             jsonWriter.write(jsonUserStatuses);
  62         } else {
  63             final var writer = (PlainTextWriter) outputWriter;
  64
  65             for (var entry : registered.entrySet()) {
  66                 final var uuid = entry.getValue().second();
  67                 writer.println("{}: {}", entry.getKey(), uuid != null);
  68             }
  69         }
  70     }
  71
  72     private record JsonUserStatus(String recipient, String number, String uuid, boolean isRegistered) {}
  73 }