Update README.md

.github/workflows/PKGBUILD

@@ -0,0 +1,33 @@
+# Maintainer: Alexandr Bogomyakov (ultradesu) <ab@hexor.ru>
+
+pkgname=tmux-helper
+pkgver=0.3.4
+pkgrel=1
+pkgdesc="Tmux helper"
+url="https://github.com/house-of-vanity/tmux-helper.git"
+arch=($CARCH)
+license=(WTFPL custom)
+depends=(tmux dbus)
+makedepends=(cargo git dbus)
+source=("git+https://github.com/house-of-vanity/$pkgname")
+sha512sums=('SKIP')
+
+pkgver() {
+  cd "$srcdir/$pkgname"
+  git describe --long --tags | awk -F '-' '{print $1}'| sed 's/^v//;s/\([^-]*-g\)/r\1/;s/-/./g'
+}
+
+prepare() {
+  cd "$srcdir/$pkgname"
+  cargo fetch --target $CARCH-unknown-linux-gnu
+}
+
+build() {
+  cd "$srcdir/$pkgname"
+  cargo build --release --frozen --all-targets
+}
+
+package() {
+  cd "$srcdir/$pkgname"
+  install -Dt "$pkgdir/usr/bin" target/release/$pkgname
+}

.github/workflows/build.yml

@@ -0,0 +1,95 @@
+name: Build and publish
+
+on:
+  push:
+    tags:
+      - 'v*'
+
+env:
+  BINARY_NAME: tmux-helper
+
+jobs:
+  build:
+    name: Build binaries
+    runs-on: ${{ matrix.os }}
+    strategy:
+      matrix:
+        os: [ubuntu-latest, macos-latest]
+        include:
+          - os: ubuntu-latest
+            platform_name: linux-amd64
+          - os: macos-latest
+            platform_name: macos-arm64
+    steps:
+      - uses: actions/checkout@v4
+
+      - name: Pre-build (Linux)
+        if: matrix.os == 'ubuntu-latest'
+        run: sudo apt install -y libdbus-1-dev pkg-config libdbus-1-3 libfuse-dev
+
+      - name: Pre-build (macOS)
+        if: matrix.os == 'macos-latest'
+        run: brew install dbus
+
+      - name: Build binary
+        run: cargo build --release
+
+      - name: Rename binary
+        run: |
+          mkdir -p ${{ env.BINARY_NAME }}_${{ matrix.platform_name }}
+          cp target/release/${{ env.BINARY_NAME }} ${{ env.BINARY_NAME }}_${{ matrix.platform_name }}/${{ env.BINARY_NAME }}
+
+      - uses: actions/upload-artifact@v4
+        with:
+          name: ${{ env.BINARY_NAME }}_${{ matrix.platform_name }}
+          path: ${{ env.BINARY_NAME }}_${{ matrix.platform_name }}
+
+  release:
+    name: Create Release Page
+    needs: build
+    runs-on: ubuntu-latest
+    permissions:
+      contents: write
+    outputs:
+      upload_url: ${{ steps.create_release.outputs.upload_url }}
+    steps:
+      - uses: actions/checkout@v4
+      - uses: ncipollo/release-action@v1
+        id: create_release
+        env:
+          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+        with:
+          allowUpdates: true
+          tag: ${{ github.ref_name }}
+
+  upload:
+    name: Upload Release Assets
+    needs: release
+    runs-on: ${{ matrix.os }}
+    strategy:
+      matrix:
+        os: [ubuntu-latest, macos-latest]
+        include:
+          - os: ubuntu-latest
+            platform_name: linux-amd64
+          - os: macos-latest
+            platform_name: macos-arm64
+    steps:
+      - uses: actions/checkout@v4
+
+      - uses: actions/download-artifact@v4
+        name: Download ${{ matrix.platform_name }} artifact
+        with:
+          name: ${{ env.BINARY_NAME }}_${{ matrix.platform_name }}
+          path: ${{ env.BINARY_NAME }}_${{ matrix.platform_name }}
+
+      - name: Upload Release Asset
+        uses: actions/upload-release-asset@v1
+        env:
+          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+        with:
+          upload_url: ${{ needs.release.outputs.upload_url }}
+          asset_path: ${{ env.BINARY_NAME }}_${{ matrix.platform_name }}/${{ env.BINARY_NAME }}
+          asset_name: ${{ env.BINARY_NAME }}_${{ matrix.platform_name }}
+          asset_content_type: application/octet-stream
+

