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

   1 package org.asamk.signal;
   2
   3 import org.asamk.Signal;
   4 import org.asamk.signal.manager.Manager;
   5 import org.asamk.signal.manager.groups.GroupUtils;
   6 import org.freedesktop.dbus.connections.impl.DBusConnection;
   7 import org.freedesktop.dbus.exceptions.DBusException;
   8 import org.whispersystems.signalservice.api.messages.SignalServiceContent;
   9 import org.whispersystems.signalservice.api.messages.SignalServiceDataMessage;
  10 import org.whispersystems.signalservice.api.messages.SignalServiceEnvelope;
  11 import org.whispersystems.signalservice.api.messages.SignalServiceGroup;
  12
  13 import java.util.ArrayList;
  14 import java.util.List;
  15
  16 import static org.asamk.signal.util.Util.getLegacyIdentifier;
  17
  18 public class DbusReceiveMessageHandler implements Manager.ReceiveMessageHandler {
  19
  20     private final Manager m;
  21     private final DBusConnection conn;
  22     private final String objectPath;
  23
  24     public DbusReceiveMessageHandler(Manager m, DBusConnection conn, final String objectPath) {
  25         this.m = m;
  26         this.conn = conn;
  27         this.objectPath = objectPath;
  28     }
  29
  30     @Override
  31     public void handleMessage(SignalServiceEnvelope envelope, SignalServiceContent content, Throwable exception) {
  32         if (envelope.isReceipt()) {
  33             try {
  34                 conn.sendMessage(new Signal.ReceiptReceived(objectPath, envelope.getTimestamp(),
  35                         // A receipt envelope always has a source address
  36                         getLegacyIdentifier(envelope.getSourceAddress())));
  37             } catch (DBusException e) {
  38                 e.printStackTrace();
  39             }
  40         } else if (content != null) {
  41             final var sender = !envelope.isUnidentifiedSender() && envelope.hasSourceUuid()
  42                     ? envelope.getSourceAddress()
  43                     : content.getSender();
  44             if (content.getReceiptMessage().isPresent()) {
  45                 final var receiptMessage = content.getReceiptMessage().get();
  46                 if (receiptMessage.isDeliveryReceipt()) {
  47                     for (long timestamp : receiptMessage.getTimestamps()) {
  48                         try {
  49                             conn.sendMessage(new Signal.ReceiptReceived(objectPath,
  50                                     timestamp,
  51                                     getLegacyIdentifier(sender)));
  52                         } catch (DBusException e) {
  53                             e.printStackTrace();
  54                         }
  55                     }
  56                 }
  57             } else if (content.getDataMessage().isPresent()) {
  58                 var message = content.getDataMessage().get();
  59
  60                 var groupId = getGroupId(message);
  61                 if (!message.isEndSession() && (
  62                         groupId == null
  63                                 || message.getGroupContext().get().getGroupV1Type() == null
  64                                 || message.getGroupContext().get().getGroupV1Type() == SignalServiceGroup.Type.DELIVER
  65                 )) {
  66                     try {
  67                         conn.sendMessage(new Signal.MessageReceived(objectPath,
  68                                 message.getTimestamp(),
  69                                 getLegacyIdentifier(sender),
  70                                 groupId != null ? groupId : new byte[0],
  71                                 message.getBody().isPresent() ? message.getBody().get() : "",
  72                                 getAttachments(message, m)));
  73                     } catch (DBusException e) {
  74                         e.printStackTrace();
  75                     }
  76                 }
  77             } else if (content.getSyncMessage().isPresent()) {
  78                 var sync_message = content.getSyncMessage().get();
  79                 if (sync_message.getSent().isPresent()) {
  80                     var transcript = sync_message.getSent().get();
  81
  82                     if (transcript.getDestination().isPresent() || transcript.getMessage()
  83                             .getGroupContext()
  84                             .isPresent()) {
  85                         var message = transcript.getMessage();
  86                         var groupId = getGroupId(message);
  87
  88                         try {
  89                             conn.sendMessage(new Signal.SyncMessageReceived(objectPath,
  90                                     transcript.getTimestamp(),
  91                                     getLegacyIdentifier(sender),
  92                                     transcript.getDestination().isPresent()
  93                                             ? getLegacyIdentifier(transcript.getDestination().get())
  94                                             : "",
  95                                     groupId != null ? groupId : new byte[0],
  96                                     message.getBody().isPresent() ? message.getBody().get() : "",
  97                                     getAttachments(message, m)));
  98                         } catch (DBusException e) {
  99                             e.printStackTrace();
 100                         }
 101                     }
 102                 }
 103             }
 104         }
 105     }
 106
 107     private static byte[] getGroupId(final SignalServiceDataMessage message) {
 108         return message.getGroupContext().isPresent() ? GroupUtils.getGroupId(message.getGroupContext().get())
 109                 .serialize() : null;
 110     }
 111
 112     static private List<String> getAttachments(SignalServiceDataMessage message, Manager m) {
 113         var attachments = new ArrayList<String>();
 114         if (message.getAttachments().isPresent()) {
 115             for (var attachment : message.getAttachments().get()) {
 116                 if (attachment.isPointer()) {
 117                     attachments.add(m.getAttachmentFile(attachment.asPointer().getRemoteId()).getAbsolutePath());
 118                 }
 119             }
 120         }
 121         return attachments;
 122     }
 123 }