views: Show SpoilerButton in message info view if present.

This commit wraps-up the PR by appending the SpoilerButton to message info view if present. Fixes zulip#688. Co-authored-by: Preet Mishra <[email protected]>
Ezio-Sarthak · Jul 3, 2021 · 8c5e15a · 8c5e15a
1 parent 80e21b6
commit 8c5e15a
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/zulipterminal/ui_tools/views.py b/zulipterminal/ui_tools/views.py
@@ -32,6 +32,7 @@
     MentionedButton,
     MessageLinkButton,
     PMButton,
+    SpoilerButton,
     StarredButton,
     StreamButton,
     TopicButton,
@@ -1338,6 +1339,8 @@ def __init__(
             msg_info.append(("Message Links", []))
         if time_mentions:
             msg_info.append(("Time mentions", time_mentions))
+        if spoilers:
+            msg_info.append(("Spoilers", []))
         if msg["reactions"]:
             reactions = sorted(
                 (reaction["emoji_name"], reaction["user"]["full_name"])
@@ -1388,6 +1391,25 @@ def __init__(
             )
             popup_width = max(popup_width, message_link_width)
 
+        if spoilers:
+            spoiler_buttons = []
+            spoiler_width = 0
+            for index, (header_len, header, content) in enumerate(spoilers):
+                spoiler_width = max(header_len, spoiler_width)
+                display_attr = None if index % 2 else "popup_contrast"
+                spoiler_buttons.append(
+                    SpoilerButton(controller, header_len, header, content, display_attr)
+                )
+
+            # slice_index = Number of labels before message links + 1 newline
+            #               + 1 'Spoilers' category label.
+            slice_index = len(msg_info[0][1]) + 2
+            slice_index += sum([len(w) + 2 for w in button_widgets])
+            button_widgets.append(spoiler_buttons)
+
+            widgets = widgets[:slice_index] + spoiler_buttons + widgets[slice_index:]
+            popup_width = max(popup_width, spoiler_width)
+
         super().__init__(controller, widgets, "MSG_INFO", popup_width, title)
 
     @staticmethod