from __future__ import annotations
from typing import Union
import weechat
from slack.log import print_error
from slack.shared import shared
from slack.slack_workspace import SlackWorkspace
from slack.util import get_callback_name
from slack.weechat_config import (
WeeChatColor,
WeeChatConfig,
WeeChatOption,
WeeChatOptionType,
WeeChatSection,
)
class SlackConfigSectionColor:
def __init__(self, weechat_config: WeeChatConfig):
self._section = WeeChatSection(weechat_config, "color")
self.disconnected = WeeChatOption(
self._section,
"disconnected",
"text color for the disconnected text",
WeeChatColor("red"),
)
self.loading = WeeChatOption(
self._section,
"loading",
"text color for the loading text",
WeeChatColor("yellow"),
)
self.reaction_suffix = WeeChatOption(
self._section,
"reaction_suffix",
"text color for the [:wave:(@user)] suffix on messages that have"
" reactions attached to them.",
WeeChatColor("darkgray"),
)
self.user_mention_color = WeeChatOption(
self._section,
"user_mention_color",
"text color for mentioned user names in the chat",
WeeChatColor("blue"),
)
self.usergroup_mention_color = WeeChatOption(
self._section,
"usergroup_mention_color",
"text color for mentioned user group names in the chat",
WeeChatColor("blue"),
)
class SlackConfigSectionLook:
def __init__(self, weechat_config: WeeChatConfig):
self._section = WeeChatSection(weechat_config, "look")
self.bot_user_suffix = WeeChatOption(
self._section,
"bot_user_suffix",
"the suffix appended to nicks to indicate a bot",
" :]",
)
self.external_user_suffix = WeeChatOption(
self._section,
"external_user_suffix",
"the suffix appended to nicks to indicate external users",
"*",
)
class SlackConfigSectionWorkspace:
def __init__(
self,
section: WeeChatSection,
workspace_name: Union[str, None],
parent_config: Union[SlackConfigSectionWorkspace, None],
):
self._section = section
self._workspace_name = workspace_name
self._parent_config = parent_config
self.api_token = self._create_option(
"api_token",
"",
"",
)
self.api_cookies = self._create_option(
"api_cookies",
"",
"",
)
self.autoconnect = self._create_option(
"autoconnect",
"automatically connect to workspace when WeeChat is starting",
False,
)
self.network_timeout = self._create_option(
"network_timeout",
"timeout (in seconds) for network requests",
30,
)
self.use_real_names = self._create_option(
"use_real_names",
"use real names as the nicks for all users. When this is"
" false, display names will be used if set, with a fallback"
" to the real name if display name is not set",
False,
)
def _create_option(
self,
name: str,
description: str,
default_value: WeeChatOptionType,
min_value: Union[int, None] = None,
max_value: Union[int, None] = None,
string_values: Union[str, None] = None,
) -> WeeChatOption[WeeChatOptionType]:
if self._workspace_name:
option_name = f"{self._workspace_name}.{name}"
else:
option_name = name
if self._parent_config:
parent_option = getattr(self._parent_config, name, None)
else:
parent_option = None
return WeeChatOption(
self._section,
option_name,
description,
default_value,
min_value,
max_value,
string_values,
parent_option,
)
def config_section_workspace_read_cb(
data: str, config_file: str, section: str, option_name: str, value: Union[str, None]
) -> int:
option_split = option_name.split(".", 1)
if len(option_split) < 2:
return weechat.WEECHAT_CONFIG_OPTION_SET_ERROR
workspace_name, name = option_split
if not workspace_name or not name:
return weechat.WEECHAT_CONFIG_OPTION_SET_ERROR
if workspace_name not in shared.workspaces:
shared.workspaces[workspace_name] = SlackWorkspace(workspace_name)
option = getattr(shared.workspaces[workspace_name].config, name, None)
if option is None:
return weechat.WEECHAT_CONFIG_OPTION_SET_OPTION_NOT_FOUND
if not isinstance(option, WeeChatOption):
return weechat.WEECHAT_CONFIG_OPTION_SET_ERROR
if value is None or (
shared.weechat_version < 0x3080000
and value == ""
and option.weechat_type != "string"
):
rc = option.value_set_null()
else:
rc = option.value_set_as_str(value)
if rc == weechat.WEECHAT_CONFIG_OPTION_SET_ERROR:
print_error(f'error creating workspace option "{option_name}"')
return rc
def config_section_workspace_write_for_old_weechat_cb(
data: str, config_file: str, section_name: str
) -> int:
if not weechat.config_write_line(config_file, section_name, ""):
return weechat.WEECHAT_CONFIG_WRITE_ERROR
for workspace in shared.workspaces.values():
for option in vars(workspace.config).values():
if isinstance(option, WeeChatOption):
if (
option.weechat_type != "string"
or not weechat.config_option_is_null(
option._pointer # pyright: ignore [reportPrivateUsage]
)
):
if not weechat.config_write_option(
config_file,
option._pointer, # pyright: ignore [reportPrivateUsage]
):
return weechat.WEECHAT_CONFIG_WRITE_ERROR
return weechat.WEECHAT_CONFIG_WRITE_OK
class SlackConfig:
def __init__(self):
self.weechat_config = WeeChatConfig("slack")
self.color = SlackConfigSectionColor(self.weechat_config)
self.look = SlackConfigSectionLook(self.weechat_config)
self._section_workspace_default = WeeChatSection(
self.weechat_config, "workspace_default"
)
# WeeChat < 3.8 sends null as an empty string to callback_read, so in
# order to distinguish them, don't write the null values to the config
# See https://github.com/weechat/weechat/pull/1843
callback_write = (
get_callback_name(config_section_workspace_write_for_old_weechat_cb)
if shared.weechat_version < 0x3080000
else ""
)
self._section_workspace = WeeChatSection(
self.weechat_config,
"workspace",
callback_read=get_callback_name(config_section_workspace_read_cb),
callback_write=callback_write,
)
self._workspace_default = SlackConfigSectionWorkspace(
self._section_workspace_default, None, None
)
def config_read(self):
weechat.config_read(self.weechat_config.pointer)
def create_workspace_config(self, workspace_name: str):
if workspace_name in shared.workspaces:
raise Exception(
f"Failed to create workspace config, already exists: {workspace_name}"
)
return SlackConfigSectionWorkspace(
self._section_workspace, workspace_name, self._workspace_default
)
