mirror of
https://github.com/trycua/computer.git
synced 2025-12-31 10:29:59 -06:00
69 lines
2.9 KiB
Python
69 lines
2.9 KiB
Python
import platform
|
|
import subprocess
|
|
from typing import Tuple, Type
|
|
from .base import BaseAccessibilityHandler, BaseAutomationHandler, BaseFileHandler
|
|
from computer_server.diorama.base import BaseDioramaHandler
|
|
|
|
# Conditionally import platform-specific handlers
|
|
system = platform.system().lower()
|
|
if system == 'darwin':
|
|
from .macos import MacOSAccessibilityHandler, MacOSAutomationHandler
|
|
from computer_server.diorama.macos import MacOSDioramaHandler
|
|
elif system == 'linux':
|
|
from .linux import LinuxAccessibilityHandler, LinuxAutomationHandler
|
|
elif system == 'windows':
|
|
from .windows import WindowsAccessibilityHandler, WindowsAutomationHandler
|
|
|
|
from .generic import GenericFileHandler
|
|
|
|
class HandlerFactory:
|
|
"""Factory for creating OS-specific handlers."""
|
|
|
|
@staticmethod
|
|
def _get_current_os() -> str:
|
|
"""Determine the current OS.
|
|
|
|
Returns:
|
|
str: The OS type ('darwin' for macOS, 'linux' for Linux, or 'windows' for Windows)
|
|
|
|
Raises:
|
|
RuntimeError: If unable to determine the current OS
|
|
"""
|
|
try:
|
|
# Use platform.system() as primary method
|
|
system = platform.system().lower()
|
|
if system in ['darwin', 'linux', 'windows']:
|
|
return system
|
|
|
|
# Fallback to uname if platform.system() doesn't return expected values (Unix-like systems only)
|
|
result = subprocess.run(['uname', '-s'], capture_output=True, text=True)
|
|
if result.returncode == 0:
|
|
return result.stdout.strip().lower()
|
|
|
|
raise RuntimeError(f"Unsupported OS: {system}")
|
|
except Exception as e:
|
|
raise RuntimeError(f"Failed to determine current OS: {str(e)}")
|
|
|
|
@staticmethod
|
|
def create_handlers() -> Tuple[BaseAccessibilityHandler, BaseAutomationHandler, BaseDioramaHandler, BaseFileHandler]:
|
|
"""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
|
|
"""
|
|
os_type = HandlerFactory._get_current_os()
|
|
|
|
if os_type == 'darwin':
|
|
return MacOSAccessibilityHandler(), MacOSAutomationHandler(), MacOSDioramaHandler(), GenericFileHandler()
|
|
elif os_type == 'linux':
|
|
return LinuxAccessibilityHandler(), LinuxAutomationHandler(), BaseDioramaHandler(), GenericFileHandler()
|
|
elif os_type == 'windows':
|
|
return WindowsAccessibilityHandler(), WindowsAutomationHandler(), BaseDioramaHandler(), GenericFileHandler()
|
|
else:
|
|
raise NotImplementedError(f"OS '{os_type}' is not supported")
|