Compare commits

...

5 commits

Author SHA1 Message Date
MrDulfin
9457c5c996 fixed a function and removed Album Art function 2024-05-19 22:29:30 -04:00
MrDulfin
b04a166c1b Merge branch 'Library-Rework' into controller 2024-05-19 22:25:41 -04:00
MrDulfin
56040bfd28 Updated Controller and added Player Trait 2024-05-19 18:48:29 -04:00
MrDulfin
94e6c25219 made a few minor changes 2024-05-07 18:28:03 -04:00
MrDulfin
f02f5bca41 first draft of library Rework 2024-04-03 01:00:52 -04:00
13 changed files with 541 additions and 758 deletions

View file

@ -12,7 +12,15 @@ keywords = []
categories = [] categories = []
[dependencies] [dependencies]
file-format = { version = "0.23.0", features = ["reader-asf", "reader-ebml", "reader-mp4", "reader-rm", "reader-txt", "reader-xml", "serde"] } file-format = { version = "0.23.0", features = [
"reader-asf",
"reader-ebml",
"reader-mp4",
"reader-rm",
"reader-txt",
"reader-xml",
"serde",
] }
lofty = "0.18.2" lofty = "0.18.2"
serde = { version = "1.0.195", features = ["derive"] } serde = { version = "1.0.195", features = ["derive"] }
walkdir = "2.4.0" walkdir = "2.4.0"
@ -38,3 +46,5 @@ deunicode = "1.4.2"
opener = { version = "0.7.0", features = ["reveal"] } opener = { version = "0.7.0", features = ["reveal"] }
tempfile = "3.10.1" tempfile = "3.10.1"
listenbrainz = "0.7.0" listenbrainz = "0.7.0"
discord-rpc-client = "0.4.0"
nestify = "0.3.3"

View file

