Rewrite it in Rust
This commit is contained in:
parent
5e5a10650e
commit
36593527ca
|
@ -0,0 +1,146 @@
|
|||
#!/usr/bin/env rust-script
|
||||
//!
|
||||
//! ```cargo
|
||||
//! [dependencies]
|
||||
//! serde = { version = "1.0", features = ["derive"] }
|
||||
//! serde_json = "1.0"
|
||||
//! anyhow = "1.0"
|
||||
//! handlebars = "3"
|
||||
//! ```
|
||||
|
||||
use serde::Deserialize;
|
||||
use anyhow::Result;
|
||||
use std::process::Command;
|
||||
use std::{fs, io};
|
||||
use handlebars::Handlebars;
|
||||
use std::collections::HashMap;
|
||||
|
||||
#[derive(Deserialize, Debug)]
|
||||
struct Monitor {
|
||||
id: u32,
|
||||
name: String,
|
||||
width: u32,
|
||||
height: u32,
|
||||
}
|
||||
|
||||
impl Monitor {
|
||||
fn list() -> Result<Vec<Self>> {
|
||||
let output = Command::new("hyprctl")
|
||||
.arg("monitors")
|
||||
.arg("-j")
|
||||
.output()?;
|
||||
|
||||
assert!(output.status.success());
|
||||
|
||||
let monitors_json = std::str::from_utf8(&output.stdout)?;
|
||||
|
||||
Ok(serde_json::from_str(monitors_json)?)
|
||||
}
|
||||
|
||||
fn launch_bar(&self, wl_display: &str, home: &str) -> Result<()> {
|
||||
let bar_name = format!("bar-{wl_display}-{}", self.id);
|
||||
let dir = format!("/tmp/vbar/{wl_display}/{}", self.id);
|
||||
|
||||
match fs::create_dir_all(&dir) {
|
||||
Ok(()) => {},
|
||||
Err(e) if e.kind() == io::ErrorKind::AlreadyExists => {
|
||||
let output = Command::new("rm")
|
||||
.arg("-rf")
|
||||
.arg(&dir)
|
||||
.output()?;
|
||||
assert!(output.status.success());
|
||||
|
||||
fs::create_dir_all(&dir)?;
|
||||
},
|
||||
Err(e) => return Err(e.into()),
|
||||
}
|
||||
|
||||
self.generate_bar_config(
|
||||
&bar_name,
|
||||
&home,
|
||||
&dir,
|
||||
&format!("{home}/.config/eww/templates/bar"),
|
||||
&dir,
|
||||
)?;
|
||||
|
||||
let output = Command::new("eww")
|
||||
.arg("-c")
|
||||
.arg(&dir)
|
||||
.arg("reload")
|
||||
.output()?;
|
||||
assert!(output.status.success());
|
||||
|
||||
let output = Command::new("eww")
|
||||
.arg("-c")
|
||||
.arg(&dir)
|
||||
.arg("open")
|
||||
.arg(bar_name)
|
||||
.output()?;
|
||||
assert!(output.status.success());
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn generate_bar_config(&self, bar_name: &str, home: &str, base_dir: &str, source_dir: &str, dest_dir: &str) -> Result<()> {
|
||||
for entry in fs::read_dir(source_dir)? {
|
||||
let entry = entry?;
|
||||
let path = entry.path();
|
||||
let relpath = path.strip_prefix(&source_dir)?
|
||||
.to_str().unwrap();
|
||||
|
||||
if path.is_dir() {
|
||||
fs::create_dir_all(&format!("{dest_dir}/{relpath}"))?;
|
||||
|
||||
self.generate_bar_config(
|
||||
&bar_name,
|
||||
&home,
|
||||
&base_dir,
|
||||
&path.to_str().unwrap(),
|
||||
&format!("{source_dir}/{relpath}"),
|
||||
)?;
|
||||
} else if path.is_file() {
|
||||
let data = fs::read_to_string(&path)?;
|
||||
let output = self.template(&data, &bar_name, &home, &base_dir)?;
|
||||
|
||||
fs::write(format!("{dest_dir}/{relpath}"), output)?;
|
||||
}
|
||||
}
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn template(&self, source: &str, bar_name: &str, home: &str, base_dir: &str) -> Result<String> {
|
||||
let height: u32 = (self.height as f32 * 0.02) as u32;
|
||||
let id_str = self.id.to_string();
|
||||
let height_str = height.to_string();
|
||||
|
||||
let values: HashMap<&str, &str> = HashMap::from([
|
||||
("monitor", id_str.as_str()),
|
||||
("monitor_name", self.name.as_str()),
|
||||
("bar_name", bar_name),
|
||||
("home", home),
|
||||
("base_dir", base_dir),
|
||||
("height", height_str.as_str()),
|
||||
]);
|
||||
|
||||
let mut handlebars = Handlebars::new();
|
||||
handlebars.register_template_string("tpl", source)?;
|
||||
Ok(handlebars.render("tpl", &values)?)
|
||||
}
|
||||
}
|
||||
|
||||
fn main() -> Result<()> {
|
||||
let wl_display = std::env::var_os("WAYLAND_DISPLAY").unwrap()
|
||||
.into_string().unwrap();
|
||||
|
||||
let home = std::env::var_os("HOME").unwrap()
|
||||
.into_string().unwrap();
|
||||
|
||||
let monitors = Monitor::list()?;
|
||||
println!("{:?}", monitors);
|
||||
for monitor in monitors {
|
||||
monitor.launch_bar(&wl_display, &home)?;
|
||||
}
|
||||
|
||||
Ok(())
|
||||
}
|
|
@ -1,10 +1,10 @@
|
|||
@import "$base_dir/modules/workspaces.scss";
|
||||
@import "$base_dir/modules/sound.scss";
|
||||
@import "$base_dir/modules/clock.scss";
|
||||
@import "$base_dir/modules/weather.scss";
|
||||
@import "$base_dir/modules/widget.scss";
|
||||
@import "$base_dir/modules/management.scss";
|
||||
@import "$base_dir/variables.scss";
|
||||
@import "{{ base_dir }}/modules/workspaces.scss";
|
||||
@import "{{ base_dir }}/modules/sound.scss";
|
||||
@import "{{ base_dir }}/modules/clock.scss";
|
||||
@import "{{ base_dir }}/modules/weather.scss";
|
||||
@import "{{ base_dir }}/modules/widget.scss";
|
||||
@import "{{ base_dir }}/modules/management.scss";
|
||||
@import "{{ base_dir }}/variables.scss";
|
||||
|
||||
*{
|
||||
all: unset;
|
||||
|
|
|
@ -4,12 +4,12 @@
|
|||
; bar_name: The bar name (should be unique)
|
||||
; monitor: The Hyprland monitor id
|
||||
; monitor_name: The Hyprland monitor name
|
||||
(include "$base_dir/modules/bar.yuck")
|
||||
(include "{{ base_dir }}/modules/bar.yuck")
|
||||
|
||||
(defwindow $bar_name
|
||||
(defwindow {{ bar_name }}
|
||||
:geometry (geometry :x "0" :y "15"
|
||||
:anchor "bottom center"
|
||||
:height "$height" :width "60%")
|
||||
:height "{{ height }}" :width "60%")
|
||||
:exclusive true
|
||||
:monitor $monitor
|
||||
:monitor {{ monitor }}
|
||||
(bar ))
|
||||
|
|
|
@ -1,3 +1,3 @@
|
|||
(deflisten activeworkspaceid
|
||||
:initial "2"
|
||||
"$home/.config/eww/scripts/get-activeworkspace-id")
|
||||
"{{ home }}/.config/eww/scripts/get-activeworkspace-id")
|
||||
|
|
|
@ -1,7 +1,7 @@
|
|||
(include "$base_dir/modules/workspaces.yuck")
|
||||
(include "$base_dir/modules/activeworkspaceid.yuck")
|
||||
(include "$base_dir/modules/widgets-panel.yuck")
|
||||
(include "$base_dir/modules/management.yuck")
|
||||
(include "{{ base_dir }}/modules/workspaces.yuck")
|
||||
(include "{{ base_dir }}/modules/activeworkspaceid.yuck")
|
||||
(include "{{ base_dir }}/modules/widgets-panel.yuck")
|
||||
(include "{{ base_dir }}/modules/management.yuck")
|
||||
|
||||
(defwidget bar[]
|
||||
(box
|
||||
|
|
|
@ -1,4 +1,4 @@
|
|||
@import "$base_dir/variables.scss";
|
||||
@import "{{ base_dir }}/variables.scss";
|
||||
|
||||
.clock {
|
||||
padding: .3rem;
|
||||
|
|
|
@ -1,4 +1,4 @@
|
|||
@import "$base_dir/variables.scss";
|
||||
@import "{{ base_dir }}/variables.scss";
|
||||
|
||||
.management {
|
||||
padding: .3rem;
|
||||
|
|
|
@ -10,7 +10,7 @@
|
|||
(defwidget lock[]
|
||||
(button
|
||||
:class "lock-btn btn"
|
||||
:onclick "$home/.config/eww/scripts/lock"
|
||||
:onclick "{{ home }}/.config/eww/scripts/lock"
|
||||
""))
|
||||
|
||||
(defwidget reboot[]
|
||||
|
|
|
@ -1,4 +1,4 @@
|
|||
@import "$base_dir/variables.scss";
|
||||
@import "{{ base_dir }}/variables.scss";
|
||||
|
||||
.volume-widget {
|
||||
padding: .3rem;
|
||||
|
|
|
@ -1,6 +1,6 @@
|
|||
(deflisten volume
|
||||
:initial `{"value": 100, "is_muted": false}`
|
||||
"$home/.config/eww/scripts/get-volume")
|
||||
"{{ home }}/.config/eww/scripts/get-volume")
|
||||
|
||||
(defwidget volume-widget[]
|
||||
(box
|
||||
|
|
|
@ -1,4 +1,4 @@
|
|||
@import "$base_dir/variables.scss";
|
||||
@import "{{ base_dir }}/variables.scss";
|
||||
|
||||
.weather {
|
||||
padding: .3rem;
|
||||
|
|
|
@ -1,4 +1,4 @@
|
|||
@import "$base_dir/variables.scss";
|
||||
@import "{{ base_dir }}/variables.scss";
|
||||
|
||||
.widget {
|
||||
font-size: 1rem;
|
||||
|
|
|
@ -1,6 +1,6 @@
|
|||
(include "$base_dir/modules/sound.yuck")
|
||||
(include "$base_dir/modules/clock.yuck")
|
||||
(include "$base_dir/modules/weather.yuck")
|
||||
(include "{{ base_dir }}/modules/sound.yuck")
|
||||
(include "{{ base_dir }}/modules/clock.yuck")
|
||||
(include "{{ base_dir }}/modules/weather.yuck")
|
||||
|
||||
(defwidget widgets-panel[]
|
||||
(box
|
||||
|
|
|
@ -1,4 +1,4 @@
|
|||
@import "$base_dir/variables.scss";
|
||||
@import "{{ base_dir }}/variables.scss";
|
||||
|
||||
.workspaces-widget {
|
||||
margin-left: .4rem;
|
||||
|
|
|
@ -1,6 +1,6 @@
|
|||
(deflisten workspaces
|
||||
:initial "[]"
|
||||
"$home/.config/eww/scripts/get-workspaces $monitor_name")
|
||||
"{{ home }}/.config/eww/scripts/get-workspaces {{ monitor_name }}")
|
||||
|
||||
(defwidget workspaces-widget[]
|
||||
(box
|
||||
|
|
Loading…
Reference in New Issue