from typing import Any

from talon import Context, Module, actions

from .command_client import NoFileServerException, NotSet, run_command

mod = Module()

ctx = Context()
linux_ctx = Context()

ctx.matches = r"""
app: vscode
"""
linux_ctx.matches = r"""
os: linux
app: vscode
"""

ctx.tags = ["user.command_client"]


def command_server_or_client_fallback(command_id: str, wait: bool):
    """Execute command via command server, falling back to command palette if directory not present."""
    try:
        run_command(command_id, wait_for_finish=wait)
    except NoFileServerException:
        actions.user.command_palette()
        actions.user.paste(command_id)
        actions.key("enter")
        print(
            "Command server directory not found; falling back to command palette.  For better performance, install the VSCode extension for Talon: https://marketplace.visualstudio.com/items?itemName=pokey.talon"
        )


@ctx.action_class("user")
class VsCodeAction:
    def command_server_directory() -> str:
        return "vscode-command-server"


@mod.action_class
class Actions:
    def vscode(command_id: str):
        """Execute command via vscode command server, if available, or fallback
        to command palette."""
        command_server_or_client_fallback(command_id, False)

    def vscode_and_wait(command_id: str):
        """Execute command via vscode command server, if available, and wait
        for command to finish.  If command server not available, uses command
        palette and doesn't guarantee that it will wait for command to
        finish."""
        command_server_or_client_fallback(command_id, True)

    # These commands are shims, to provide backwards compatibility, they may be removed in the fuuture.
    # Prefer the run_command... version in command_client.
    def vscode_with_plugin(
        command_id: str,
        arg1: Any = NotSet,
        arg2: Any = NotSet,
        arg3: Any = NotSet,
        arg4: Any = NotSet,
        arg5: Any = NotSet,
    ):
        """Execute command via vscode command server."""
        actions.user.run_rpc_command(
            command_id,
            arg1,
            arg2,
            arg3,
            arg4,
            arg5,
        )

    def vscode_with_plugin_and_wait(
        command_id: str,
        arg1: Any = NotSet,
        arg2: Any = NotSet,
        arg3: Any = NotSet,
        arg4: Any = NotSet,
        arg5: Any = NotSet,
    ):
        """Execute command via vscode command server and wait for command to finish."""
        actions.user.run_rpc_command_and_wait(command_id, arg1, arg2, arg3, arg4, arg5)

    def vscode_get(
        command_id: str,
        arg1: Any = NotSet,
        arg2: Any = NotSet,
        arg3: Any = NotSet,
        arg4: Any = NotSet,
        arg5: Any = NotSet,
    ) -> Any:
        """Execute command via vscode command server and return command output."""
        return actions.user.run_rpc_command_get(
            command_id, arg1, arg2, arg3, arg4, arg5
        )


@linux_ctx.action_class("user")
class LinuxUserActions:
    def trigger_command_server_command_execution():
        # Work around bug with upper f-keys in VSCode on Linux. See
        # https://github.com/pokey/command-server/issues/9#issuecomment-963733930
        actions.key("ctrl-shift-alt-p")