Rewrite it in Rust

This commit is contained in:
Victor Timofei 2023-06-14 15:25:23 +03:00
parent 5e5a10650e
commit 36593527ca
Signed by: vtimofei
GPG Key ID: B790DCEBE281403A
15 changed files with 174 additions and 28 deletions

146
.config/eww/scripts/launch.rs Executable file
View File

@ -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(())
}

View File

@ -1,10 +1,10 @@
@import "$base_dir/modules/workspaces.scss"; @import "{{ base_dir }}/modules/workspaces.scss";
@import "$base_dir/modules/sound.scss"; @import "{{ base_dir }}/modules/sound.scss";
@import "$base_dir/modules/clock.scss"; @import "{{ base_dir }}/modules/clock.scss";
@import "$base_dir/modules/weather.scss"; @import "{{ base_dir }}/modules/weather.scss";
@import "$base_dir/modules/widget.scss"; @import "{{ base_dir }}/modules/widget.scss";
@import "$base_dir/modules/management.scss"; @import "{{ base_dir }}/modules/management.scss";
@import "$base_dir/variables.scss"; @import "{{ base_dir }}/variables.scss";
*{ *{
all: unset; all: unset;

View File

@ -4,12 +4,12 @@
; bar_name: The bar name (should be unique) ; bar_name: The bar name (should be unique)
; monitor: The Hyprland monitor id ; monitor: The Hyprland monitor id
; monitor_name: The Hyprland monitor name ; 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" :geometry (geometry :x "0" :y "15"
:anchor "bottom center" :anchor "bottom center"
:height "$height" :width "60%") :height "{{ height }}" :width "60%")
:exclusive true :exclusive true
:monitor $monitor :monitor {{ monitor }}
(bar )) (bar ))

View File

@ -1,3 +1,3 @@
(deflisten activeworkspaceid (deflisten activeworkspaceid
:initial "2" :initial "2"
"$home/.config/eww/scripts/get-activeworkspace-id") "{{ home }}/.config/eww/scripts/get-activeworkspace-id")

View File

@ -1,7 +1,7 @@
(include "$base_dir/modules/workspaces.yuck") (include "{{ base_dir }}/modules/workspaces.yuck")
(include "$base_dir/modules/activeworkspaceid.yuck") (include "{{ base_dir }}/modules/activeworkspaceid.yuck")
(include "$base_dir/modules/widgets-panel.yuck") (include "{{ base_dir }}/modules/widgets-panel.yuck")
(include "$base_dir/modules/management.yuck") (include "{{ base_dir }}/modules/management.yuck")
(defwidget bar[] (defwidget bar[]
(box (box

View File

@ -1,4 +1,4 @@
@import "$base_dir/variables.scss"; @import "{{ base_dir }}/variables.scss";
.clock { .clock {
padding: .3rem; padding: .3rem;

View File

@ -1,4 +1,4 @@
@import "$base_dir/variables.scss"; @import "{{ base_dir }}/variables.scss";
.management { .management {
padding: .3rem; padding: .3rem;

View File

@ -10,7 +10,7 @@
(defwidget lock[] (defwidget lock[]
(button (button
:class "lock-btn btn" :class "lock-btn btn"
:onclick "$home/.config/eww/scripts/lock" :onclick "{{ home }}/.config/eww/scripts/lock"
"󰌾")) "󰌾"))
(defwidget reboot[] (defwidget reboot[]

View File

@ -1,4 +1,4 @@
@import "$base_dir/variables.scss"; @import "{{ base_dir }}/variables.scss";
.volume-widget { .volume-widget {
padding: .3rem; padding: .3rem;

View File

@ -1,6 +1,6 @@
(deflisten volume (deflisten volume
:initial `{"value": 100, "is_muted": false}` :initial `{"value": 100, "is_muted": false}`
"$home/.config/eww/scripts/get-volume") "{{ home }}/.config/eww/scripts/get-volume")
(defwidget volume-widget[] (defwidget volume-widget[]
(box (box

View File

@ -1,4 +1,4 @@
@import "$base_dir/variables.scss"; @import "{{ base_dir }}/variables.scss";
.weather { .weather {
padding: .3rem; padding: .3rem;

View File

@ -1,4 +1,4 @@
@import "$base_dir/variables.scss"; @import "{{ base_dir }}/variables.scss";
.widget { .widget {
font-size: 1rem; font-size: 1rem;

View File

@ -1,6 +1,6 @@
(include "$base_dir/modules/sound.yuck") (include "{{ base_dir }}/modules/sound.yuck")
(include "$base_dir/modules/clock.yuck") (include "{{ base_dir }}/modules/clock.yuck")
(include "$base_dir/modules/weather.yuck") (include "{{ base_dir }}/modules/weather.yuck")
(defwidget widgets-panel[] (defwidget widgets-panel[]
(box (box

View File

@ -1,4 +1,4 @@
@import "$base_dir/variables.scss"; @import "{{ base_dir }}/variables.scss";
.workspaces-widget { .workspaces-widget {
margin-left: .4rem; margin-left: .4rem;

View File

@ -1,6 +1,6 @@
(deflisten workspaces (deflisten workspaces
:initial "[]" :initial "[]"
"$home/.config/eww/scripts/get-workspaces $monitor_name") "{{ home }}/.config/eww/scripts/get-workspaces {{ monitor_name }}")
(defwidget workspaces-widget[] (defwidget workspaces-widget[]
(box (box