From 2658a4b432110e5f3c9db290282e75bade06a17f Mon Sep 17 00:00:00 2001
From: kali <kalikaneko@riseup.net>
Date: Sat, 2 May 2020 21:27:58 +0200
Subject: add group info to message

---
 signald/main.py  | 3 ++-
 signald/types.py | 1 +
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/signald/main.py b/signald/main.py
index e82f3e7..d4ece1d 100644
--- a/signald/main.py
+++ b/signald/main.py
@@ -107,7 +107,7 @@ class Signal:
                 continue
 
             message = message["data"]
-            data_message = message["dataMessage"] if message.get("dataMessage") else {}
+            data_message = message.get("dataMessage", {})
 
             yield Message(
                 username=message["username"],
@@ -118,6 +118,7 @@ class Signal:
                 timestamp_iso=message["timestampISO"],
                 expiration_secs=data_message.get("expiresInSeconds"),
                 is_receipt=message["isReceipt"],
+                group_info=data_message.get("groupInfo", {}),
                 attachments=[
                     Attachment(
                         content_type=attachment["contentType"],
diff --git a/signald/types.py b/signald/types.py
index b1c4b51..1b605b3 100644
--- a/signald/types.py
+++ b/signald/types.py
@@ -21,3 +21,4 @@ class Message:
     is_receipt = attr.ib(type=bool, default=False)
     attachments = attr.ib(type=list, default=[])
     quote = attr.ib(type=str, default=None)
+    group_info = attr.ib(type=dict, default={})
-- 
cgit v1.2.3-54-g00ecf