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.commands.exceptions.RateLimitErrorException;
   9 import org.asamk.signal.manager.Manager;
  10 import org.asamk.signal.manager.api.RateLimitException;
  11 import org.asamk.signal.manager.api.UserStatus;
  12 import org.asamk.signal.output.JsonWriter;
  13 import org.asamk.signal.output.OutputWriter;
  14 import org.asamk.signal.output.PlainTextWriter;
  15 import org.asamk.signal.util.CommandUtil;
  16 import org.slf4j.Logger;
  17 import org.slf4j.LoggerFactory;
  18
  19 import java.io.IOException;
  20 import java.util.HashSet;
  21 import java.util.Map;
  22
  23 public class GetUserStatusCommand implements JsonRpcLocalCommand {
  24
  25     private final static Logger logger = LoggerFactory.getLogger(GetUserStatusCommand.class);
  26
  27     @Override
  28     public String getName() {
  29         return "getUserStatus";
  30     }
  31
  32     @Override
  33     public void attachToSubparser(final Subparser subparser) {
  34         subparser.help("Check if the specified phone number/s have been registered");
  35         subparser.addArgument("recipient").help("Phone number").nargs("+");
  36     }
  37
  38     @Override
  39     public void handleCommand(
  40             final Namespace ns, final Manager m, final OutputWriter outputWriter
  41     ) throws CommandException {
  42         // Get a map of registration statuses
  43         Map<String, UserStatus> registered;
  44         try {
  45             registered = m.getUserStatus(new HashSet<>(ns.getList("recipient")));
  46         } catch (RateLimitException e) {
  47             final var message = CommandUtil.getRateLimitMessage(e);
  48             throw new RateLimitErrorException(message, e);
  49         } catch (IOException e) {
  50             throw new IOErrorException("Unable to check if users are registered: "
  51                     + e.getMessage()
  52                     + " ("
  53                     + e.getClass().getSimpleName()
  54                     + ")", e);
  55         }
  56
  57         // Output
  58         if (outputWriter instanceof JsonWriter jsonWriter) {
  59
  60             var jsonUserStatuses = registered.entrySet().stream().map(entry -> {
  61                 final var number = entry.getValue().number();
  62                 final var uuid = entry.getValue().uuid();
  63                 return new JsonUserStatus(entry.getKey(), number, uuid == null ? null : uuid.toString(), uuid != null);
  64             }).toList();
  65
  66             jsonWriter.write(jsonUserStatuses);
  67         } else {
  68             final var writer = (PlainTextWriter) outputWriter;
  69
  70             for (var entry : registered.entrySet()) {
  71                 final var userStatus = entry.getValue();
  72                 writer.println("{}: {}{}",
  73                         entry.getKey(),
  74                         userStatus.uuid() != null,
  75                         userStatus.unrestrictedUnidentifiedAccess() ? " (unrestricted sealed sender)" : "");
  76             }
  77         }
  78     }
  79
  80     private record JsonUserStatus(String recipient, String number, String uuid, boolean isRegistered) {}
  81 }