Show a loading indicator in the input

1 files changed, 28 insertions, 10 deletions

diff --git a/slack/init.py b/slack/init.py
index 3964a51..e8e8ec3 100644
--- a/slack/init.py
+++ b/slack/init.py
@@ -1,10 +1,11 @@
 import weechat
 
+from slack.api import get_conversation_from_buffer_pointer
 from slack.commands import register_commands
 from slack.config import SlackConfig
 from slack.shared import shared
 from slack.task import create_task, sleep
-from slack.util import get_callback_name
+from slack.util import get_callback_name, with_color
 
 SCRIPT_AUTHOR = "Trygve Aaberge <trygveaa@gmail.com>"
 SCRIPT_LICENSE = "MIT"
@@ -17,18 +18,30 @@ def shutdown_cb():
     return weechat.WEECHAT_RC_OK
 
 
-def signal_buffer_switch_cb(data: str, signal: str, signal_data: str) -> int:
-    current_conversation = None
-    for workspace in shared.workspaces.values():
-        conversation = workspace.get_conversation_from_buffer_pointer(signal_data)
-        if conversation:
-            current_conversation = conversation
-
-    if current_conversation:
-        create_task(current_conversation.fill_history())
+def signal_buffer_switch_cb(data: str, signal: str, buffer_pointer: str) -> int:
+    conversation = get_conversation_from_buffer_pointer(buffer_pointer)
+    if conversation:
+        create_task(conversation.fill_history())
     return weechat.WEECHAT_RC_OK
 
 
+def modifier_input_text_display_with_cursor_cb(
+    data: str, modifier: str, buffer_pointer: str, string: str
+) -> str:
+    conversation = get_conversation_from_buffer_pointer(buffer_pointer)
+    if conversation and conversation.loading:
+        input_delim_color = weechat.config_string(
+            weechat.config_get("weechat.bar.input.color_delim")
+        )
+        return (
+            f"{with_color(input_delim_color, '[')}"
+            f"{with_color(shared.config.color.loading.value, 'loading')}"
+            f"{with_color(input_delim_color, ']')}"
+            f" {string}"
+        )
+    return string
+
+
 async def init():
     auto_connect = weechat.info_get("auto_connect", "") == "1"
     if auto_connect:
@@ -60,5 +73,10 @@ def main():
         weechat.hook_signal(
             "window_switch", get_callback_name(signal_buffer_switch_cb), ""
         )
+        weechat.hook_modifier(
+            "input_text_display_with_cursor",
+            get_callback_name(modifier_input_text_display_with_cursor_cb),
+            "",
+        )
 
         create_task(init(), final=True)