Support adding and removing reactions

5 files changed, 89 insertions, 4 deletions

diff --git a/slack/register.py b/slack/register.py
index 23a4d55..b907c49 100644
--- a/slack/register.py
+++ b/slack/register.py
@@ -119,6 +119,9 @@ def register():
         shared.weechat_version = int(weechat.info_get("version_number", "") or 0)
         shared.current_buffer_pointer = weechat.current_buffer()
         shared.standard_emojis = load_standard_emojis()
+        shared.standard_emojis_inverse = {
+            value["unicode"]: value for value in shared.standard_emojis.values()
+        }
         shared.workspaces = {}
         shared.config = SlackConfig()
         shared.config.config_read()
diff --git a/slack/shared.py b/slack/shared.py
index eeb6df5..32bad27 100644
--- a/slack/shared.py
+++ b/slack/shared.py
@@ -29,6 +29,7 @@ class Shared:
         self.config: SlackConfig
         self.uncaught_errors: List[UncaughtError] = []
         self.standard_emojis: Dict[str, Emoji]
+        self.standard_emojis_inverse: Dict[str, Emoji]
         self.highlight_tag = "highlight"
 
 
diff --git a/slack/slack_api.py b/slack/slack_api.py
index 7fff321..784bea6 100644
--- a/slack/slack_api.py
+++ b/slack/slack_api.py
@@ -26,6 +26,7 @@ if TYPE_CHECKING:
     from slack_api.slack_users_prefs import SlackUsersPrefsGetResponse
     from slack_edgeapi.slack_usergroups_info import SlackEdgeUsergroupsInfoResponse
     from slack_edgeapi.slack_users_search import SlackUsersSearchResponse
+    from typing_extensions import Literal, assert_never
 
     from slack.slack_conversation import SlackConversation
     from slack.slack_workspace import SlackWorkspace
@@ -317,3 +318,27 @@ class SlackApi(SlackApiCommon):
         if response["ok"] is False:
             raise SlackApiError(self.workspace, method, response, params)
         return response
+
+    async def reactions_change(
+        self,
+        conversation: SlackConversation,
+        ts: SlackTs,
+        name: str,
+        change_type: Literal["+", "-"],
+    ):
+        method = (
+            "reactions.add"
+            if change_type == "+"
+            else "reactions.remove"
+            if change_type == "-"
+            else assert_never(change_type)
+        )
+        params: Params = {
+            "channel": conversation.id,
+            "timestamp": ts,
+            "name": name,
+        }
+        response: SlackGenericResponse = await self._fetch(method, params)
+        if response["ok"] is False:
+            raise SlackApiError(self.workspace, method, response, params)
+        return response
diff --git a/slack/slack_buffer.py b/slack/slack_buffer.py
index 4fa5981..7a6faee 100644
--- a/slack/slack_buffer.py
+++ b/slack/slack_buffer.py
@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import re
 import time
 from abc import ABC, abstractmethod
 from contextlib import contextmanager
@@ -7,6 +8,7 @@ from typing import TYPE_CHECKING, Dict, List, Mapping, Optional, Set, Tuple
 
 import weechat
 
+from slack.log import print_error
 from slack.shared import shared
 from slack.slack_message import SlackMessage, SlackTs
 from slack.task import run_async
@@ -19,6 +21,17 @@ if TYPE_CHECKING:
     from slack.slack_conversation import SlackConversation
     from slack.slack_workspace import SlackWorkspace
 
