Added function to add items to the queue.

currently does not work with songs that have the `Played` State
This commit is contained in:
MrDulfin 2024-02-26 20:17:37 -05:00
parent 0e6a676e09
commit 79ec5aa1ef
3 changed files with 102 additions and 20 deletions

View file

@ -27,6 +27,8 @@ use crate::{
music_controller::queue::Queue, music_controller::queue::Queue,
}; };
use super::queue::{QueueItem, QueueState};
pub struct Controller { pub struct Controller {
// queues: Vec<Queue>, // queues: Vec<Queue>,
config: Arc<RwLock<Config>>, config: Arc<RwLock<Config>>,
@ -74,7 +76,7 @@ enum QueueCmd {
Test, Test,
Play, Play,
Pause, Pause,
SetSongs(Vec<Song>), // SetSongs(Vec<QueueItem<QueueState>>),
// SetLocation(URI), // SetLocation(URI),
Enqueue(URI), Enqueue(URI),
SetVolume(f64), SetVolume(f64),
@ -248,10 +250,10 @@ impl Controller {
Err(_) => todo!() Err(_) => todo!()
} }
}, },
SetSongs(songs) => { // SetSongs(songs) => {
queue.set_tracks(songs); // queue.set_tracks(songs);
in_thread.send(QueueResponse::Default).unwrap(); // in_thread.send(QueueResponse::Default).unwrap();
}, // },
Enqueue(uri) => { Enqueue(uri) => {
queue.player.enqueue_next(&uri).unwrap(); queue.player.enqueue_next(&uri).unwrap();
@ -287,12 +289,12 @@ impl Controller {
Ok(()) Ok(())
} }
fn q_set_songs(&self, index: usize, songs: Vec<Song>) -> Result<(), Box<dyn Error>> { // fn q_set_songs(&self, index: usize, songs: Vec<QueueItem<QueueState>>) -> Result<(), Box<dyn Error>> {
let mail = &self.queue_mail[index]; // let mail = &self.queue_mail[index];
mail.send(QueueCmd::SetSongs(songs))?; // mail.send(QueueCmd::SetSongs(songs))?;
dbg!(mail.recv()?); // dbg!(mail.recv()?);
Ok(()) // Ok(())
} // }
fn q_enqueue(&self, index: usize, uri: URI) -> Result<(), Box<dyn Error>> { fn q_enqueue(&self, index: usize, uri: URI) -> Result<(), Box<dyn Error>> {
let mail = &self.queue_mail[index]; let mail = &self.queue_mail[index];

View file

@ -1,25 +1,104 @@
use crate::{music_player::Player, music_storage::library::Song}; use uuid::Uuid;
use std::error::Error;
pub struct Queue { use crate::{music_player::Player, music_storage::library::{Album, Song}};
use std::{error::Error, path::Path};
#[derive(Debug, PartialEq, Clone, Copy)]
pub enum QueueState {
Played,
Current,
AddHere,
None,
}
#[derive(Debug, Clone)]
pub enum QueueItemType<'a> {
Song(Uuid),
Album(Album<'a>)
}
#[derive(Debug, Clone)]
#[non_exhaustive]
pub struct QueueItem<'a> {
item: QueueItemType<'a>,
state: QueueState
}
#[derive(Debug)]
pub struct Queue<'a> {
pub player: Player, pub player: Player,
pub name: String, pub name: String,
pub songs: Vec<Song>, pub items: Vec<QueueItem<'a>>,
} }
impl Queue {
impl<'a> Queue<'a> {
pub fn new() -> Result<Self, Box<dyn Error>> { pub fn new() -> Result<Self, Box<dyn Error>> {
Ok( Ok(
Queue { Queue {
player: Player::new()?, player: Player::new()?,
name: String::new(), name: String::new(),
songs: Vec::new() items: Vec::new()
} }
) )
} }
pub fn set_tracks(&mut self, tracks: Vec<Song>) { pub fn set_items(&mut self, tracks: Vec<QueueItem<'a>>) {
let mut tracks = tracks; let mut tracks = tracks;
self.songs.clear(); self.items.clear();
self.songs.append(&mut tracks); self.items.append(&mut tracks);
}
pub fn current_index(&self) -> i16 {
let e = self.items.iter().filter(|song| song.state == QueueState::Played ).collect::<Vec<&QueueItem>>().len();
dbg!(&e);
e as i16 - 1
}
pub fn add_item(&mut self, item: QueueItemType<'a>) -> Result<(), Box<dyn Error>> {
use QueueState::*;
let ind = self.current_index();
let mut i: i16 = 1;
self.items = self.items.iter().enumerate().map(|(j, item_)| {
let mut item_ = item_.to_owned();
if item_.state == AddHere {
i = j as i16 + 2;
item_.state = None;
}
if item_.state == Current {
i = j as i16 + 2;
}
item_
}).collect::<Vec<QueueItem>>();
let pos = (ind + i) as usize;
dbg!(&pos);
self.items.insert(
pos,
QueueItem {
item: item.clone(),
state: if pos == self.items.len() && i == 1 {
Current
}else {
AddHere
}
}
);
Ok(())
} }
} }
#[test]
fn itemaddtest() {
let mut q = Queue::new().unwrap();
q.add_item(QueueItemType::Song(Uuid::new_v4())).unwrap();
dbg!(&q.items);
q.add_item(QueueItemType::Song(Uuid::new_v4())).unwrap();
dbg!(&q.items);
q.add_item(QueueItemType::Song(Uuid::new_v4())).unwrap();
dbg!(&q.items);
q.add_item(QueueItemType::Song(Uuid::new_v4())).unwrap();
dbg!(&q.items);
q.add_item(QueueItemType::Song(Uuid::new_v4())).unwrap();
dbg!(&q.items);
q.add_item(QueueItemType::Song(Uuid::new_v4())).unwrap();
dbg!(&q.items);
}

View file

@ -90,6 +90,7 @@ enum PlaybackStats {
} }
/// An instance of a music player with a GStreamer backend /// An instance of a music player with a GStreamer backend
#[derive(Debug)]
pub struct Player { pub struct Player {
source: Option<URI>, source: Option<URI>,
//pub message_tx: Sender<PlayerCmd>, //pub message_tx: Sender<PlayerCmd>,