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

   1 package org.asamk.signal.util;
   2
   3 import org.asamk.signal.commands.exceptions.CommandException;
   4 import org.asamk.signal.commands.exceptions.IOErrorException;
   5 import org.asamk.signal.manager.api.RecipientIdentifier;
   6 import org.slf4j.Logger;
   7 import org.slf4j.LoggerFactory;
   8 import org.whispersystems.signalservice.api.messages.SendMessageResult;
   9 import org.whispersystems.signalservice.api.push.exceptions.ProofRequiredException;
  10
  11 import java.util.ArrayList;
  12 import java.util.Collection;
  13 import java.util.List;
  14 import java.util.Map;
  15 import java.util.stream.Collectors;
  16
  17 import static org.asamk.signal.util.Util.getLegacyIdentifier;
  18
  19 public class ErrorUtils {
  20
  21     private final static Logger logger = LoggerFactory.getLogger(ErrorUtils.class);
  22
  23     private ErrorUtils() {
  24     }
  25
  26     public static void handleSendMessageResults(
  27             Map<RecipientIdentifier, List<SendMessageResult>> mapResults
  28     ) throws CommandException {
  29         List<String> errors = getErrorMessagesFromSendMessageResults(mapResults);
  30         handleSendMessageResultErrors(errors);
  31     }
  32
  33     public static void handleSendMessageResults(
  34             Collection<SendMessageResult> results
  35     ) throws CommandException {
  36         var errors = getErrorMessagesFromSendMessageResults(results);
  37         handleSendMessageResultErrors(errors);
  38     }
  39
  40     public static List<String> getErrorMessagesFromSendMessageResults(final Map<RecipientIdentifier, List<SendMessageResult>> mapResults) {
  41         return mapResults.values()
  42                 .stream()
  43                 .flatMap(results -> getErrorMessagesFromSendMessageResults(results).stream())
  44                 .collect(Collectors.toList());
  45     }
  46
  47     public static List<String> getErrorMessagesFromSendMessageResults(Collection<SendMessageResult> results) {
  48         var errors = new ArrayList<String>();
  49         for (var result : results) {
  50             var error = getErrorMessageFromSendMessageResult(result);
  51             if (error != null) {
  52                 errors.add(error);
  53             }
  54         }
  55
  56         return errors;
  57     }
  58
  59     public static String getErrorMessageFromSendMessageResult(SendMessageResult result) {
  60         var identifier = getLegacyIdentifier(result.getAddress());
  61         if (result.isNetworkFailure()) {
  62             return String.format("Network failure for \"%s\"", identifier);
  63         } else if (result.isUnregisteredFailure()) {
  64             return String.format("Unregistered user \"%s\"", identifier);
  65         } else if (result.getIdentityFailure() != null) {
  66             return String.format("Untrusted Identity for \"%s\"", identifier);
  67         } else if (result.getProofRequiredFailure() != null) {
  68             final var failure = result.getProofRequiredFailure();
  69             return String.format(
  70                     "CAPTCHA proof required for sending to \"%s\", available options \"%s\" with token \"%s\", or wait \"%d\" seconds",
  71                     identifier,
  72                     failure.getOptions()
  73                             .stream()
  74                             .map(ProofRequiredException.Option::toString)
  75                             .collect(Collectors.joining(", ")),
  76                     failure.getToken(),
  77                     failure.getRetryAfterSeconds());
  78         }
  79         return null;
  80     }
  81
  82     private static void handleSendMessageResultErrors(List<String> errors) throws CommandException {
  83         if (errors.size() == 0) {
  84             return;
  85         }
  86         var message = new StringBuilder();
  87         message.append("Failed to send (some) messages:\n");
  88         for (var error : errors) {
  89             message.append(error).append("\n");
  90         }
  91         throw new IOErrorException(message.toString(), null);
  92     }
  93 }