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