diff options
Diffstat (limited to 'slack/commands.py')
-rw-r--r-- | slack/commands.py | 367 |
1 files changed, 10 insertions, 357 deletions
diff --git a/slack/commands.py b/slack/commands.py index e066fa4..849a346 100644 --- a/slack/commands.py +++ b/slack/commands.py @@ -5,7 +5,6 @@ import pprint import re from dataclasses import dataclass from functools import wraps -from itertools import chain from typing import ( TYPE_CHECKING, Any, @@ -24,15 +23,15 @@ import weechat from slack.error import SlackError, SlackRtmError, UncaughtError from slack.log import open_debug_buffer, print_error -from slack.python_compatibility import format_exception, removeprefix, removesuffix -from slack.shared import MESSAGE_ID_REGEX_STRING, REACTION_CHANGE_REGEX_STRING, shared +from slack.python_compatibility import format_exception, removeprefix +from slack.shared import shared from slack.slack_buffer import SlackBuffer from slack.slack_conversation import SlackConversation from slack.slack_thread import SlackThread -from slack.slack_user import SlackUser, get_user_nick, name_from_user_info +from slack.slack_user import SlackUser from slack.slack_workspace import SlackWorkspace -from slack.task import Future, gather, run_async, sleep -from slack.util import get_callback_name, with_color +from slack.task import gather, run_async, sleep +from slack.util import get_callback_name, get_resolved_futures, with_color from slack.weechat_config import WeeChatOption, WeeChatOptionTypes if TYPE_CHECKING: @@ -46,16 +45,6 @@ if TYPE_CHECKING: T = TypeVar("T") -REACTION_PREFIX_REGEX_STRING = ( - rf"{MESSAGE_ID_REGEX_STRING}?{REACTION_CHANGE_REGEX_STRING}" -) - -commands: Dict[str, Command] = {} - - -def get_resolved_futures(futures: Iterable[Future[T]]) -> List[T]: - return [future.result() for future in futures if future.done_with_result()] - # def parse_help_docstring(cmd): # doc = textwrap.dedent(cmd.__doc__).strip().split("\n", 1) @@ -114,7 +103,7 @@ def weechat_command( run_async(result) return - commands[cmd] = Command(cmd, top_level, "", "", "", completion, wrapper) + shared.commands[cmd] = Command(cmd, top_level, "", "", "", completion, wrapper) return wrapper @@ -546,16 +535,6 @@ async def command_slack_status(buffer: str, args: List[str], options: Options): print_error("Run the command in a slack buffer") -def completion_slack_workspaces_cb( - data: str, completion_item: str, buffer: str, completion: str -) -> int: - for workspace_name in shared.workspaces: - weechat.completion_list_add( - completion, workspace_name, 0, weechat.WEECHAT_LIST_POS_SORT - ) - return weechat.WEECHAT_RC_OK - - def find_command(start_cmd: str, args: str) -> Optional[Tuple[Command, str]]: args_parts = re.finditer("[^ ]+", args) cmd = start_cmd @@ -563,7 +542,7 @@ def find_command(start_cmd: str, args: str) -> Optional[Tuple[Command, str]]: for part in args_parts: next_cmd = f"{cmd} {part.group(0)}" - if next_cmd not in commands: + if next_cmd not in shared.commands: cmd_args_startpos = part.start(0) break cmd = next_cmd @@ -571,8 +550,8 @@ def find_command(start_cmd: str, args: str) -> Optional[Tuple[Command, str]]: cmd_args_startpos = len(args) cmd_args = args[cmd_args_startpos:] - if cmd in commands: - return commands[cmd], cmd_args + if cmd in shared.commands: + return shared.commands[cmd], cmd_args return None @@ -590,258 +569,6 @@ def command_cb(data: str, buffer: str, args: str) -> int: return weechat.WEECHAT_RC_OK -def completion_list_add_expand( - completion: str, word: str, nick_completion: int, where: str, buffer: str -): - if word == "%(slack_workspaces)": - completion_slack_workspaces_cb("", "slack_workspaces", buffer, completion) - elif word == "%(nicks)": - completion_nicks_cb("", "nicks", buffer, completion) - elif word == "%(threads)": - completion_thread_hashes_cb("", "threads", buffer, completion) - else: - weechat.completion_list_add(completion, word, nick_completion, where) - - -def completion_slack_workspace_commands_cb( - data: str, completion_item: str, buffer: str, completion: str -) -> int: - base_command = weechat.completion_get_string(completion, "base_command") - base_word = weechat.completion_get_string(completion, "base_word") - args = weechat.completion_get_string(completion, "args") - args_without_base_word = removesuffix(args, base_word) - - found_cmd_with_args = find_command(base_command, args_without_base_word) - if found_cmd_with_args: - command = found_cmd_with_args[0] - matching_cmds = [ - removeprefix(cmd, command.cmd).lstrip() - for cmd in commands - if cmd.startswith(command.cmd) and cmd != command.cmd - ] - if len(matching_cmds) > 1: - for match in matching_cmds: - cmd_arg = match.split(" ") - completion_list_add_expand( - completion, cmd_arg[0], 0, weechat.WEECHAT_LIST_POS_SORT, buffer - ) - else: - for arg in command.completion.split("|"): - completion_list_add_expand( - completion, arg, 0, weechat.WEECHAT_LIST_POS_SORT, buffer - ) - - return weechat.WEECHAT_RC_OK - - -def completion_emojis_cb( - data: str, completion_item: str, buffer: str, completion: str -) -> int: - slack_buffer = shared.buffers.get(buffer) - if slack_buffer is None: - return weechat.WEECHAT_RC_OK - - base_word = weechat.completion_get_string(completion, "base_word") - reaction = re.match(REACTION_PREFIX_REGEX_STRING + ":", base_word) - prefix = reaction.group(0) if reaction else ":" - - emoji_names = chain( - shared.standard_emojis.keys(), slack_buffer.workspace.custom_emojis.keys() - ) - for emoji_name in emoji_names: - if "::skin-tone-" not in emoji_name: - weechat.completion_list_add( - completion, - f"{prefix}{emoji_name}:", - 0, - weechat.WEECHAT_LIST_POS_SORT, - ) - return weechat.WEECHAT_RC_OK - - -def completion_slack_channels_cb( - data: str, completion_item: str, buffer: str, completion: str -) -> int: - slack_buffer = shared.buffers.get(buffer) - if slack_buffer is None: - return weechat.WEECHAT_RC_OK - - conversations = slack_buffer.workspace.open_conversations.values() - for conversation in conversations: - if conversation.buffer_type == "channel": - weechat.completion_list_add( - completion, - conversation.name_with_prefix("short_name_without_padding"), - 0, - weechat.WEECHAT_LIST_POS_SORT, - ) - return weechat.WEECHAT_RC_OK - - -def completion_nicks_cb( - data: str, completion_item: str, buffer: str, completion: str -) -> int: - slack_buffer = shared.buffers.get(buffer) - if slack_buffer is None: - return weechat.WEECHAT_RC_OK - - all_users = get_resolved_futures(slack_buffer.workspace.users.values()) - all_nicks = sorted([user.nick.raw_nick for user in all_users], key=str.casefold) - for nick in all_nicks: - weechat.completion_list_add( - completion, - f"@{nick}", - 1, - weechat.WEECHAT_LIST_POS_END, - ) - weechat.completion_list_add( - completion, - nick, - 1, - weechat.WEECHAT_LIST_POS_END, - ) - - buffer_nicks = sorted( - [nick.raw_nick for nick in slack_buffer.members], key=str.casefold, reverse=True - ) - for nick in buffer_nicks: - weechat.completion_list_add( - completion, - nick, - 1, - weechat.WEECHAT_LIST_POS_BEGINNING, - ) - weechat.completion_list_add( - completion, - f"@{nick}", - 1, - weechat.WEECHAT_LIST_POS_BEGINNING, - ) - - senders = [ - m.sender_user_id - for m in slack_buffer.messages.values() - if m.sender_user_id and m.subtype in [None, "me_message", "thread_broadcast"] - ] - unique_senders = list(dict.fromkeys(senders)) - sender_users = get_resolved_futures( - [slack_buffer.workspace.users[sender] for sender in unique_senders] - ) - nicks = [user.nick.raw_nick for user in sender_users] - for nick in nicks: - weechat.completion_list_add( - completion, - nick, - 1, - weechat.WEECHAT_LIST_POS_BEGINNING, - ) - weechat.completion_list_add( - completion, - f"@{nick}", - 1, - weechat.WEECHAT_LIST_POS_BEGINNING, - ) - - my_user_nick = slack_buffer.workspace.my_user.nick.raw_nick - weechat.completion_list_add( - completion, - f"@{my_user_nick}", - 1, - weechat.WEECHAT_LIST_POS_END, - ) - weechat.completion_list_add( - completion, - my_user_nick, - 1, - weechat.WEECHAT_LIST_POS_END, - ) - - return weechat.WEECHAT_RC_OK - - -def completion_thread_hashes_cb( - data: str, completion_item: str, buffer: str, completion: str -) -> int: - slack_buffer = shared.buffers.get(buffer) - if not isinstance(slack_buffer, SlackConversation): - return weechat.WEECHAT_RC_OK - - message_tss = sorted(slack_buffer.message_hashes.keys()) - messages = [slack_buffer.messages.get(ts) for ts in message_tss] - thread_messages = [ - message - for message in messages - if message is not None and message.is_thread_parent - ] - for message in thread_messages: - weechat.completion_list_add( - completion, message.hash, 0, weechat.WEECHAT_LIST_POS_BEGINNING - ) - for message in thread_messages: - weechat.completion_list_add( - completion, f"${message.hash}", 0, weechat.WEECHAT_LIST_POS_BEGINNING - ) - return weechat.WEECHAT_RC_OK - - -def complete_input(buffer: str, slack_buffer: SlackBuffer, query: str): - if ( - slack_buffer.completion_context == "ACTIVE_COMPLETION" - and slack_buffer.completion_values - ): - input_value = weechat.buffer_get_string(buffer, "input") - input_pos = weechat.buffer_get_integer(buffer, "input_pos") - result = slack_buffer.completion_values[slack_buffer.completion_index] - input_before = removesuffix(input_value[:input_pos], query) - input_after = input_value[input_pos:] - new_input = input_before + result + input_after - new_pos = input_pos - len(query) + len(result) - - with slack_buffer.completing(): - weechat.buffer_set(buffer, "input", new_input) - weechat.buffer_set(buffer, "input_pos", str(new_pos)) - - -def nick_suffix(): - return weechat.config_string( - weechat.config_get("weechat.completion.nick_completer") - ) - - -async def complete_user_next( - buffer: str, slack_buffer: SlackBuffer, query: str, is_first_word: bool -): - if slack_buffer.completion_context == "NO_COMPLETION": - slack_buffer.completion_context = "PENDING_COMPLETION" - search = await slack_buffer.workspace.api.edgeapi.fetch_users_search(query) - if slack_buffer.completion_context != "PENDING_COMPLETION": - return - slack_buffer.completion_context = "ACTIVE_COMPLETION" - suffix = nick_suffix() if is_first_word else " " - slack_buffer.completion_values = [ - get_user_nick(name_from_user_info(slack_buffer.workspace, user)).raw_nick - + suffix - for user in search["results"] - ] - slack_buffer.completion_index = 0 - elif slack_buffer.completion_context == "ACTIVE_COMPLETION": - slack_buffer.completion_index += 1 - if slack_buffer.completion_index >= len(slack_buffer.completion_values): - slack_buffer.completion_index = 0 - - complete_input(buffer, slack_buffer, query) - - -def complete_previous(buffer: str, slack_buffer: SlackBuffer, query: str) -> int: - if slack_buffer.completion_context == "ACTIVE_COMPLETION": - slack_buffer.completion_index -= 1 - if slack_buffer.completion_index < 0: - slack_buffer.completion_index = len(slack_buffer.completion_values) - 1 - complete_input(buffer, slack_buffer, query) - return weechat.WEECHAT_RC_OK_EAT - return weechat.WEECHAT_RC_OK - - async def mark_read(slack_buffer: SlackBuffer): # Sleep so the read marker is updated before we run slack_buffer.mark_read await sleep(1) @@ -855,41 +582,7 @@ def buffer_set_unread_cb(data: str, buffer: str, command: str) -> int: return weechat.WEECHAT_RC_OK -def input_complete_cb(data: str, buffer: str, command: str) -> int: - slack_buffer = shared.buffers.get(buffer) - if slack_buffer: - input_value = weechat.buffer_get_string(buffer, "input") - input_pos = weechat.buffer_get_integer(buffer, "input_pos") - input_before_cursor = input_value[:input_pos] - - word_index = ( - -2 if slack_buffer.completion_context == "ACTIVE_COMPLETION" else -1 - ) - word_until_cursor = " ".join(input_before_cursor.split(" ")[word_index:]) - - if word_until_cursor.startswith("@"): - query = word_until_cursor[1:] - is_first_word = word_until_cursor == input_before_cursor - - if command == "/input complete_next": - run_async( - complete_user_next(buffer, slack_buffer, query, is_first_word) - ) - return weechat.WEECHAT_RC_OK_EAT - else: - return complete_previous(buffer, slack_buffer, query) - return weechat.WEECHAT_RC_OK - - def register_commands(): - if shared.weechat_version < 0x02090000: - weechat.completion_get_string = ( - weechat.hook_completion_get_string # pyright: ignore [reportUnknownMemberType, reportGeneralTypeIssues] - ) - weechat.completion_list_add = ( - weechat.hook_completion_list_add # pyright: ignore [reportUnknownMemberType, reportGeneralTypeIssues] - ) - weechat.hook_command_run( "/buffer set unread", get_callback_name(buffer_set_unread_cb), "" ) @@ -899,48 +592,8 @@ def register_commands(): weechat.hook_command_run( "/input set_unread_current_buffer", get_callback_name(buffer_set_unread_cb), "" ) - # Disable until working properly - # weechat.hook_command_run( - # "/input complete_*", get_callback_name(input_complete_cb), "" - # ) - weechat.hook_completion( - "slack_workspaces", - "Slack workspaces (internal names)", - get_callback_name(completion_slack_workspaces_cb), - "", - ) - weechat.hook_completion( - "slack_commands", - "completions for Slack commands", - get_callback_name(completion_slack_workspace_commands_cb), - "", - ) - weechat.hook_completion( - "slack_channels", - "conversations in the current Slack workspace", - get_callback_name(completion_slack_channels_cb), - "", - ) - weechat.hook_completion( - "slack_emojis", - "Emoji names known to Slack", - get_callback_name(completion_emojis_cb), - "", - ) - weechat.hook_completion( - "nicks", - "nicks in the current Slack buffer", - get_callback_name(completion_nicks_cb), - "", - ) - weechat.hook_completion( - "threads", - "complete thread ids for slack", - get_callback_name(completion_thread_hashes_cb), - "", - ) - for cmd, command in commands.items(): + for cmd, command in shared.commands.items(): if command.top_level: weechat.hook_command( command.cmd, |