mirror of
https://github.com/Dangoware/dmp-core.git
synced 2025-04-19 09:42:53 -05:00
added remove_item()
function to Queue
This commit is contained in:
parent
79ec5aa1ef
commit
e175fe7337
2 changed files with 71 additions and 27 deletions
|
@ -1,7 +1,7 @@
|
|||
use uuid::Uuid;
|
||||
|
||||
use crate::{music_player::Player, music_storage::library::{Album, Song}};
|
||||
use std::{error::Error, path::Path};
|
||||
use crate::{music_player::Player, music_storage::library::{Album, Song, URI}};
|
||||
use std::{error::Error, ops::Add, path::Path};
|
||||
|
||||
#[derive(Debug, PartialEq, Clone, Copy)]
|
||||
pub enum QueueState {
|
||||
|
@ -11,15 +11,31 @@ pub enum QueueState {
|
|||
None,
|
||||
}
|
||||
#[derive(Debug, Clone)]
|
||||
#[non_exhaustive]
|
||||
pub enum QueueItemType<'a> {
|
||||
Song(Uuid),
|
||||
Album(Album<'a>)
|
||||
ExternalSong(URI),
|
||||
Album{
|
||||
album: Album<'a>,
|
||||
shuffled: bool,
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone)]
|
||||
#[non_exhaustive]
|
||||
pub enum QueueSource {
|
||||
Library,
|
||||
Playlist(Uuid),
|
||||
Search,
|
||||
Queue,
|
||||
File,
|
||||
}
|
||||
#[derive(Debug, Clone)]
|
||||
#[non_exhaustive]
|
||||
pub struct QueueItem<'a> {
|
||||
item: QueueItemType<'a>,
|
||||
state: QueueState
|
||||
state: QueueState,
|
||||
source: QueueSource
|
||||
}
|
||||
|
||||
|
||||
|
@ -47,29 +63,34 @@ impl<'a> Queue<'a> {
|
|||
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);
|
||||
pub fn current_index(&mut self) -> i16 {
|
||||
let mut i = 1;
|
||||
let mut e = self.items.iter().filter(|song| song.state == QueueState::Played ).collect::<Vec<&QueueItem>>().len();
|
||||
while e >= 51 {
|
||||
self.items.remove(0);
|
||||
e = self.items.iter().filter(|song| song.state == QueueState::Played ).collect::<Vec<&QueueItem>>().len();
|
||||
i+=1;
|
||||
}
|
||||
e as i16 - 1
|
||||
}
|
||||
|
||||
pub fn add_item(&mut self, item: QueueItemType<'a>) -> Result<(), Box<dyn Error>> {
|
||||
pub fn add_item(&mut self, item: QueueItemType<'a>, source: QueueSource) -> 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;
|
||||
i = j as i16 + 1 - ind;
|
||||
item_.state = None;
|
||||
}
|
||||
if item_.state == Current {
|
||||
i = j as i16 + 2;
|
||||
i = j as i16 + 1 - ind;
|
||||
}
|
||||
item_
|
||||
}).collect::<Vec<QueueItem>>();
|
||||
let pos = (ind + i) as usize;
|
||||
dbg!(&pos);
|
||||
// dbg!(&pos, &i, &ind);
|
||||
self.items.insert(
|
||||
pos,
|
||||
QueueItem {
|
||||
|
@ -78,27 +99,48 @@ impl<'a> Queue<'a> {
|
|||
Current
|
||||
}else {
|
||||
AddHere
|
||||
}
|
||||
},
|
||||
source
|
||||
}
|
||||
);
|
||||
Ok(())
|
||||
}
|
||||
|
||||
pub fn remove_item(&mut self, index: usize) -> Result<(), Box<dyn Error>> {
|
||||
use QueueState::*;
|
||||
let ind = (self.current_index() + index as i16 + 1) as usize;
|
||||
|
||||
if ind < self.items.len() {
|
||||
// update the state of the next item to replace the item being removed
|
||||
if self.items.get(ind + 1).is_some() {
|
||||
self.items[ind + 1].state = self.items[ind].state;
|
||||
}
|
||||
self.items[ind].state = None;
|
||||
self.items.remove(ind);
|
||||
|
||||
Ok(())
|
||||
}else {
|
||||
Err("No Songs to remove!".into())
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
#[test]
|
||||
fn itemaddtest() {
|
||||
fn item_add_test() {
|
||||
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);
|
||||
for _ in 0..5 {
|
||||
q.items.push(QueueItem { item: QueueItemType::Song(Uuid::new_v4()), state: QueueState::Played, source: QueueSource::Queue });
|
||||
}
|
||||
for _ in 0..3 {
|
||||
q.add_item(QueueItemType::Song(Uuid::new_v4()), QueueSource::Library).unwrap();
|
||||
}
|
||||
dbg!(&q.items, &q.items.len());
|
||||
|
||||
for _ in 0..1 {
|
||||
q.remove_item(0).inspect_err(|e| println!("{e:?}"));
|
||||
dbg!(&q.items.len());
|
||||
}
|
||||
|
||||
dbg!(&q.items, &q.items.len());
|
||||
}
|
|
@ -24,6 +24,7 @@ pub enum SortOrder {
|
|||
}
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct Playlist {
|
||||
uuid: Uuid,
|
||||
title: String,
|
||||
cover: Option<AlbumArt>,
|
||||
tracks: Vec<Uuid>,
|
||||
|
@ -152,6 +153,7 @@ impl Playlist {
|
|||
impl Default for Playlist {
|
||||
fn default() -> Self {
|
||||
Playlist {
|
||||
uuid: Uuid::new_v4(),
|
||||
title: String::default(),
|
||||
cover: None,
|
||||
tracks: Vec::default(),
|
||||
|
|
Loading…
Reference in a new issue