from windows_mcp.tree.views import TreeState,BoundingBox
from dataclasses import dataclass
from tabulate import tabulate
from typing import Optional
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:Optional[Window]
windows:list[Window]
screenshot:Optional[Image]=None
tree_state:Optional[TreeState]=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")
