cmus-notify/src/cmus/player_settings.rs

143 lines
3.9 KiB
Rust

use std::num::ParseIntError;
use std::str::FromStr;
use crate::cmus::CmusError;
#[derive(Debug, PartialEq)]
pub struct PlayerSettings {
pub repeat: bool,
pub shuffle: Shuffle,
pub aa_mode: AAAMode,
pub volume: Volume,
}
#[derive(Debug, PartialEq, Default)]
pub enum Shuffle {
#[default]
Off,
Tracks,
Albums,
}
#[derive(Debug, PartialEq, Default)]
pub struct Volume {
pub left: u8,
pub right: u8,
}
#[derive(Debug, PartialEq, Default)]
pub enum AAAMode {
#[default]
All,
Album,
Artist,
}
impl FromStr for AAAMode {
type Err = CmusError;
fn from_str(s: &str) -> Result<Self, Self::Err> {
match s {
"all" => Ok(Self::All),
"album" => Ok(Self::Album),
"artist" => Ok(Self::Artist),
_ => Err(CmusError::UnknownAAAMode(s.to_string()))
}
}
}
impl FromStr for Shuffle {
type Err = CmusError;
fn from_str(s: &str) -> Result<Self, Self::Err> {
match s {
"off" => Ok(Self::Off),
"tracks" => Ok(Self::Tracks),
"albums" => Ok(Self::Albums),
_ => Err(CmusError::UnknownShuffleMode(s.to_string()))
}
}
}
impl FromStr for PlayerSettings {
type Err = CmusError;
fn from_str(s: &str) -> Result<Self, Self::Err> {
let mut repeat = false;
let mut shuffle = Shuffle::default();
let mut aa_mode = AAAMode::default();
let mut volume = Volume::default();
for line in s.lines() {
if line.starts_with("set ") {
let line = &line[4..];
let (key, value) = line.split_once(" ").ok_or(CmusError::UnknownError("Corrupted cmus response".to_string()))?;
match key {
"repeat" => repeat = value == "true",
"shuffle" => shuffle = Shuffle::from_str(value)?,
"aaa_mode" => aa_mode = AAAMode::from_str(value)?,
"vol_left" => volume.left = value.parse().map_err(|e: ParseIntError| CmusError::UnknownError(e.to_string()))?,
"vol_right" => volume.right = value.parse().map_err(|e: ParseIntError| CmusError::UnknownError(e.to_string()))?,
_ => {}
}
}
}
Ok(Self {
repeat,
shuffle,
aa_mode,
volume,
})
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_parse_aaamode_from_str() {
let all = AAAMode::from_str("all");
let album = AAAMode::from_str("album");
let artist = AAAMode::from_str("artist");
let unknown = AAAMode::from_str("unknown");
assert_eq!(all, Ok(AAAMode::All));
assert_eq!(album, Ok(AAAMode::Album));
assert_eq!(artist, Ok(AAAMode::Artist));
assert_eq!(unknown, Err(CmusError::UnknownAAAMode("unknown".to_string())));
}
#[test]
fn test_parse_shuffle_mode_from_str() {
let off = Shuffle::from_str("off");
let tracks = Shuffle::from_str("tracks");
let albums = Shuffle::from_str("albums");
let unknown = Shuffle::from_str("unknown");
assert_eq!(off, Ok(Shuffle::Off));
assert_eq!(tracks, Ok(Shuffle::Tracks));
assert_eq!(albums, Ok(Shuffle::Albums));
assert_eq!(unknown, Err(CmusError::UnknownShuffleMode("unknown".to_string())));
}
#[test]
fn test_parse_player_settings_from_str() {
let setting_sample = include_str!(
"../../tests/samples/player_settings_mode-artist_vol-46_repeat-false_repeat_current-false_shuffle-tracks.txt");
let settings = PlayerSettings::from_str(setting_sample);
assert_eq!(settings, Ok(PlayerSettings {
repeat: false,
shuffle: Shuffle::Tracks,
aa_mode: AAAMode::Artist,
volume: Volume {
left: 46,
right: 46,
}
}));
}
}