Implement updateGroup

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

diff --git a/src/main/java/cli/Manager.java b/src/main/java/cli/Manager.java
index 2258c99d9f1b3a443a5f2aaeb028251aefde362d..a0fb5b01e67faded00d17d65b25aed6759dc17ae 100644 (file)
--- a/src/main/java/cli/Manager.java
+++ b/src/main/java/cli/Manager.java
@@ -44,8 +44,7 @@ import org.whispersystems.textsecure.api.util.InvalidNumberException;
 import org.whispersystems.textsecure.api.util.PhoneNumberFormatter;
 
 import java.io.*;
 import org.whispersystems.textsecure.api.util.PhoneNumberFormatter;
 
 import java.io.*;

-import java.util.LinkedList;
-import java.util.List;
+import java.util.*;

 import java.util.concurrent.TimeUnit;
 import java.util.concurrent.TimeoutException;
 
 import java.util.concurrent.TimeUnit;
 import java.util.concurrent.TimeoutException;
 


@@ -242,14 +241,32 @@ class Manager {
         accountManager.setPreKeys(axolotlStore.getIdentityKeyPair().getPublicKey(), lastResortKey, signedPreKeyRecord, oneTimePreKeys);
     }
 
         accountManager.setPreKeys(axolotlStore.getIdentityKeyPair().getPublicKey(), lastResortKey, signedPreKeyRecord, oneTimePreKeys);
     }
 

-    public void sendMessage(List<TextSecureAddress> recipients, TextSecureDataMessage message)
+    public void sendMessage(List<String> recipients, TextSecureDataMessage message)

             throws IOException, EncapsulatedExceptions {
         TextSecureMessageSender messageSender = new TextSecureMessageSender(URL, TRUST_STORE, username, password,
                 axolotlStore, USER_AGENT, Optional.<TextSecureMessageSender.EventListener>absent());
             throws IOException, EncapsulatedExceptions {
         TextSecureMessageSender messageSender = new TextSecureMessageSender(URL, TRUST_STORE, username, password,
                 axolotlStore, USER_AGENT, Optional.<TextSecureMessageSender.EventListener>absent());

-        messageSender.sendMessage(recipients, message);
+
+        Set<TextSecureAddress> recipientsTS = new HashSet<>(recipients.size());
+        for (String recipient : recipients) {
+            try {
+                recipientsTS.add(getPushAddress(recipient));
+            } catch (InvalidNumberException e) {
+                System.err.println("Failed to add recipient \"" + recipient + "\": " + e.getMessage());
+                System.err.println("Aborting sending.");
+                return;
+            }
+        }
+
+        messageSender.sendMessage(new ArrayList<>(recipientsTS), message);
+
+        if (message.isEndSession()) {
+            for (TextSecureAddress recipient : recipientsTS) {
+                handleEndSession(recipient.getNumber());
+            }
+        }

     }
 
     }
 

