commit
0c3f87531e
4 changed files with 214 additions and 0 deletions
@ -0,0 +1 @@ |
||||
/target |
@ -0,0 +1,12 @@ |
||||
[package] |
||||
name = "swayr" |
||||
version = "0.0.1" |
||||
authors = ["Tassilo Horn"] |
||||
edition = "2018" |
||||
|
||||
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html |
||||
|
||||
[dependencies] |
||||
serde = { version = "1.0.117", features = ["derive"] } |
||||
serde_json = "1.0.59" |
||||
clap = "3.0.0-beta.2" |
@ -0,0 +1,189 @@ |
||||
extern crate serde; |
||||
extern crate serde_json; |
||||
|
||||
use serde::Deserialize; |
||||
|
||||
use std::process as proc; |
||||
|
||||
pub type Id = u32; |
||||
pub type Dim = u16; |
||||
pub type Pid = u16; |
||||
|
||||
#[derive(Deserialize)] |
||||
pub struct Rect { |
||||
x: Dim, |
||||
y: Dim, |
||||
width: Dim, |
||||
height: Dim, |
||||
} |
||||
|
||||
// TODO: Maybe there are more?
|
||||
#[derive(Deserialize)] |
||||
pub enum Border { |
||||
#[serde(rename = "none")] |
||||
None, |
||||
#[serde(rename = "pixel")] |
||||
Pixel, |
||||
#[serde(rename = "csd")] |
||||
Csd, |
||||
} |
||||
|
||||
// TODO: Maybe there are more?
|
||||
#[derive(Deserialize)] |
||||
pub enum Layout { |
||||
#[serde(rename = "splith")] |
||||
SplitH, |
||||
#[serde(rename = "splitv")] |
||||
SplitV, |
||||
#[serde(rename = "tabbed")] |
||||
Tabbed, |
||||
#[serde(rename = "stacked")] |
||||
Stacked, |
||||
#[serde(rename = "output")] |
||||
Output, |
||||
#[serde(rename = "none")] |
||||
None, |
||||
} |
||||
|
||||
#[derive(Deserialize)] |
||||
pub enum Orientation { |
||||
#[serde(rename = "horizontal")] |
||||
Horizontal, |
||||
#[serde(rename = "vertical")] |
||||
Vertical, |
||||
#[serde(rename = "none")] |
||||
None, |
||||
} |
||||
|
||||
#[derive(Deserialize, PartialEq)] |
||||
pub enum NodeType { |
||||
#[serde(rename = "root")] |
||||
Root, |
||||
#[serde(rename = "workspace")] |
||||
Workspace, |
||||
#[serde(rename = "output")] |
||||
Output, |
||||
#[serde(rename = "con")] |
||||
Con, |
||||
#[serde(rename = "floating_con")] |
||||
FloatingCon, |
||||
} |
||||
|
||||
#[derive(Deserialize)] |
||||
pub enum ShellType { |
||||
#[serde(rename = "xdg_shell")] |
||||
XdgShell, |
||||
#[serde(rename = "xwayland")] |
||||
XWayland, |
||||
} |
||||
|
||||
#[derive(Deserialize)] |
||||
pub struct Node { |
||||
id: Id, |
||||
name: Option<String>, |
||||
rect: Rect, |
||||
focused: bool, |
||||
focus: Vec<Id>, |
||||
border: Border, |
||||
current_border_width: Dim, |
||||
layout: Layout, |
||||
orientation: Orientation, |
||||
percent: Option<f64>, |
||||
window_rect: Rect, |
||||
deco_rect: Rect, |
||||
geometry: Rect, |
||||
window: Option<Id>, |
||||
urgent: bool, |
||||
marks: Vec<String>, |
||||
fullscreen_mode: u8, // TODO: actually, it's 0 or 1, i.e., a bool
|
||||
nodes: Vec<Node>, |
||||
floating_nodes: Vec<Node>, |
||||
sticky: bool, |
||||
r#type: NodeType, |
||||
app_id: Option<String>, |
||||
visible: Option<bool>, |
||||
max_render_time: Option<i32>, |
||||
pid: Option<Pid>, |
||||
shell: Option<ShellType>, |
||||
} |
||||
|
||||
impl Node { |
||||
fn add_children<'a>(&'a self, v: &'a mut Vec<&'a Node>) { |
||||
for n in self.nodes.iter() { |
||||
v.push(&n); |
||||
n.add_children(v); |
||||
} |
||||
} |
||||
|
||||
fn iter(&self) -> NodeIter { |
||||
let mut v = vec![self]; |
||||
self.add_children(&mut v); |
||||
NodeIter { stack: v } |
||||
} |
||||
} |
||||
|
||||
struct NodeIter<'a> { |
||||
stack: Vec<&'a Node>, |
||||
} |
||||
|
||||
impl<'a> Iterator for NodeIter<'a> { |
||||
type Item = &'a Node; |
||||
|
||||
fn next(&mut self) -> Option<<Self as Iterator>::Item> { |
||||
self.stack.pop() |
||||
} |
||||
} |
||||
|
||||
pub fn get_tree() -> Node { |
||||
let output = proc::Command::new("swaymsg") |
||||
.arg("-t") |
||||
.arg("get_tree") |
||||
.output() |
||||
.expect("Error running swaymsg!"); |
||||
let result = serde_json::from_str( |
||||
String::from_utf8(output.stdout) |
||||
.expect("Wrong string data!") |
||||
.as_str(), |
||||
); |
||||
|
||||
match result { |
||||
Ok(node) => node, |
||||
Err(e) => { |
||||
eprintln!("Error: {}", e); |
||||
panic!() |
||||
} |
||||
} |
||||
} |
||||
|
||||
pub struct Con<'a> { |
||||
name: &'a str, |
||||
id: Id, |
||||
app_id: Option<&'a str>, |
||||
} |
||||
|
||||
impl<'a> std::fmt::Display for Con<'a> { |
||||
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> Result<(), std::fmt::Error> { |
||||
write!(f, "{} {}", self.name, self.app_id.unwrap_or("")) |
||||
} |
||||
} |
||||
|
||||
/// Gets all cons (aka, application windows) of the tree.
|
||||
pub fn get_cons<'a>(tree: &'a Node) -> Vec<Con<'a>> { |
||||
let mut v = vec![]; |
||||
for n in tree.iter() { |
||||
if n.r#type == NodeType::Con || n.r#type == NodeType::FloatingCon { |
||||
v.push(Con { |
||||
name: &n |
||||
.name |
||||
.as_ref() |
||||
.expect(format!("Con without name. id = {}", n.id).as_str()), |
||||
id: n.id, |
||||
app_id: match &n.app_id { |
||||
Some(s) => Some(s.as_ref()), |
||||
None => None, |
||||
}, |
||||
}) |
||||
} |
||||
} |
||||
v |
||||
} |
@ -0,0 +1,12 @@ |
||||
use std::fs; |
||||
use std::io; |
||||
|
||||
use swayr::Node; |
||||
|
||||
fn main() { |
||||
let root_node = swayr::get_tree(); |
||||
for con in swayr::get_cons(&root_node) { |
||||
println!("{}\n", con); |
||||
} |
||||
println!("Yes!") |
||||
} |
Loading…
Reference in new issue