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.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.multidevice.SignalServiceSyncMessage;
  13 import org.whispersystems.signalservice.api.messages.multidevice.SentTranscriptMessage;
  14 import org.whispersystems.signalservice.api.push.SignalServiceAddress;
  15
  16 import java.util.ArrayList;
  17 import java.util.List;
  18
  19 public class JsonDbusReceiveMessageHandler extends JsonReceiveMessageHandler {
  20
  21     private final DBusConnection conn;
  22
  23     private final String objectPath;
  24
  25     public JsonDbusReceiveMessageHandler(Manager m, DBusConnection conn, final String objectPath) {
  26         super(m);
  27         this.conn = conn;
  28         this.objectPath = objectPath;
  29     }
  30
  31     static void sendReceivedMessageToDbus(SignalServiceEnvelope envelope, SignalServiceContent content, DBusConnection conn, final String objectPath, Manager m) {
  32         if (envelope.isReceipt()) {
  33             try {
  34                 conn.sendSignal(new Signal.ReceiptReceived(
  35                         objectPath,
  36                         envelope.getTimestamp(),
  37                         !envelope.isUnidentifiedSender() && envelope.hasSource() ? envelope.getSourceE164().get() : content.getSender().getNumber().get()
  38                 ));
  39             } catch (DBusException e) {
  40                 e.printStackTrace();
  41             }
  42         } else if (content != null) {
  43             if (content.getDataMessage().isPresent()) {
  44                 SignalServiceDataMessage message = content.getDataMessage().get();
  45
  46                 if (message.getBody().isPresent())
  47                     System.out.println(message.getBody().get());
  48
  49                 if (!message.isEndSession() &&
  50                         !(message.getGroupContext().isPresent() &&
  51                                 message.getGroupContext().get().getGroupV1Type() != SignalServiceGroup.Type.DELIVER)) {
  52                     try {
  53                         conn.sendSignal(new Signal.MessageReceived(
  54                                 objectPath,
  55                                 message.getTimestamp(),
  56                                 envelope.isUnidentifiedSender() || !envelope.hasSource() ? content.getSender().getNumber().get() : envelope.getSourceE164().get(),
  57                                 message.getGroupContext().isPresent() && message.getGroupContext().get().getGroupV1().isPresent()
  58                                         ? message.getGroupContext().get().getGroupV1().get().getGroupId() : new byte[0],
  59                                 message.getBody().isPresent() ? message.getBody().get() : "",
  60                                 JsonDbusReceiveMessageHandler.getAttachments(message, m)));
  61                     } catch (DBusException e) {
  62                         e.printStackTrace();
  63                     }
  64                 }
  65             } else if (content.getSyncMessage().isPresent()) {
  66                 SignalServiceSyncMessage sync_message = content.getSyncMessage().get();
  67                 if (sync_message.getSent().isPresent()) {
  68                     SentTranscriptMessage transcript = sync_message.getSent().get();
  69
  70                     if (!envelope.isUnidentifiedSender() && envelope.hasSource() && (transcript.getDestination().isPresent() || transcript.getMessage().getGroupContext().isPresent())) {
  71                         SignalServiceDataMessage message = transcript.getMessage();
  72
  73                         try {
  74                             conn.sendSignal(new Signal.SyncMessageReceived(
  75                                     objectPath,
  76                                     transcript.getTimestamp(),
  77                                     envelope.getSourceAddress().getNumber().get(),
  78                                     transcript.getDestination().isPresent() ? transcript.getDestination().get().getNumber().get() : "",
  79                                     message.getGroupContext().isPresent() && message.getGroupContext().get().getGroupV1().isPresent()
  80                                             ? message.getGroupContext().get().getGroupV1().get().getGroupId() : new byte[0],
  81                                     message.getBody().isPresent() ? message.getBody().get() : "",
  82                                     JsonDbusReceiveMessageHandler.getAttachments(message, m)));
  83                         } catch (DBusException e) {
  84                             e.printStackTrace();
  85                         }
  86                     }
  87                 }
  88             }
  89         }
  90     }
  91
  92     static private List<String> getAttachments(SignalServiceDataMessage message, Manager m) {
  93         List<String> attachments = new ArrayList<>();
  94         if (message.getAttachments().isPresent()) {
  95             for (SignalServiceAttachment attachment : message.getAttachments().get()) {
  96                 if (attachment.isPointer()) {
  97                     attachments.add(m.getAttachmentFile(attachment.asPointer().getId()).getAbsolutePath());
  98                 }
  99             }
 100         }
 101         return attachments;
 102     }
 103
 104     @Override
 105     public void handleMessage(SignalServiceEnvelope envelope, SignalServiceContent content, Throwable exception) {
 106         super.handleMessage(envelope, content, exception);
 107
 108         sendReceivedMessageToDbus(envelope, content, conn, objectPath, m);
 109     }
 110 }