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