Initial commit
This commit is contained in:
commit
d77d8a5b22
16
Cargo.toml
Normal file
16
Cargo.toml
Normal file
@ -0,0 +1,16 @@
|
||||
[package]
|
||||
name = "autoname_workspaces"
|
||||
version = "0.1.0"
|
||||
authors = ["Julian Ospald <hasufell@posteo.de>"]
|
||||
edition = "2018"
|
||||
|
||||
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
|
||||
|
||||
[dependencies]
|
||||
counter = "0.4.3"
|
||||
i3ipc = "0.10.1"
|
||||
lazy_static = "1.4"
|
||||
log = "0.4"
|
||||
regex = "1"
|
||||
simple_logger = "1.5.0"
|
||||
signal-hook = "0.1.13"
|
||||
357
src/main.rs
Normal file
357
src/main.rs
Normal file
@ -0,0 +1,357 @@
|
||||
extern crate counter;
|
||||
extern crate i3ipc;
|
||||
extern crate signal_hook;
|
||||
#[macro_use(lazy_static)]
|
||||
extern crate lazy_static;
|
||||
#[macro_use]
|
||||
extern crate log;
|
||||
extern crate simple_logger;
|
||||
|
||||
use counter::Counter;
|
||||
use i3ipc::I3Connection;
|
||||
use log::{info, trace, warn};
|
||||
use regex::Regex;
|
||||
use signal_hook::{iterator::Signals, SIGINT, SIGTERM};
|
||||
use std::collections::HashMap;
|
||||
use std::process::Command;
|
||||
use std::sync::{Arc, Mutex};
|
||||
use std::vec::Vec;
|
||||
use std::{error::Error, thread, time::Duration};
|
||||
|
||||
const SUPERSCRIPT: &'static [&'static str; 10] =
|
||||
&["⁰", "¹", "²", "³", "⁴", "⁵", "⁶", "⁷", "⁸", "⁹"];
|
||||
const SUBSCRIPT: &'static [&'static str; 10] = &["₀", "₁", "₂", "₃", "₄", "₅", "₆", "₇", "₈", "₉"];
|
||||
const DIGITS: &'static [&'static str; 10] = &["0", "1", "2", "3", "4", "5", "6", "7", "8", "9"];
|
||||
|
||||
lazy_static! {
|
||||
static ref WINDOW_ICONS: HashMap<&'static str, &'static str> = vec![
|
||||
("NeovimGtk", "\u{f121}"),
|
||||
("alacritty", "\u{f120}"),
|
||||
("atom", "\u{f121}"),
|
||||
("banshee", "\u{f04b}"),
|
||||
("blender", "\u{f1b2}"),
|
||||
("chromium", "\u{f268}"),
|
||||
("cura", "\u{f1b2}"),
|
||||
("darktable", "\u{f03e}"),
|
||||
("discord", "\u{f075}"),
|
||||
("eclipse", "\u{f121}"),
|
||||
("emacs", "\u{f121}"),
|
||||
("eog", "\u{f03e}"),
|
||||
("evince", "\u{f1c1}"),
|
||||
("evolution", "\u{f0e0}"),
|
||||
("feh", "\u{f03e}"),
|
||||
("file-roller", "\u{f066}"),
|
||||
("filezilla", "\u{f233}"),
|
||||
("firefox", "\u{f269}"),
|
||||
("firefox-esr", "\u{f269}"),
|
||||
("gimp-2.8", "\u{f03e}"),
|
||||
("git-gui", "\u{f1d3}"),
|
||||
("gitk", "\u{f1d3}"),
|
||||
("gnome-control-center", "\u{f205}"),
|
||||
("gnome-terminal-server", "\u{f120}"),
|
||||
("google-chrome", "\u{f268}"),
|
||||
("gpick", "\u{f1fb}"),
|
||||
("gvim", "\u{f121}"),
|
||||
("imv", "\u{f03e}"),
|
||||
("java", "\u{f121}"),
|
||||
("jetbrains-idea", "\u{f121}"),
|
||||
("jetbrains-studio", "\u{f121}"),
|
||||
("keepassxc", "\u{f084}"),
|
||||
("keybase", "\u{f084}"),
|
||||
("kicad", "\u{f2db}"),
|
||||
("kitty", "\u{f120}"),
|
||||
("libreoffice", "\u{f15c}"),
|
||||
("lua5.1", "\u{f186}"),
|
||||
("mpv", "\u{f26c}"),
|
||||
("mupdf", "\u{f1c1}"),
|
||||
("mysql-workbench-bin", "\u{f1c0}"),
|
||||
("nautilus", "\u{f0c5}"),
|
||||
("nemo", "\u{f0c5}"),
|
||||
("openscad", "\u{f1b2}"),
|
||||
("pavucontrol", "\u{f028}"),
|
||||
("pidgin", "\u{f075}"),
|
||||
("postman", "\u{f197}"),
|
||||
("rhythmbox", "\u{f04b}"),
|
||||
("robo3t", "\u{f1c0}"),
|
||||
("sakura", "\u{f120}"),
|
||||
("slack", "\u{f198}"),
|
||||
("slic3r.pl", "\u{f1b2}"),
|
||||
("spacefm", "\u{f0c5}"),
|
||||
("spotify", "\u{f001}"),
|
||||
("steam", "\u{f1b6}"),
|
||||
("subl", "\u{f15c}"),
|
||||
("subl3", "\u{f15c}"),
|
||||
("sublime_text", "\u{f15c}"),
|
||||
("thunar", "\u{f0c5}"),
|
||||
("thunderbird", "\u{f0e0}"),
|
||||
("totem", "\u{f04b}"),
|
||||
("urxvt", "\u{f120}"),
|
||||
("vim", "\u{f121}"),
|
||||
("xfce4-terminal", "\u{f120}"),
|
||||
("xournal", "\u{f15c}"),
|
||||
("yelp", "\u{f121}"),
|
||||
("zenity", "\u{f2d0}"),
|
||||
("zoom", "\u{f075}"),
|
||||
]
|
||||
.into_iter()
|
||||
.collect();
|
||||
}
|
||||
|
||||
fn main() {
|
||||
simple_logger::init().unwrap();
|
||||
|
||||
// establish a connection to i3 over a unix socket
|
||||
let connection = Arc::new(Mutex::new(I3Connection::connect().unwrap()));
|
||||
|
||||
let signals = Signals::new(&[SIGINT, SIGTERM]).unwrap();
|
||||
|
||||
let con = Arc::clone(&connection);
|
||||
thread::spawn(move || {
|
||||
for sig in signals.forever() {
|
||||
match sig {
|
||||
signal_hook::SIGINT => on_exit(con.clone()),
|
||||
signal_hook::SIGTERM => on_exit(con.clone()),
|
||||
_ => unreachable!(),
|
||||
}
|
||||
}
|
||||
});
|
||||
|
||||
match rename_workspaces(connection.clone()) {
|
||||
Ok(_) => info!("Successfully renamed workspaces"),
|
||||
Err(err) => warn!("Error renaming workspaces: {}", err),
|
||||
}
|
||||
|
||||
// TODO: callback
|
||||
}
|
||||
|
||||
fn on_exit(con: Arc<Mutex<I3Connection>>) {
|
||||
let mut c = con.lock().unwrap();
|
||||
let ws = c
|
||||
.get_workspaces()
|
||||
.unwrap_or(i3ipc::reply::Workspaces {
|
||||
workspaces: Vec::new(),
|
||||
})
|
||||
.workspaces;
|
||||
let mut i: u32 = 1;
|
||||
|
||||
for workspace in ws {
|
||||
let name_parts = match parse_workspace_name(workspace.name.as_str()) {
|
||||
Some(np) => np,
|
||||
None => NameParts {
|
||||
num: i.to_string(),
|
||||
shortname: String::from(""),
|
||||
icons: String::from(""),
|
||||
},
|
||||
};
|
||||
let new_name: String = construct_workspace_name(&name_parts);
|
||||
i += 1;
|
||||
|
||||
if workspace.name == new_name {
|
||||
continue;
|
||||
}
|
||||
|
||||
c.run_command(format!("rename workspace {} to {}", workspace.name, new_name).as_str());
|
||||
}
|
||||
|
||||
std::process::exit(0);
|
||||
}
|
||||
|
||||
struct NameParts {
|
||||
num: String,
|
||||
shortname: String,
|
||||
icons: String,
|
||||
}
|
||||
|
||||
fn parse_workspace_name(name: &str) -> Option<NameParts> {
|
||||
let re = Regex::new(r"(?P<num>\d+):?(?P<shortname>\w+)? ?(?P<icons>.+)?").unwrap();
|
||||
let matches = re
|
||||
.find_iter(name)
|
||||
.map(|m| m.as_str())
|
||||
.collect::<Vec<&str>>();
|
||||
if matches.len() == 3 {
|
||||
return Some(NameParts {
|
||||
num: String::from(matches[0]),
|
||||
shortname: String::from(matches[1]),
|
||||
icons: String::from(matches[2]),
|
||||
});
|
||||
} else {
|
||||
return None;
|
||||
}
|
||||
}
|
||||
|
||||
fn construct_workspace_name(np: &NameParts) -> String {
|
||||
let first_part = [np.num.as_str(), ":"].concat();
|
||||
let last_part = if !np.shortname.is_empty() || !np.icons.is_empty() {
|
||||
if !np.icons.is_empty() {
|
||||
[np.shortname.as_str(), " ", np.icons.as_str()].concat()
|
||||
} else {
|
||||
String::from(np.shortname.as_str())
|
||||
}
|
||||
} else {
|
||||
String::from(" ")
|
||||
};
|
||||
|
||||
return [first_part, last_part].concat();
|
||||
}
|
||||
|
||||
fn rename_workspaces(con: Arc<Mutex<I3Connection>>) -> Result<(), i3ipc::MessageError> {
|
||||
let mut c = con.lock().unwrap();
|
||||
let ws_infos = (c.get_workspaces()?).workspaces;
|
||||
let mut prev_output: Option<String> = None;
|
||||
let mut n: u32 = 1;
|
||||
let tree = c.get_tree()?;
|
||||
let workspaces = find_focused_workspace(&tree);
|
||||
|
||||
for (ws_index, workspace) in workspaces.iter().enumerate() {
|
||||
let ws_info = &ws_infos[ws_index];
|
||||
let name_parts = match workspace
|
||||
.name
|
||||
.as_ref()
|
||||
.and_then(|n| parse_workspace_name(n.as_str()))
|
||||
{
|
||||
Some(n) => n,
|
||||
None => NameParts {
|
||||
num: n.to_string(),
|
||||
shortname: String::from(""),
|
||||
icons: String::from(""),
|
||||
},
|
||||
};
|
||||
let mut icon_list: Vec<String> = Vec::new();
|
||||
for leave in leaves(workspace) {
|
||||
icon_list.push(icon_for_window(leave));
|
||||
}
|
||||
let new_icons = format_icon_list(icon_list);
|
||||
|
||||
match prev_output.as_ref() {
|
||||
Some(o) => {
|
||||
if ws_info.output != *o {
|
||||
n += 1;
|
||||
}
|
||||
}
|
||||
_ => (),
|
||||
}
|
||||
prev_output = Some(ws_info.output.clone());
|
||||
|
||||
// TODO: renumber workspaces
|
||||
let new_num = name_parts.num;
|
||||
n += 1;
|
||||
|
||||
let new_name = construct_workspace_name(&NameParts {
|
||||
num: new_num,
|
||||
shortname: name_parts.shortname,
|
||||
icons: new_icons,
|
||||
});
|
||||
|
||||
match workspace.name.as_ref() {
|
||||
Some(n) => {
|
||||
info!("rename workspace {} to {}", n, new_name);
|
||||
c.run_command(format!("rename workspace {} to {}", n, new_name).as_str())?;
|
||||
}
|
||||
None => warn!("Could not find workspace name"),
|
||||
}
|
||||
}
|
||||
|
||||
return Ok(());
|
||||
}
|
||||
|
||||
fn find_focused_workspace<'a>(node: &'a i3ipc::reply::Node) -> Option<&'a i3ipc::reply::Node> {
|
||||
let mut work_node: Option<&'a i3ipc::reply::Node> = None;
|
||||
return find_focused_workspace_rec(node, &mut work_node);
|
||||
}
|
||||
|
||||
fn find_focused_workspace_rec<'a>(
|
||||
node: &'a i3ipc::reply::Node,
|
||||
work_node: &mut Option<&'a i3ipc::reply::Node>,
|
||||
) -> Option<&'a i3ipc::reply::Node> {
|
||||
if node.nodetype == i3ipc::reply::NodeType::Workspace {
|
||||
*work_node = Some(node);
|
||||
}
|
||||
|
||||
if node.focused {
|
||||
return *work_node;
|
||||
} else {
|
||||
if let Some(&want) = node.focus.get(0) {
|
||||
let child = node.nodes.iter().find(|n| want == n.id).unwrap();
|
||||
return find_focused_workspace_rec(child, work_node);
|
||||
} else {
|
||||
return None;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fn leaves(node: &i3ipc::reply::Node) -> Vec<&i3ipc::reply::Node> {
|
||||
let mut vec: Vec<&i3ipc::reply::Node> = Vec::new();
|
||||
for n in &node.nodes {
|
||||
if n.nodes.is_empty() {
|
||||
vec.push(&n);
|
||||
} else {
|
||||
let child_leaves = leaves(&n);
|
||||
vec.extend(child_leaves);
|
||||
}
|
||||
}
|
||||
|
||||
return vec;
|
||||
}
|
||||
|
||||
fn icon_for_window(node: &i3ipc::reply::Node) -> String {
|
||||
let classes = node.window.and_then(|w| xprop(w, "WM_CLASS"));
|
||||
match classes {
|
||||
Some(c) => {
|
||||
if c.len() > 0 {
|
||||
for class in c {
|
||||
match WINDOW_ICONS.get(class.to_ascii_lowercase().as_str()) {
|
||||
Some(m) => return String::from(*m),
|
||||
None => (),
|
||||
}
|
||||
}
|
||||
}
|
||||
return String::from("*");
|
||||
}
|
||||
None => return String::from("*"),
|
||||
}
|
||||
}
|
||||
|
||||
fn xprop(win_id: i32, property: &str) -> Option<Vec<String>> {
|
||||
return Command::new("xprop")
|
||||
.arg("-id")
|
||||
.arg(win_id.to_string())
|
||||
.arg(property)
|
||||
.output()
|
||||
.ok()
|
||||
.and_then(|r| String::from_utf8(r.stdout).ok())
|
||||
.map(|prop| {
|
||||
let re = Regex::new(r#"([^"]*)"#).unwrap();
|
||||
return re
|
||||
.find_iter(prop.as_str())
|
||||
.map(|m| String::from(m.as_str()))
|
||||
.collect::<Vec<String>>();
|
||||
});
|
||||
}
|
||||
|
||||
// TODO: support for superscript and normal numbers
|
||||
fn format_icon_list(icons: Vec<String>) -> String {
|
||||
let mut new_list: Vec<String> = Vec::new();
|
||||
let icon_count = icons.into_iter().collect::<Counter<_>>();
|
||||
for (icon, count) in icon_count.iter() {
|
||||
if *count > 1 {
|
||||
new_list.push(
|
||||
[
|
||||
icon.to_string(),
|
||||
encode_base_10_number(*count as usize, SUPERSCRIPT),
|
||||
]
|
||||
.concat(),
|
||||
);
|
||||
} else {
|
||||
new_list.push(icon.to_string());
|
||||
}
|
||||
}
|
||||
|
||||
return new_list.join(" ");
|
||||
}
|
||||
|
||||
fn encode_base_10_number(n: usize, symbols: &[&str; 10]) -> String {
|
||||
n.to_string()
|
||||
.chars()
|
||||
.map(|c| symbols[c.to_digit(10).unwrap() as usize])
|
||||
.collect()
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user