#!/bin/env python3

import signal, subprocess, re, os
from dataclasses import dataclass
import sys
from time import sleep

@dataclass
class WindowInfo:
    window_id:str
    desktop_id:str
    className:str
    machine:str
    title:str

class ExitReceiver:
    def __init__(self) -> None:
        self.should_exit = False
        signal.signal(signal.SIGINT, self.exit_received)
        signal.signal(signal.SIGTERM, self.exit_received)
    
    def exit_received(self, code, *args):
        print('exit received:', code, args)
        self.should_exit = True

def run_wmctrl(args:list[str]) -> tuple[int,str]:
    try:
        with subprocess.Popen(["wmctrl"] + args, stdout=subprocess.PIPE) as p:
            output = p.communicate()[0].decode()[:-1]  # Drop trailing newline
            returncode = p.returncode
            return returncode, output
    except FileNotFoundError:
        return 1, 'ERROR: Please install wmctrl'

# window_id | desktop_id | class | machine | title
rx = re.compile(r'^(\w+)\s+(\d+)\s+(\S+)\s+(\S+)\s+(.+)$')
def find_window(title="", className="") -> list[WindowInfo]:
    matches = []
    code, msg = run_wmctrl(["-l", "-x"])
    if code != 0:
        raise Exception(msg)
    for line in msg.split("\n"):
        match = rx.search(line)
        if not match:
            continue
        groups = match.groups()
        if title and title not in groups[-1]:
            continue
        if className and className not in groups[2]:
            continue
        matches.append(WindowInfo(*groups))
    return matches

if __name__ == "__main__":
    matched_before = []
    loop = ExitReceiver()
    while not loop.should_exit:
        windows = find_window("Save As", "codium.Codium")
        if windows.__len__() > 0:
            while matched_before.__len__() >= 5:
                matched_before.pop(0)
            for win in windows:
                if win.window_id in matched_before:
                    continue
                if run_wmctrl(["-i", "-R", win.window_id])[0] == 0:
                    matched_before.append(windows[0].window_id)
        sleep(0.25)