nmode's Git Repositories - signal-cli/blob - src/main/java/org/asamk/signal/commands/ListContactsCommand.java

   1 package org.asamk.signal.commands;
   2
   3 import net.sourceforge.argparse4j.impl.Arguments;
   4 import net.sourceforge.argparse4j.inf.Namespace;
   5 import net.sourceforge.argparse4j.inf.Subparser;
   6
   7 import org.asamk.signal.commands.exceptions.CommandException;
   8 import org.asamk.signal.manager.Manager;
   9 import org.asamk.signal.manager.storage.recipients.Contact;
  10 import org.asamk.signal.manager.storage.recipients.Profile;
  11 import org.asamk.signal.output.JsonWriter;
  12 import org.asamk.signal.output.OutputWriter;
  13 import org.asamk.signal.output.PlainTextWriter;
  14 import org.asamk.signal.util.CommandUtil;
  15
  16 import java.util.Base64;
  17 import java.util.Optional;
  18 import java.util.UUID;
  19
  20 public class ListContactsCommand implements JsonRpcLocalCommand {
  21
  22     @Override
  23     public String getName() {
  24         return "listContacts";
  25     }
  26
  27     @Override
  28     public void attachToSubparser(final Subparser subparser) {
  29         subparser.help("Show a list of known contacts with names and profiles.");
  30         subparser.addArgument("recipient").help("Specify one ore more phone numbers to show.").nargs("*");
  31         subparser.addArgument("-a", "--all-recipients")
  32                 .action(Arguments.storeTrue())
  33                 .help("Include all known recipients, not only contacts.");
  34         subparser.addArgument("--blocked")
  35                 .type(Boolean.class)
  36                 .help("Specify if only blocked or unblocked contacts should be shown (default: all contacts)");
  37         subparser.addArgument("--name").help("Find contacts with the given contact or profile name.");
  38     }
  39
  40     @Override
  41     public void handleCommand(
  42             final Namespace ns, final Manager m, final OutputWriter outputWriter
  43     ) throws CommandException {
  44         final var allRecipients = Boolean.TRUE.equals(ns.getBoolean("all-recipients"));
  45         final var blocked = ns.getBoolean("blocked");
  46         final var recipientStrings = ns.<String>getList("recipient");
  47         final var recipientIdentifiers = CommandUtil.getSingleRecipientIdentifiers(recipientStrings, m.getSelfNumber());
  48         final var name = ns.getString("name");
  49         final var recipients = m.getRecipients(!allRecipients,
  50                 Optional.ofNullable(blocked),
  51                 recipientIdentifiers,
  52                 Optional.ofNullable(name));
  53
  54         if (outputWriter instanceof PlainTextWriter writer) {
  55             for (var r : recipients) {
  56                 final var contact = r.getContact() == null ? Contact.newBuilder().build() : r.getContact();
  57                 final var profile = r.getProfile() == null ? Profile.newBuilder().build() : r.getProfile();
  58                 writer.println("Number: {} Name: {} Profile name: {} Color: {} Blocked: {} Message expiration: {}",
  59                         r.getAddress().getLegacyIdentifier(),
  60                         contact.getName(),
  61                         profile.getDisplayName(),
  62                         contact.getColor(),
  63                         contact.isBlocked(),
  64                         contact.getMessageExpirationTime() == 0
  65                                 ? "disabled"
  66                                 : contact.getMessageExpirationTime() + "s");
  67             }
  68         } else {
  69             final var writer = (JsonWriter) outputWriter;
  70             final var jsonContacts = recipients.stream().map(r -> {
  71                 final var address = r.getAddress();
  72                 final var contact = r.getContact() == null ? Contact.newBuilder().build() : r.getContact();
  73                 return new JsonContact(address.number().orElse(null),
  74                         address.uuid().map(UUID::toString).orElse(null),
  75                         contact.getName(),
  76                         contact.getColor(),
  77                         contact.isBlocked(),
  78                         contact.getMessageExpirationTime(),
  79                         r.getProfile() == null
  80                                 ? null
  81                                 : new JsonContact.JsonProfile(r.getProfile().getLastUpdateTimestamp(),
  82                                         r.getProfile().getGivenName(),
  83                                         r.getProfile().getFamilyName(),
  84                                         r.getProfile().getAbout(),
  85                                         r.getProfile().getAboutEmoji(),
  86                                         r.getProfile().getMobileCoinAddress() == null
  87                                                 ? null
  88                                                 : Base64.getEncoder()
  89                                                         .encodeToString(r.getProfile().getMobileCoinAddress())));
  90             }).toList();
  91
  92             writer.write(jsonContacts);
  93         }
  94     }
  95
  96     private record JsonContact(
  97             String number,
  98             String uuid,
  99             String name,
 100             String color,
 101             boolean isBlocked,
 102             int messageExpirationTime,
 103             JsonProfile profile
 104     ) {
 105
 106         private record JsonProfile(
 107                 long lastUpdateTimestamp,
 108                 String givenName,
 109                 String familyName,
 110                 String about,
 111                 String aboutEmoji,
 112                 String mobileCoinAddress
 113         ) {}
 114     }
 115 }