.github/workflows/push-build.yml

@@ -1,20 +0,0 @@
-name: Rust
-
-on: push
-
-jobs:
-  build:
-
-    runs-on: ubuntu-latest
-
-    steps:
-    - uses: actions/checkout@v1
-    - name: Pre-build
-      run: sudo apt install -y libdbus-1-dev pkg-config libdbus-1-3
-    - name: Build
-      run: cargo build --verbose --release
-      env:
-        RUST_BACKTRACE: 1
-    - name: Strip
-      run: strip target/release/tmux-helper
-

.github/workflows/release-upload.yml

@@ -1,25 +0,0 @@
-name: Rust
-
-on: release
-
-jobs:
-  build:
-
-    runs-on: ubuntu-latest
-
-    steps:
-    - uses: actions/checkout@v1
-    - name: Pre-build
-      run: sudo apt install -y libdbus-1-dev pkg-config libdbus-1-3
-    - name: Build
-      run: cargo build --verbose --release
-      env:
-        RUST_BACKTRACE: 1
-    - name: Strip
-      run: strip target/release/tmux-helper
-    - name: Upload to release
-      uses: JasonEtco/upload-to-release@master
-      with:
-        args: target/release/tmux-helper application/x-pie-executable
-      env:
-        GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}

Cargo.toml

@@ -1,9 +1,14 @@
 [package]
 name = "tmux-helper"
-version = "0.1.0"
+version = "0.4.0"
+description = "Utility for printing system info for tmux status line."
 authors = ["Ultra Desu <ultradesu@hexor.ru>"]
-edition = "2018"
+edition = "2021"
 
 [dependencies]
-sys-info = "*"
-dbus = "*"
+sysinfo = "0.36.0"
+dbus = "0.9"
+chrono = "0.4"
+mpd = "0.1"
+clap = "4"
+size_format = "1"

README.md

@@ -1,16 +1,59 @@
 # Tmux helper
-Small app that perform system check and print TMUX friendly output.
+Small app that perform system check and print TMUX friendly output. Prebuilded for MacOS M chip and Linux AMD64
 
-![Preview](.github/prev.png)
+<img width="1495" height="1264" alt="image" src="https://github.com/user-attachments/assets/7b9ffc97-0b59-4028-9b5d-f29347d16000" />
 
-### Features:
-1. `tmux-helper -cb` print cpu load with bar.
-2. `tmux-helper -mb` print memory usage with bar.
-3. `tmux-helper -p` show current player status using d-bus.
 
 ### Building 
 `cargo build --release`
 or get binary on release page
 
