nmode's Git Repositories - signal-cli/blob - src/main/java/org/asamk/signal/commands/ListIdentitiesCommand.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.manager.Manager;
   8 import org.asamk.signal.manager.api.Identity;
   9 import org.asamk.signal.output.JsonWriter;
  10 import org.asamk.signal.output.OutputWriter;
  11 import org.asamk.signal.output.PlainTextWriter;
  12 import org.asamk.signal.util.CommandUtil;
  13 import org.asamk.signal.util.Hex;
  14 import org.asamk.signal.util.Util;
  15 import org.slf4j.Logger;
  16 import org.slf4j.LoggerFactory;
  17
  18 import java.util.Base64;
  19 import java.util.List;
  20 import java.util.UUID;
  21
  22 public class ListIdentitiesCommand implements JsonRpcLocalCommand {
  23
  24     private final static Logger logger = LoggerFactory.getLogger(ListIdentitiesCommand.class);
  25
  26     @Override
  27     public String getName() {
  28         return "listIdentities";
  29     }
  30
  31     private static void printIdentityFingerprint(PlainTextWriter writer, Identity theirId) {
  32         writer.println("{}: {} Added: {} Fingerprint: {} Safety Number: {}",
  33                 theirId.recipient().getLegacyIdentifier(),
  34                 theirId.trustLevel(),
  35                 theirId.dateAdded(),
  36                 Hex.toString(theirId.getFingerprint()),
  37                 Util.formatSafetyNumber(theirId.safetyNumber()));
  38     }
  39
  40     @Override
  41     public void attachToSubparser(final Subparser subparser) {
  42         subparser.help("List all known identity keys and their trust status, fingerprint and safety number.");
  43         subparser.addArgument("-n", "--number").help("Only show identity keys for the given phone number.");
  44     }
  45
  46     @Override
  47     public void handleCommand(
  48             final Namespace ns, final Manager m, final OutputWriter outputWriter
  49     ) throws CommandException {
  50         var number = ns.getString("number");
  51
  52         List<Identity> identities;
  53         if (number == null) {
  54             identities = m.getIdentities();
  55         } else {
  56             identities = m.getIdentities(CommandUtil.getSingleRecipientIdentifier(number, m.getSelfNumber()));
  57         }
  58
  59         if (outputWriter instanceof PlainTextWriter writer) {
  60             for (var id : identities) {
  61                 printIdentityFingerprint(writer, id);
  62             }
  63         } else {
  64             final var writer = (JsonWriter) outputWriter;
  65             final var jsonIdentities = identities.stream().map(id -> {
  66                 final var address = id.recipient();
  67                 var safetyNumber = Util.formatSafetyNumber(id.safetyNumber());
  68                 var scannableSafetyNumber = id.scannableSafetyNumber();
  69                 return new JsonIdentity(address.number().orElse(null),
  70                         address.uuid().map(UUID::toString).orElse(null),
  71                         Hex.toString(id.getFingerprint()),
  72                         safetyNumber,
  73                         scannableSafetyNumber == null
  74                                 ? null
  75                                 : Base64.getEncoder().encodeToString(scannableSafetyNumber),
  76                         id.trustLevel().name(),
  77                         id.dateAdded().getTime());
  78             }).toList();
  79
  80             writer.write(jsonIdentities);
  81         }
  82     }
  83
  84     private record JsonIdentity(
  85             String number,
  86             String uuid,
  87             String fingerprint,
  88             String safetyNumber,
  89             String scannableSafetyNumber,
  90             String trustLevel,
  91             long addedTimestamp
  92     ) {}
  93 }