@ -14,7 +14,10 @@ pub mod music_controller{
pub mod queue; pub mod queue;
} }
pub mod music_player; pub mod music_player {
pub mod gstreamer;
pub mod player;
}
#[allow(clippy::module_inception)] #[allow(clippy::module_inception)]
pub mod config { pub mod config {
pub mod config; pub mod config;

View file

@ -1,95 +1,103 @@
use std::{ // use std::{
sync::{Arc, RwLock}, // sync::{Arc, RwLock},
error::Error, // error::Error,
}; // };
use listenbrainz::ListenBrainz; // use discord_rpc_client::Client;
use uuid::Uuid; // use listenbrainz::ListenBrainz;
// use uuid::Uuid;
use crate::{ // use crate::{
config::config::Config, music_controller::controller::{Controller, QueueCmd, QueueResponse}, music_storage::library::{MusicLibrary, Song, Tag} // config::config::Config, music_controller::controller::{Controller, QueueCmd, QueueResponse}, music_storage::library::{MusicLibrary, Song, Tag}
}; // };
use super::controller::DatabaseResponse; // use super::controller::DatabaseResponse;
impl Controller { // impl Controller {
pub fn listenbrainz_authenticate(&mut self) -> Result<ListenBrainz, Box<dyn Error>> { // pub fn listenbrainz_authenticate(&mut self) -> Result<ListenBrainz, Box<dyn Error>> {
let config = &self.config.read().unwrap(); // let config = &self.config.read().unwrap();
let mut client = ListenBrainz::new(); // let mut client = ListenBrainz::new();
let lbz_token = match &config.connections.listenbrainz_token { // let lbz_token = match &config.connections.listenbrainz_token {
Some(token) => token, // Some(token) => token,
None => todo!("No ListenBrainz token in config") // None => todo!("No ListenBrainz token in config")
}; // };
if !client.is_authenticated() { // if !client.is_authenticated() {
client.authenticate(lbz_token)?; // client.authenticate(lbz_token)?;
} // }
Ok(client) // Ok(client)
} // }
pub fn lbz_scrobble(&self, client: ListenBrainz, uuid: Uuid) -> Result<(), Box<dyn Error>> { // pub fn lbz_scrobble(&self, client: ListenBrainz, uuid: Uuid) -> Result<(), Box<dyn Error>> {
let config = &self.config.read().unwrap(); // let config = &self.config.read().unwrap();
&self.db_mail.send(super::controller::DatabaseCmd::QueryUuid(uuid)); // &self.db_mail.send(super::controller::DatabaseCmd::QueryUuid(uuid));
let res = &self.db_mail.recv()?; // let res = &self.db_mail.recv()?;
let song = match res { // let song = match res {
DatabaseResponse::Song(song) => song, // DatabaseResponse::Song(song) => song,
_ => todo!() // _ => todo!()
}; // };
let unknown = &"unknown".to_string(); // let unknown = &"unknown".to_string();
let artist = song.get_tag(&Tag::Artist).unwrap_or(unknown); // let artist = song.get_tag(&Tag::Artist).unwrap_or(unknown);
let track = song.get_tag(&Tag::Title).unwrap_or(unknown); // let track = song.get_tag(&Tag::Title).unwrap_or(unknown);
let release = song.get_tag(&Tag::Album).map(|rel| rel.as_str()); // let release = song.get_tag(&Tag::Album).map(|rel| rel.as_str());
client.listen(artist, track, release)?; // client.listen(artist, track, release)?;
Ok(()) // Ok(())
} // }
pub fn lbz_now_playing(&self, client: ListenBrainz, uuid: Uuid) -> Result<(), Box<dyn Error>> { // pub fn lbz_now_playing(&self, client: ListenBrainz, uuid: Uuid) -> Result<(), Box<dyn Error>> {
let config = &self.config.read().unwrap(); // let config = &self.config.read().unwrap();
&self.db_mail.send(super::controller::DatabaseCmd::QueryUuid(uuid)); // &self.db_mail.send(super::controller::DatabaseCmd::QueryUuid(uuid));
let res = &self.db_mail.recv()?; // let res = &self.db_mail.recv()?;
let song = match res { // let song = match res {
DatabaseResponse::Song(song) => song, // DatabaseResponse::Song(song) => song,
_ => todo!() // _ => todo!()
}; // };
let unknown = &"unknown".to_string(); // let unknown = &"unknown".to_string();
let artist = song.get_tag(&Tag::Artist).unwrap_or(unknown); // let artist = song.get_tag(&Tag::Artist).unwrap_or(unknown);
let track = song.get_tag(&Tag::Title).unwrap_or(unknown); // let track = song.get_tag(&Tag::Title).unwrap_or(unknown);
let release = song.get_tag(&Tag::Album).map(|rel| rel.as_str()); // let release = song.get_tag(&Tag::Album).map(|rel| rel.as_str());
client.listen(artist, track, release)?; // client.listen(artist, track, release)?;
Ok(()) // Ok(())
} // }
}
#[cfg(test)] // pub fn discord_song_change(client: &mut Client,song: Song) {
mod test_super { // client.set_activity(|a| {
use std::{thread::sleep, time::Duration}; // a.state(format!("Listening to {}", song.get_tag(&Tag::Title).unwrap()))
// .into()
// });
// }
// }
use super::*; // #[cfg(test)]
use crate::config::config::tests::read_config_lib; // mod test_super {
// use std::{thread::sleep, time::Duration};
#[test] // use super::*;
fn listenbrainz() { // use crate::config::config::tests::read_config_lib;
let mut c = Controller::start(".\\test-config\\config_test.json").unwrap();
let client = c.listenbrainz_authenticate().unwrap(); // #[test]
// fn listenbrainz() {
// let mut c = Controller::start(".\\test-config\\config_test.json").unwrap();
c.q_new().unwrap(); // let client = c.listenbrainz_authenticate().unwrap();
c.queue_mail[0].send(QueueCmd::SetVolume(0.04)).unwrap();
let songs = c.lib_get_songs(); // c.q_new().unwrap();
// c.queue_mail[0].send(QueueCmd::SetVolume(0.04)).unwrap();
c.q_enqueue(0, songs[1].location.to_owned()).unwrap(); // let songs = c.lib_get_songs();
c.q_play(0).unwrap();
// c.q_enqueue(0, songs[1].location.to_owned()).unwrap();
// c.q_play(0).unwrap();
sleep(Duration::from_secs(100)); // sleep(Duration::from_secs(100));
c.lbz_scrobble(client, songs[1].uuid).unwrap(); // c.lbz_scrobble(client, songs[1].uuid).unwrap();
} // }
} // }

View file

@ -2,89 +2,37 @@
//! player. It manages queues, playback, library access, and //! player. It manages queues, playback, library access, and
//! other functions //! other functions
use crossbeam_channel;
use crossbeam_channel::{Receiver, Sender};
use listenbrainz::ListenBrainz;
use std::path::PathBuf; use std::path::PathBuf;
use std::sync::{Arc, RwLock}; use std::sync::{Arc, RwLock};
use crossbeam_channel::{Sender, Receiver};
use crossbeam_channel;
use listenbrainz::ListenBrainz;
use std::thread::spawn; use std::thread::spawn;
use std::error::Error;
use crossbeam_channel::unbounded; use crossbeam_channel::unbounded;
use std::error::Error;
use uuid::Uuid; use uuid::Uuid;
use crate::music_controller::queue::{QueueItem, QueueItemType}; use crate::music_controller::queue::QueueItem;
use crate::music_player::gstreamer::GStreamer;
use crate::music_storage::library::{Tag, URI}; use crate::music_storage::library::{Tag, URI};
use crate::{ use crate::{
music_storage::library::{MusicLibrary, Song},
config::config::Config, config::config::Config,
music_controller::queue::Queue, music_controller::queue::Queue,
music_storage::library::{MusicLibrary, Song},
}; };
pub struct Controller { pub struct Controller {
// queues: Vec<Queue>, pub queue: Queue,
pub config: Arc<RwLock<Config>>, pub config: Arc<RwLock<Config>>,
// library: MusicLibrary, pub library: MusicLibrary,
pub(super) controller_mail: MailMan<ControllerCmd, ControllerResponse>, player_mail: MailMan<PlayerCmd, PlayerRes>,
pub(super) db_mail: MailMan<DatabaseCmd, DatabaseResponse>,
pub(super) queue_mail: Vec<MailMan<QueueCmd, QueueResponse>>,
}
#[derive(Debug)]
pub(super) enum ControllerCmd {
Default,
Test
}
#[derive(Debug)]
pub(super) enum ControllerResponse {
Empty,
QueueMailMan(MailMan<QueueCmd, QueueResponse>),
}
#[derive(Debug)]
pub(super) enum DatabaseCmd {
Default,
Test,
SaveLibrary,
GetSongs,
QueryUuid(Uuid),
QueryUuids(Vec<Uuid>),
ReadFolder(String),
}
#[derive(Debug)]
pub(super) enum DatabaseResponse {
Empty,
Song(Song),
Songs(Vec<Song>),
Library(MusicLibrary),
}
#[derive(Debug)]
pub(super) enum QueueCmd {
Default,
Test,
Play,
Pause,
// SetSongs(Vec<QueueItem<QueueState>>),
// SetLocation(URI),
Enqueue(URI),
SetVolume(f64),
}
#[derive(Debug)]
pub(super) enum QueueResponse {
Default,
Test,
Index(i32),
Uuid(Uuid),
} }
#[derive(Debug)] #[derive(Debug)]
pub(super) struct MailMan<T: Send, U: Send> { pub(super) struct MailMan<T: Send, U: Send> {
pub tx: Sender<T>, pub tx: Sender<T>,
rx: Receiver<U> rx: Receiver<U>,
} }
impl<T: Send> MailMan<T, T> { impl<T: Send> MailMan<T, T> {
@ -98,10 +46,7 @@ impl<T: Send, U: Send> MailMan<T, U> {
let (tx, rx) = unbounded::<T>(); let (tx, rx) = unbounded::<T>();
let (tx1, rx1) = unbounded::<U>(); let (tx1, rx1) = unbounded::<U>();
( (MailMan { tx, rx: rx1 }, MailMan { tx: tx1, rx })
MailMan { tx, rx: rx1 },
MailMan { tx: tx1, rx }
)
} }
pub fn send(&self, mail: T) -> Result<(), Box<dyn Error>> { pub fn send(&self, mail: T) -> Result<(), Box<dyn Error>> {
@ -115,192 +60,61 @@ impl<T: Send, U: Send> MailMan<T, U> {
} }
} }
enum PlayerCmd {
Test(URI),
}
enum PlayerRes {
Test,
}
#[allow(unused_variables)] #[allow(unused_variables)]
impl Controller { impl Controller {
pub fn start<P>(config_path: P) -> Result<Self, Box<dyn Error>> pub fn start<P>(config_path: P) -> Result<Self, Box<dyn Error>>
where std::path::PathBuf: std::convert::From<P> where
std::path::PathBuf: std::convert::From<P>,
{ {
let config_path = PathBuf::from(config_path); let config_path = PathBuf::from(config_path);
let config = Config::read_file(config_path)?; let config = Config::read_file(config_path)?;
let uuid = config.libraries.get_default()?.uuid; let uuid = config.libraries.get_default()?.uuid;
let config_ = Arc::new(RwLock::from(config)); let config_ = Arc::new(RwLock::from(config));
let mut lib = MusicLibrary::init(config_.clone(), uuid)?; let library = MusicLibrary::init(config_.clone(), uuid)?;
let config = config_.clone(); let (player_mail, in_thread) = MailMan::<PlayerCmd, PlayerRes>::double();
let (out_thread_controller, in_thread) = MailMan::double();
let monitor_thread = spawn(move || {
use ControllerCmd::*;
loop {
let command = in_thread.recv().unwrap();
match command { spawn(move || {
Default => (), let mut player = GStreamer::new().unwrap();
Test => {
in_thread.send(ControllerResponse::Empty).unwrap(); while true {
}, match in_thread.recv().unwrap() {
PlayerCmd::Test(uri) => {
&player.set_volume(0.04);
_ = &player.enqueue_next(&uri).unwrap();
_ = &player.play();
in_thread.send(PlayerRes::Test).unwrap();
}
} }
} }
}); });
let config = config_.clone(); Ok(Controller {
let (out_thread_db, in_thread) = MailMan::double(); queue: Queue::new(),
let db_monitor = spawn(move || {
use DatabaseCmd::*;
loop {
let command = in_thread.recv().unwrap();
match command {
Default => {},
Test => {
in_thread.send(DatabaseResponse::Empty).unwrap();
},
GetSongs => {
let songs = lib.query_tracks(&String::from(""), &(vec![Tag::Title]), &(vec![Tag::Title])).unwrap().iter().cloned().cloned().collect();
in_thread.send(DatabaseResponse::Songs(songs)).unwrap();
},
SaveLibrary => {
//TODO: make this send lib ref to the function to save instead
lib.save(config.read().unwrap().to_owned()).unwrap();
},
QueryUuid(uuid) => {
match lib.query_uuid(&uuid) {
Some(song) => in_thread.send(DatabaseResponse::Song(song.0.clone())).unwrap(),
None => in_thread.send(DatabaseResponse::Empty).unwrap(),
}
},
QueryUuids(uuids) => {
let mut vec = Vec::new();
for uuid in uuids {
match lib.query_uuid(&uuid) {
Some(song) => vec.push(song.0.clone()),
None => unimplemented!()
}
}
in_thread.send(DatabaseResponse::Songs(vec)).unwrap();
},
ReadFolder(folder) => {
lib.scan_folder(&folder).unwrap();
in_thread.send(DatabaseResponse::Empty).unwrap();
}
}
}
});
Ok(
Controller {
// queues: Vec::new(),
config: config_.clone(), config: config_.clone(),
controller_mail: out_thread_controller, library,
db_mail: out_thread_db, player_mail,
queue_mail: Vec::new(), })
}
)
} }
pub fn lib_get_songs(&self) -> Vec<Song> { pub fn q_add(&mut self, item: &Uuid, source: super::queue::PlayerLocation, by_human: bool) {
self.db_mail.send(DatabaseCmd::GetSongs); let item = self.library.query_uuid(item).unwrap().0.to_owned();
match self.db_mail.recv().unwrap() { self.queue.add_item(item, source, by_human)
DatabaseResponse::Songs(songs) => songs,
_ => Vec::new()
} }
} }
pub fn lib_scan_folder(&self, folder: String) -> Result<(), Box<dyn Error>> {
let mail = &self.db_mail;
mail.send(DatabaseCmd::ReadFolder(folder))?;
dbg!(mail.recv()?);
Ok(())
}
pub fn lib_save(&self) -> Result<(), Box<dyn Error>> {
self.db_mail.send(DatabaseCmd::SaveLibrary);
Ok(())
}
pub fn q_new(&mut self) -> Result<usize, Box<dyn Error>> {
let (out_thread_queue, in_thread) = MailMan::<QueueCmd, QueueResponse>::double();
let queues_monitor = spawn(move || {
use QueueCmd::*;
let mut queue = Queue::new().unwrap();
loop {
let command = in_thread.recv().unwrap();
match command {
Default => {},
Test => { in_thread.send(QueueResponse::Test).unwrap() },
Play => {
match queue.player.play() {
Ok(_) => in_thread.send(QueueResponse::Default).unwrap(),
Err(_) => todo!()
};
},
Pause => {
match queue.player.pause() {
Ok(_) => in_thread.send(QueueResponse::Default).unwrap(),
Err(_) => todo!()
}
},
// SetSongs(songs) => {
// queue.set_tracks(songs);
// in_thread.send(QueueResponse::Default).unwrap();
// },
Enqueue(uri) => {
queue.player.enqueue_next(&uri).unwrap();
// in_thread.send(QueueResponse::Default).unwrap();
},
SetVolume(vol) => {
queue.player.set_volume(vol);
}
}
}
});
self.queue_mail.push(out_thread_queue);
Ok(self.queue_mail.len() - 1)
}
pub fn q_play(&self, index: usize) -> Result<(), Box<dyn Error>> {
let mail = &self.queue_mail[index];
mail.send(QueueCmd::Play)?;
dbg!(mail.recv()?);
Ok(())
}
pub fn q_pause(&self, index: usize) -> Result<(), Box<dyn Error>> {
let mail = &self.queue_mail[index];
mail.send(QueueCmd::Pause)?;
dbg!(mail.recv()?);
Ok(())
}
pub fn q_set_volume(&self, index: usize, volume: f64) -> Result<(), Box<dyn Error>> {
let mail = &self.queue_mail[index];
mail.send(QueueCmd::SetVolume(volume))?;
Ok(())
}
// fn q_set_songs(&self, index: usize, songs: Vec<QueueItem<QueueState>>) -> Result<(), Box<dyn Error>> {
// let mail = &self.queue_mail[index];
// mail.send(QueueCmd::SetSongs(songs))?;
// dbg!(mail.recv()?);
// Ok(())
// }
pub fn q_enqueue(&self, index: usize, uri: URI) -> Result<(), Box<dyn Error>> {
let mail = &self.queue_mail[index];
mail.send(QueueCmd::Enqueue(uri))?;
// dbg!(mail.recv()?);
Ok(())
}
}
#[cfg(test)] #[cfg(test)]
mod tests { mod test_super {
use std::{thread::sleep, time::Duration}; use std::{thread::sleep, time::Duration};
use super::Controller; use super::Controller;
@ -309,33 +123,18 @@ mod tests {
fn play_test() { fn play_test() {
let mut a = match Controller::start("test-config/config_test.json".to_string()) { let mut a = match Controller::start("test-config/config_test.json".to_string()) {
Ok(c) => c, Ok(c) => c,
Err(e) => panic!("{e}") Err(e) => panic!("{e}"),
}; };
sleep(Duration::from_millis(500)); sleep(Duration::from_millis(500));
let i = a.q_new().unwrap();
a.q_set_volume(i, 0.04);
// a.new_queue();
let songs = a.lib_get_songs();
a.q_enqueue(i, songs[2].location.clone());
// a.enqueue(1, songs[2].location.clone());
a.q_play(i).unwrap();
// a.play(1).unwrap();
sleep(Duration::from_secs(10));
a.q_pause(i);
sleep(Duration::from_secs(10));
a.q_play(i);
sleep(Duration::from_secs(1000));
} }
#[test] #[test]
fn test_() { fn test_() {
let a = match Controller::start("test-config/config_test.json".to_string()) { let c = Controller::start(
Ok(c) => c, "F:\\Dangoware\\Dango Music Player\\dmp-core\\test-config\\config_test.json",
Err(e) => panic!("{e}") )
}; .unwrap();
a.lib_scan_folder("F:/Music/Mp3".to_string());
a.lib_save(); sleep(Duration::from_secs(60));
} }
} }

View file

@ -1,12 +1,9 @@
use uuid::Uuid; use crate::music_storage::library::{MusicLibrary, Song, URI};
use crate::{
music_player::{Player, PlayerError},
music_storage::library::{Album, MusicLibrary, URI}
};
use std::{ use std::{
error::Error, error::Error,
sync::{Arc, RwLock} sync::{Arc, RwLock},
}; };
use uuid::Uuid;
use thiserror::Error; use thiserror::Error;
@ -15,8 +12,7 @@ pub enum QueueError {
#[error("Index out of bounds! Index {0} is over len {1}")] #[error("Index out of bounds! Index {0} is over len {1}")]
OutOfBounds(usize, usize), OutOfBounds(usize, usize),
#[error("The Queue is empty!")] #[error("The Queue is empty!")]
EmptyQueue EmptyQueue,
} }
#[derive(Debug, PartialEq, Clone, Copy)] #[derive(Debug, PartialEq, Clone, Copy)]
@ -27,54 +23,6 @@ pub enum QueueState {
NoState, NoState,
} }
#[derive(Debug, Clone, PartialEq)]
#[non_exhaustive]
pub enum QueueItemType<'a> {
Song(Uuid),
ExternalSong(URI),
Album{
album: Album<'a>,
shuffled: bool,
order: Option<Vec<Uuid>>,
// disc #, track #
current: (i32, i32)
},
Playlist {
uuid: Uuid,
shuffled: bool,
order: Option<Vec<Uuid>>,
current: Uuid
},
None,
Test
}
impl QueueItemType<'_> {
fn get_uri(&self, lib: Arc<RwLock<MusicLibrary>>) -> Option<URI> {
use QueueItemType::*;
let lib = lib.read().unwrap();
match self {
Song(uuid) => {
if let Some((song, _)) = lib.query_uuid(uuid) {
Some(song.location.clone())
}else {
Option::None
}
},
Album{album, shuffled, current: (disc, index), ..} => {
if !shuffled {
Some(album.track(*disc as usize, *index as usize).unwrap().location.clone())
}else {
todo!()
}
},
ExternalSong(uri) => { Some(uri.clone()) },
_ => { Option::None }
}
}
}
// TODO: move this to a different location to be used elsewhere // TODO: move this to a different location to be used elsewhere
#[derive(Debug, Clone, PartialEq)] #[derive(Debug, Clone, PartialEq)]
#[non_exhaustive] #[non_exhaustive]
@ -83,74 +31,80 @@ pub enum PlayerLocation {
Library, Library,
Playlist(Uuid), Playlist(Uuid),
File, File,
Custom Custom,
} }
#[derive(Debug, Clone, PartialEq)] #[derive(Debug, Clone, PartialEq)]
#[non_exhaustive] #[non_exhaustive]
pub struct QueueItem<'a> { pub struct QueueItem {
pub(super) item: QueueItemType<'a>, pub(super) item: Song,
pub(super) state: QueueState, pub(super) state: QueueState,
pub(super) source: PlayerLocation, pub(super) source: PlayerLocation,
pub(super) by_human: bool pub(super) by_human: bool,
} }
impl QueueItem<'_> { impl From<Song> for QueueItem {
fn new() -> Self { fn from(song: Song) -> Self {
QueueItem { QueueItem {
item: QueueItemType::None, item: song,
state: QueueState::NoState, state: QueueState::NoState,
source: PlayerLocation::Library, source: PlayerLocation::Library,
by_human: false by_human: false,
} }
} }
} }
#[derive(Debug)] #[derive(Debug)]
pub struct Queue<'a> { pub struct Queue {
pub player: Player, pub items: Vec<QueueItem>,
pub name: String, pub played: Vec<QueueItem>,
pub items: Vec<QueueItem<'a>>, pub loop_: bool,
pub played: Vec<QueueItem<'a>>, pub shuffle: bool,
pub loop_: bool
} }
impl<'a> Queue<'a> { impl Queue {
fn has_addhere(&self) -> bool { fn has_addhere(&self) -> bool {
for item in &self.items { for item in &self.items {
if item.state == QueueState::AddHere { if item.state == QueueState::AddHere {
return true return true;
} }
} }
false false
} }
fn dbg_items(&self) { fn dbg_items(&self) {
dbg!(self.items.iter().map(|item| item.item.clone() ).collect::<Vec<QueueItemType>>(), self.items.len()); dbg!(
self.items
.iter()
.map(|item| item.item.clone())
.collect::<Vec<Song>>(),
self.items.len()
);
} }
pub fn new() -> Result<Self, PlayerError> { pub fn new() -> Self {
Ok( //TODO: Make the queue take settings from config/state if applicable
Queue { Queue {
player: Player::new()?,
name: String::new(),
items: Vec::new(), items: Vec::new(),
played: Vec::new(), played: Vec::new(),
loop_: false, loop_: false,
shuffle: false,
} }
)
} }
pub fn set_items(&mut self, tracks: Vec<QueueItem<'a>>) { pub fn set_items(&mut self, tracks: Vec<QueueItem>) {
let mut tracks = tracks; let mut tracks = tracks;
self.items.clear(); self.items.clear();
self.items.append(&mut tracks); self.items.append(&mut tracks);
} }
pub fn add_item(&mut self, item: QueueItemType<'a>, source: PlayerLocation, by_human: bool) { pub fn add_item(&mut self, item: Song, source: PlayerLocation, by_human: bool) {
let mut i: usize = 0; let mut i: usize = 0;
self.items = self.items.iter().enumerate().map(|(j, item_)| { self.items = self
.items
.iter()
.enumerate()
.map(|(j, item_)| {
let mut item_ = item_.to_owned(); let mut item_ = item_.to_owned();
// get the index of the current AddHere item and give it to i // get the index of the current AddHere item and give it to i
if item_.state == QueueState::AddHere { if item_.state == QueueState::AddHere {
@ -158,17 +112,21 @@ impl<'a> Queue<'a> {
item_.state = QueueState::NoState; item_.state = QueueState::NoState;
} }
item_ item_
}).collect::<Vec<QueueItem>>(); })
.collect::<Vec<QueueItem>>();
self.items.insert(i + if self.items.is_empty() { 0 } else { 1 }, QueueItem { self.items.insert(
i + if self.items.is_empty() { 0 } else { 1 },
QueueItem {
item, item,
state: QueueState::AddHere, state: QueueState::AddHere,
source, source,
by_human by_human,
}); },
);
} }
pub fn add_item_next(&mut self, item: QueueItemType<'a>, source: PlayerLocation) { pub fn add_item_next(&mut self, item: Song, source: PlayerLocation) {
use QueueState::*; use QueueState::*;
let empty = self.items.is_empty(); let empty = self.items.is_empty();
@ -176,19 +134,23 @@ impl<'a> Queue<'a> {
(if empty { 0 } else { 1 }), (if empty { 0 } else { 1 }),
QueueItem { QueueItem {
item, item,
state: if (self.items.get(1).is_none() || (!self.has_addhere() && self.items.get(1).is_some()) || empty) { AddHere } else { NoState }, state: if (self.items.get(1).is_none()
|| !self.has_addhere() && self.items.get(1).is_some())
|| empty
{
AddHere
} else {
NoState
},
source, source,
by_human: true by_human: true,
} },
) )
} }
pub fn add_multi(&mut self, items: Vec<QueueItemType>, source: PlayerLocation, by_human: bool) { pub fn add_multi(&mut self, items: Vec<Song>, source: PlayerLocation, by_human: bool) {}
}
pub fn remove_item(&mut self, remove_index: usize) -> Result<(), QueueError> { pub fn remove_item(&mut self, remove_index: usize) -> Result<(), QueueError> {
// dbg!(/*&remove_index, self.current_index(), &index,*/ &self.items[remove_index]); // dbg!(/*&remove_index, self.current_index(), &index,*/ &self.items[remove_index]);
if remove_index < self.items.len() { if remove_index < self.items.len() {
@ -233,21 +195,19 @@ impl<'a> Queue<'a> {
self.played.clear(); self.played.clear();
} }
// TODO: uh, fix this? // TODO: uh, fix this?
fn move_to(&mut self, index: usize) -> Result<(), QueueError> { fn move_to(&mut self, index: usize) -> Result<(), QueueError> {
use QueueState::*; use QueueState::*;
let empty = self.items.is_empty(); let empty = self.items.is_empty();
let nothing_error = Err(QueueError::EmptyQueue);
let index = if !empty { index } else { return nothing_error; }; let index = if !empty {
index
} else {
return Err(QueueError::EmptyQueue);
};
if !empty && index < self.items.len() { if !empty && index < self.items.len() {
let position = self.player.position();
if position.is_some_and(|dur| !dur.is_zero() ) {
self.played.push(self.items[0].clone());
}
let to_item = self.items[index].clone(); let to_item = self.items[index].clone();
loop { loop {
@ -259,17 +219,19 @@ impl<'a> Queue<'a> {
self.items[1].state = AddHere; self.items[1].state = AddHere;
} }
if let Err(e) = self.remove_item(0) { if let Err(e) = self.remove_item(0) {
dbg!(&e); self.dbg_items(); return Err(e); dbg!(&e);
self.dbg_items();
return Err(e);
} }
// dbg!(&to_item.item, &self.items[ind].item); // dbg!(&to_item.item, &self.items[ind].item);
} else if empty { } else if empty {
return nothing_error; return Err(QueueError::EmptyQueue);
} else { } else {
break; break;
} }
} }
} else { } else {
return Err(QueueError::EmptyQueue.into()); return Err(QueueError::EmptyQueue);
} }
Ok(()) Ok(())
} }
@ -287,10 +249,7 @@ impl<'a> Queue<'a> {
} }
#[allow(clippy::should_implement_trait)] #[allow(clippy::should_implement_trait)]
pub fn next(&mut self, lib: Arc<RwLock<MusicLibrary>>) -> Result<OutQueue, Box<dyn Error>> { pub fn next(&mut self) -> Result<&QueueItem, Box<dyn Error>> {
if self.items.is_empty() { if self.items.is_empty() {
if self.loop_ { if self.loop_ {
return Err(QueueError::EmptyQueue.into()); // TODO: add function to loop the queue return Err(QueueError::EmptyQueue.into()); // TODO: add function to loop the queue
@ -300,119 +259,20 @@ impl<'a> Queue<'a> {
} }
// TODO: add an algorithm to detect if the song should be skipped // TODO: add an algorithm to detect if the song should be skipped
let item = self.items[0].clone(); let item = self.items[0].clone();
let uri: URI = match &self.items[1].item {
QueueItemType::Song(uuid) => {
// TODO: Refactor later for multiple URIs
match &lib.read().unwrap().query_uuid(uuid) {
Some(song) => song.0.location.clone(),
None => return Err("Uuid does not exist!".into()),
}
},
QueueItemType::Album { album, current, ..} => {
let (disc, track) = (current.0 as usize, current.1 as usize);
match album.track(disc, track) {
Some(track) => track.location.clone(),
None => return Err(format!("Track in Album {} at disc {} track {} does not exist!", album.title(), disc, track).into())
}
},
QueueItemType::Playlist { current, .. } => {
// TODO: Refactor later for multiple URIs
match &lib.read().unwrap().query_uuid(current) {
Some(song) => song.0.location.clone(),
None => return Err("Uuid does not exist!".into()),
}
},
_ => todo!()
};
if !self.player.is_paused() {
self.player.enqueue_next(&uri)?;
self.player.play()?
}
if self.items[0].state == QueueState::AddHere || !self.has_addhere() { if self.items[0].state == QueueState::AddHere || !self.has_addhere() {
self.items[1].state = QueueState::AddHere; self.items[1].state = QueueState::AddHere;
} }
self.played.push(item); self.played.push(item);
self.items.remove(0); self.items.remove(0);
Ok(todo!()) Ok(&self.items[1])
} }
pub fn prev() {} pub fn prev() {}
pub fn enqueue_item(&mut self, item: QueueItem, lib: Arc<RwLock<MusicLibrary>>) -> Result<(), Box<dyn Error>> {
if let Some(uri) = item.item.get_uri(lib) {
self.player.enqueue_next(&uri)?;
}else {
return Err("this item does not exist!".into());
}
Ok(())
}
pub fn check_played(&mut self) { pub fn check_played(&mut self) {
while self.played.len() > 50 { while self.played.len() > 50 {
self.played.remove(0); self.played.remove(0);
} }
} }
} }
pub struct OutQueue {
}
pub enum OutQueueItem {
}
#[test]
fn item_add_test() {
let mut q = Queue::new().unwrap();
for _ in 0..5 {
// dbg!("tick!");
q.add_item(QueueItemType::Song(Uuid::new_v4()), PlayerLocation::Library, true);
// dbg!(&q.items, &q.items.len());
}
for _ in 0..1 {
q.remove_item(0).inspect_err(|e| println!("{e:?}"));
}
for _ in 0..2 {
q.items.push(QueueItem { item: QueueItemType::Test, state: QueueState::NoState, source: PlayerLocation::Library, by_human: false });
}
dbg!(5);
q.add_item_next(QueueItemType::Test, PlayerLocation::Test);
dbg!(6);
dbg!(&q.items, &q.items.len());
}
#[test]
fn test_() {
let mut q = Queue::new().unwrap();
for _ in 0..400 {
q.items.push(QueueItem { item: QueueItemType::Song(Uuid::new_v4()), state: QueueState::NoState, source: PlayerLocation::File, by_human: false });
}
for _ in 0..50000 {
q.add_item(QueueItemType::Song(Uuid::new_v4()), PlayerLocation::Library, true);
}
// q.add_item_next(QueueItemType::Test, PlayerLocation::File);
// dbg!(&q.items, &q.items.len());
}
#[test]
fn move_test() {
let mut q = Queue::new().unwrap();
for _ in 0..5 {
q.add_item(QueueItemType::Song(Uuid::new_v4()), PlayerLocation::Library, true);
}
// q.add_item(QueueItemType::Test, QueueSource::Library, true).unwrap();
dbg!(&q.items, &q.items.len());
q.move_to(3).inspect_err(|e| {dbg!(e);});
dbg!(&q.items, &q.items.len());
// q.dbg_items();
}

View file

@ -15,8 +15,10 @@ use gstreamer::prelude::*;
use chrono::Duration; use chrono::Duration;
use thiserror::Error; use thiserror::Error;
use super::player::{Player, PlayerError};
#[derive(Debug)] #[derive(Debug)]
pub enum PlayerCmd { pub enum GstCmd {
Play, Play,
Pause, Pause,
Eos, Eos,
@ -24,7 +26,7 @@ pub enum PlayerCmd {
} }
#[derive(Debug, PartialEq, Eq)] #[derive(Debug, PartialEq, Eq)]
pub enum PlayerState { pub enum GstState {
Playing, Playing,
Paused, Paused,
Ready, Ready,
@ -33,7 +35,7 @@ pub enum PlayerState {
VoidPending, VoidPending,
} }
impl From<gst::State> for PlayerState { impl From<gst::State> for GstState {
fn from(value: gst::State) -> Self { fn from(value: gst::State) -> Self {
match value { match value {
gst::State::VoidPending => Self::VoidPending, gst::State::VoidPending => Self::VoidPending,
@ -45,7 +47,7 @@ impl From<gst::State> for PlayerState {
} }
} }
impl TryInto<gst::State> for PlayerState { impl TryInto<gst::State> for GstState {
fn try_into(self) -> Result<gst::State, Box<dyn Error>> { fn try_into(self) -> Result<gst::State, Box<dyn Error>> {
match self { match self {
Self::VoidPending => Ok(gst::State::VoidPending), Self::VoidPending => Ok(gst::State::VoidPending),
@ -60,24 +62,6 @@ impl TryInto<gst::State> for PlayerState {
type Error = Box<dyn Error>; type Error = Box<dyn Error>;
} }
#[derive(Error, Debug)]
pub enum PlayerError {
#[error("player initialization failed")]
Init(#[from] glib::Error),
#[error("element factory failed to create playbin3")]
Factory(#[from] glib::BoolError),
#[error("could not change playback state")]
StateChange(#[from] gst::StateChangeError),
#[error("the file or source is not found")]
NotFound,
#[error("failed to build gstreamer item")]
Build,
#[error("poison error")]
Poison,
#[error("general player error")]
General,
}
#[derive(Debug, PartialEq, Eq)] #[derive(Debug, PartialEq, Eq)]
enum PlaybackStats { enum PlaybackStats {
Idle, Idle,
@ -91,10 +75,10 @@ enum PlaybackStats {
/// An instance of a music player with a GStreamer backend /// An instance of a music player with a GStreamer backend
#[derive(Debug)] #[derive(Debug)]
pub struct Player { pub struct GStreamer {
source: Option<URI>, source: Option<URI>,
//pub message_tx: Sender<PlayerCmd>, //pub message_tx: Sender<PlayerCmd>,
pub message_rx: crossbeam::channel::Receiver<PlayerCmd>, pub message_rx: crossbeam::channel::Receiver<GstCmd>,
playback_tx: crossbeam::channel::Sender<PlaybackStats>, playback_tx: crossbeam::channel::Sender<PlaybackStats>,
playbin: Arc<RwLock<Element>>, playbin: Arc<RwLock<Element>>,
@ -105,7 +89,7 @@ pub struct Player {
position: Arc<RwLock<Option<Duration>>>, position: Arc<RwLock<Option<Duration>>>,
} }
impl Player { impl GStreamer {
pub fn new() -> Result<Self, PlayerError> { pub fn new() -> Result<Self, PlayerError> {
// Initialize GStreamer, maybe figure out how to nicely fail here // Initialize GStreamer, maybe figure out how to nicely fail here
gst::init()?; gst::init()?;
@ -162,14 +146,14 @@ impl Player {
// Check if the current playback position is close to the end // Check if the current playback position is close to the end
let finish_point = end - Duration::milliseconds(250); let finish_point = end - Duration::milliseconds(250);
if pos_temp.unwrap() >= end { if pos_temp.unwrap() >= end {
let _ = playback_tx.try_send(PlayerCmd::Eos); let _ = playback_tx.try_send(GstCmd::Eos);
playbin_arc playbin_arc
.write() .write()
.unwrap() .unwrap()
.set_state(gst::State::Ready) .set_state(gst::State::Ready)
.expect("Unable to set the pipeline state"); .expect("Unable to set the pipeline state");
} else if pos_temp.unwrap() >= finish_point { } else if pos_temp.unwrap() >= finish_point {
let _ = playback_tx.try_send(PlayerCmd::AboutToFinish); let _ = playback_tx.try_send(GstCmd::AboutToFinish);
} }
// This has to be done AFTER the current time in the file // This has to be done AFTER the current time in the file
@ -468,7 +452,7 @@ impl Player {
} }
/// Get the current state of the playback /// Get the current state of the playback
pub fn state(&mut self) -> PlayerState { pub fn state(&mut self) -> GstState {
self.playbin().unwrap().current_state().into() self.playbin().unwrap().current_state().into()
/* /*
match *self.buffer.read().unwrap() { match *self.buffer.read().unwrap() {
@ -498,7 +482,9 @@ impl Player {
} }
} }
impl Drop for Player { // impl Player for GStreamer {}
impl Drop for GStreamer {
/// Cleans up the `GStreamer` pipeline and the monitoring /// Cleans up the `GStreamer` pipeline and the monitoring
/// thread when [Player] is dropped. /// thread when [Player] is dropped.
fn drop(&mut self) { fn drop(&mut self) {

0
src/music_player/kira.rs Normal file
View file

View file

@ -0,0 +1,57 @@
use chrono::Duration;
use thiserror::Error;
use gstreamer as gst;
use crate::music_storage::library::URI;
#[derive(Error, Debug)]
pub enum PlayerError {
#[error("player initialization failed")]
Init(#[from] glib::Error),
#[error("element factory failed to create playbin3")]
Factory(#[from] glib::BoolError),
#[error("could not change playback state")]
StateChange(#[from] gst::StateChangeError),
#[error("the file or source is not found")]
NotFound,
#[error("failed to build gstreamer item")]
Build,
#[error("poison error")]
Poison,
#[error("general player error")]
General,
}
pub trait Player {
fn source(&self) -> &Option<URI>;
fn enqueue_next(&mut self, next_track: &URI) -> Result<(), PlayerError>;
fn set_volume(&mut self, volume: f64);
fn volume(&mut self) -> f64;
fn ready(&mut self) -> Result<(), PlayerError>;
fn play(&mut self) -> Result<(), PlayerError>;
fn resume(&mut self) -> Result<(), PlayerError>;
fn pause(&mut self) -> Result<(), PlayerError>;
fn stop(&mut self) -> Result<(), PlayerError>;
fn is_paused(&mut self) -> bool;
fn position(&mut self) -> Option<Duration>;
fn duration(&mut self) -> Option<Duration>;
fn raw_duration(&self) -> Option<Duration>;
fn seek_by(&mut self, seek_amount: Duration) -> Result<(), PlayerError>;
fn seek_to(&mut self, target_pos: Duration) -> Result<(), PlayerError>;
}

View file

@ -176,14 +176,15 @@ pub struct FoobarPlaylistTrack {
impl FoobarPlaylistTrack { impl FoobarPlaylistTrack {
fn find_song(&self) -> Song { fn find_song(&self) -> Song {
let location = URI::Local(self.file_name.clone().into()); let location = URI::Local(self.file_name.clone().into());
let internal_tags = Vec::new();
Song { Song {
location, location: vec![location],
uuid: Uuid::new_v4(), uuid: Uuid::new_v4(),
plays: 0, plays: 0,
skips: 0, skips: 0,
favorited: false, favorited: false,
// banned: None, banned: None,
rating: None, rating: None,
format: None, format: None,
duration: self.duration, duration: self.duration,
@ -193,6 +194,7 @@ impl FoobarPlaylistTrack {
date_modified: None, date_modified: None,
album_art: Vec::new(), album_art: Vec::new(),
tags: BTreeMap::new(), tags: BTreeMap::new(),
internal_tags,
} }
} }
} }

View file

@ -13,7 +13,7 @@ use std::vec::Vec;
use chrono::prelude::*; use chrono::prelude::*;
use crate::music_storage::db_reader::extern_library::ExternalLibrary; use crate::music_storage::db_reader::extern_library::ExternalLibrary;
use crate::music_storage::library::{AlbumArt, Service, Song, Tag, URI}; use crate::music_storage::library::{AlbumArt, BannedType, Service, Song, Tag, URI};
use crate::music_storage::utils; use crate::music_storage::utils;
use urlencoding::decode; use urlencoding::decode;
@ -166,17 +166,19 @@ impl ExternalLibrary for ITunesLibrary {
}; };
let play_time_ = StdDur::from_secs(track.plays as u64 * dur.as_secs()); let play_time_ = StdDur::from_secs(track.plays as u64 * dur.as_secs());
let internal_tags = Vec::new(); // TODO: handle internal tags generation
let ny: Song = Song { let ny: Song = Song {
location, location: vec![location],
uuid: Uuid::new_v4(), uuid: Uuid::new_v4(),
plays: track.plays, plays: track.plays,
skips: 0, skips: 0,
favorited: track.favorited, favorited: track.favorited,
// banned: if track.banned { banned: if track.banned {
// Some(BannedType::All) Some(BannedType::All)
// }else { }else {
// None None
// }, },
rating: track.rating, rating: track.rating,
format: match FileFormat::from_file(PathBuf::from(&loc)) { format: match FileFormat::from_file(PathBuf::from(&loc)) {
Ok(e) => Some(e), Ok(e) => Some(e),
@ -192,6 +194,7 @@ impl ExternalLibrary for ITunesLibrary {
Err(_) => Vec::new(), Err(_) => Vec::new(),
}, },
tags: tags_, tags: tags_,
internal_tags,
}; };
// dbg!(&ny.tags); // dbg!(&ny.tags);
bun.push(ny); bun.push(ny);

View file

@ -1,3 +1,4 @@
use super::playlist::PlaylistFolder;
// Crate things // Crate things
use super::utils::{find_images, normalize, read_file, write_file}; use super::utils::{find_images, normalize, read_file, write_file};
use crate::config::config::Config; use crate::config::config::Config;
@ -8,16 +9,15 @@ use std::error::Error;
use std::io::Write; use std::io::Write;
use std::ops::ControlFlow::{Break, Continue}; use std::ops::ControlFlow::{Break, Continue};
// Files // Files
use file_format::{FileFormat, Kind}; use file_format::{FileFormat, Kind};
use glib::filename_to_uri; use glib::filename_to_uri;
use lofty::{AudioFile, ItemKey, ItemValue, ParseOptions, Probe, TagType, TaggedFileExt}; use lofty::{AudioFile, ItemKey, ItemValue, ParseOptions, Probe, TagType, TaggedFileExt};
use rcue::parser::parse_from_file; use rcue::parser::parse_from_file;
use uuid::Uuid;
use std::fs::{self, File}; use std::fs::{self, File};
use tempfile::TempDir;
use std::path::{Path, PathBuf}; use std::path::{Path, PathBuf};
use tempfile::TempDir;
use uuid::Uuid;
use walkdir::WalkDir; use walkdir::WalkDir;
// Time // Time
@ -117,35 +117,58 @@ impl ToString for Field {
} }
} }
#[derive(Debug, Clone, Deserialize, Serialize, PartialEq, Eq, PartialOrd, Ord)]
#[non_exhaustive]
pub enum InternalTag {
DoNotTrack(DoNotTrack),
SongType(SongType),
SongLink(Uuid, SongType),
// Volume Adjustment from -100% to 100%
VolumeAdjustment(i8),
}
#[derive(Debug, Clone, Deserialize, Serialize, PartialEq)] #[derive(Debug, Clone, Deserialize, Serialize, PartialEq)]
#[non_exhaustive]
pub enum BannedType { pub enum BannedType {
Shuffle, Shuffle,
All, All,
} }
#[derive(Debug, Clone, Deserialize, Serialize, PartialEq)] #[derive(Debug, Clone, Deserialize, Serialize, PartialEq, Eq, PartialOrd, Ord)]
#[non_exhaustive]
pub enum DoNotTrack { pub enum DoNotTrack {
// TODO: add services to not track // TODO: add services to not track
LastFM,
LibreFM,
MusicBrainz,
Discord,
} }
#[derive(Debug, Clone, Deserialize, Serialize, PartialEq)] #[derive(Debug, Clone, Deserialize, Serialize, PartialEq, Eq, PartialOrd, Ord)]
enum SongType { #[non_exhaustive]
// TODO: add song types pub enum SongType {
// TODO: add MORE?! song types
Main, Main,
Instrumental, Instrumental,
Remix, Remix,
Custom(String) Custom(String),
}
impl Default for SongType {
fn default() -> Self {
SongType::Main
}
} }
/// Stores information about a single song /// Stores information about a single song
#[derive(Debug, Clone, Deserialize, Serialize, PartialEq)] #[derive(Debug, Clone, Deserialize, Serialize, PartialEq)]
pub struct Song { pub struct Song {
pub location: URI, pub location: Vec<URI>,
pub uuid: Uuid, pub uuid: Uuid,
pub plays: i32, pub plays: i32,
pub skips: i32, pub skips: i32,
pub favorited: bool, pub favorited: bool,
// pub banned: Option<BannedType>, pub banned: Option<BannedType>,
pub rating: Option<u8>, pub rating: Option<u8>,
pub format: Option<FileFormat>, pub format: Option<FileFormat>,
pub duration: Duration, pub duration: Duration,
@ -158,9 +181,9 @@ pub struct Song {
pub date_modified: Option<DateTime<Utc>>, pub date_modified: Option<DateTime<Utc>>,
pub album_art: Vec<AlbumArt>, pub album_art: Vec<AlbumArt>,
pub tags: BTreeMap<Tag, String>, pub tags: BTreeMap<Tag, String>,
pub internal_tags: Vec<InternalTag>,
} }
impl Song { impl Song {
/// Get a tag's value /// Get a tag's value
/// ///
@ -180,7 +203,7 @@ impl Song {
pub fn get_field(&self, target_field: &str) -> Option<Field> { pub fn get_field(&self, target_field: &str) -> Option<Field> {
let lower_target = target_field.to_lowercase(); let lower_target = target_field.to_lowercase();
match lower_target.as_str() { match lower_target.as_str() {
"location" => Some(Field::Location(self.location.clone())), "location" => Some(Field::Location(self.primary_uri().unwrap().0.clone())), //TODO: make this not unwrap()
"plays" => Some(Field::Plays(self.plays)), "plays" => Some(Field::Plays(self.plays)),
"skips" => Some(Field::Skips(self.skips)), "skips" => Some(Field::Skips(self.skips)),
"favorited" => Some(Field::Favorited(self.favorited)), "favorited" => Some(Field::Favorited(self.favorited)),
@ -279,13 +302,15 @@ impl Song {
// TODO: Fix error handling // TODO: Fix error handling
let binding = fs::canonicalize(target_file).unwrap(); let binding = fs::canonicalize(target_file).unwrap();
// TODO: Handle creation of internal tag: Song Type and Song Links
let internal_tags = { Vec::new() };
let new_song = Song { let new_song = Song {
location: URI::Local(binding), location: vec![URI::Local(binding)],
uuid: Uuid::new_v4(), uuid: Uuid::new_v4(),
plays: 0, plays: 0,
skips: 0, skips: 0,
favorited: false, favorited: false,
// banned: None, banned: None,
rating: None, rating: None,
format, format,
duration, duration,
@ -295,6 +320,7 @@ impl Song {
date_modified: Some(chrono::offset::Utc::now()), date_modified: Some(chrono::offset::Utc::now()),
tags, tags,
album_art, album_art,
internal_tags,
}; };
Ok(new_song) Ok(new_song)
} }
@ -314,7 +340,6 @@ impl Song {
for file in cue_data.files.iter() { for file in cue_data.files.iter() {
let audio_location = &parent_dir.join(file.file.clone()); let audio_location = &parent_dir.join(file.file.clone());
if !audio_location.exists() { if !audio_location.exists() {
continue; continue;
} }
@ -399,17 +424,17 @@ impl Song {
let album_art = find_images(&audio_location.to_path_buf()).unwrap(); let album_art = find_images(&audio_location.to_path_buf()).unwrap();
let new_song = Song { let new_song = Song {
location: URI::Cue { location: vec![URI::Cue {
location: audio_location.clone(), location: audio_location.clone(),
index: i, index: i,
start, start,
end, end,
}, }],
uuid: Uuid::new_v4(), uuid: Uuid::new_v4(),
plays: 0, plays: 0,
skips: 0, skips: 0,
favorited: false, favorited: false,
// banned: None, banned: None,
rating: None, rating: None,
format, format,
duration, duration,
@ -419,6 +444,7 @@ impl Song {
date_modified: Some(chrono::offset::Utc::now()), date_modified: Some(chrono::offset::Utc::now()),
tags, tags,
album_art, album_art,
internal_tags: Vec::new(),
}; };
tracks.push((new_song, audio_location.clone())); tracks.push((new_song, audio_location.clone()));
} }
@ -426,71 +452,31 @@ impl Song {
Ok(tracks) Ok(tracks)
} }
/// Takes the AlbumArt[index] and opens it in the native file viewer /// Returns a reference to the first valid URI in the song, and any invalid URIs that come before it, or errors if there are no valid URIs
#[allow(clippy::type_complexity)]
pub fn primary_uri(&self) -> Result<(&URI, Option<Vec<&URI>>), Box<dyn Error>> {
let mut invalid_uris = Vec::new();
let mut valid_uri = None;
pub fn open_album_art(&self, index: usize, temp_dir: &TempDir) -> Result<(), Box<dyn Error>> { for uri in &self.location {
use opener::open; if uri.exists()? {
use urlencoding::decode; valid_uri = Some(uri);
break;
if index >= self.album_art.len() { } else {
return Err("Index out of bounds".into()); invalid_uris.push(uri);
}
let uri = match &self.album_art[index] {
AlbumArt::External(uri) => {
PathBuf::from(decode(match uri.as_uri().strip_prefix("file:///") {
Some(e) => e,
None => return Err("Invalid path?".into())
})?.to_owned().to_string())
},
AlbumArt::Embedded(_) => {
let normal_options = ParseOptions::new().parsing_mode(lofty::ParsingMode::Relaxed);
let blank_tag = &lofty::Tag::new(TagType::Id3v2);
let tagged_file: lofty::TaggedFile;
#[cfg(target_family = "windows")]
let uri = urlencoding::decode(
match self.location.as_uri().strip_prefix("file:///") {
Some(str) => str,
None => return Err("invalid path.. again?".into())
})?.into_owned();
#[cfg(target_family = "unix")]
let uri = urlencoding::decode(
match self.location.as_uri().strip_prefix("file://") {
Some(str) => str,
None => return Err("invalid path.. again?".into())
})?.into_owned();
let tag = match Probe::open(uri)?.options(normal_options).read() {
Ok(file) => {
tagged_file = file;
match tagged_file.primary_tag() {
Some(primary_tag) => primary_tag,
None => match tagged_file.first_tag() {
Some(first_tag) => first_tag,
None => blank_tag,
},
} }
} }
match valid_uri {
Err(_) => blank_tag, Some(uri) => Ok((
}; uri,
if !invalid_uris.is_empty() {
let data = tag.pictures()[index].data(); Some(invalid_uris)
} else {
let fmt = FileFormat::from_bytes(data); None
let file_path = temp_dir.path().join(format!("{}_{index}.{}", self.uuid, fmt.extension()));
File::create(&file_path)?.write_all(data)?;
file_path
}, },
}; )),
dbg!(open(dbg!(uri))?); None => Err("No valid URIs for this song".into()),
Ok(()) }
} }
} }
@ -652,14 +638,8 @@ pub struct MusicLibrary {
pub name: String, pub name: String,
pub uuid: Uuid, pub uuid: Uuid,
pub library: Vec<Song>, pub library: Vec<Song>,
} pub playlists: PlaylistFolder,
pub backup_songs: Vec<Song>, // maybe move this to the config instead?
#[test]
fn library_init() {
let config = Config::read_file(PathBuf::from("test_config/config_test.json")).unwrap();
let target_uuid = config.libraries.libraries[0].uuid;
let a = MusicLibrary::init(Arc::new(RwLock::from(config)), target_uuid).unwrap();
dbg!(a);
} }
impl MusicLibrary { impl MusicLibrary {
@ -669,6 +649,8 @@ impl MusicLibrary {
name, name,
uuid, uuid,
library: Vec::new(), library: Vec::new(),
playlists: PlaylistFolder::new(),
backup_songs: Vec::new(),
} }
} }
@ -737,9 +719,12 @@ impl MusicLibrary {
.par_iter() .par_iter()
.enumerate() .enumerate()
.try_for_each(|(i, track)| { .try_for_each(|(i, track)| {
if path == &track.location { for location in &track.location {
//TODO: check that this works
if path == location {
return std::ops::ControlFlow::Break((track, i)); return std::ops::ControlFlow::Break((track, i));
} }
}
Continue(()) Continue(())
}); });
@ -774,7 +759,8 @@ impl MusicLibrary {
fn query_path(&self, path: PathBuf) -> Option<Vec<&Song>> { fn query_path(&self, path: PathBuf) -> Option<Vec<&Song>> {
let result: Arc<Mutex<Vec<&Song>>> = Arc::new(Mutex::new(Vec::new())); let result: Arc<Mutex<Vec<&Song>>> = Arc::new(Mutex::new(Vec::new()));
self.library.par_iter().for_each(|track| { self.library.par_iter().for_each(|track| {
if path == track.location.path() { if path == track.primary_uri().unwrap().0.path() {
//TODO: make this also not unwrap
result.clone().lock().unwrap().push(track); result.clone().lock().unwrap().push(track);
} }
}); });
@ -786,10 +772,7 @@ impl MusicLibrary {
} }
/// Finds all the audio files within a specified folder /// Finds all the audio files within a specified folder
pub fn scan_folder( pub fn scan_folder(&mut self, target_path: &str) -> Result<i32, Box<dyn std::error::Error>> {
&mut self,
target_path: &str,
) -> Result<i32, Box<dyn std::error::Error>> {
let mut total = 0; let mut total = 0;
let mut errors = 0; let mut errors = 0;
for target_file in WalkDir::new(target_path) for target_file in WalkDir::new(target_path)
@ -852,7 +835,6 @@ impl MusicLibrary {
} }
pub fn add_file(&mut self, target_file: &Path) -> Result<(), Box<dyn Error>> { pub fn add_file(&mut self, target_file: &Path) -> Result<(), Box<dyn Error>> {
let new_song = Song::from_file(target_file)?; let new_song = Song::from_file(target_file)?;
match self.add_song(new_song) { match self.add_song(new_song) {
Ok(_) => (), Ok(_) => (),
@ -868,14 +850,13 @@ impl MusicLibrary {
let tracks = Song::from_cue(cuesheet)?; let tracks = Song::from_cue(cuesheet)?;
let mut tracks_added = tracks.len() as i32; let mut tracks_added = tracks.len() as i32;
for (new_song, location) in tracks { for (new_song, location) in tracks {
// Try to remove the original audio file from the db if it exists // Try to remove the original audio file from the db if it exists
if self.remove_uri(&URI::Local(location.clone())).is_ok() { if self.remove_uri(&URI::Local(location.clone())).is_ok() {
tracks_added -= 1 tracks_added -= 1
} }
match self.add_song(new_song) { match self.add_song(new_song) {
Ok(_) => {}, Ok(_) => {}
Err(_error) => { Err(_error) => {
//println!("{}", _error); //println!("{}", _error);
continue; continue;
@ -886,13 +867,14 @@ impl MusicLibrary {
} }
pub fn add_song(&mut self, new_song: Song) -> Result<(), Box<dyn Error>> { pub fn add_song(&mut self, new_song: Song) -> Result<(), Box<dyn Error>> {
if self.query_uri(&new_song.location).is_some() { let location = new_song.primary_uri()?.0;
return Err(format!("URI already in database: {:?}", new_song.location).into()); if self.query_uri(location).is_some() {
return Err(format!("URI already in database: {:?}", location).into());
} }
match new_song.location { match location {
URI::Local(_) if self.query_path(new_song.location.path()).is_some() => { URI::Local(_) if self.query_path(location.path()).is_some() => {
return Err(format!("Location exists for {:?}", new_song.location).into()) return Err(format!("Location exists for {:?}", location).into())
} }
_ => (), _ => (),
} }
@ -915,17 +897,18 @@ impl MusicLibrary {
} }
/// Scan the song by a location and update its tags /// Scan the song by a location and update its tags
// TODO: change this to work with multiple uris
pub fn update_uri( pub fn update_uri(
&mut self, &mut self,
target_uri: &URI, target_uri: &URI,
new_tags: Vec<Tag>, new_tags: Vec<Tag>,
) -> Result<(), Box<dyn std::error::Error>> { ) -> Result<(), Box<dyn std::error::Error>> {
let target_song = match self.query_uri(target_uri) { let (target_song, _) = match self.query_uri(target_uri) {
Some(song) => song, Some(song) => song,
None => return Err("URI not in database!".to_string().into()), None => return Err("URI not in database!".to_string().into()),
}; };
println!("{:?}", target_song.0.location); println!("{:?}", target_song.location);
for tag in new_tags { for tag in new_tags {
println!("{:?}", tag); println!("{:?}", tag);
@ -1143,12 +1126,18 @@ impl MusicLibrary {
#[cfg(test)] #[cfg(test)]
mod test { mod test {
use std::{path::Path, thread::sleep, time::{Duration, Instant}}; use std::{
path::{Path, PathBuf},
sync::{Arc, RwLock},
thread::sleep,
time::{Duration, Instant},
};
use tempfile::TempDir; use tempfile::TempDir;
use super::Song; use crate::{config::config::Config, music_storage::library::MusicLibrary};
use super::Song;
#[test] #[test]
fn get_art_test() { fn get_art_test() {
@ -1162,4 +1151,12 @@ mod test {
sleep(Duration::from_secs(20)); sleep(Duration::from_secs(20));
} }
#[test]
fn library_init() {
let config = Config::read_file(PathBuf::from("test_config/config_test.json")).unwrap();
let target_uuid = config.libraries.libraries[0].uuid;
let a = MusicLibrary::init(Arc::new(RwLock::from(config)), target_uuid).unwrap();
dbg!(a);
}
} }

View file

@ -1,21 +1,52 @@
use std::{fs::File, io::Read, path:: PathBuf, sync::{Arc, RwLock}};
use std::error::Error; use std::error::Error;
use std::{
fs::File,
io::Read,
path::PathBuf,
sync::{Arc, RwLock},
};
use std::time::Duration; use std::time::Duration;
// use chrono::Duration;
use super::library::{AlbumArt, MusicLibrary, Song, Tag, URI};
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
use uuid::Uuid; use uuid::Uuid;
use super::library::{AlbumArt, MusicLibrary, Song, Tag, URI};
use m3u8_rs::{MediaPlaylist, MediaPlaylistType, MediaSegment, Playlist as List2}; use m3u8_rs::{MediaPlaylist, MediaPlaylistType, MediaSegment, Playlist as List2};
use nestify::nest;
use rayon::prelude::*; use rayon::prelude::*;
#[derive(Debug, Clone, Serialize, Deserialize)] #[derive(Debug, Clone, Serialize, Deserialize)]
pub enum SortOrder { pub enum SortOrder {
Manual, Manual,
Tag(Vec<Tag>) Tag(Vec<Tag>),
} }
#[derive(Debug, Clone, Serialize, Deserialize)]
nest! {
#[derive(Debug, Clone, Deserialize, Serialize)]*
pub struct PlaylistFolder {
name: String,
items: Vec<
pub enum PlaylistFolderItem {
Folder(PlaylistFolder),
List(Playlist)
}
>
}
}
impl PlaylistFolder {
pub fn new() -> Self {
PlaylistFolder {
name: String::new(),
items: Vec::new(),
}
}
}
#[derive(Debug, Clone, Deserialize, Serialize)]
pub struct Playlist { pub struct Playlist {
uuid: Uuid, uuid: Uuid,
title: String, title: String,
@ -75,11 +106,16 @@ impl Playlist {
// } // }
// None // None
// } // }
pub fn contains_value(&self, tag: &Tag, value: &String, lib: Arc<RwLock<MusicLibrary>>) -> bool { pub fn contains_value(
&self,
tag: &Tag,
value: &String,
lib: Arc<RwLock<MusicLibrary>>,
) -> bool {
let lib = lib.read().unwrap(); let lib = lib.read().unwrap();
let items = match lib.query_tracks(value, &vec![tag.to_owned()], &vec![tag.to_owned()]) { let items = match lib.query_tracks(value, &vec![tag.to_owned()], &vec![tag.to_owned()]) {
Some(e) => e, Some(e) => e,
None => return false None => return false,
}; };
for item in items { for item in items {
@ -102,23 +138,35 @@ impl Playlist {
super::utils::read_file(PathBuf::from(path)) super::utils::read_file(PathBuf::from(path))
} }
pub fn to_m3u8(&mut self, lib: Arc<RwLock<MusicLibrary>>, location: &str) -> Result<(), Box<dyn Error>> { pub fn to_m3u8(
&mut self,
lib: Arc<RwLock<MusicLibrary>>,
location: &str,
) -> Result<(), Box<dyn Error>> {
let lib = lib.read().unwrap(); let lib = lib.read().unwrap();
let seg = self.tracks let seg = self
.tracks
.iter() .iter()
.filter_map(|uuid| { .filter_map(|uuid| {
// TODO: The Unwraps need to be handled here
if let Some((track, _)) = lib.query_uuid(uuid) { if let Some((track, _)) = lib.query_uuid(uuid) {
if let URI::Local(_) = track.location { if let URI::Local(_) = track.primary_uri().unwrap().0 {
Some(MediaSegment { Some(MediaSegment {
uri: track.location.to_string(), uri: track.primary_uri().unwrap().0.to_string(),
duration: track.duration.as_millis() as f32, duration: track.duration.as_millis() as f32,
title: track.tags.get_key_value(&Tag::Title).map(|tag| tag.1.into()), title: track
.tags
.get_key_value(&Tag::Title)
.map(|tag| tag.1.into()),
..Default::default() ..Default::default()
}) })
}else { None } } else {
}else { None } None
} }
) } else {
None
}
})
.collect::<Vec<MediaSegment>>(); .collect::<Vec<MediaSegment>>();
let m3u8 = MediaPlaylist { let m3u8 = MediaPlaylist {
@ -142,7 +190,10 @@ impl Playlist {
Ok(()) Ok(())
} }
pub fn from_m3u8(path: &str, lib: Arc<RwLock<MusicLibrary>>) -> Result<Playlist, Box<dyn Error>> { pub fn from_m3u8(
path: &str,
lib: Arc<RwLock<MusicLibrary>>,
) -> Result<Playlist, Box<dyn Error>> {
let mut file = match File::open(path) { let mut file = match File::open(path) {
Ok(file) => file, Ok(file) => file,
Err(e) => return Err(e.into()), Err(e) => return Err(e.into()),
@ -158,7 +209,9 @@ impl Playlist {
}; };
match playlist { match playlist {
List2::MasterPlaylist(_) => Err("This is a Master Playlist!\nPlase input a Media Playlist".into()), List2::MasterPlaylist(_) => {
Err("This is a Master Playlist!\nPlase input a Media Playlist".into())
}
List2::MediaPlaylist(playlist_) => { List2::MediaPlaylist(playlist_) => {
let mut uuids = Vec::new(); let mut uuids = Vec::new();
for seg in playlist_.segments { for seg in playlist_.segments {
@ -179,7 +232,8 @@ impl Playlist {
#[cfg(target_family = "windows")] #[cfg(target_family = "windows")]
{ {
playlist.title = path.split("\\") playlist.title = path
.split("\\")
.last() .last()
.unwrap_or_default() .unwrap_or_default()
.strip_suffix(".m3u8") .strip_suffix(".m3u8")
@ -188,7 +242,8 @@ impl Playlist {
} }
#[cfg(target_family = "unix")] #[cfg(target_family = "unix")]
{ {
playlist.title = path.split("/") playlist.title = path
.split("/")
.last() .last()
.unwrap_or_default() .unwrap_or_default()
.strip_suffix(".m3u8") .strip_suffix(".m3u8")
@ -202,7 +257,6 @@ impl Playlist {
} }
} }
pub fn out_tracks(&self, lib: Arc<RwLock<MusicLibrary>>) -> (Vec<Song>, Vec<&Uuid>) { pub fn out_tracks(&self, lib: Arc<RwLock<MusicLibrary>>) -> (Vec<Song>, Vec<&Uuid>) {
let lib = lib.read().unwrap(); let lib = lib.read().unwrap();
let mut songs = vec![]; let mut songs = vec![];
@ -223,10 +277,12 @@ impl Playlist {
for (i, sort_option) in sort_by.iter().enumerate() { for (i, sort_option) in sort_by.iter().enumerate() {
dbg!(&i); dbg!(&i);
let tag_a = match sort_option { let tag_a = match sort_option {
Tag::Field(field_selection) => match a.get_field(field_selection.as_str()) { Tag::Field(field_selection) => {
match a.get_field(field_selection.as_str()) {
Some(field_value) => field_value.to_string(), Some(field_value) => field_value.to_string(),
None => continue, None => continue,
}, }
}
_ => match a.get_tag(sort_option) { _ => match a.get_tag(sort_option) {
Some(tag_value) => tag_value.to_owned(), Some(tag_value) => tag_value.to_owned(),
None => continue, None => continue,
@ -266,7 +322,6 @@ impl Playlist {
} }
} }
impl Default for Playlist { impl Default for Playlist {
fn default() -> Self { fn default() -> Self {
Playlist { Playlist {
@ -276,7 +331,7 @@ impl Default for Playlist {
tracks: Vec::default(), tracks: Vec::default(),
sort_order: SortOrder::Manual, sort_order: SortOrder::Manual,
play_count: 0, play_count: 0,
play_time: Duration::from_millis(0), play_time: Duration::from_secs(0),
} }
} }
} }
@ -293,14 +348,17 @@ mod test_super {
let tracks = lib.library.iter().map(|track| track.uuid).collect(); let tracks = lib.library.iter().map(|track| track.uuid).collect();
playlist.set_tracks(tracks); playlist.set_tracks(tracks);
_ = playlist.to_m3u8(Arc::new(RwLock::from(lib)), ".\\test-config\\playlists\\playlist.m3u8"); _ = playlist.to_m3u8(
Arc::new(RwLock::from(lib)),
".\\test-config\\playlists\\playlist.m3u8",
);
} }
fn m3u8_to_list() -> Playlist { fn m3u8_to_list() -> Playlist {
let (_, lib) = read_config_lib(); let (_, lib) = read_config_lib();
let arc = Arc::new(RwLock::from(lib)); let arc = Arc::new(RwLock::from(lib));
let playlist = Playlist::from_m3u8(".\\test-config\\playlists\\playlist.m3u8", arc).unwrap(); let playlist =
Playlist::from_m3u8(".\\test-config\\playlists\\playlist.m3u8", arc).unwrap();
playlist.to_file(".\\test-config\\playlists\\playlist"); playlist.to_file(".\\test-config\\playlists\\playlist");
dbg!(playlist) dbg!(playlist)

View file

@ -25,12 +25,12 @@ pub(super) fn normalize(input_string: &str) -> String {
/// Write any data structure which implements [serde::Serialize] /// Write any data structure which implements [serde::Serialize]
/// out to a [bincode] encoded file compressed using [snap] /// out to a [bincode] encoded file compressed using [snap]
pub(super) fn write_file<T: serde::Serialize>( pub(super) fn write_file<T: serde::Serialize, U: std::convert::AsRef<Path>+std::convert::AsRef<std::ffi::OsStr>+Clone>(
library: T, library: T,
path: PathBuf, path: U,
) -> Result<(), Box<dyn Error>> { ) -> Result<(), Box<dyn Error>> {
// Create a temporary name for writing out // Create a temporary name for writing out
let mut writer_name = path.clone(); let mut writer_name = PathBuf::from(&path);
writer_name.set_extension("tmp"); writer_name.set_extension("tmp");
// Create a new BufWriter on the file and a snap frame encoder // Create a new BufWriter on the file and a snap frame encoder