codex-wrangler 1.3.3

Linux/X11/i3 tray switcher for live coding-agent TUI sessions
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82

use std::process::Command;

use anyhow::{Context as _, Result, bail};
use serde::Deserialize;
use x11rb::protocol::xproto::Window;

#[derive(Deserialize)]
struct Node {
    window: Option<Window>,
    #[serde(default)]
    floating: Floating,
    #[serde(default)]
    nodes: Vec<Self>,
    #[serde(default)]
    floating_nodes: Vec<Self>,
}

#[derive(Clone, Copy, Default, Deserialize)]
#[serde(rename_all = "snake_case")]
enum Floating {
    AutoOn,
    UserOn,
    #[default]
    AutoOff,
    UserOff,
}

impl Node {
    fn floating(&self, quarry: Window, inherited: bool) -> Option<bool> {
        let floating = inherited || matches!(self.floating, Floating::AutoOn | Floating::UserOn);
        if self.window == Some(quarry) {
            return Some(floating);
        }
        self.nodes
            .iter()
            .find_map(|node| node.floating(quarry, floating))
            .or_else(|| {
                self.floating_nodes
                    .iter()
                    .find_map(|node| node.floating(quarry, true))
            })
    }
}

pub fn window_floating(window: Window) -> Result<Option<bool>> {
    let output = Command::new("i3-msg")
        .args(["-r", "-t", "get_tree"])
        .output()
        .context("query i3 layout tree")?;
    if !output.status.success() {
        bail!(
            "i3 layout query failed: {}",
            String::from_utf8_lossy(&output.stderr).trim()
        );
    }
    let tree: Node = serde_json::from_slice(&output.stdout).context("decode i3 layout tree")?;
    Ok(tree.floating(window, false))
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn floating_branch_dominates_a_stale_child_flag() {
        let tree: Node = serde_json::from_str(
            r#"{
              "window": null,
              "nodes": [{"window": 7, "floating": "user_off"}],
              "floating_nodes": [{
                "window": null,
                "floating": "auto_on",
                "nodes": [{"window": 9, "floating": "user_off"}]
              }]
            }"#,
        )
        .expect("tree");
        assert_eq!(tree.floating(7, false), Some(false));
        assert_eq!(tree.floating(9, false), Some(true));
        assert_eq!(tree.floating(11, false), None);
    }
}