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 static final 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         switch (outputWriter) {
  59             case JsonWriter writer -> {
  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(),
  64                             number,
  65                             uuid == null ? null : uuid.toString(),
  66                             uuid != null);
  67                 }).toList();
  68                 writer.write(jsonUserStatuses);
  69             }
  70             case PlainTextWriter writer -> {
  71                 for (var entry : registered.entrySet()) {
  72                     final var userStatus = entry.getValue();
  73                     writer.println("{}: {}{}",
  74                             entry.getKey(),
  75                             userStatus.uuid() != null,
  76                             userStatus.unrestrictedUnidentifiedAccess() ? " (unrestricted sealed sender)" : "");
  77                 }
  78             }
  79         }
  80     }
  81
  82     private record JsonUserStatus(String recipient, String number, String uuid, boolean isRegistered) {}
  83 }