From: AsamK <asamk@gmx.de>
Date: Tue, 7 Jul 2015 09:42:24 +0000 (+0200)
Subject: Canonicalize recipient addresses
X-Git-Tag: v0.0.2~8
X-Git-Url: https://git.nmode.ca/signal-cli/commitdiff_plain/960e97bc122dd3cd020452038b0abb4781496042

Canonicalize recipient addresses
---

diff --git a/src/main/java/cli/Main.java b/src/main/java/cli/Main.java
index 9a3e5b36..adeb5262 100644
--- a/src/main/java/cli/Main.java
+++ b/src/main/java/cli/Main.java
@@ -24,7 +24,7 @@ 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.util.InvalidNumberException;
 
 import java.io.File;
 import java.io.FileInputStream;
@@ -152,9 +152,9 @@ public class Main {
                 TextSecureDataMessage message = messageBuilder.build();
                 for (String recipient : ns.<String>getList("recipient")) {
                     try {
-                        messageSender.sendMessage(new TextSecureAddress(recipient), message);
-                    } catch (UntrustedIdentityException | IOException e) {
-                        System.out.println("Send message: " + e.getMessage());
+                        messageSender.sendMessage(m.getPushAddress(recipient), message);
+                    } catch (UntrustedIdentityException | IOException | InvalidNumberException e) {
+                        System.out.println("Failed to send message  to \"" + recipient + "\": " + e.getMessage());
                     }
                 }
                 break;
diff --git a/src/main/java/cli/Manager.java b/src/main/java/cli/Manager.java
index cd5f6305..3da46f1d 100644
--- a/src/main/java/cli/Manager.java
+++ b/src/main/java/cli/Manager.java
@@ -206,4 +206,14 @@ public class Manager {
                 messagePipe.shutdown();
         }
     }
+
+    public String canonicalizeNumber(String number) throws InvalidNumberException {
+        String localNumber = username;
+        return PhoneNumberFormatter.formatNumber(number, localNumber);
+    }
+
+    protected TextSecureAddress getPushAddress(String number) throws InvalidNumberException {
+        String e164number = canonicalizeNumber(number);
+        return new TextSecureAddress(e164number);
+    }
 }