-### Customization
-Colours are hardcoded but it's easy to change it. It defined as CONST in very beginning of main.rs file.
+### Fetures
+```shell
+Utility for printing system info for tmux status line.
+
+Usage: tmux-helper [OPTIONS]
+
+Options:
+  -c, --cpu
+          Print cpu load bar.
+  -m, --mem
+          Print mem usage bar.
+      --low <low>
+          Low threshold (0.0 - 1.0) [default: 0.7]
+      --mid <mid>
+          Mid threshold (0.0 - 1.0) [default: 0.9]
+  -p, --mpris
+          Show player info using MPRIS2 interface.
+  -d, --mpd
+          Show mpd player using MPD native protocol.
+  -l, --localtime [<localtime>]
+          Local time
+  -u, --utctime [<utctime>]
+          UTC time
+  -s, --symbol [<bar_symbol>]
+          Symbol to build bar [default: ▮]
+  -e, --empty-symbol [<bar_empty_symbol>]
+          Symbol to represent the empty part of the bar [default: ▯]
+  -a, --mpd-address <mpd_address>
+          <ADDR>:<PORT> of MPD server. [default: 127.0.0.1:6600]
+      --COLOR_LOW <COLOR_LOW>
+          CPU and MEM bar color while low usage. [default: 119]
+      --COLOR_MID <COLOR_MID>
+          CPU and MEM bar color while mid usage. [default: 220]
+      --COLOR_HIGH <COLOR_HIGH>
+          CPU and MEM bar color while high usage. [default: 197]
+      --COLOR_TRACK_NAME <COLOR_TRACK_NAME>
+          Color of track name filed. [default: 46]
+      --COLOR_TRACK_ARTIST <COLOR_TRACK_ARTIST>
+          Color of artist name filed. [default: 46]
+      --COLOR_TRACK_TIME <COLOR_TRACK_TIME>
+          Color of playing time field. [default: 153]
+      --COLOR_END <COLOR_END>
+          Default color using to terminate others. [default: 153]
+  -h, --help
+          Print help
+  -V, --version
+          Print version
+```
+

src/config.rs

@@ -0,0 +1,245 @@
+use clap::{Arg, Command};
+
+#[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 bar_symbol: Option<String>,
+    pub bar_empty_symbol: Option<String>,
+    pub low_threshold: f32,
+    pub mid_threshold: f32,
+    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 {
+    let cli_args = Command::new(env!("CARGO_PKG_NAME"))
+        .version(env!("CARGO_PKG_VERSION"))
+        .author(env!("CARGO_PKG_AUTHORS"))
+        .about(env!("CARGO_PKG_DESCRIPTION"))
+        .arg(
+            Arg::new("cpu")
+                .short('c')
+                .long("cpu")
+                .help("Print cpu load bar.")
+                .action(clap::ArgAction::SetTrue)
+                .conflicts_with_all(["mem", "mpris", "mpd", "localtime", "utctime"]),
+        )
+        .arg(
+            Arg::new("mem")
+                .short('m')
+                .long("mem")
+                .action(clap::ArgAction::SetTrue)
+                .help("Print mem usage bar."),
+        )
+        .arg(
+            Arg::new("low")
+                .long("low")
+                .help("Low threshold (0.0 - 1.0)")
+                .value_parser(clap::value_parser!(f32))
+                .default_value("0.7"),
+        )
+        .arg(
+            Arg::new("mid")
+                .long("mid")
+                .help("Mid threshold (0.0 - 1.0)")
+                .value_parser(clap::value_parser!(f32))
+                .default_value("0.9"),
+        )
+        .arg(
+            Arg::new("mpris")
+                .short('p')
+                .long("mpris")
+                .action(clap::ArgAction::SetTrue)
+                .help("Show player info using MPRIS2 interface."),
+        )
+        .arg(
+            Arg::new("mpd")
+                .short('d')
+                .long("mpd")
+                .action(clap::ArgAction::SetTrue)
+                .help("Show mpd player using MPD native protocol."),
+        )
+        .arg(
+            Arg::new("localtime")
+                .short('l')
+                .long("localtime")
+                .help("Local time")
+                .num_args(0..=1)
+                .default_missing_value("%H:%M"),
+        )
+        .arg(
+            Arg::new("utctime")
+                .short('u')
+                .long("utctime")
+                .help("UTC time")
+                .num_args(0..=1)
+                .default_missing_value("%H:%M"),
+        )
+        .arg(
+            Arg::new("bar_symbol")
+                .short('s')
+                .long("symbol")
+                .help("Symbol to build bar")
+                .num_args(0..=1)
+                .default_value("▮"),
+        )
+        .arg(
+            Arg::new("bar_empty_symbol")
+                .short('e')
+                .long("empty-symbol")
+                .help("Symbol to represent the empty part of the bar")
+                .num_args(0..=1)
+                .default_value("▯"),
+        )
+        .arg(
+            Arg::new("mpd_address")
+                .short('a')
+                .long("mpd-address")
+                .help("<ADDR>:<PORT> of MPD server.")
+                .default_value("127.0.0.1:6600"),
+        )
+        .arg(
+            Arg::new("COLOR_LOW")
+                .long("COLOR_LOW")
+                .help("CPU and MEM bar color while low usage.")
+                .default_value("119"),
+        )
+        .arg(
+            Arg::new("COLOR_MID")
+                .long("COLOR_MID")
+                .help("CPU and MEM bar color while mid usage.")
+                .default_value("220"),
+        )
+        .arg(
+            Arg::new("COLOR_HIGH")
+                .long("COLOR_HIGH")
+                .help("CPU and MEM bar color while high usage.")
+                .default_value("197"),
+        )
+        .arg(
+            Arg::new("COLOR_TRACK_NAME")
+                .long("COLOR_TRACK_NAME")
+                .help("Color of track name filed.")
+                .default_value("46"),
+        )
+        .arg(
+            Arg::new("COLOR_TRACK_ARTIST")
+                .long("COLOR_TRACK_ARTIST")
+                .help("Color of artist name filed.")
+                .default_value("46"),
+        )
+        .arg(
+            Arg::new("COLOR_TRACK_TIME")
+                .long("COLOR_TRACK_TIME")
+                .help("Color of playing time field.")
+                .default_value("153"),
+        )
+        .arg(
+            Arg::new("COLOR_END")
+                .long("COLOR_END")
+                .help("Default color using to terminate others.")
+                .default_value("153"),
+        )
+        .get_matches();
+
+    let lt_format = cli_args
+        .get_one::<String>("localtime")
+        .map(|s| s.to_string());
+    let ut_format = cli_args.get_one::<String>("utctime").map(|s| s.to_string());
+    let bar_symbol = cli_args
+        .get_one::<String>("bar_symbol")
+        .map(|s| s.to_string());
+    let bar_empty_symbol = cli_args
+        .get_one::<String>("bar_empty_symbol")
+        .map(|s| s.to_string());
+
+    let mut cfg = Config {
+        action: Action::Cpu,
+        mpd_server: cli_args
+            .get_one::<String>("mpd_address")
+            .unwrap()
+            .to_string(),
+        lt_format,
+        ut_format,
+        bar_symbol,
+        bar_empty_symbol,
+        low_threshold: *cli_args.get_one::<f32>("low").unwrap(),
+        mid_threshold: *cli_args.get_one::<f32>("mid").unwrap(),
+        color_low: colorize(cli_args.get_one::<String>("COLOR_LOW").unwrap().to_string()),
+        color_mid: colorize(cli_args.get_one::<String>("COLOR_MID").unwrap().to_string()),
+        color_high: colorize(
+            cli_args
+                .get_one::<String>("COLOR_HIGH")
+                .unwrap()
+                .to_string(),
+        ),
+        color_track_name: colorize(
+            cli_args
+                .get_one::<String>("COLOR_TRACK_NAME")
+                .unwrap()
+                .to_string(),
+        ),
+        color_track_artist: colorize(
+            cli_args
+                .get_one::<String>("COLOR_TRACK_ARTIST")
+                .unwrap()
+                .to_string(),
+        ),
+        color_track_time: colorize(
+            cli_args
+                .get_one::<String>("COLOR_TRACK_TIME")
+                .unwrap()
+                .to_string(),
+        ),
+        color_end: colorize(cli_args.get_one::<String>("COLOR_END").unwrap().to_string()),
+    };
+
+    if cli_args.get_flag("cpu") {
+        cfg.action = Action::Cpu;
+    }
+    if cli_args.get_flag("mem") {
+        cfg.action = Action::Mem;
+    }
+    if cli_args.contains_id("localtime") {
+        cfg.action = Action::Localtime;
+    }
+    if cli_args.contains_id("utctime") {
+        cfg.action = Action::Utctime;
+    }
+    if cli_args.get_flag("mpris") {
+        cfg.action = Action::Mpris;
+    }
+    if cli_args.get_flag("mpd") {
+        cfg.action = Action::Mpd;
+    }
+
+    cfg
+}

