from windows_mcp.tree.views import TreeState, BoundingBox
from dataclasses import dataclass
from tabulate import tabulate
from PIL.Image import Image
from enum import Enum
class Browser(Enum):
CHROME = "chrome"
EDGE = "msedge"
FIREFOX = "firefox"
@classmethod
def has_process(cls, process_name: str) -> bool:
if not hasattr(cls, "_process_names"):
cls._process_names = {f"{b.value}.exe" for b in cls}
return process_name.lower() in cls._process_names
class Status(Enum):
MAXIMIZED = "Maximized"
MINIMIZED = "Minimized"
NORMAL = "Normal"
HIDDEN = "Hidden"
@dataclass
class Window:
name: str
is_browser: bool
depth: int
status: Status
bounding_box: BoundingBox
handle: int
process_id: int
def to_row(self):
return [
self.name,
self.depth,
self.status.value,
self.bounding_box.width,
self.bounding_box.height,
self.handle,
]
@dataclass
class Size:
width: int
height: int
def to_string(self):
return f"({self.width},{self.height})"
@dataclass
class DesktopState:
active_desktop: dict
all_desktops: list[dict]
active_window: Window | None
windows: list[Window]
screenshot: Image | None = None
tree_state: TreeState | None = None
def active_desktop_to_string(self):
desktop_name = self.active_desktop.get("name")
headers = ["Name"]
return tabulate([[desktop_name]], headers=headers, tablefmt="simple")
def desktops_to_string(self):
headers = ["Name"]
rows = [[desktop.get("name")] for desktop in self.all_desktops]
return tabulate(rows, headers=headers, tablefmt="simple")
def active_window_to_string(self):
if not self.active_window:
return "No active window found"
headers = ["Name", "Depth", "Status", "Width", "Height", "Handle"]
return tabulate([self.active_window.to_row()], headers=headers, tablefmt="simple")
def windows_to_string(self):
if not self.windows:
return "No windows found"
headers = ["Name", "Depth", "Status", "Width", "Height", "Handle"]
rows = [window.to_row() for window in self.windows]
return tabulate(rows, headers=headers, tablefmt="simple")
