Fixed move_to() function and added more queue functions. also created ControllerError

This commit is contained in:
MrDulfin 2024-05-20 23:51:44 -04:00
parent 48c7e29eca
commit 5b94c7950f
3 changed files with 187 additions and 36 deletions

View file

@ -54,7 +54,6 @@ pub struct ConfigLibraries {
} }
impl ConfigLibraries { impl ConfigLibraries {
//TODO: Add new function for test tube
pub fn set_default(mut self, uuid: &Uuid) { pub fn set_default(mut self, uuid: &Uuid) {
self.default_library = *uuid; self.default_library = *uuid;
} }

View file

@ -6,16 +6,20 @@ use crossbeam_channel;
use crossbeam_channel::{Receiver, Sender}; use crossbeam_channel::{Receiver, Sender};
use std::path::PathBuf; use std::path::PathBuf;
use std::sync::{Arc, RwLock}; use std::sync::{Arc, RwLock};
use thiserror::Error;
use crossbeam_channel::unbounded; use crossbeam_channel::unbounded;
use std::error::Error; use std::error::Error;
use uuid::Uuid; use uuid::Uuid;
use crate::music_player::player::Player; use crate::config::config::ConfigError;
use crate::music_player::player::{Player, PlayerError};
use crate::{ use crate::{
config::config::Config, music_controller::queue::Queue, music_storage::library::MusicLibrary, config::config::Config, music_controller::queue::Queue, music_storage::library::MusicLibrary,
}; };
use super::queue::QueueError;
pub struct Controller<P: Player> { pub struct Controller<P: Player> {
pub queue: Queue, pub queue: Queue,
pub config: Arc<RwLock<Config>>, pub config: Arc<RwLock<Config>>,
@ -23,6 +27,16 @@ pub struct Controller<P: Player> {
pub player: Box<P>, pub player: Box<P>,
} }
#[derive(Error, Debug)]
pub enum ControllerError {
#[error("{0:?}")]
QueueError(#[from] QueueError),
#[error("{0:?}")]
PlayerError(#[from] PlayerError),
#[error("{0:?}")]
ConfigError(#[from] ConfigError),
}
#[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>,

View file

@ -1,4 +1,5 @@
use crate::music_storage::library::Song; use crate::music_storage::library::Song;
use chrono::format::Item;
use std::error::Error; use std::error::Error;
use uuid::Uuid; use uuid::Uuid;
@ -6,10 +7,12 @@ use thiserror::Error;
#[derive(Error, Debug)] #[derive(Error, Debug)]
pub enum QueueError { pub enum QueueError {
#[error("Index out of bounds! Index {0} is over len {1}")] #[error("Index out of bounds! Index {index} is over len {len}")]
OutOfBounds(usize, usize), OutOfBounds { index: usize, len: usize },
#[error("The Queue is empty!")] #[error("The Queue is empty!")]
EmptyQueue, EmptyQueue,
#[error("There are no past played songs!")]
EmptyPlayed,
} }
#[derive(Debug, PartialEq, Clone, Copy)] #[derive(Debug, PartialEq, Clone, Copy)]
@ -21,7 +24,7 @@ pub enum QueueState {
} }
// 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, Copy, PartialEq)]
#[non_exhaustive] #[non_exhaustive]
pub enum PlayerLocation { pub enum PlayerLocation {
Test, Test,
@ -39,6 +42,7 @@ pub struct QueueItem {
pub(super) source: PlayerLocation, pub(super) source: PlayerLocation,
pub(super) by_human: bool, pub(super) by_human: bool,
} }
impl From<Song> for QueueItem { impl From<Song> for QueueItem {
fn from(song: Song) -> Self { fn from(song: Song) -> Self {
QueueItem { QueueItem {
@ -58,6 +62,7 @@ pub struct Queue {
pub shuffle: bool, pub shuffle: bool,
} }
// TODO: HAndle the First QueueState
impl Queue { impl Queue {
fn has_addhere(&self) -> bool { fn has_addhere(&self) -> bool {
for item in &self.items { for item in &self.items {
@ -72,8 +77,8 @@ impl Queue {
dbg!( dbg!(
self.items self.items
.iter() .iter()
.map(|item| item.item.clone()) .map(|item| (item.item.uuid, item.state))
.collect::<Vec<Song>>(), .collect::<Vec<(Uuid, QueueState)>>(),
self.items.len() self.items.len()
); );
} }
@ -84,6 +89,7 @@ impl Queue {
self.items.append(&mut tracks); self.items.append(&mut tracks);
} }
/// Inserts an item after the AddHere item
pub fn add_item(&mut self, item: Song, 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;
@ -113,6 +119,7 @@ impl Queue {
); );
} }
/// Inserts an item after the currently playing item
pub fn add_item_next(&mut self, item: Song, 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();
@ -135,9 +142,70 @@ impl Queue {
) )
} }
pub fn add_multi(&mut self, items: Vec<Song>, source: PlayerLocation, by_human: bool) {} pub fn add_multi(&mut self, items: Vec<Song>, source: PlayerLocation, by_human: bool) {
let mut i: usize = 0;
pub fn remove_item(&mut self, remove_index: usize) -> Result<(), QueueError> { self.items = self
.items
.iter()
.enumerate()
.map(|(j, item_)| {
let mut item_ = item_.to_owned();
// get the index of the current AddHere item and give it to i
if item_.state == QueueState::AddHere {
i = j;
item_.state = QueueState::NoState;
}
item_
})
.collect::<Vec<QueueItem>>();
let empty = self.items.is_empty();
let len = items.len();
for item in items {
self.items.insert(
i + if empty { 0 } else { 1 },
QueueItem {
item,
state: QueueState::NoState,
source,
by_human,
},
);
}
self.items[i + len - if empty { 1 } else { 0 }].state = QueueState::AddHere;
}
/// Add multiple Songs after the currently playing Song
pub fn add_multi_next(&mut self, items: Vec<Song>, source: PlayerLocation, by_human: bool) {
use QueueState::*;
let empty = self.items.is_empty();
let add_here = (self.items.get(1).is_none()
|| !self.has_addhere() && self.items.get(1).is_some())
|| empty;
let len = items.len();
for item in items {
self.items.insert(
if empty { 0 } else { 1 },
QueueItem {
item,
state: NoState,
source,
by_human: true,
},
)
}
if add_here {
self.items[len - if empty { 1 } else { 0 }].state = QueueState::AddHere;
}
}
pub fn remove_item(&mut self, remove_index: usize) -> Result<QueueItem, 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() {
@ -145,19 +213,36 @@ impl Queue {
if self.items.get(remove_index + 1).is_some() { if self.items.get(remove_index + 1).is_some() {
self.items[remove_index + 1].state = self.items[remove_index].state; self.items[remove_index + 1].state = self.items[remove_index].state;
} }
self.items[remove_index].state = QueueState::NoState; Ok(self.items.remove(remove_index))
self.items.remove(remove_index);
Ok(())
} else { } else {
Err(QueueError::EmptyQueue) Err(QueueError::EmptyQueue)
} }
} }
pub fn insert<T>(&mut self, index: usize, new_item: T, addhere: bool)
where
QueueItem: std::convert::From<T>,
{
if addhere {
let mut new_item = QueueItem::from(new_item);
for item in &mut self.items {
if item.state == QueueState::AddHere {
item.state = QueueState::NoState
}
}
new_item.state = QueueState::AddHere;
self.items.insert(index, new_item);
} else {
let new_item = QueueItem::from(new_item);
self.items.insert(index, new_item);
}
}
pub fn clear(&mut self) { pub fn clear(&mut self) {
self.items.clear(); self.items.clear();
} }
pub fn clear_except(&mut self, index: usize) -> Result<(), Box<dyn Error>> { pub fn clear_except(&mut self, index: usize) -> Result<(), QueueError> {
use QueueState::*; use QueueState::*;
let empty = self.items.is_empty(); let empty = self.items.is_empty();
@ -166,9 +251,12 @@ impl Queue {
self.items.retain(|item| *item == i); self.items.retain(|item| *item == i);
self.items[0].state = AddHere; self.items[0].state = AddHere;
} else if empty { } else if empty {
return Err("Queue is empty!".into()); return Err(QueueError::EmptyQueue);
} else { } else {
return Err("index out of bounds!".into()); return Err(QueueError::OutOfBounds {
index,
len: self.items.len(),
});
} }
Ok(()) Ok(())
} }
@ -182,8 +270,7 @@ impl Queue {
self.played.clear(); self.played.clear();
} }
// TODO: uh, fix this? pub 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();
@ -194,22 +281,20 @@ impl Queue {
return Err(QueueError::EmptyQueue); return Err(QueueError::EmptyQueue);
}; };
if !empty && index < self.items.len() { if !empty && dbg!(index < self.items.len()) {
let to_item = self.items[index].clone(); let to_item = self.items[index].clone();
loop { loop {
let empty = !self.items.is_empty(); let empty = self.items.is_empty();
let item = self.items[0].item.to_owned(); let item = self.items[0].item.to_owned();
if item != to_item.item && !empty { if item != to_item.item && !empty {
if self.items[0].state == AddHere && self.items.get(1).is_some() { if self.items[0].state == AddHere && self.items.get(1).is_some() {
self.items[1].state = AddHere; self.items[1].state = AddHere;
} }
if let Err(e) = self.remove_item(0) { let item = self.items.remove(0);
dbg!(&e); self.played.push(item);
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 Err(QueueError::EmptyQueue); return Err(QueueError::EmptyQueue);
@ -227,35 +312,52 @@ impl Queue {
self.items.swap(a, b) self.items.swap(a, b)
} }
pub fn move_item(&mut self, a: usize, b: usize) { pub fn move_item(&mut self, from: usize, to: usize) {
let item = self.items[a].to_owned(); let item = self.items[from].to_owned();
if a != b { if from != to {
self.items.remove(a); self.items.remove(from);
} }
self.items.insert(b, item); self.items.insert(to, item);
} }
#[allow(clippy::should_implement_trait)] #[allow(clippy::should_implement_trait)]
pub fn next(&mut self) -> Result<&QueueItem, Box<dyn Error>> { pub fn next(&mut self) -> Result<&QueueItem, QueueError> {
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 unimplemented!() // TODO: add function to loop the queue
} else { } else {
return Err(QueueError::EmptyQueue.into()); return Err(QueueError::EmptyQueue);
} }
} }
// TODO: add an algorithm to detect if the song should be skipped
let item = self.items[0].clone(); let item = self.items.remove(0);
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);
Ok(&self.items[1]) Ok(&self.items[1])
} }
pub fn prev() {} pub fn prev(&mut self) -> Result<&QueueItem, QueueError> {
if self.items[0].state == QueueState::First && self.loop_ {
todo!()
}
if let Some(item) = self.played.pop() {
self.items.insert(0, item);
Ok(&self.items[0])
} else {
Err(QueueError::EmptyPlayed)
}
}
pub fn now_playing(&self) -> Result<&QueueItem, QueueError> {
if !self.items.is_empty() {
Ok(&self.items[0])
} else {
Err(QueueError::EmptyQueue)
}
}
pub fn check_played(&mut self) { pub fn check_played(&mut self) {
while self.played.len() > 50 { while self.played.len() > 50 {
@ -263,3 +365,39 @@ impl Queue {
} }
} }
} }
#[cfg(test)]
mod test_super {
#![allow(unused)]
use crate::{
config::config::tests::{new_config_lib, read_config_lib},
music_storage::library,
};
use super::*;
#[test]
fn move_test() {
let (_, library) = read_config_lib();
let mut q = Queue::default();
q.add_multi(library.library.clone(), PlayerLocation::Library, true);
q.add_multi_next(library.library, PlayerLocation::Library, true);
q.dbg_items();
dbg!(&q.played);
// q.move_to(2).inspect_err(|e| println!("{e:?}"));
// q.dbg_items();
// dbg!(&q.played.iter().map(|i| i.item.uuid).collect::<Vec<_>>());
// let a = q
// .prev()
// .inspect_err(|e| println!("{e:?}"))
// .unwrap()
// .item
// .uuid;
// q.dbg_items();
// dbg!(a, &q.played.iter().map(|i| i.item.uuid).collect::<Vec<_>>());
}
}