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.PlainTextWriter;
   4 import org.asamk.signal.commands.exceptions.CommandException;
   5 import org.asamk.signal.commands.exceptions.IOErrorException;
   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.List;
  13 import java.util.stream.Collectors;
  14
  15 public class ErrorUtils {
  16
  17     private final static Logger logger = LoggerFactory.getLogger(ErrorUtils.class);
  18
  19     private ErrorUtils() {
  20     }
  21
  22     public static void handleTimestampAndSendMessageResults(
  23             PlainTextWriter writer, long timestamp, List<SendMessageResult> results
  24     ) throws CommandException {
  25         if (timestamp != 0) {
  26             writer.println("{}", timestamp);
  27         }
  28         var errors = getErrorMessagesFromSendMessageResults(results);
  29         handleSendMessageResultErrors(errors);
  30     }
  31
  32     public static List<String> getErrorMessagesFromSendMessageResults(List<SendMessageResult> results) {
  33         var errors = new ArrayList<String>();
  34         for (var result : results) {
  35             var error = getErrorMessageFromSendMessageResult(result);
  36             if (error != null) {
  37                 errors.add(error);
  38             }
  39         }
  40
  41         return errors;
  42     }
  43
  44     public static String getErrorMessageFromSendMessageResult(SendMessageResult result) {
  45         if (result.isNetworkFailure()) {
  46             return String.format("Network failure for \"%s\"", result.getAddress().getLegacyIdentifier());
  47         } else if (result.isUnregisteredFailure()) {
  48             return String.format("Unregistered user \"%s\"", result.getAddress().getLegacyIdentifier());
  49         } else if (result.getIdentityFailure() != null) {
  50             return String.format("Untrusted Identity for \"%s\"", result.getAddress().getLegacyIdentifier());
  51         } else if (result.getProofRequiredFailure() != null) {
  52             final var failure = result.getProofRequiredFailure();
  53             return String.format(
  54                     "CAPTCHA proof required for sending to \"%s\", available options \"%s\" with token \"%s\", or wait \"%d\" seconds",
  55                     result.getAddress().getLegacyIdentifier(),
  56                     failure.getOptions()
  57                             .stream()
  58                             .map(ProofRequiredException.Option::toString)
  59                             .collect(Collectors.joining(", ")),
  60                     failure.getToken(),
  61                     failure.getRetryAfterSeconds());
  62         }
  63         return null;
  64     }
  65
  66     private static void handleSendMessageResultErrors(List<String> errors) throws CommandException {
  67         if (errors.size() == 0) {
  68             return;
  69         }
  70         var message = new StringBuilder();
  71         message.append("Failed to send (some) messages:\n");
  72         for (var error : errors) {
  73             message.append(error).append("\n");
  74         }
  75         throw new IOErrorException(message.toString());
  76     }
  77 }