Fix player printing.

Cargo.toml

@@ -1,6 +1,7 @@
 [package]
 name = "tmux-helper"
-version = "0.2.2"
+version = "0.3.2"
+description = "Utility for printing system info for tmux status line."
 authors = ["Ultra Desu <ultradesu@hexor.ru>"]
 edition = "2018"
 
@@ -8,3 +9,5 @@ edition = "2018"
 sys-info = "*"
 dbus = "*"
 chrono = "*"
+mpd = "*"
+clap = "*"

src/config.rs

@@ -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
+}

src/main.rs

@@ -1,19 +1,9 @@
 extern crate chrono;
 extern crate dbus;
+extern crate mpd;
 
-use crate::dbus::blocking::stdintf::org_freedesktop_dbus::Properties;
-use chrono::{DateTime, Local, Utc};
-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]";
+mod config;
+mod utils;
 
 #[derive(Debug, Clone)]
 struct TrackInfo {
@@ -24,233 +14,20 @@ struct TrackInfo {
     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() {
-    let args: Vec<String> = env::args().collect();
-    let help_text: &str = "Available commands -mb, -cb, -tl <TIME FORMAT>, -tu <TIME FORMAT>, -p";
-    match args.len() {
-        1 => {
-            panic!(help_text);
-        }
-        2 => match args[1].as_ref() {
-            "-cb" => cpu_load_bar(15),
-            "-mb" => mem_load_bar(15),
-            "-tl" => get_time(false, ""),
-            "-tu" => get_time(true, ""),
-            "-p" => match player_info(get_player().unwrap()) {
-                Ok(mut track_info) => {
-                    let mut title_len = 30;
-                    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);
-        }
+    let conf = config::read();
+    // 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
+    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,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 shorten(line: String, max_len: usize, max_shift: usize) -> String {
+    let mut new_line = String::new();
+    let len = if max_len + max_shift >= line.chars().count() {
+        line.chars().count()
+    } else {
+        max_len
+    };
+    if line.len() > len {
+        let mut counter = 0;
+        for ch in line.chars() {
+            if counter == len {
+                new_line.push_str("..");
+                break;
+            }
+            new_line.push(ch);
+            counter += 1;
+        }
+    }
+    else {
+        new_line = line;
+    }
+    new_line
+}
+
+fn format_player(track_info: TrackInfo, config: &config::Config) {
+    let mut separator: String = " — ".to_string();
+    let mut max_len = 30;
+    if track_info.artist.chars().count() == 0 {
+        separator = "".to_string();
+        max_len = max_len * 2;
+    }
+    let artist_line = shorten(track_info.artist, max_len, 6);
+    let title_line = shorten(track_info.title, max_len, 6);
+    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(),
+    };
+//  println!("{:?}", conn.currentsong());
+    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 there is no tags and title.
+        if track_info.artist == track_info.title {
+            if let Some(name) = song.name {
+                track_info.title = name
+            }
+        }
+    }
+    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)
+}