+MESSAGE_ID_REGEX_STRING = r"(?P<msg_id>\d+|\$[0-9a-z]{3,})"
+REACTION_PREFIX_REGEX_STRING = rf"{MESSAGE_ID_REGEX_STRING}?(?P<reaction_change>\+|-)"
+
+EMOJI_CHAR_REGEX_STRING = "(?P<emoji_char>[\U00000080-\U0010ffff]+)"
+EMOJI_NAME_REGEX_STRING = (
+    ":(?P<emoji_name>[a-z0-9_+-]+(?:::skin-tone-[2-6](?:-[2-6])?)?):"
+)
+EMOJI_CHAR_OR_NAME_REGEX_STRING = (
+    f"({EMOJI_CHAR_REGEX_STRING}|{EMOJI_NAME_REGEX_STRING})"
+)
+
 
 def hdata_line_ts(line_pointer: str) -> Optional[SlackTs]:
     data = weechat.hdata_pointer(weechat.hdata_get("line"), line_pointer, "data")
@@ -311,14 +324,57 @@ class SlackBuffer(ABC):
             weechat.buffer_set(self.buffer_pointer, "hotlist", "-1")
             self.hotlist_tss.clear()
 
+    def ts_from_hash(self, ts_hash: str) -> Optional[SlackTs]:
+        return self.conversation.message_hashes.get_ts(ts_hash)
+
+    def ts_from_index(self, index: int) -> Optional[SlackTs]:
+        lines = weechat.hdata_pointer(
+            weechat.hdata_get("buffer"), self.buffer_pointer, "lines"
+        )
+        line = weechat.hdata_pointer(weechat.hdata_get("lines"), lines, "last_line")
+        move = -index + 1
+        if move:
+            line = weechat.hdata_move(weechat.hdata_get("line"), line, move)
+        return hdata_line_ts(line)
+
+    def ts_from_hash_or_index(self, hash_or_index: str) -> Optional[SlackTs]:
+        if not hash_or_index:
+            return self.ts_from_index(1)
+        elif hash_or_index.isdigit():
+            return self.ts_from_index(int(hash_or_index))
+        else:
+            return self.ts_from_hash(hash_or_index)
+
     @abstractmethod
     async def post_message(self, text: str) -> None:
         raise NotImplementedError()
 
+    async def send_change_reaction(
+        self, ts: SlackTs, emoji_char: str, change_type: Literal["+", "-"]
+    ) -> None:
+        emoji = shared.standard_emojis_inverse.get(emoji_char)
+        emoji_name = emoji["name"] if emoji else emoji_char
+        await self._api.reactions_change(self.conversation, ts, emoji_name, change_type)
+
     async def process_input(self, input_data: str):
-        if input_data.startswith(("//", " ")):
-            input_data = input_data[1:]
-        await self.post_message(htmlescape(input_data))
+        reaction = re.match(
+            rf"{REACTION_PREFIX_REGEX_STRING}{EMOJI_CHAR_OR_NAME_REGEX_STRING}\s*$",
+            input_data,
+        )
+        if reaction:
+            msg_id = reaction.group("msg_id")
+            ts = self.ts_from_hash_or_index(msg_id)
+            if ts is None:
+                print_error(f"No slack message found for message id or index {msg_id}")
+                return
+            emoji = reaction.group("emoji_char") or reaction.group("emoji_name")
+            reaction_change_type = reaction.group("reaction_change")
+            if reaction_change_type == "+" or reaction_change_type == "-":
+                await self.send_change_reaction(ts, emoji, reaction_change_type)
+        else:
+            if input_data.startswith(("//", " ")):
+                input_data = input_data[1:]
+            await self.post_message(htmlescape(input_data))
 
     def _buffer_input_cb(self, data: str, buffer: str, input_data: str) -> int:
         run_async(self.process_input(input_data))
diff --git a/slack/slack_conversation.py b/slack/slack_conversation.py
index 652c8f2..2ac8f21 100644
--- a/slack/slack_conversation.py
+++ b/slack/slack_conversation.py
@@ -613,7 +613,7 @@ class SlackConversation(SlackBuffer):
                 )
 
     async def open_thread(self, thread_hash: str, switch: bool = False):
-        thread_ts = self.message_hashes.get_ts(thread_hash)
+        thread_ts = self.ts_from_hash(thread_hash)
         if thread_ts:
             thread_message = self.messages.get(thread_ts)
             if thread_message is None: