mirror of
https://github.com/house-of-vanity/tmux-helper.git
synced 2026-02-04 17:57:58 +00:00
Compare commits
7 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
6e08864d99 | ||
|
|
6a9a871006 | ||
|
|
bab994fc41 | ||
|
|
fc1c2f539d | ||
|
|
0f5a1aaaed | ||
|
|
1030e69932 | ||
|
|
474d48d15c |
BIN
.github/prev.png
vendored
BIN
.github/prev.png
vendored
Binary file not shown.
|
Before Width: | Height: | Size: 19 KiB After Width: | Height: | Size: 25 KiB |
@@ -1,6 +1,7 @@
|
|||||||
[package]
|
[package]
|
||||||
name = "tmux-helper"
|
name = "tmux-helper"
|
||||||
version = "0.2.2"
|
version = "0.3.1"
|
||||||
|
description = "Utility for printing system info for tmux status line."
|
||||||
authors = ["Ultra Desu <ultradesu@hexor.ru>"]
|
authors = ["Ultra Desu <ultradesu@hexor.ru>"]
|
||||||
edition = "2018"
|
edition = "2018"
|
||||||
|
|
||||||
@@ -8,3 +9,5 @@ edition = "2018"
|
|||||||
sys-info = "*"
|
sys-info = "*"
|
||||||
dbus = "*"
|
dbus = "*"
|
||||||
chrono = "*"
|
chrono = "*"
|
||||||
|
mpd = "*"
|
||||||
|
clap = "*"
|
||||||
|
|||||||
226
src/config.rs
Normal file
226
src/config.rs
Normal file
@@ -0,0 +1,226 @@
|
|||||||
|
use clap::{App, Arg};
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
pub enum Action {
|
||||||
|
Mem,
|
||||||
|
Cpu,
|
||||||
|
Mpris,
|
||||||
|
Mpd,
|
||||||
|
Localtime,
|
||||||
|
Utctime,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Default for Action {
|
||||||
|
fn default() -> Action {
|
||||||
|
Action::Cpu
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Default, Debug)]
|
||||||
|
pub struct Config {
|
||||||
|
pub action: Action,
|
||||||
|
pub mpd_server: String,
|
||||||
|
pub lt_format: Option<String>,
|
||||||
|
pub ut_format: Option<String>,
|
||||||
|
pub color_low: String,
|
||||||
|
pub color_mid: String,
|
||||||
|
pub color_high: String,
|
||||||
|
pub color_track_name: String,
|
||||||
|
pub color_track_artist: String,
|
||||||
|
pub color_track_time: String,
|
||||||
|
pub color_end: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
fn colorize(color: String) -> String {
|
||||||
|
format!("#[fg=colour{}]", color)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn read() -> Config {
|
||||||
|
// Parse opts and args
|
||||||
|
let cli_args = App::new(env!("CARGO_PKG_NAME"))
|
||||||
|
.version(env!("CARGO_PKG_VERSION"))
|
||||||
|
.author(env!("CARGO_PKG_AUTHORS"))
|
||||||
|
.about(env!("CARGO_PKG_DESCRIPTION"))
|
||||||
|
// Flags
|
||||||
|
.arg(
|
||||||
|
Arg::with_name("cpu")
|
||||||
|
.short("c")
|
||||||
|
.long("cpu")
|
||||||
|
.help("Print cpu load bar.")
|
||||||
|
.conflicts_with_all(&["mem", "mpris", "mpd", "localtime", "utctime"])
|
||||||
|
.required(false),
|
||||||
|
)
|
||||||
|
.arg(
|
||||||
|
Arg::with_name("mem")
|
||||||
|
.short("m")
|
||||||
|
.long("mem")
|
||||||
|
.help("Print mem usage bar.")
|
||||||
|
// .conflicts_with("cpu")
|
||||||
|
// .conflicts_with("mpris")
|
||||||
|
// .conflicts_with("mpd")
|
||||||
|
// .conflicts_with("localtime")
|
||||||
|
// .conflicts_with("utctime")
|
||||||
|
.required(false),
|
||||||
|
)
|
||||||
|
.arg(
|
||||||
|
Arg::with_name("mpris")
|
||||||
|
.short("p")
|
||||||
|
.long("mpris")
|
||||||
|
.help("Show player info using MPRIS2 interface.")
|
||||||
|
// .conflicts_with("cpu")
|
||||||
|
// .conflicts_with("mem")
|
||||||
|
// .conflicts_with("localtime")
|
||||||
|
// .conflicts_with("mpd")
|
||||||
|
// .conflicts_with("utctime")
|
||||||
|
.required(false),
|
||||||
|
)
|
||||||
|
.arg(
|
||||||
|
Arg::with_name("mpd")
|
||||||
|
.short("d")
|
||||||
|
.long("mpd")
|
||||||
|
.help("Show mpd player using MPD native protocol.")
|
||||||
|
// .conflicts_with("cpu")
|
||||||
|
// .conflicts_with("mem")
|
||||||
|
// .conflicts_with("localtime")
|
||||||
|
// .conflicts_with("mpris")
|
||||||
|
// .conflicts_with("utctime")
|
||||||
|
.required(false),
|
||||||
|
)
|
||||||
|
// Options
|
||||||
|
.arg(
|
||||||
|
Arg::with_name("localtime")
|
||||||
|
.short("l")
|
||||||
|
.long("localtime")
|
||||||
|
.help("Local time")
|
||||||
|
// .conflicts_with_all(&["mem", "mpris", "mpd", "cpu", "utctime"])
|
||||||
|
.takes_value(true)
|
||||||
|
.required(false),
|
||||||
|
)
|
||||||
|
.arg(
|
||||||
|
Arg::with_name("utctime")
|
||||||
|
.short("u")
|
||||||
|
.long("utctime")
|
||||||
|
.help("UTC time")
|
||||||
|
// .conflicts_with_all(&["mem", "mpris", "mpd", "cpu", "localtime"])
|
||||||
|
.takes_value(true)
|
||||||
|
.required(false),
|
||||||
|
)
|
||||||
|
.arg(
|
||||||
|
Arg::with_name("mpd_address")
|
||||||
|
.short("a")
|
||||||
|
.long("mpd-address")
|
||||||
|
.help("<ADDR>:<PORT> of MPD server.")
|
||||||
|
.takes_value(true)
|
||||||
|
.default_value("127.0.0.1:6600")
|
||||||
|
.required(false),
|
||||||
|
)
|
||||||
|
.arg(
|
||||||
|
Arg::with_name("COLOR_LOW")
|
||||||
|
.long("COLOR_LOW")
|
||||||
|
.help("CPU and MEM bar color while low usage.")
|
||||||
|
.takes_value(true)
|
||||||
|
.default_value("119")
|
||||||
|
.required(false),
|
||||||
|
)
|
||||||
|
.arg(
|
||||||
|
Arg::with_name("COLOR_MID")
|
||||||
|
.long("COLOR_MID")
|
||||||
|
.help("CPU and MEM bar color while mid usage.")
|
||||||
|
.takes_value(true)
|
||||||
|
.default_value("220")
|
||||||
|
.required(false),
|
||||||
|
)
|
||||||
|
.arg(
|
||||||
|
Arg::with_name("COLOR_HIGH")
|
||||||
|
.long("COLOR_HIGH")
|
||||||
|
.help("CPU and MEM bar color while high usage.")
|
||||||
|
.takes_value(true)
|
||||||
|
.default_value("197")
|
||||||
|
.required(false),
|
||||||
|
)
|
||||||
|
.arg(
|
||||||
|
Arg::with_name("COLOR_TRACK_NAME")
|
||||||
|
.long("COLOR_TRACK_NAME")
|
||||||
|
.help("Color of track name filed.")
|
||||||
|
.takes_value(true)
|
||||||
|
.default_value("46")
|
||||||
|
.required(false),
|
||||||
|
)
|
||||||
|
.arg(
|
||||||
|
Arg::with_name("COLOR_TRACK_ARTIST")
|
||||||
|
.long("COLOR_TRACK_ARTIST")
|
||||||
|
.help("Color of artist name filed.")
|
||||||
|
.takes_value(true)
|
||||||
|
.default_value("46")
|
||||||
|
.required(false),
|
||||||
|
)
|
||||||
|
.arg(
|
||||||
|
Arg::with_name("COLOR_TRACK_TIME")
|
||||||
|
.long("COLOR_TRACK_TIME")
|
||||||
|
.help("Color of playing time field.")
|
||||||
|
.takes_value(true)
|
||||||
|
.default_value("153")
|
||||||
|
.required(false),
|
||||||
|
)
|
||||||
|
.arg(
|
||||||
|
Arg::with_name("COLOR_END")
|
||||||
|
.long("COLOR_END")
|
||||||
|
.help("Default color using to terminate others.")
|
||||||
|
.takes_value(true)
|
||||||
|
.default_value("153")
|
||||||
|
.required(false),
|
||||||
|
)
|
||||||
|
.get_matches();
|
||||||
|
|
||||||
|
// cpu - cpu usage bar
|
||||||
|
// mem - mem usage bar
|
||||||
|
// mpris - player info using MPRIS2 interface
|
||||||
|
// mpd - player info using MPD native interface
|
||||||
|
// utctime - utc time
|
||||||
|
// localtime - local time
|
||||||
|
// lt_format - local time format
|
||||||
|
// ut_format - utc time format
|
||||||
|
|
||||||
|
let lt_format = Some(match cli_args.value_of("localtime") {
|
||||||
|
Some(format) => format.to_string(),
|
||||||
|
None => "%H:%M".to_string(),
|
||||||
|
});
|
||||||
|
let ut_format = Some(match cli_args.value_of("utctime") {
|
||||||
|
Some(format) => format.to_string(),
|
||||||
|
None => "%H:%M".to_string(),
|
||||||
|
});
|
||||||
|
|
||||||
|
let mut cfg = Config {
|
||||||
|
action: Action::Cpu,
|
||||||
|
mpd_server: cli_args.value_of("mpd_address").unwrap().to_string(),
|
||||||
|
lt_format: lt_format,
|
||||||
|
ut_format: ut_format,
|
||||||
|
color_low: colorize(cli_args.value_of("COLOR_LOW").unwrap().to_string()),
|
||||||
|
color_mid: colorize(cli_args.value_of("COLOR_MID").unwrap().to_string()),
|
||||||
|
color_high: colorize(cli_args.value_of("COLOR_HIGH").unwrap().to_string()),
|
||||||
|
color_track_name: colorize(cli_args.value_of("COLOR_TRACK_NAME").unwrap().to_string()),
|
||||||
|
color_track_artist: colorize(cli_args.value_of("COLOR_TRACK_ARTIST").unwrap().to_string()),
|
||||||
|
color_track_time: colorize(cli_args.value_of("COLOR_TRACK_TIME").unwrap().to_string()),
|
||||||
|
color_end: colorize(cli_args.value_of("COLOR_END").unwrap().to_string()),
|
||||||
|
};
|
||||||
|
|
||||||
|
if cli_args.is_present("cpu") {
|
||||||
|
cfg.action = Action::Cpu;
|
||||||
|
}
|
||||||
|
if cli_args.is_present("mem") {
|
||||||
|
cfg.action = Action::Mem;
|
||||||
|
}
|
||||||
|
if cli_args.is_present("localtime") {
|
||||||
|
cfg.action = Action::Localtime;
|
||||||
|
}
|
||||||
|
if cli_args.is_present("utctime") {
|
||||||
|
cfg.action = Action::Utctime;
|
||||||
|
}
|
||||||
|
if cli_args.is_present("mpris") {
|
||||||
|
cfg.action = Action::Mpris;
|
||||||
|
}
|
||||||
|
if cli_args.is_present("mpd") {
|
||||||
|
cfg.action = Action::Mpd;
|
||||||
|
}
|
||||||
|
cfg
|
||||||
|
}
|
||||||
257
src/main.rs
257
src/main.rs
@@ -1,19 +1,9 @@
|
|||||||
extern crate chrono;
|
extern crate chrono;
|
||||||
extern crate dbus;
|
extern crate dbus;
|
||||||
|
extern crate mpd;
|
||||||
|
|
||||||
use crate::dbus::blocking::stdintf::org_freedesktop_dbus::Properties;
|
mod config;
|
||||||
use chrono::{DateTime, Local, Utc};
|
mod utils;
|
||||||
use dbus::{arg, blocking::Connection};
|
|
||||||
use std::{env, time::Duration};
|
|
||||||
use sys_info;
|
|
||||||
|
|
||||||
const LOW: &str = "#[fg=colour119]";
|
|
||||||
const MID: &str = "#[fg=colour220]";
|
|
||||||
const HIGH: &str = "#[fg=colour197]";
|
|
||||||
const END: &str = "#[fg=colour153]";
|
|
||||||
const TRACK_NAME: &str = "#[fg=colour46]";
|
|
||||||
const TRACK_ARTIST: &str = "#[fg=colour46]";
|
|
||||||
const TRACK_TIME: &str = "#[fg=colour153]";
|
|
||||||
|
|
||||||
#[derive(Debug, Clone)]
|
#[derive(Debug, Clone)]
|
||||||
struct TrackInfo {
|
struct TrackInfo {
|
||||||
@@ -24,233 +14,20 @@ struct TrackInfo {
|
|||||||
status: String,
|
status: String,
|
||||||
}
|
}
|
||||||
|
|
||||||
fn to_bar(value: i32, max: i32, low: f32, mid: f32) {
|
|
||||||
let mut bar = "".to_string();
|
|
||||||
let bar_sym = "▮".to_string();
|
|
||||||
if (value as f32) / (max as f32) < low {
|
|
||||||
bar.push_str(LOW);
|
|
||||||
} else if (value as f32) / (max as f32) < mid {
|
|
||||||
bar.push_str(MID);
|
|
||||||
} else {
|
|
||||||
bar.push_str(HIGH);
|
|
||||||
}
|
|
||||||
for i in 0..max {
|
|
||||||
if i < value as i32 {
|
|
||||||
bar.push_str(&bar_sym);
|
|
||||||
} else {
|
|
||||||
bar.push_str(" ")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
bar.push_str(END);
|
|
||||||
bar.push_str("|");
|
|
||||||
print!("{}", bar)
|
|
||||||
}
|
|
||||||
|
|
||||||
fn mem_load_bar(bar_len: i32) {
|
|
||||||
let memory;
|
|
||||||
match sys_info::mem_info() {
|
|
||||||
Err(w) => panic!("{:?}", w),
|
|
||||||
Ok(mem_data) => memory = mem_data,
|
|
||||||
}
|
|
||||||
let len =
|
|
||||||
((memory.total - memory.avail) as f32 / (memory.total as f32) * bar_len as f32) as i32;
|
|
||||||
to_bar(len, bar_len, 0.7, 0.9);
|
|
||||||
print!("{:.0} MiB#[default]", memory.avail / 1024);
|
|
||||||
}
|
|
||||||
|
|
||||||
fn cpu_load_bar(bar_len: i32) {
|
|
||||||
let cpu_count = match sys_info::cpu_num() {
|
|
||||||
Ok(c) => c,
|
|
||||||
Err(e) => panic!("{:?}", e),
|
|
||||||
};
|
|
||||||
let la_one: f32 = match sys_info::loadavg() {
|
|
||||||
Ok(l) => l.one as f32,
|
|
||||||
Err(e) => panic!("{:?}", e),
|
|
||||||
};
|
|
||||||
let len: f32 = la_one as f32 / cpu_count as f32 * bar_len as f32;
|
|
||||||
to_bar(len as i32, bar_len, 0.3, 0.7);
|
|
||||||
print!("{:.2} LA1#[default]", la_one);
|
|
||||||
}
|
|
||||||
|
|
||||||
fn get_player() -> Result<Vec<String>, Box<dyn std::error::Error>> {
|
|
||||||
let conn = Connection::new_session()?;
|
|
||||||
let proxy = conn.with_proxy("org.freedesktop.DBus", "/", Duration::from_millis(5000));
|
|
||||||
let (names,): (Vec<String>,) = proxy.method_call("org.freedesktop.DBus", "ListNames", ())?;
|
|
||||||
let mut players: Vec<String> = Vec::new();
|
|
||||||
for name in names {
|
|
||||||
if name.contains("org.mpris.MediaPlayer2") {
|
|
||||||
players.push(name);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(players)
|
|
||||||
}
|
|
||||||
|
|
||||||
fn player_info(players: Vec<String>) -> Result<TrackInfo, Box<dyn std::error::Error>> {
|
|
||||||
let mut players_vec: Vec<TrackInfo> = Vec::new();
|
|
||||||
for player in players {
|
|
||||||
let mut track_info = TrackInfo {
|
|
||||||
artist: "".to_string(),
|
|
||||||
title: "".to_string(),
|
|
||||||
position: "".to_string(),
|
|
||||||
duration: "".to_string(),
|
|
||||||
status: "".to_string(),
|
|
||||||
};
|
|
||||||
let conn = Connection::new_session()?;
|
|
||||||
let proxy = conn.with_proxy(
|
|
||||||
player,
|
|
||||||
"/org/mpris/MediaPlayer2",
|
|
||||||
Duration::from_millis(5000),
|
|
||||||
);
|
|
||||||
let metadata: Box<dyn arg::RefArg> =
|
|
||||||
proxy.get("org.mpris.MediaPlayer2.Player", "Metadata")?;
|
|
||||||
let mut iter = metadata.as_iter().unwrap();
|
|
||||||
while let Some(key) = iter.next() {
|
|
||||||
if key.as_str() == Some("xesam:title") {
|
|
||||||
if let Some(title) = iter.next().unwrap().as_str() {
|
|
||||||
track_info.title = title.to_string();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if key.as_str() == Some("mpris:length") {
|
|
||||||
if let Some(length) = iter.next().unwrap().as_i64() {
|
|
||||||
track_info.duration = format_time(length / 1000000);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if key.as_str() == Some("xesam:artist") {
|
|
||||||
if let Some(mut artists) = iter.next().unwrap().as_iter() {
|
|
||||||
while let Some(artist) = artists.next() {
|
|
||||||
if let Some(mut line) = artist.as_iter() {
|
|
||||||
track_info.artist = line.next().unwrap().as_str().unwrap().to_string();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
let position: Box<dyn arg::RefArg> =
|
|
||||||
proxy.get("org.mpris.MediaPlayer2.Player", "Position")?;
|
|
||||||
track_info.position = format_time(position.as_i64().unwrap() / 1000000);
|
|
||||||
// ugly
|
|
||||||
let _status_text_box: Box<dyn arg::RefArg> =
|
|
||||||
proxy.get("org.mpris.MediaPlayer2.Player", "PlaybackStatus")?;
|
|
||||||
let _status_text = _status_text_box.as_str().unwrap();
|
|
||||||
match _status_text.as_ref() {
|
|
||||||
"Playing" => track_info.status = "▶".to_string(),
|
|
||||||
"Paused" => track_info.status = "⏸".to_string(),
|
|
||||||
_ => track_info.status = "⏹".to_string(),
|
|
||||||
};
|
|
||||||
players_vec.push(track_info);
|
|
||||||
}
|
|
||||||
for player in &players_vec {
|
|
||||||
if player.status == "▶".to_string() {
|
|
||||||
return Ok(player.clone());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Ok(players_vec[players_vec.len() - 1].clone())
|
|
||||||
}
|
|
||||||
|
|
||||||
fn format_time(sec: i64) -> String {
|
|
||||||
let minutes = sec / 60;
|
|
||||||
let secondes = sec % 60;
|
|
||||||
let result = format!("{:02}:{:02}", minutes, secondes);
|
|
||||||
result.to_string()
|
|
||||||
}
|
|
||||||
|
|
||||||
fn get_time(utc: bool, mut format: &str) {
|
|
||||||
// Format reference: https://docs.rs/chrono/0.4.10/chrono/format/strftime/index.html
|
|
||||||
if format.len() == 0 {
|
|
||||||
format = "%H:%M";
|
|
||||||
}
|
|
||||||
if utc {
|
|
||||||
let local_time = Local::now();
|
|
||||||
let utc_time = DateTime::<Utc>::from_utc(local_time.naive_utc(), Utc);
|
|
||||||
println!("{}", utc_time.format(format));
|
|
||||||
} else {
|
|
||||||
let local_time = Local::now();
|
|
||||||
println!("{}", local_time.format(format));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn main() {
|
fn main() {
|
||||||
let args: Vec<String> = env::args().collect();
|
let conf = config::read();
|
||||||
let help_text: &str = "Available commands -mb, -cb, -tl <TIME FORMAT>, -tu <TIME FORMAT>, -p";
|
// cpu - cpu usage bar
|
||||||
match args.len() {
|
// mem - mem usage bar
|
||||||
1 => {
|
// mpris - player info using MPRIS2 interface
|
||||||
panic!(help_text);
|
// mpd - player info using MPD native interface
|
||||||
}
|
// utctime - utc time
|
||||||
2 => match args[1].as_ref() {
|
// localtime - local time
|
||||||
"-cb" => cpu_load_bar(15),
|
match conf.action {
|
||||||
"-mb" => mem_load_bar(15),
|
config::Action::Cpu => utils::cpu_load_bar(15, &conf),
|
||||||
"-tl" => get_time(false, ""),
|
config::Action::Mem => utils::mem_load_bar(15, &conf),
|
||||||
"-tu" => get_time(true, ""),
|
config::Action::Mpris => utils::mpris(&conf),
|
||||||
"-p" => match player_info(get_player().unwrap()) {
|
config::Action::Utctime => utils::get_time(true, conf.ut_format),
|
||||||
Ok(mut track_info) => {
|
config::Action::Localtime => utils::get_time(false, conf.lt_format),
|
||||||
let mut title_len = 30;
|
config::Action::Mpd => utils::mpd(&conf),
|
||||||
let mut artist_len = 30;
|
|
||||||
let mut separator: String = " — ".to_string();
|
|
||||||
let max_shift = 6;
|
|
||||||
if track_info.artist.chars().count() == 0 {
|
|
||||||
separator = "".to_string();
|
|
||||||
title_len += artist_len;
|
|
||||||
}
|
|
||||||
if artist_len + max_shift >= track_info.artist.chars().count() {
|
|
||||||
artist_len = track_info.artist.chars().count()
|
|
||||||
}
|
|
||||||
if track_info.artist.len() > artist_len {
|
|
||||||
let mut artist: String = String::new();
|
|
||||||
let mut counter = 0;
|
|
||||||
for ch in track_info.artist.chars() {
|
|
||||||
if counter == artist_len {
|
|
||||||
artist.push_str("..");
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
artist.push(ch);
|
|
||||||
counter += 1;
|
|
||||||
}
|
|
||||||
track_info.artist = artist;
|
|
||||||
}
|
|
||||||
if title_len + max_shift >= track_info.title.chars().count() {
|
|
||||||
title_len = track_info.title.chars().count()
|
|
||||||
}
|
|
||||||
if track_info.title.len() > title_len {
|
|
||||||
let mut title: String = String::new();
|
|
||||||
let mut counter = 0;
|
|
||||||
for ch in track_info.title.chars() {
|
|
||||||
if counter == title_len {
|
|
||||||
title.push_str("..");
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
title.push(ch);
|
|
||||||
counter += 1;
|
|
||||||
}
|
|
||||||
track_info.title = title;
|
|
||||||
}
|
|
||||||
println!(
|
|
||||||
"#[none]#[bold]{}{}{}#[none]{}{}{}{} {}[{}/{}] {} {}#[default]",
|
|
||||||
TRACK_NAME,
|
|
||||||
track_info.title,
|
|
||||||
END,
|
|
||||||
separator,
|
|
||||||
TRACK_ARTIST,
|
|
||||||
track_info.artist,
|
|
||||||
END,
|
|
||||||
TRACK_TIME,
|
|
||||||
track_info.position,
|
|
||||||
track_info.duration,
|
|
||||||
track_info.status,
|
|
||||||
END,
|
|
||||||
);
|
|
||||||
}
|
|
||||||
Err(_e) => println!("No music playing"),
|
|
||||||
},
|
|
||||||
_ => panic!(help_text),
|
|
||||||
},
|
|
||||||
3 => match args[1].as_ref() {
|
|
||||||
"-tl" => get_time(false, args[2].as_ref()),
|
|
||||||
"-tu" => get_time(true, args[2].as_ref()),
|
|
||||||
_ => panic!(help_text),
|
|
||||||
},
|
|
||||||
_ => {
|
|
||||||
panic!(help_text);
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
267
src/utils.rs
Normal file
267
src/utils.rs
Normal file
@@ -0,0 +1,267 @@
|
|||||||
|
use crate::config;
|
||||||
|
use crate::dbus::blocking::stdintf::org_freedesktop_dbus::Properties;
|
||||||
|
use chrono::{DateTime, Local, Utc};
|
||||||
|
use dbus::{arg, blocking::Connection};
|
||||||
|
use std::time::Duration;
|
||||||
|
use sys_info;
|
||||||
|
use mpd::Client;
|
||||||
|
use std::process;
|
||||||
|
|
||||||
|
#[derive(Debug, Clone)]
|
||||||
|
pub struct TrackInfo {
|
||||||
|
title: String,
|
||||||
|
artist: String,
|
||||||
|
position: String,
|
||||||
|
duration: String,
|
||||||
|
status: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn to_bar(value: i32, max: i32, low: f32, mid: f32, config: &config::Config) {
|
||||||
|
let mut bar = "".to_string();
|
||||||
|
let bar_sym = "▮".to_string();
|
||||||
|
if (value as f32) / (max as f32) < low {
|
||||||
|
bar.push_str(&config.color_low);
|
||||||
|
} else if (value as f32) / (max as f32) < mid {
|
||||||
|
bar.push_str(&config.color_mid);
|
||||||
|
} else {
|
||||||
|
bar.push_str(&config.color_high);
|
||||||
|
}
|
||||||
|
for i in 0..max {
|
||||||
|
if i < value as i32 {
|
||||||
|
bar.push_str(&bar_sym);
|
||||||
|
} else {
|
||||||
|
bar.push_str(" ")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
bar.push_str(&config.color_end);
|
||||||
|
bar.push_str("|");
|
||||||
|
print!("{}", bar)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn mem_load_bar(bar_len: i32, config: &config::Config) {
|
||||||
|
let memory;
|
||||||
|
match sys_info::mem_info() {
|
||||||
|
Err(w) => panic!("{:?}", w),
|
||||||
|
Ok(mem_data) => memory = mem_data,
|
||||||
|
}
|
||||||
|
let len =
|
||||||
|
((memory.total - memory.avail) as f32 / (memory.total as f32) * bar_len as f32) as i32;
|
||||||
|
to_bar(len, bar_len, 0.7, 0.9, config);
|
||||||
|
print!("{:.0} MiB#[default]", memory.avail / 1024);
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn cpu_load_bar(bar_len: i32, config: &config::Config) {
|
||||||
|
let cpu_count = match sys_info::cpu_num() {
|
||||||
|
Ok(c) => c,
|
||||||
|
Err(e) => panic!("{:?}", e),
|
||||||
|
};
|
||||||
|
let la_one: f32 = match sys_info::loadavg() {
|
||||||
|
Ok(l) => l.one as f32,
|
||||||
|
Err(e) => panic!("{:?}", e),
|
||||||
|
};
|
||||||
|
let len: f32 = la_one as f32 / cpu_count as f32 * bar_len as f32;
|
||||||
|
to_bar(len as i32, bar_len, 0.3, 0.7, config);
|
||||||
|
print!("{:.2} LA1#[default]", la_one);
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn get_player() -> Result<Vec<String>, Box<dyn std::error::Error>> {
|
||||||
|
let conn = Connection::new_session()?;
|
||||||
|
let proxy = conn.with_proxy("org.freedesktop.DBus", "/", Duration::from_millis(5000));
|
||||||
|
let (names,): (Vec<String>,) = proxy.method_call("org.freedesktop.DBus", "ListNames", ())?;
|
||||||
|
let mut players: Vec<String> = Vec::new();
|
||||||
|
for name in names {
|
||||||
|
if name.contains("org.mpris.MediaPlayer2") {
|
||||||
|
players.push(name);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(players)
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn player_info(players: Vec<String>) -> Result<TrackInfo, Box<dyn std::error::Error>> {
|
||||||
|
let mut players_vec: Vec<TrackInfo> = Vec::new();
|
||||||
|
for player in players {
|
||||||
|
let mut track_info = TrackInfo {
|
||||||
|
artist: "".to_string(),
|
||||||
|
title: "".to_string(),
|
||||||
|
position: "".to_string(),
|
||||||
|
duration: "".to_string(),
|
||||||
|
status: "".to_string(),
|
||||||
|
};
|
||||||
|
let conn = Connection::new_session()?;
|
||||||
|
let proxy = conn.with_proxy(
|
||||||
|
player,
|
||||||
|
"/org/mpris/MediaPlayer2",
|
||||||
|
Duration::from_millis(5000),
|
||||||
|
);
|
||||||
|
let metadata: Box<dyn arg::RefArg> =
|
||||||
|
proxy.get("org.mpris.MediaPlayer2.Player", "Metadata")?;
|
||||||
|
let mut iter = metadata.as_iter().unwrap();
|
||||||
|
while let Some(key) = iter.next() {
|
||||||
|
if key.as_str() == Some("xesam:title") {
|
||||||
|
if let Some(title) = iter.next().unwrap().as_str() {
|
||||||
|
track_info.title = title.to_string();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if key.as_str() == Some("mpris:length") {
|
||||||
|
if let Some(length) = iter.next().unwrap().as_i64() {
|
||||||
|
track_info.duration = format_time(length / 1000000);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if key.as_str() == Some("xesam:artist") {
|
||||||
|
if let Some(mut artists) = iter.next().unwrap().as_iter() {
|
||||||
|
while let Some(artist) = artists.next() {
|
||||||
|
if let Some(mut line) = artist.as_iter() {
|
||||||
|
track_info.artist = line.next().unwrap().as_str().unwrap().to_string();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
let position: Box<dyn arg::RefArg> =
|
||||||
|
proxy.get("org.mpris.MediaPlayer2.Player", "Position")?;
|
||||||
|
track_info.position = format_time(position.as_i64().unwrap() / 1000000);
|
||||||
|
// ugly
|
||||||
|
let _status_text_box: Box<dyn arg::RefArg> =
|
||||||
|
proxy.get("org.mpris.MediaPlayer2.Player", "PlaybackStatus")?;
|
||||||
|
let _status_text = _status_text_box.as_str().unwrap();
|
||||||
|
match _status_text.as_ref() {
|
||||||
|
"Playing" => track_info.status = "▶".to_string(),
|
||||||
|
"Paused" => track_info.status = "⏸".to_string(),
|
||||||
|
_ => track_info.status = "⏹".to_string(),
|
||||||
|
};
|
||||||
|
players_vec.push(track_info);
|
||||||
|
}
|
||||||
|
for player in &players_vec {
|
||||||
|
if player.status == "▶".to_string() {
|
||||||
|
return Ok(player.clone());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Ok(players_vec[players_vec.len() - 1].clone())
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn format_time(sec: i64) -> String {
|
||||||
|
let minutes = sec / 60;
|
||||||
|
let secondes = sec % 60;
|
||||||
|
let result = format!("{:02}:{:02}", minutes, secondes);
|
||||||
|
result.to_string()
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn get_time(utc: bool, format: Option<String>) {
|
||||||
|
// Format reference: https://docs.rs/chrono/0.4.10/chrono/format/strftime/index.html
|
||||||
|
let fmt = match format {
|
||||||
|
Some(format) => format,
|
||||||
|
None => "%H:%M".to_string(),
|
||||||
|
};
|
||||||
|
|
||||||
|
if utc {
|
||||||
|
let local_time = Local::now();
|
||||||
|
let utc_time = DateTime::<Utc>::from_utc(local_time.naive_utc(), Utc);
|
||||||
|
println!("{}", utc_time.format(&fmt));
|
||||||
|
} else {
|
||||||
|
let local_time = Local::now();
|
||||||
|
println!("{}", local_time.format(&fmt));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn format_player(track_info: TrackInfo, config: &config::Config) {
|
||||||
|
let mut title_len = 30;
|
||||||
|
let mut artist_len = 30;
|
||||||
|
let mut artist_line: String = String::new();
|
||||||
|
let mut title_line: String = String::new();
|
||||||
|
let mut separator: String = " — ".to_string();
|
||||||
|
let max_shift = 6;
|
||||||
|
if track_info.artist.chars().count() == 0 {
|
||||||
|
separator = "".to_string();
|
||||||
|
title_len += artist_len;
|
||||||
|
}
|
||||||
|
if artist_len + max_shift >= track_info.artist.chars().count() {
|
||||||
|
artist_len = track_info.artist.chars().count()
|
||||||
|
}
|
||||||
|
if track_info.artist.len() > artist_len {
|
||||||
|
let mut counter = 0;
|
||||||
|
for ch in track_info.artist.chars() {
|
||||||
|
if counter == artist_len {
|
||||||
|
artist_line.push_str("..");
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
artist_line.push(ch);
|
||||||
|
counter += 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
if title_len + max_shift >= track_info.title.chars().count() {
|
||||||
|
title_len = track_info.title.chars().count()
|
||||||
|
}
|
||||||
|
if track_info.title.len() > title_len {
|
||||||
|
let mut counter = 0;
|
||||||
|
for ch in track_info.title.chars() {
|
||||||
|
if counter == title_len {
|
||||||
|
title_line.push_str("..");
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
title_line.push(ch);
|
||||||
|
counter += 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
println!(
|
||||||
|
"#[none]#[bold]{}{}{}#[none]{}{}{}{} {}[{}/{}] {} {}#[default]",
|
||||||
|
config.color_track_name,
|
||||||
|
title_line,
|
||||||
|
config.color_end,
|
||||||
|
separator,
|
||||||
|
config.color_track_artist,
|
||||||
|
artist_line,
|
||||||
|
config.color_end,
|
||||||
|
config.color_track_time,
|
||||||
|
track_info.position,
|
||||||
|
track_info.duration,
|
||||||
|
track_info.status,
|
||||||
|
config.color_end,
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn mpris(config: &config::Config) {
|
||||||
|
match player_info(get_player().unwrap()) {
|
||||||
|
Ok(track_info) => format_player(track_info, config),
|
||||||
|
Err(_e) => println!("No music playing"),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn mpd(config: &config::Config) {
|
||||||
|
let mut conn = match Client::connect(&config.mpd_server) {
|
||||||
|
Ok(conn) => conn,
|
||||||
|
Err(e) => {println!("Can't connect to MPD server. {}", e); process::exit(0x0001)}
|
||||||
|
};
|
||||||
|
let mut track_info = TrackInfo {
|
||||||
|
title: String::new(),
|
||||||
|
artist: String::new(),
|
||||||
|
position: String::new(),
|
||||||
|
duration: String::new(),
|
||||||
|
status: String::new(),
|
||||||
|
};
|
||||||
|
if let Some(song) = conn.currentsong().unwrap() {
|
||||||
|
if let Some(title) = song.title {
|
||||||
|
track_info.title = title
|
||||||
|
}
|
||||||
|
if let Some(artist) = song.tags.get("Artist") {
|
||||||
|
track_info.artist = artist.to_string()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if let Some(time) = conn.status().unwrap().time {
|
||||||
|
track_info.position = format_time(time.0.num_seconds() as i64);
|
||||||
|
track_info.duration = format_time(time.1.num_seconds() as i64);
|
||||||
|
}
|
||||||
|
let status = match conn.status() {
|
||||||
|
Ok(status) => {
|
||||||
|
match status.state {
|
||||||
|
mpd::State::Play => "▶".to_string(),
|
||||||
|
mpd::State::Pause => "⏸".to_string(),
|
||||||
|
mpd::State::Stop => "⏹".to_string(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Err(_) => {"⏹".to_string()},
|
||||||
|
};
|
||||||
|
track_info.status = status;
|
||||||
|
format_player(track_info, config)
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user