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                 if (!message.isEndSession() &&
  65                         !(message.getGroupContext().isPresent() &&
  66                                 message.getGroupContext().get().getGroupV1Type() != SignalServiceGroup.Type.DELIVER)) {
  67                     try {
  68                         conn.sendMessage(new Signal.MessageReceived(
  69                                 objectPath,
  70                                 message.getTimestamp(),
  71                                 sender.getLegacyIdentifier(),
  72                                 message.getGroupContext().isPresent() && message.getGroupContext().get().getGroupV1().isPresent()
  73                                         ? message.getGroupContext().get().getGroupV1().get().getGroupId() : new byte[0],
  74                                 message.getBody().isPresent() ? message.getBody().get() : "",
  75                                 JsonDbusReceiveMessageHandler.getAttachments(message, m)));
  76                     } catch (DBusException e) {
  77                         e.printStackTrace();
  78                     }
  79                 }
  80             } else if (content.getSyncMessage().isPresent()) {
  81                 SignalServiceSyncMessage sync_message = content.getSyncMessage().get();
  82                 if (sync_message.getSent().isPresent()) {
  83                     SentTranscriptMessage transcript = sync_message.getSent().get();
  84
  85                     if (transcript.getDestination().isPresent() || transcript.getMessage().getGroupContext().isPresent()) {
  86                         SignalServiceDataMessage message = transcript.getMessage();
  87
  88                         try {
  89                             conn.sendMessage(new Signal.SyncMessageReceived(
  90                                     objectPath,
  91                                     transcript.getTimestamp(),
  92                                     sender.getLegacyIdentifier(),
  93                                     transcript.getDestination().isPresent() ? transcript.getDestination().get().getLegacyIdentifier() : "",
  94                                     message.getGroupContext().isPresent() && message.getGroupContext().get().getGroupV1().isPresent()
  95                                             ? message.getGroupContext().get().getGroupV1().get().getGroupId() : new byte[0],
  96                                     message.getBody().isPresent() ? message.getBody().get() : "",
  97                                     JsonDbusReceiveMessageHandler.getAttachments(message, m)));
  98                         } catch (DBusException e) {
  99                             e.printStackTrace();
 100                         }
 101                     }
 102                 }
 103             }
 104         }
 105     }
 106
 107     static private List<String> getAttachments(SignalServiceDataMessage message, Manager m) {
 108         List<String> attachments = new ArrayList<>();
 109         if (message.getAttachments().isPresent()) {
 110             for (SignalServiceAttachment attachment : message.getAttachments().get()) {
 111                 if (attachment.isPointer()) {
 112                     attachments.add(m.getAttachmentFile(attachment.asPointer().getRemoteId()).getAbsolutePath());
 113                 }
 114             }
 115         }
 116         return attachments;
 117     }
 118
 119     @Override
 120     public void handleMessage(SignalServiceEnvelope envelope, SignalServiceContent content, Throwable exception) {
 121         super.handleMessage(envelope, content, exception);
 122
 123         sendReceivedMessageToDbus(envelope, content, conn, objectPath, m);
 124     }
 125 }