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

   1 package org.asamk.signal.util;
   2
   3 import net.sourceforge.argparse4j.inf.Namespace;
   4
   5 import org.asamk.signal.commands.exceptions.UserErrorException;
   6 import org.asamk.signal.manager.Manager;
   7 import org.asamk.signal.manager.api.CaptchaRequiredException;
   8 import org.asamk.signal.manager.api.GroupId;
   9 import org.asamk.signal.manager.api.GroupIdFormatException;
  10 import org.asamk.signal.manager.api.InvalidNumberException;
  11 import org.asamk.signal.manager.api.RateLimitException;
  12 import org.asamk.signal.manager.api.ReceiveConfig;
  13 import org.asamk.signal.manager.api.RecipientIdentifier;
  14
  15 import java.util.Collection;
  16 import java.util.HashSet;
  17 import java.util.List;
  18 import java.util.Set;
  19
  20 public class CommandUtil {
  21
  22     private CommandUtil() {
  23     }
  24
  25     public static Set<RecipientIdentifier> getRecipientIdentifiers(
  26             final Manager m,
  27             final boolean isNoteToSelf,
  28             final List<String> recipientStrings,
  29             final List<String> groupIdStrings,
  30             final List<String> usernameStrings
  31     ) throws UserErrorException {
  32         final var recipientIdentifiers = new HashSet<RecipientIdentifier>();
  33         if (isNoteToSelf) {
  34             recipientIdentifiers.add(RecipientIdentifier.NoteToSelf.INSTANCE);
  35         }
  36         if (recipientStrings != null) {
  37             final var localNumber = m.getSelfNumber();
  38             recipientIdentifiers.addAll(CommandUtil.getSingleRecipientIdentifiers(recipientStrings, localNumber));
  39         }
  40         if (groupIdStrings != null) {
  41             recipientIdentifiers.addAll(CommandUtil.getGroupIdentifiers(groupIdStrings));
  42         }
  43         if (usernameStrings != null) {
  44             recipientIdentifiers.addAll(CommandUtil.getUsernameIdentifiers(usernameStrings));
  45         }
  46
  47         if (recipientIdentifiers.isEmpty()) {
  48             throw new UserErrorException("No recipients given");
  49         }
  50         return recipientIdentifiers;
  51     }
  52
  53     public static Set<RecipientIdentifier.Group> getGroupIdentifiers(Collection<String> groupIdStrings) throws UserErrorException {
  54         if (groupIdStrings == null) {
  55             return Set.of();
  56         }
  57         final var groupIds = new HashSet<RecipientIdentifier.Group>();
  58         for (final var groupIdString : groupIdStrings) {
  59             groupIds.add(new RecipientIdentifier.Group(getGroupId(groupIdString)));
  60         }
  61         return groupIds;
  62     }
  63
  64     public static Set<GroupId> getGroupIds(Collection<String> groupIdStrings) throws UserErrorException {
  65         if (groupIdStrings == null) {
  66             return Set.of();
  67         }
  68         final var groupIds = new HashSet<GroupId>();
  69         for (final var groupIdString : groupIdStrings) {
  70             groupIds.add(getGroupId(groupIdString));
  71         }
  72         return groupIds;
  73     }
  74
  75     public static GroupId getGroupId(String groupId) throws UserErrorException {
  76         if (groupId == null) {
  77             return null;
  78         }
  79         try {
  80             return GroupId.fromBase64(groupId);
  81         } catch (GroupIdFormatException e) {
  82             throw new UserErrorException("Invalid group id: " + e.getMessage());
  83         }
  84     }
  85
  86     public static Set<RecipientIdentifier.Single> getSingleRecipientIdentifiers(
  87             final Collection<String> recipientStrings,
  88             final String localNumber
  89     ) throws UserErrorException {
  90         if (recipientStrings == null) {
  91             return Set.of();
  92         }
  93         final var identifiers = new HashSet<RecipientIdentifier.Single>();
  94         for (var recipientString : recipientStrings) {
  95             identifiers.add(getSingleRecipientIdentifier(recipientString, localNumber));
  96         }
  97         return identifiers;
  98     }
  99
 100     public static RecipientIdentifier.Single getSingleRecipientIdentifier(
 101             final String recipientString,
 102             final String localNumber
 103     ) throws UserErrorException {
 104         try {
 105             return RecipientIdentifier.Single.fromString(recipientString, localNumber);
 106         } catch (InvalidNumberException e) {
 107             throw new UserErrorException("Invalid phone number '" + recipientString + "': " + e.getMessage(), e);
 108         }
 109     }
 110
 111     public static Set<RecipientIdentifier.Username> getUsernameIdentifiers(Collection<String> usernameIdStrings) {
 112         if (usernameIdStrings == null) {
 113             return Set.of();
 114         }
 115         final var usernameIds = new HashSet<RecipientIdentifier.Username>();
 116         for (final var usernameIdString : usernameIdStrings) {
 117             usernameIds.add(new RecipientIdentifier.Username(usernameIdString));
 118         }
 119         return usernameIds;
 120     }
 121
 122     public static String getCaptchaRequiredMessage(final CaptchaRequiredException e, final boolean captchaProvided) {
 123         String message;
 124         if (!captchaProvided) {
 125             message = """
 126                       Captcha required for verification, use --captcha CAPTCHA
 127                       To get the token, go to https://signalcaptchas.org/registration/generate.html
 128                       After solving the captcha, right-click on the "Open Signal" link and copy the link.""";
 129         } else {
 130             message = "Invalid captcha given.";
 131         }
 132         if (e.getNextAttemptTimestamp() > 0) {
 133             message += "\nNext Captcha may be provided at " + DateUtils.formatTimestamp(e.getNextAttemptTimestamp());
 134         }
 135         return message;
 136     }
 137
 138     public static String getRateLimitMessage(final RateLimitException e) {
 139         String message = "Rate limit reached";
 140         if (e.getNextAttemptTimestamp() > 0) {
 141             message += "\nNext attempt may be tried at " + DateUtils.formatTimestamp(e.getNextAttemptTimestamp());
 142         }
 143         return message;
 144     }
 145
 146     public static ReceiveConfig getReceiveConfig(final Namespace ns) {
 147         final var ignoreAttachments = Boolean.TRUE.equals(ns.getBoolean("ignore-attachments"));
 148         final var ignoreStories = Boolean.TRUE.equals(ns.getBoolean("ignore-stories"));
 149         final var sendReadReceipts = Boolean.TRUE.equals(ns.getBoolean("send-read-receipts"));
 150
 151         return new ReceiveConfig(ignoreAttachments, ignoreStories, sendReadReceipts);
 152     }
 153 }