52 lines
1.2 KiB
Rust
52 lines
1.2 KiB
Rust
use std::{fmt::Write, path::PathBuf};
|
|
use dim_sdk::{plugin_info, Context, DimPlugin};
|
|
use std::fs::read_to_string;
|
|
|
|
plugin_info!(
|
|
Plug, // Your main global structs name that implements `DimPlugin`
|
|
"battery", // Plugin name
|
|
"1.0.0", // Plugin Version (leave empty for none)
|
|
"GPLv3" // Plugin license (leave empty for none)
|
|
);
|
|
|
|
#[derive(Debug, Clone)]
|
|
struct Plug {
|
|
}
|
|
|
|
impl Plug {
|
|
pub fn new() -> Self {
|
|
Self {
|
|
}
|
|
}
|
|
}
|
|
|
|
const BATT0: &'static str = "/sys/class/power_supply/BAT0/capacity";
|
|
const BATT1: &'static str = "/sys/class/power_supply/BAT1/capacity";
|
|
|
|
impl DimPlugin for Plug {
|
|
fn init(&mut self, _ctx: Context) {
|
|
}
|
|
fn poll(&mut self, f: &mut dim_sdk::CBuffer) -> dim_sdk::Result<()> {
|
|
//TODO: Quick fix, make this better, more portable
|
|
let path = if PathBuf::from(BATT0).exists() { BATT0 } else { BATT1 };
|
|
|
|
let contents = read_to_string(path)?;
|
|
let cleaned_contents: String = contents.chars().filter(|c| c.is_digit(10)).collect();
|
|
|
|
write!(f, "Battery: {}%", cleaned_contents)?;
|
|
Ok(())
|
|
}
|
|
fn pre_reload(&mut self) {
|
|
}
|
|
fn post_reload(&mut self) {
|
|
|
|
}
|
|
fn free(&mut self) {
|
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
|