from __future__ import annotations from typing import TYPE_CHECKING import weechat from slack.error import SlackError from slack.shared import shared from slack.slack_emoji import get_emoji from slack.util import with_color if TYPE_CHECKING: from slack_api.slack_bots_info import SlackBotInfo from slack_api.slack_usergroups_info import SlackUsergroupInfo from slack_api.slack_users_info import SlackUserInfo from slack.slack_workspace import SlackWorkspace def nick_color(nick: str) -> str: return weechat.info_get("nick_color_name", nick) # TODO: Probably need to do some mapping here based on the existing users, in case some has been changed to avoid duplicate names def _name_from_user_info(workspace: SlackWorkspace, info: SlackUserInfo) -> str: display_name = info["profile"].get("display_name") if display_name and not workspace.config.use_real_names: return display_name return info["profile"].get("display_name") or info.get("real_name") or info["name"] def name_from_user_info_without_spaces( workspace: SlackWorkspace, info: SlackUserInfo ) -> str: return _name_from_user_info(workspace, info).replace( " ", shared.config.look.replace_space_in_nicks_with.value ) def format_bot_nick(nick: str, colorize: bool = False, only_nick: bool = False) -> str: nick = nick.replace(" ", shared.config.look.replace_space_in_nicks_with.value) if colorize: nick = with_color(nick_color(nick), nick) if not only_nick: nick = nick + shared.config.look.bot_user_suffix.value return nick class SlackUser: def __init__(self, workspace: SlackWorkspace, info: SlackUserInfo): self.workspace = workspace self._info = info @classmethod async def create(cls, workspace: SlackWorkspace, id: str): info_response = await workspace.api.fetch_user_info(id) return cls(workspace, info_response["user"]) @property def id(self) -> str: return self._info["id"] @property def is_self(self) -> bool: return self.id == self.workspace.my_user.id @property def is_external(self) -> bool: return self._info["profile"]["team"] != self.workspace.id @property def status_text(self) -> str: return self._info["profile"].get("status_text", "") or "" @property def status_emoji(self) -> str: status_emoji = self._info["profile"].get("status_emoji") if not status_emoji: return "" return get_emoji(status_emoji.strip(":")) def nick(self, colorize: bool = False, only_nick: bool = False) -> str: nick = self._name_without_spaces() if colorize: nick = with_color(self.nick_color(), nick) if not only_nick and self.is_external: nick += shared.config.look.external_user_suffix.value return nick def _name_without_spaces(self) -> str: return name_from_user_info_without_spaces(self.workspace, self._info) def nick_color(self) -> str: if self.is_self: return weechat.config_string( weechat.config_get("weechat.color.chat_nick_self") ) return nick_color(self._name_without_spaces()) def update_info_json(self, info_json: SlackUserInfo): self._info.update(info_json) # pyright: ignore [reportGeneralTypeIssues] self._rendered_prefix = None self._rendered_message = None self._parsed_message = None class SlackBot: def __init__(self, workspace: SlackWorkspace, info: SlackBotInfo): self.workspace = workspace self._info = info @classmethod async def create(cls, workspace: SlackWorkspace, id: str): info_response = await workspace.api.fetch_bot_info(id) return cls(workspace, info_response["bot"]) def nick(self, colorize: bool = False, only_nick: bool = False) -> str: return format_bot_nick( self._info["name"], colorize=colorize, only_nick=only_nick ) def nick_color(self): return nick_color(self.nick(colorize=False, only_nick=True)) class SlackUsergroup: def __init__(self, workspace: SlackWorkspace, info: SlackUsergroupInfo): self.workspace = workspace self._info = info @classmethod async def create(cls, workspace: SlackWorkspace, id: str): info_response = await workspace.api.edgeapi.fetch_usergroups_info([id]) if not info_response["results"] or info_response["results"][0]["id"] != id: raise SlackError(workspace, "usergroup_not_found") return cls(workspace, info_response["results"][0]) def handle(self) -> str: return self._info["handle"]