cueward-adapter-macos 0.3.1

use serde::{Deserialize, Serialize};
use std::io::Write;
use std::process::Command;

use crate::MacosError;

#[derive(Debug, Clone, Copy, Serialize, Deserialize, PartialEq, Eq)]
pub struct WindowBounds {
    pub x: i32,
    pub y: i32,
    pub width: i32,
    pub height: i32,
}

#[derive(Debug, Clone, Serialize, PartialEq, Eq)]
pub struct CapturableWindow {
    pub window_id: u32,
    pub app: String,
    pub title: String,
    pub owner_pid: i32,
    pub is_frontmost: bool,
    pub bounds: WindowBounds,
}

#[derive(Debug, Clone, Deserialize, PartialEq)]
pub(crate) struct WindowCatalogEntry {
    pub window_id: u32,
    pub app: String,
    pub title: String,
    pub owner_pid: i32,
    pub layer: i32,
    pub alpha: f64,
    pub is_onscreen: bool,
    pub is_frontmost: bool,
    pub bounds: WindowBounds,
}

pub(crate) fn parse_window_list_payload(payload: &str) -> Result<Vec<WindowCatalogEntry>, MacosError> {
    serde_json::from_str(payload)
        .map_err(|error| MacosError::Other(format!("failed to parse window list: {error}")))
}

pub(crate) fn select_capturable_windows(entries: Vec<WindowCatalogEntry>) -> Vec<CapturableWindow> {
    let mut windows = entries
        .into_iter()
        .filter(|entry| !entry.app.trim().is_empty())
        .filter(|entry| !entry.title.trim().is_empty())
        .filter(|entry| entry.bounds.width > 0 && entry.bounds.height > 0)
        .filter(|entry| entry.is_onscreen)
        .filter(|entry| entry.alpha > 0.0)
        .filter(|entry| entry.layer == 0)
        .filter(|entry| !is_noise_window(entry))
        .map(|entry| CapturableWindow {
            window_id: entry.window_id,
            app: entry.app,
            title: entry.title,
            owner_pid: entry.owner_pid,
            is_frontmost: entry.is_frontmost,
            bounds: entry.bounds,
        })
        .collect::<Vec<_>>();

    windows.sort_by(|left, right| {
        right
            .is_frontmost
            .cmp(&left.is_frontmost)
            .then_with(|| left.app.to_ascii_lowercase().cmp(&right.app.to_ascii_lowercase()))
            .then_with(|| left.title.to_ascii_lowercase().cmp(&right.title.to_ascii_lowercase()))
            .then_with(|| left.window_id.cmp(&right.window_id))
    });

    windows
}

fn is_noise_window(entry: &WindowCatalogEntry) -> bool {
    matches!(
        (entry.app.as_str(), entry.title.as_str()),
        ("WindowManager", "App Icon Window")
            | ("WindowManager", "Gesture Blocking Overlay")
    )
}

#[cfg_attr(not(test), allow(dead_code))]
pub(crate) fn find_capturable_window(
    windows: &[CapturableWindow],
    window_id: u32,
) -> Result<&CapturableWindow, MacosError> {
    windows
        .iter()
        .find(|window| window.window_id == window_id)
        .ok_or_else(|| MacosError::Other(format!("window id not found: {window_id}")))
}

/// List on-screen windows that can be captured, sorted with frontmost first.
pub fn list_capturable_windows() -> Result<Vec<CapturableWindow>, MacosError> {
    let script = r#"import AppKit
import CoreGraphics
import Foundation

let frontmostPid = NSWorkspace.shared.frontmostApplication?.processIdentifier ?? 0
let options: CGWindowListOption = [.optionOnScreenOnly, .excludeDesktopElements]
guard let raw = CGWindowListCopyWindowInfo(options, kCGNullWindowID) as? [[String: Any]] else {
    fputs("failed to read window list\n", stderr)
    exit(1)
}

let payload: [[String: Any]] = raw.compactMap { item in
    guard let windowId = item[kCGWindowNumber as String] as? NSNumber else { return nil }
    let ownerPid = (item[kCGWindowOwnerPID as String] as? NSNumber)?.intValue ?? 0
    let layer = (item[kCGWindowLayer as String] as? NSNumber)?.intValue ?? -1
    let alpha = (item[kCGWindowAlpha as String] as? NSNumber)?.doubleValue ?? 1.0
    let isOnscreen = (item[kCGWindowIsOnscreen as String] as? NSNumber)?.boolValue ?? false
    let owner = (item[kCGWindowOwnerName as String] as? String) ?? ""
    let title = (item[kCGWindowName as String] as? String) ?? ""
    let boundsDict = (item[kCGWindowBounds as String] as? NSDictionary) ?? [:]
    let rect = CGRect(dictionaryRepresentation: boundsDict) ?? .zero

    return [
        "window_id": windowId.uint32Value,
        "app": owner,
        "title": title,
        "owner_pid": ownerPid,
        "layer": layer,
        "alpha": alpha,
        "is_onscreen": isOnscreen,
        "is_frontmost": ownerPid == frontmostPid,
        "bounds": [
            "x": Int(rect.origin.x),
            "y": Int(rect.origin.y),
            "width": Int(rect.size.width),
            "height": Int(rect.size.height),
        ],
    ]
}

let data = try JSONSerialization.data(withJSONObject: payload, options: [])
if let text = String(data: data, encoding: .utf8) {
    print(text)
}"#;

    let mut file = tempfile::NamedTempFile::with_suffix(".swift")
        .map_err(|error| MacosError::Other(format!("failed to create swift temp file: {error}")))?;
    file.write_all(script.as_bytes())
        .map_err(|error| MacosError::Other(format!("failed to write swift temp file: {error}")))?;

    let output = Command::new("swift")
        .arg(file.path())
        .output()
        .map_err(|error| MacosError::Other(format!("swift: {error}")))?;

    if !output.status.success() {
        let stderr = String::from_utf8_lossy(&output.stderr);
        return Err(MacosError::Other(format!("failed to list windows: {stderr}")));
    }

    let stdout = String::from_utf8_lossy(&output.stdout);
    let parsed = parse_window_list_payload(&stdout)?;
    Ok(select_capturable_windows(parsed))
}