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

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