Support showing thread replies in the channel

1 files changed, 33 insertions, 1 deletions

diff --git a/slack/slack_message.py b/slack/slack_message.py
index 3537e1d..1d61950 100644
--- a/slack/slack_message.py
+++ b/slack/slack_message.py
@@ -107,6 +107,7 @@ class SlackMessage:
         self._rendered_message = None
         self.conversation = conversation
         self.ts = SlackTs(message_json["ts"])
+        self.replies: List[SlackMessage] = []
         self._deleted = False
 
     @property
@@ -118,6 +119,28 @@ class SlackMessage:
         return self.conversation.message_hashes[self.ts]
 
     @property
+    def thread_ts(self) -> Optional[SlackTs]:
+        return (
+            SlackTs(self._message_json["thread_ts"])
+            if "thread_ts" in self._message_json
+            else None
+        )
+
+    @property
+    def is_thread_parent(self) -> bool:
+        return self.thread_ts == self.ts
+
+    @property
+    def is_reply(self) -> bool:
+        return self.thread_ts is not None and not self.is_thread_parent
+
+    @property
+    def parent_message(self) -> Optional[SlackMessage]:
+        if not self.is_reply or self.thread_ts is None:
+            return None
+        return self.conversation.get_message(self.thread_ts)
+
+    @property
     def is_bot_message(self) -> bool:
         return (
             "subtype" in self._message_json
@@ -341,6 +364,7 @@ class SlackMessage:
         if self._deleted:
             return with_color(shared.config.color.deleted_message.value, "(deleted)")
         else:
+            thread_prefix = self._create_thread_prefix()
             text = await self._render_message_text()
             text_edited = (
                 f" {with_color(shared.config.color.edited_message_suffix.value, '(edited)')}"
@@ -349,7 +373,7 @@ class SlackMessage:
             )
             reactions = await self._create_reactions_string()
             thread = self._create_thread_string()
-            return text + text_edited + reactions + thread
+            return thread_prefix + text + text_edited + reactions + thread
 
     async def render_message(self, rerender: bool = False) -> str:
         if self._rendered_message is not None and not rerender:
@@ -517,6 +541,14 @@ class SlackMessage:
         else:
             return ""
 
+    def _create_thread_prefix(self) -> str:
+        parent_message = self.parent_message
+        if not parent_message:
+            return ""
+
+        text = f"[{parent_message.hash}]"
+        return with_color(nick_color(str(parent_message.hash)), text) + " "
+
     def _create_thread_string(self) -> str:
         if "reply_count" not in self._message_json:
             return ""