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.manager.Manager;
   7 import org.asamk.signal.storage.protocol.JsonIdentityKeyStore;
   8 import org.asamk.signal.util.Hex;
   9 import org.asamk.signal.util.Util;
  10 import org.whispersystems.libsignal.util.Pair;
  11 import org.whispersystems.signalservice.api.util.InvalidNumberException;
  12
  13 import java.util.List;
  14 import java.util.Map;
  15
  16 public class ListIdentitiesCommand implements LocalCommand {
  17
  18     private static void printIdentityFingerprint(Manager m, String theirUsername, JsonIdentityKeyStore.Identity theirId) {
  19         String digits = Util.formatSafetyNumber(m.computeSafetyNumber(theirUsername, theirId.getIdentityKey()));
  20         System.out.println(String.format("%s: %s Added: %s Fingerprint: %s Safety Number: %s", theirUsername,
  21                 theirId.getTrustLevel(), theirId.getDateAdded(), Hex.toString(theirId.getFingerprint()), digits));
  22     }
  23
  24     @Override
  25     public void attachToSubparser(final Subparser subparser) {
  26         subparser.addArgument("-n", "--number")
  27                 .help("Only show identity keys for the given phone number.");
  28     }
  29
  30     @Override
  31     public int handleCommand(final Namespace ns, final Manager m) {
  32         if (!m.isRegistered()) {
  33             System.err.println("User is not registered.");
  34             return 1;
  35         }
  36         if (ns.get("number") == null) {
  37             for (Map.Entry<String, List<JsonIdentityKeyStore.Identity>> keys : m.getIdentities().entrySet()) {
  38                 for (JsonIdentityKeyStore.Identity id : keys.getValue()) {
  39                     printIdentityFingerprint(m, keys.getKey(), id);
  40                 }
  41             }
  42         } else {
  43             String number = ns.getString("number");
  44             try {
  45                 Pair<String, List<JsonIdentityKeyStore.Identity>> key = m.getIdentities(number);
  46                 for (JsonIdentityKeyStore.Identity id : key.second()) {
  47                     printIdentityFingerprint(m, key.first(), id);
  48                 }
  49             } catch (InvalidNumberException e) {
  50                 System.out.println("Invalid number: " + e.getMessage());
  51             }
  52         }
  53         return 0;
  54     }
  55 }