src/main.rs

@@ -1,203 +1,24 @@
-extern crate dbus;
-use crate::dbus::blocking::stdintf::org_freedesktop_dbus::Properties;
-use dbus::{arg, blocking::Connection};
-use std::{env, fs, time::Duration};
-use sys_info;
-
-const LOW: &str = "#[fg=colour2]";
-const MID: &str = "#[fg=colour3]";
-const HIGH: &str = "#[fg=colour1]";
-const END: &str = "#[fg=colour7]";
-const TRACK_NAME: &str = "#[fg=colour3]";
-const TRACK_ARTIST: &str = "#[fg=colour3]";
-const TRACK_TIME: &str = "#[bg=colour252 fg=colour235 bold]";
+mod config;
+mod utils;
 
 #[derive(Debug, Clone)]
-struct TrackInfo {
-    title: String,
-    artist: String,
-    position: String,
-    duration: String,
-    status: String,
-}
-
-fn read_file(file_path: &str) -> String {
-    fs::read_to_string(file_path).expect("Cant read file.")
-}
-
-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", memory.avail / 1024);
-}
-
-fn cpu_load_bar(bar_len: i32) {
-    let load = read_file("/proc/loadavg");
-    let load_data = load.split_whitespace().collect::<Vec<&str>>();
-    let _cpu_count = read_file("/proc/cpuinfo");
-    let cpu_count = _cpu_count.matches("model name").count();
-    let one: f32 = load_data[0].parse().unwrap();
-    let len: f32 = 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", 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);
-            //println!("{}", 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()
+pub struct TrackInfo {
+    pub title: String,
+    pub artist: String,
+    pub position: String,
+    pub duration: String,
+    pub status: String,
 }
 
 fn main() {
-    let args: Vec<String> = env::args().collect();
-    let help_text: &str = "Available commands -mb, -cb";
-    match args.len() {
-        1 => {
-            panic!(help_text);
-        }
-        2 => match args[1].as_ref() {
-            "-cb" => cpu_load_bar(15),
-            "-mb" => mem_load_bar(15),
-            "-p" => match player_info(get_player().unwrap()) {
-                Ok(mut track_info) => {
-                    let title_len = 30;
-                    let artist_len = 30;
-                    if track_info.title.len() >= title_len {
-                        track_info.title.truncate(title_len);
-                        track_info.title.push_str("..");
-                    }
-                    println!("{}, {}", track_info.artist, track_info.artist.len());
-                    if track_info.artist.len() >= artist_len {
-                        track_info.artist.truncate(artist_len);
-                        track_info.artist.push_str("..");
-                    }
-                    println!(
-                        "#[none]#[bold]{}{}{}#[none]{} - {}{} {}[{}/{}] {} {}",
-                        TRACK_NAME,
-                        track_info.title,
-                        END,
-                        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),
-        },
-        _ => {
-            panic!(help_text);
-        }
+    let conf = config::read();
+
+    match conf.action {
+        config::Action::Cpu => utils::cpu_load_bar(15, &conf),
+        config::Action::Mem => utils::mem_load_bar(15, &conf),
+        config::Action::Mpris => utils::mpris(&conf),
+        config::Action::Utctime => utils::get_time(true, conf.ut_format),
+        config::Action::Localtime => utils::get_time(false, conf.lt_format),
+        config::Action::Mpd => utils::mpd(&conf),
     }
 }

src/utils.rs

@@ -0,0 +1,272 @@
+use crate::config;
+use chrono::{Local, Utc};
+use dbus::arg::RefArg;
+use dbus::blocking::stdintf::org_freedesktop_dbus::Properties;
+use dbus::{arg, blocking::Connection};
+use mpd::Client;
+use size_format::SizeFormatterBinary;
+use std::process;
+use std::thread;
+use std::time::Duration;
+use sysinfo::System;
+
+#[derive(Debug, Clone)]
+pub struct TrackInfo {
+    pub title: String,
+    pub artist: String,
+    pub position: String,
+    pub duration: String,
+    pub status: String,
+}
+
+pub fn to_bar(value: i32, max: i32, low: f32, mid: f32, config: &config::Config) {
+    let mut bar = "".to_string();
+    let ratio = (value as f32) / (max as f32);
+    bar.push_str(if ratio < low {
+        &config.color_low
+    } else if ratio < mid {
+        &config.color_mid
+    } else {
+        &config.color_high
+    });
+    let symbol = config.bar_symbol.as_deref().unwrap_or("█");
+    let empty = config.bar_empty_symbol.as_deref().unwrap_or(" ");
+    for i in 0..max {
+        bar.push_str(if i < value { symbol } else { empty });
+    }
+    bar.push_str(&config.color_end);
+    bar.push('|');
+    print!("{}", bar);
+}
+
+pub fn mem_load_bar(bar_len: i32, config: &config::Config) {
+    let mut sys = System::new_all();
+    sys.refresh_memory();
+    
+    let total_memory = sys.total_memory();
+    let used_memory = sys.used_memory();
+    
+    // On macOS sysinfo.used_memory() includes caches and compressed memory
+    // Try a more conservative estimate
+    // Usually real used memory is about 30-50% of what sysinfo shows
+    #[cfg(target_os = "macos")]
+    let actual_used = used_memory * 40 / 100; // Approximately 40% of sysinfo.used_memory
+    
+    #[cfg(not(target_os = "macos"))]
+    let actual_used = used_memory;
+    
+    let actual_free = total_memory - actual_used;
+    
+    let used_ratio = actual_used as f32 / total_memory as f32;
+    let len = (used_ratio * bar_len as f32) as i32;
+    
+    to_bar(
+        len,
+        bar_len,
+        config.low_threshold,
+        config.mid_threshold,
+        config,
+    );
+    
+    // Show: used/free
+    print!(
+        "{}/{} #[default]",
+        SizeFormatterBinary::new(actual_used),
+        SizeFormatterBinary::new(actual_free)
+    );
+}
+
+pub fn cpu_load_bar(bar_len: i32, config: &config::Config) {
+    let mut sys = System::new_all();
+    
+    // Update CPU information
+    sys.refresh_cpu_all();
+    
+    // Wait a bit to get accurate CPU usage data
+    thread::sleep(sysinfo::MINIMUM_CPU_UPDATE_INTERVAL);
+    sys.refresh_cpu_all();
+    
+    let cpu_count = sys.cpus().len();
+    
+    // Get average CPU usage
+    let cpu_usage: f32 = sys.cpus().iter().map(|cpu| cpu.cpu_usage()).sum::<f32>() / cpu_count as f32;
+    let cpu_load_ratio = cpu_usage / 100.0; // sysinfo returns percentages
+    
+    let len = (cpu_load_ratio * bar_len as f32).round() as i32;
+    
+    to_bar(
+        len,
+        bar_len,
+        config.low_threshold,
+        config.mid_threshold,
+        config,
+    );
+
+    print!("{:.1}%#[default]", cpu_usage);
+}
+
+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_secs(5));
+    let (names,): (Vec<String>,) = proxy.method_call("org.freedesktop.DBus", "ListNames", ())?;
+    Ok(names
+        .into_iter()
+        .filter(|n| n.contains("org.mpris.MediaPlayer2"))
+        .collect())
+}
+
+pub fn player_info(players: Vec<String>) -> Result<TrackInfo, Box<dyn std::error::Error>> {
+    for player in players {
+        let conn = Connection::new_session()?;
+        let proxy = conn.with_proxy(player, "/org/mpris/MediaPlayer2", Duration::from_secs(5));
+        let metadata: arg::PropMap = proxy.get("org.mpris.MediaPlayer2.Player", "Metadata")?;
+
+        let title = metadata
+            .get("xesam:title")
+            .and_then(|v| v.as_str())
+            .unwrap_or("")
+            .to_string();
+        let artist = metadata
+            .get("xesam:artist")
+            .and_then(|v| v.as_iter())
+            .and_then(|mut artists| artists.next().and_then(|a| a.as_str()))
+            .unwrap_or("")
+            .to_string();
+        let duration_us = metadata
+            .get("mpris:length")
+            .and_then(|v| v.as_i64())
+            .unwrap_or(0);
+        let position_us: i64 = proxy.get("org.mpris.MediaPlayer2.Player", "Position")?;
+        let status_text: String = proxy.get("org.mpris.MediaPlayer2.Player", "PlaybackStatus")?;
+
+        let status = match status_text.as_str() {
+            "Playing" => "▶",
+            "Paused" => "⏸",
+            _ => "⏹",
+        }
+        .to_string();
+
+        let track_info = TrackInfo {
+            title,
+            artist,
+            position: format_time(position_us / 1_000_000),
+            duration: format_time(duration_us / 1_000_000),
+            status,
+        };
+
+        if track_info.status == "▶" {
+            return Ok(track_info);
+        }
+    }
+    Err("No active player".into())
+}
+
+pub fn format_time(sec: i64) -> String {
+    format!("{:02}:{:02}", sec / 60, sec % 60)
+}
+
+pub fn get_time(utc: bool, format: Option<String>) {
+    let fmt = format.unwrap_or_else(|| "%H:%M".to_string());
+    let now = if utc {
+        Utc::now().format(&fmt)
+    } else {
+        Local::now().format(&fmt)
+    };
+    println!("{}", now);
+}
+
+fn shorten(line: &str, max_len: usize) -> String {
+    if line.chars().count() > max_len {
+        format!("{}..", line.chars().take(max_len).collect::<String>())
+    } else {
+        line.to_string()
+    }
+}
+
+fn format_player(track_info: TrackInfo, config: &config::Config) {
+    let separator = if track_info.artist.is_empty() {
+        ""
+    } else {
+        " — "
+    };
+    let max_len = if track_info.artist.is_empty() { 60 } else { 30 };
+
+    let artist_line = shorten(&track_info.artist, max_len);
+    let title_line = shorten(&track_info.title, max_len);
+
+    if track_info.position == "00:00" || track_info.duration.is_empty() {
+        println!(
+            "#[bold]{}{}{}{}{}{} {}{} {}#[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.status
+        );
+    } else {
+        println!(
+            "#[bold]{}{}{}{}{}{} {}[{}/{}] {}{}{}#[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_or_default()) {
+        Ok(track_info) => format_player(track_info, config),
+        Err(_) => println!("No music playing"),
+    }
+}
+
+pub fn mpd(config: &config::Config) {
+    let mut conn = Client::connect(&config.mpd_server).unwrap_or_else(|e| {
+        println!("Can't connect to MPD server. {}", e);
+        process::exit(1);
+    });
+
+    let song = conn.currentsong().unwrap_or(None);
+    let status = conn.status().unwrap();
+
+    let artist = song
+        .as_ref()
+        .and_then(|s| s.tags.iter().find(|(k, _)| k == "Artist").map(|(_, v)| v))
+        .cloned()
+        .unwrap_or_default();
+
+    let title = song
+        .as_ref()
+        .and_then(|s| s.title.clone().or_else(|| s.name.clone()))
+        .unwrap_or_default();
+
+    let (position, duration) = status.time.unwrap_or_default();
+
+    let track_info = TrackInfo {
+        title,
+        artist,
+        position: format_time(position.as_secs() as i64),
+        duration: format_time(duration.as_secs() as i64),
+        status: match status.state {
+            mpd::State::Play => "▶",
+            mpd::State::Pause => "⏸",
+            mpd::State::Stop => "⏹",
+        }
+        .to_string(),
+    };
+
+    format_player(track_info, config);
+}