-    public TextSecureContent decryptMessage(TextSecureEnvelope envelope) {
+    private TextSecureContent decryptMessage(TextSecureEnvelope envelope) {

         TextSecureCipher cipher = new TextSecureCipher(new TextSecureAddress(username), axolotlStore);
         try {
             return cipher.decrypt(envelope);
         TextSecureCipher cipher = new TextSecureCipher(new TextSecureAddress(username), axolotlStore);
         try {
             return cipher.decrypt(envelope);


@@ -260,7 +277,7 @@ class Manager {
         }
     }
 
         }
     }
 

-    public void handleEndSession(String source) {
+    private void handleEndSession(String source) {

         axolotlStore.deleteAllSessions(source);
     }
 
         axolotlStore.deleteAllSessions(source);
     }
 


@@ -290,7 +307,32 @@ class Manager {
                                     TextSecureGroup groupInfo = message.getGroupInfo().get();
                                     switch (groupInfo.getType()) {
                                         case UPDATE:
                                     TextSecureGroup groupInfo = message.getGroupInfo().get();
                                     switch (groupInfo.getType()) {
                                         case UPDATE:

-                                            group = new GroupInfo(groupInfo.getGroupId(), groupInfo.getName().get(), groupInfo.getMembers().get(), groupInfo.getAvatar().get().asPointer().getId());
+                                            group = groupStore.getGroup(groupInfo.getGroupId());
+                                            if (group == null) {
+                                                group = new GroupInfo(groupInfo.getGroupId());
+                                            }
+
+                                            if (groupInfo.getAvatar().isPresent()) {
+                                                TextSecureAttachment avatar = groupInfo.getAvatar().get();
+                                                if (avatar.isPointer()) {
+                                                    long avatarId = avatar.asPointer().getId();
+                                                    try {
+                                                        retrieveAttachment(avatar.asPointer());
+                                                        group.avatarId = avatarId;
+                                                    } catch (IOException | InvalidMessageException e) {
+                                                        System.err.println("Failed to retrieve group avatar (" + avatarId + "): " + e.getMessage());
+                                                    }
+                                                }
+                                            }
+
+                                            if (groupInfo.getName().isPresent()) {
+                                                group.name = groupInfo.getName().get();
+                                            }
+
+                                            if (groupInfo.getMembers().isPresent()) {
+                                                group.members.addAll(groupInfo.getMembers().get());
+                                            }
+

                                             groupStore.updateGroup(group);
                                             break;
                                         case DELIVER:
                                             groupStore.updateGroup(group);
                                             break;
                                         case DELIVER:


@@ -304,6 +346,20 @@ class Manager {
                                             break;
                                     }
                                 }
                                             break;
                                     }
                                 }

+                                if (message.isEndSession()) {
+                                    handleEndSession(envelope.getSource());
+                                }
+                                if (message.getAttachments().isPresent()) {
+                                    for (TextSecureAttachment attachment : message.getAttachments().get()) {
+                                        if (attachment.isPointer()) {
+                                            try {
+                                                retrieveAttachment(attachment.asPointer());
+                                            } catch (IOException | InvalidMessageException e) {
+                                                System.err.println("Failed to retrieve attachment (" + attachment.asPointer().getId() + "): " + e.getMessage());
+                                            }
+                                        }
+                                    }
+                                }

                             }
                         }
                     }
                             }
                         }
                     }


@@ -322,14 +378,18 @@ class Manager {
         }
     }
 
         }
     }
 

-    public File retrieveAttachment(TextSecureAttachmentPointer pointer) throws IOException, InvalidMessageException {
+    public File getAttachmentFile(long attachmentId) {
+        return new File(attachmentsPath + "/" + attachmentId);
+    }
+
+    private File retrieveAttachment(TextSecureAttachmentPointer pointer) throws IOException, InvalidMessageException {

         final TextSecureMessageReceiver messageReceiver = new TextSecureMessageReceiver(URL, TRUST_STORE, username, password, signalingKey, USER_AGENT);
 
         File tmpFile = File.createTempFile("ts_attach_" + pointer.getId(), ".tmp");
         InputStream input = messageReceiver.retrieveAttachment(pointer, tmpFile);
 
         new File(attachmentsPath).mkdirs();
         final TextSecureMessageReceiver messageReceiver = new TextSecureMessageReceiver(URL, TRUST_STORE, username, password, signalingKey, USER_AGENT);
 
         File tmpFile = File.createTempFile("ts_attach_" + pointer.getId(), ".tmp");
         InputStream input = messageReceiver.retrieveAttachment(pointer, tmpFile);
 
         new File(attachmentsPath).mkdirs();

-        File outputFile = new File(attachmentsPath + "/" + pointer.getId());
+        File outputFile = getAttachmentFile(pointer.getId());

         OutputStream output = null;
         try {
             output = new FileOutputStream(outputFile);
         OutputStream output = null;
         try {
             output = new FileOutputStream(outputFile);


@@ -369,13 +429,25 @@ class Manager {
         return outputFile;
     }
 
         return outputFile;
     }
 

-    private String canonicalizeNumber(String number) throws InvalidNumberException {
+    public String canonicalizeNumber(String number) throws InvalidNumberException {

         String localNumber = username;
         return PhoneNumberFormatter.formatNumber(number, localNumber);
     }
 
         String localNumber = username;
         return PhoneNumberFormatter.formatNumber(number, localNumber);
     }
 

-    TextSecureAddress getPushAddress(String number) throws InvalidNumberException {
+    private TextSecureAddress getPushAddress(String number) throws InvalidNumberException {

         String e164number = canonicalizeNumber(number);
         return new TextSecureAddress(e164number);
     }
         String e164number = canonicalizeNumber(number);
         return new TextSecureAddress(e164number);
     }

+
+    public GroupInfo getGroupInfo(byte[] groupId) {
+        return groupStore.getGroup(groupId);
+    }
+
+    public void setGroupInfo(GroupInfo group) {
+        groupStore.updateGroup(group);
+    }
+
+    public String getUsername() {
+        return username;
+    }

 }
 }