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.asamk.signal.util.Util;
   7 import org.freedesktop.dbus.connections.impl.DBusConnection;
   8 import org.freedesktop.dbus.exceptions.DBusException;
   9 import org.freedesktop.dbus.types.Variant;
  10 import org.whispersystems.signalservice.api.messages.SignalServiceAttachment;
  11 import org.whispersystems.signalservice.api.messages.SignalServiceContent;
  12 import org.whispersystems.signalservice.api.messages.SignalServiceDataMessage;
  13 import org.whispersystems.signalservice.api.messages.SignalServiceEnvelope;
  14 import org.whispersystems.signalservice.api.messages.SignalServiceGroup;
  15 import org.whispersystems.signalservice.api.push.SignalServiceAddress;
  16
  17 import java.io.Serializable;
  18 import java.util.ArrayList;
  19 import java.util.HashMap;
  20 import java.util.List;
  21 import java.util.Map;
  22 import java.util.stream.Collectors;
  23
  24 import static org.asamk.signal.util.Util.getLegacyIdentifier;
  25
  26 public class DbusReceiveMessageHandler implements Manager.ReceiveMessageHandler {
  27
  28     private final Manager m;
  29     private final DBusConnection conn;
  30     private final String objectPath;
  31
  32     public DbusReceiveMessageHandler(Manager m, DBusConnection conn, final String objectPath) {
  33         this.m = m;
  34         this.conn = conn;
  35         this.objectPath = objectPath;
  36     }
  37
  38     @Override
  39     public void handleMessage(SignalServiceEnvelope envelope, SignalServiceContent content, Throwable exception) {
  40         try {
  41             sendDbusMessages(envelope, content);
  42         } catch (DBusException e) {
  43             e.printStackTrace();
  44         }
  45     }
  46
  47     private void sendDbusMessages(
  48             final SignalServiceEnvelope envelope, final SignalServiceContent content
  49     ) throws DBusException {
  50         if (envelope.isReceipt()) {
  51             conn.sendMessage(new Signal.ReceiptReceived(objectPath, envelope.getTimestamp(),
  52                     // A receipt envelope always has a source address
  53                     getLegacyIdentifier(envelope.getSourceAddress())));
  54             conn.sendMessage(new Signal.ReceiptReceivedV2(objectPath, envelope.getTimestamp(),
  55                     // A receipt envelope always has a source address
  56                     getLegacyIdentifier(envelope.getSourceAddress()), "delivery", Map.of()));
  57         } else if (content != null) {
  58             final var sender = !envelope.isUnidentifiedSender() && envelope.hasSourceUuid()
  59                     ? envelope.getSourceAddress()
  60                     : content.getSender();
  61             final var senderString = getLegacyIdentifier(sender);
  62             if (content.getReceiptMessage().isPresent()) {
  63                 final var receiptMessage = content.getReceiptMessage().get();
  64                 final var type = switch (receiptMessage.getType()) {
  65                     case READ -> "read";
  66                     case VIEWED -> "viewed";
  67                     case DELIVERY -> "delivery";
  68                     case UNKNOWN -> "unknown";
  69                 };
  70                 for (long timestamp : receiptMessage.getTimestamps()) {
  71                     conn.sendMessage(new Signal.ReceiptReceived(objectPath, timestamp, senderString));
  72                     conn.sendMessage(new Signal.ReceiptReceivedV2(objectPath,
  73                             envelope.getTimestamp(),
  74                             senderString,
  75                             type,
  76                             Map.of()));
  77                 }
  78
  79             } else if (content.getDataMessage().isPresent()) {
  80                 var message = content.getDataMessage().get();
  81
  82                 var groupId = getGroupId(message);
  83                 if (!message.isEndSession() && (
  84                         groupId == null
  85                                 || message.getGroupContext().get().getGroupV1Type() == null
  86                                 || message.getGroupContext().get().getGroupV1Type() == SignalServiceGroup.Type.DELIVER
  87                 )) {
  88                     conn.sendMessage(new Signal.MessageReceived(objectPath,
  89                             message.getTimestamp(),
  90                             senderString,
  91                             groupId != null ? groupId : new byte[0],
  92                             message.getBody().or(""),
  93                             getAttachments(message)));
  94                     conn.sendMessage(new Signal.MessageReceivedV2(objectPath,
  95                             message.getTimestamp(),
  96                             senderString,
  97                             groupId != null ? groupId : new byte[0],
  98                             message.getBody().or(""),
  99                             getMessageExtras(message)));
 100                 }
 101             } else if (content.getSyncMessage().isPresent()) {
 102                 var sync_message = content.getSyncMessage().get();
 103                 if (sync_message.getSent().isPresent()) {
 104                     var transcript = sync_message.getSent().get();
 105
 106                     if (transcript.getDestination().isPresent() || transcript.getMessage()
 107                             .getGroupContext()
 108                             .isPresent()) {
 109                         var message = transcript.getMessage();
 110                         var groupId = getGroupId(message);
 111
 112                         conn.sendMessage(new Signal.SyncMessageReceived(objectPath,
 113                                 transcript.getTimestamp(),
 114                                 senderString,
 115                                 transcript.getDestination().transform(Util::getLegacyIdentifier).or(""),
 116                                 groupId != null ? groupId : new byte[0],
 117                                 message.getBody().or(""),
 118                                 getAttachments(message)));
 119                         conn.sendMessage(new Signal.SyncMessageReceivedV2(objectPath,
 120                                 transcript.getTimestamp(),
 121                                 senderString,
 122                                 transcript.getDestination().transform(Util::getLegacyIdentifier).or(""),
 123                                 groupId != null ? groupId : new byte[0],
 124                                 message.getBody().or(""),
 125                                 getMessageExtras(message)));
 126                     }
 127                 }
 128             }
 129         }
 130     }
 131
 132     private byte[] getGroupId(final SignalServiceDataMessage message) {
 133         return message.getGroupContext().isPresent() ? GroupUtils.getGroupId(message.getGroupContext().get())
 134                 .serialize() : null;
 135     }
 136
 137     private List<String> getAttachments(SignalServiceDataMessage message) {
 138         var attachments = new ArrayList<String>();
 139         if (message.getAttachments().isPresent()) {
 140             for (var attachment : message.getAttachments().get()) {
 141                 if (attachment.isPointer()) {
 142                     attachments.add(m.getAttachmentFile(attachment.asPointer().getRemoteId()).getAbsolutePath());
 143                 }
 144             }
 145         }
 146         return attachments;
 147     }
 148
 149     private HashMap<String, Variant<?>> getMessageExtras(SignalServiceDataMessage message) {
 150         var extras = new HashMap<String, Variant<?>>();
 151         if (message.getAttachments().isPresent()) {
 152             var attachments = message.getAttachments()
 153                     .get()
 154                     .stream()
 155                     .filter(SignalServiceAttachment::isPointer)
 156                     .map(a -> getAttachmentMap(m, a))
 157                     .collect(Collectors.toList());
 158             extras.put("attachments", new Variant<>(attachments, "aa{sv}"));
 159         }
 160         if (message.getMentions().isPresent()) {
 161             var mentions = message.getMentions()
 162                     .get()
 163                     .stream()
 164                     .map(mention -> getMentionMap(m, mention))
 165                     .collect(Collectors.toList());
 166             extras.put("mentions", new Variant<>(mentions, "aa{sv}"));
 167         }
 168         extras.put("expiresInSeconds", new Variant<>(message.getExpiresInSeconds()));
 169         if (message.getQuote().isPresent()) {
 170             extras.put("quote", new Variant<>(getQuoteMap(message.getQuote().get()), "a{sv}"));
 171         }
 172         if (message.getReaction().isPresent()) {
 173             final var reaction = message.getReaction().get();
 174             extras.put("reaction", new Variant<>(getReactionMap(reaction), "a{sv}"));
 175         }
 176         if (message.getRemoteDelete().isPresent()) {
 177             extras.put("remoteDelete",
 178                     new Variant<>(Map.of("timestamp", new Variant<>(message.getRemoteDelete())), "a{sv}"));
 179         }
 180         if (message.getSticker().isPresent()) {
 181             final var sticker = message.getSticker().get();
 182             extras.put("sticker", new Variant<>(getStickerMap(sticker), "a{sv}"));
 183         }
 184         extras.put("isViewOnce", new Variant<>(message.isViewOnce()));
 185         return extras;
 186     }
 187
 188     private Map<String, Variant<?>> getQuoteMap(final SignalServiceDataMessage.Quote quote) {
 189         return Map.of("id",
 190                 new Variant<>(quote.getId()),
 191                 "author",
 192                 new Variant<>(getLegacyIdentifier(m.resolveSignalServiceAddress(quote.getAuthor()))),
 193                 "text",
 194                 new Variant<>(quote.getText()));
 195     }
 196
 197     private Map<String, Variant<? extends Serializable>> getStickerMap(final SignalServiceDataMessage.Sticker sticker) {
 198         return Map.of("packId", new Variant<>(sticker.getPackId()), "stickerId", new Variant<>(sticker.getStickerId()));
 199     }
 200
 201     private Map<String, Variant<?>> getReactionMap(final SignalServiceDataMessage.Reaction reaction) {
 202         return Map.of("emoji",
 203                 new Variant<>(reaction.getEmoji()),
 204                 "targetAuthor",
 205                 new Variant<>(getLegacyIdentifier(m.resolveSignalServiceAddress(reaction.getTargetAuthor()))),
 206                 "targetSentTimestamp",
 207                 new Variant<>(reaction.getTargetSentTimestamp()),
 208                 "isRemove",
 209                 new Variant<>(reaction.isRemove()));
 210     }
 211
 212     private Map<String, Variant<?>> getAttachmentMap(final Manager m, final SignalServiceAttachment attachment) {
 213         final var a = attachment.asPointer();
 214         final var map = new HashMap<String, Variant<?>>();
 215         map.put("file", new Variant<>(m.getAttachmentFile(a.getRemoteId()).getAbsolutePath()));
 216         map.put("remoteId", new Variant<>(a.getRemoteId().toString()));
 217         map.put("isVoiceNote", new Variant<>(a.getVoiceNote()));
 218         map.put("isBorderless", new Variant<>(a.isBorderless()));
 219         map.put("isGif", new Variant<>(a.isGif()));
 220         if (a.getCaption().isPresent()) {
 221             map.put("caption", new Variant<>(a.getCaption().get()));
 222         }
 223         if (a.getFileName().isPresent()) {
 224             map.put("fileName", new Variant<>(a.getFileName().get()));
 225         }
 226         if (a.getSize().isPresent()) {
 227             map.put("size", new Variant<>(a.getSize().get()));
 228         }
 229         if (a.getWidth() > 0 || a.getHeight() > 0) {
 230             map.put("height", new Variant<>(a.getHeight()));
 231             map.put("width", new Variant<>(a.getWidth()));
 232         }
 233         return map;
 234     }
 235
 236     private Map<String, Variant<?>> getMentionMap(
 237             final Manager m, final SignalServiceDataMessage.Mention mention
 238     ) {
 239         return Map.of("recipient",
 240                 new Variant<>(getLegacyIdentifier(m.resolveSignalServiceAddress(new SignalServiceAddress(mention.getUuid())))),
 241                 "start",
 242                 new Variant<>(mention.getStart()),
 243                 "length",
 244                 new Variant<>(mention.getLength()));
 245     }
 246 }