upstream-zulip/zerver/lib/zcommand.py

from dataclasses import dataclass

from django.utils.translation import gettext as _

from zerver.actions.user_settings import do_change_user_setting
from zerver.lib.exceptions import JsonableError
from zerver.models import UserProfile


@dataclass
class ZCommandResult:
    msg: str = ""


def process_zcommands(content: str, user_profile: UserProfile) -> ZCommandResult:
    def change_mode_setting(
        setting_name: str, switch_command: str, setting: str, setting_value: int
    ) -> str:
        msg = f"Changed to {setting_name}! To revert {setting_name}, type `/{switch_command}`."
        do_change_user_setting(
            user_profile=user_profile,
            setting_name=setting,
            setting_value=setting_value,
            acting_user=user_profile,
        )
        return msg

    if not content.startswith("/"):
        raise JsonableError(_("There should be a leading slash in the zcommand."))
    command = content.removeprefix("/")

    if command == "ping":
        return ZCommandResult()
    elif command == "dark":
        if user_profile.color_scheme == UserProfile.COLOR_SCHEME_DARK:
            return ZCommandResult(msg="You are still in dark theme.")
        return ZCommandResult(
            msg=change_mode_setting(
                setting_name="dark theme",
                switch_command="light",
                setting="color_scheme",
                setting_value=UserProfile.COLOR_SCHEME_DARK,
            )
        )
    elif command == "light":
        if user_profile.color_scheme == UserProfile.COLOR_SCHEME_LIGHT:
            return ZCommandResult(msg="You are still in light theme.")
        return ZCommandResult(
            msg=change_mode_setting(
                setting_name="light theme",
                switch_command="dark",
                setting="color_scheme",
                setting_value=UserProfile.COLOR_SCHEME_LIGHT,
            )
        )
    elif command == "fluid-width":
        if user_profile.fluid_layout_width:
            return ZCommandResult(msg="You are still in fluid width mode.")
        return ZCommandResult(
            msg=change_mode_setting(
                setting_name="fluid-width mode",
                switch_command="fixed-width",
                setting="fluid_layout_width",
                setting_value=True,
            )
        )
    elif command == "fixed-width":
        if not user_profile.fluid_layout_width:
            return ZCommandResult(msg="You are still in fixed width mode.")
        return ZCommandResult(
            msg=change_mode_setting(
                setting_name="fixed-width mode",
                switch_command="fluid-width",
                setting="fluid_layout_width",
                setting_value=False,
            )
        )
    raise JsonableError(_("No such command: {command}").format(command=command))