Exit on timeout when receiving

[signal-cli] / src / main / java / cli / Main.java

diff --git a/src/main/java/cli/Main.java b/src/main/java/cli/Main.java
index 8a6cea530c319f88fa2847333d35d67e497acc12..4cdbb806fb3394eb00250320eff3f053288a7487 100644 (file)
--- a/src/main/java/cli/Main.java
+++ b/src/main/java/cli/Main.java
@@ -21,10 +21,13 @@ import net.sourceforge.argparse4j.impl.Arguments;
 import net.sourceforge.argparse4j.inf.*;
 import org.apache.commons.io.IOUtils;
 import org.whispersystems.textsecure.api.TextSecureMessageSender;
 import net.sourceforge.argparse4j.inf.*;
 import org.apache.commons.io.IOUtils;
 import org.whispersystems.textsecure.api.TextSecureMessageSender;

+import org.whispersystems.textsecure.api.crypto.UntrustedIdentityException;

 import org.whispersystems.textsecure.api.messages.*;
 import org.whispersystems.textsecure.api.messages.multidevice.TextSecureSyncMessage;
 import org.whispersystems.textsecure.api.push.TextSecureAddress;
 import org.whispersystems.textsecure.api.push.exceptions.EncapsulatedExceptions;
 import org.whispersystems.textsecure.api.messages.*;
 import org.whispersystems.textsecure.api.messages.multidevice.TextSecureSyncMessage;
 import org.whispersystems.textsecure.api.push.TextSecureAddress;
 import org.whispersystems.textsecure.api.push.exceptions.EncapsulatedExceptions;

+import org.whispersystems.textsecure.api.push.exceptions.NetworkFailureException;
+import org.whispersystems.textsecure.api.push.exceptions.UnregisteredUserException;

 import org.whispersystems.textsecure.api.util.InvalidNumberException;
 
 import java.io.File;
 import org.whispersystems.textsecure.api.util.InvalidNumberException;
 
 import java.io.File;


@@ -162,8 +165,19 @@ public class Main {
                 }
                 try {
                     messageSender.sendMessage(recipients, message);
                 }
                 try {
                     messageSender.sendMessage(recipients, message);

-                } catch (IOException | EncapsulatedExceptions e) {
+                } catch (IOException e) {

                     System.out.println("Failed to send message: " + e.getMessage());
                     System.out.println("Failed to send message: " + e.getMessage());

+                } catch (EncapsulatedExceptions e) {
+                    System.out.println("Failed to send (some) messages:");
+                    for (NetworkFailureException n : e.getNetworkExceptions()) {
+                        System.out.println("Network failure for \"" + n.getE164number() + "\": " + n.getMessage());
+                    }
+                    for (UnregisteredUserException n : e.getUnregisteredUserExceptions()) {
+                        System.out.println("Unregistered user \"" + n.getE164Number() + "\": " + n.getMessage());
+                    }
+                    for (UntrustedIdentityException n : e.getUntrustedIdentityExceptions()) {
+                        System.out.println("Untrusted Identity for \"" + n.getE164Number() + "\": " + n.getMessage());
+                    }

                 }
                 break;
             case "receive":
                 }
                 break;
             case "receive":


@@ -172,7 +186,7 @@ public class Main {
                     System.exit(1);
                 }
                 try {
                     System.exit(1);
                 }
                 try {

-                    m.receiveMessages(new Manager.ReceiveMessageHandler() {
+                    m.receiveMessages(5, true, new Manager.ReceiveMessageHandler() {

                         @Override
                         public void handleMessage(TextSecureEnvelope envelope) {
                             System.out.println("Envelope from: " + envelope.getSource());
                         @Override
                         public void handleMessage(TextSecureEnvelope envelope) {
                             System.out.println("Envelope from: " + envelope.getSource());


@@ -219,5 +233,6 @@ public class Main {
                 break;
         }
         m.save();
                 break;
         }
         m.save();

+        System.exit(0);

     }
 }
     }
 }