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,
  50             final Manager m,
  51             final OutputWriter outputWriter
  52     ) throws CommandException {
  53         var number = ns.getString("number");
  54
  55         List<Identity> identities;
  56         if (number == null) {
  57             identities = m.getIdentities();
  58         } else {
  59             identities = m.getIdentities(CommandUtil.getSingleRecipientIdentifier(number, m.getSelfNumber()));
  60         }
  61
  62         switch (outputWriter) {
  63             case PlainTextWriter writer -> {
  64                 for (var id : identities) {
  65                     printIdentityFingerprint(writer, id);
  66                 }
  67             }
  68             case JsonWriter writer -> {
  69                 final var jsonIdentities = identities.stream().map(id -> {
  70                     final var address = id.recipient();
  71                     var safetyNumber = Util.formatSafetyNumber(id.safetyNumber());
  72                     var scannableSafetyNumber = id.scannableSafetyNumber();
  73                     return new JsonIdentity(address.number().orElse(null),
  74                             address.uuid().map(UUID::toString).orElse(null),
  75                             Hex.toString(id.fingerprint()),
  76                             safetyNumber,
  77                             scannableSafetyNumber == null
  78                                     ? null
  79                                     : Base64.getEncoder().encodeToString(scannableSafetyNumber),
  80                             id.trustLevel().name(),
  81                             id.dateAddedTimestamp());
  82                 }).toList();
  83                 writer.write(jsonIdentities);
  84             }
  85         }
  86     }
  87
  88     private record JsonIdentity(
  89             String number,
  90             String uuid,
  91             String fingerprint,
  92             String safetyNumber,
  93             String scannableSafetyNumber,
  94             String trustLevel,
  95             long addedTimestamp
  96     ) {}
  97 }