cua/libs/python/computer-server/computer_server/handlers/factory.py at 5e09f241f14f040c82b969d812efdcd9d8ebdc40 · trycua/cua · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
import logging
import os
from typing import Tuple

from computer_server.diorama.base import BaseDioramaHandler

from ..utils.helpers import get_current_os
from .base import (
    BaseAccessibilityHandler,
    BaseAutomationHandler,
    BaseDesktopHandler,
    BaseFileHandler,
    BaseWindowHandler,
)

logger = logging.getLogger(__name__)

OS_TYPE = get_current_os()

from .generic import GenericDesktopHandler, GenericFileHandler, GenericWindowHandler


class HandlerFactory:
    """Factory for creating OS-specific handlers."""

    @staticmethod
    def create_handlers() -> Tuple[
        BaseAccessibilityHandler,
        BaseAutomationHandler,
        BaseDioramaHandler,
        BaseFileHandler,
        BaseDesktopHandler,
        BaseWindowHandler,
    ]:
        """Create and return appropriate handlers for the current OS.

        Returns:
            Tuple[BaseAccessibilityHandler, BaseAutomationHandler, BaseDioramaHandler, BaseFileHandler]: A tuple containing
            the appropriate accessibility, automation, diorama, and file handlers for the current OS.

        Raises:
            NotImplementedError: If the current OS is not supported
            RuntimeError: If unable to determine the current OS
        """
        backend = os.environ.get("CUA_BACKEND", "native")
        vnc_host = os.environ.get("CUA_VNC_HOST")
        if backend == "vnc" or vnc_host:
            if not vnc_host:
                raise RuntimeError(
                    "CUA_VNC_HOST must be set when using VNC backend "
                    "(--backend=vnc requires --vnc-host)"
                )
            from .vnc import VNCAccessibilityHandler, VNCAutomationHandler

            vnc_port = int(os.environ.get("CUA_VNC_PORT", "5900"))
            vnc_password = os.environ.get("CUA_VNC_PASSWORD", "")
            logger.info(f"Using VNC backend → {vnc_host}:{vnc_port}")
            return (
                VNCAccessibilityHandler(),
                VNCAutomationHandler(host=vnc_host, port=vnc_port, password=vnc_password),
                BaseDioramaHandler(),
                GenericFileHandler(),
                GenericDesktopHandler(),
                GenericWindowHandler(),
            )
        elif OS_TYPE == "android":
            from .android import (
                AndroidAccessibilityHandler,
                AndroidAutomationHandler,
                AndroidDesktopHandler,
                AndroidFileHandler,
                AndroidWindowHandler,
            )

            return (
                AndroidAccessibilityHandler(),
                AndroidAutomationHandler(),
                BaseDioramaHandler(),
                AndroidFileHandler(),
                AndroidDesktopHandler(),
                AndroidWindowHandler(),
            )
        elif OS_TYPE == "darwin":
            from computer_server.diorama.macos import MacOSDioramaHandler

            from .macos import MacOSAccessibilityHandler, MacOSAutomationHandler

            return (
                MacOSAccessibilityHandler(),
                MacOSAutomationHandler(),
                MacOSDioramaHandler(),
                GenericFileHandler(),
                GenericDesktopHandler(),
                GenericWindowHandler(),
            )
        elif OS_TYPE == "linux":
            from .linux import LinuxAccessibilityHandler, LinuxAutomationHandler

            return (
                LinuxAccessibilityHandler(),
                LinuxAutomationHandler(),
                BaseDioramaHandler(),
                GenericFileHandler(),
                GenericDesktopHandler(),
                GenericWindowHandler(),
            )
        elif OS_TYPE == "windows":
            from .windows import WindowsAccessibilityHandler, WindowsAutomationHandler

            return (
                WindowsAccessibilityHandler(),
                WindowsAutomationHandler(),
                BaseDioramaHandler(),
                GenericFileHandler(),
                GenericDesktopHandler(),
                GenericWindowHandler(),
            )
        else:
            raise NotImplementedError(f"OS '{OS_TYPE}' is not supported")