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 uuid::Uuid;
|
||||||
|
|
||||||
use crate::{music_player::Player, music_storage::library::{Album, Song}};
|
use crate::{music_player::Player, music_storage::library::{Album, Song, URI}};
|
||||||
use std::{error::Error, path::Path};
|
use std::{error::Error, ops::Add, path::Path};
|
||||||
|
|
||||||
#[derive(Debug, PartialEq, Clone, Copy)]
|
#[derive(Debug, PartialEq, Clone, Copy)]
|
||||||
pub enum QueueState {
|
pub enum QueueState {
|
||||||
|
@ -11,15 +11,31 @@ pub enum QueueState {
|
||||||
None,
|
None,
|
||||||
}
|
}
|
||||||
#[derive(Debug, Clone)]
|
#[derive(Debug, Clone)]
|
||||||
|
#[non_exhaustive]
|
||||||
pub enum QueueItemType<'a> {
|
pub enum QueueItemType<'a> {
|
||||||
Song(Uuid),
|
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)]
|
#[derive(Debug, Clone)]
|
||||||
#[non_exhaustive]
|
#[non_exhaustive]
|
||||||
pub struct QueueItem<'a> {
|
pub struct QueueItem<'a> {
|
||||||
item: QueueItemType<'a>,
|
item: QueueItemType<'a>,
|
||||||
state: QueueState
|
state: QueueState,
|
||||||
|
source: QueueSource
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
@ -47,29 +63,34 @@ impl<'a> Queue<'a> {
|
||||||
self.items.append(&mut tracks);
|
self.items.append(&mut tracks);
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn current_index(&self) -> i16 {
|
pub fn current_index(&mut self) -> i16 {
|
||||||
let e = self.items.iter().filter(|song| song.state == QueueState::Played ).collect::<Vec<&QueueItem>>().len();
|
let mut i = 1;
|
||||||
dbg!(&e);
|
let mut e = self.items.iter().filter(|song| song.state == QueueState::Played ).collect::<Vec<&QueueItem>>().len();
|
||||||
e as i16 - 1
|
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::*;
|
use QueueState::*;
|
||||||
let ind = self.current_index();
|
let ind = self.current_index();
|
||||||
let mut i: i16 = 1;
|
let mut i: i16 = 1;
|
||||||
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();
|
||||||
if item_.state == AddHere {
|
if item_.state == AddHere {
|
||||||
i = j as i16 + 2;
|
i = j as i16 + 1 - ind;
|
||||||
item_.state = None;
|
item_.state = None;
|
||||||
}
|
}
|
||||||
if item_.state == Current {
|
if item_.state == Current {
|
||||||
i = j as i16 + 2;
|
i = j as i16 + 1 - ind;
|
||||||
}
|
}
|
||||||
item_
|
item_
|
||||||
}).collect::<Vec<QueueItem>>();
|
}).collect::<Vec<QueueItem>>();
|
||||||
let pos = (ind + i) as usize;
|
let pos = (ind + i) as usize;
|
||||||
dbg!(&pos);
|
// dbg!(&pos, &i, &ind);
|
||||||
self.items.insert(
|
self.items.insert(
|
||||||
pos,
|
pos,
|
||||||
QueueItem {
|
QueueItem {
|
||||||
|
@ -78,27 +99,48 @@ impl<'a> Queue<'a> {
|
||||||
Current
|
Current
|
||||||
}else {
|
}else {
|
||||||
AddHere
|
AddHere
|
||||||
}
|
},
|
||||||
|
source
|
||||||
}
|
}
|
||||||
);
|
);
|
||||||
Ok(())
|
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]
|
#[test]
|
||||||
fn itemaddtest() {
|
fn item_add_test() {
|
||||||
let mut q = Queue::new().unwrap();
|
let mut q = Queue::new().unwrap();
|
||||||
q.add_item(QueueItemType::Song(Uuid::new_v4())).unwrap();
|
for _ in 0..5 {
|
||||||
dbg!(&q.items);
|
q.items.push(QueueItem { item: QueueItemType::Song(Uuid::new_v4()), state: QueueState::Played, source: QueueSource::Queue });
|
||||||
q.add_item(QueueItemType::Song(Uuid::new_v4())).unwrap();
|
}
|
||||||
dbg!(&q.items);
|
for _ in 0..3 {
|
||||||
q.add_item(QueueItemType::Song(Uuid::new_v4())).unwrap();
|
q.add_item(QueueItemType::Song(Uuid::new_v4()), QueueSource::Library).unwrap();
|
||||||
dbg!(&q.items);
|
}
|
||||||
q.add_item(QueueItemType::Song(Uuid::new_v4())).unwrap();
|
dbg!(&q.items, &q.items.len());
|
||||||
dbg!(&q.items);
|
|
||||||
q.add_item(QueueItemType::Song(Uuid::new_v4())).unwrap();
|
for _ in 0..1 {
|
||||||
dbg!(&q.items);
|
q.remove_item(0).inspect_err(|e| println!("{e:?}"));
|
||||||
q.add_item(QueueItemType::Song(Uuid::new_v4())).unwrap();
|
dbg!(&q.items.len());
|
||||||
dbg!(&q.items);
|
}
|
||||||
|
|
||||||
|
dbg!(&q.items, &q.items.len());
|
||||||
}
|
}
|
|
@ -24,6 +24,7 @@ pub enum SortOrder {
|
||||||
}
|
}
|
||||||
#[derive(Debug, Clone)]
|
#[derive(Debug, Clone)]
|
||||||
pub struct Playlist {
|
pub struct Playlist {
|
||||||
|
uuid: Uuid,
|
||||||
title: String,
|
title: String,
|
||||||
cover: Option<AlbumArt>,
|
cover: Option<AlbumArt>,
|
||||||
tracks: Vec<Uuid>,
|
tracks: Vec<Uuid>,
|
||||||
|
@ -152,6 +153,7 @@ impl Playlist {
|
||||||
impl Default for Playlist {
|
impl Default for Playlist {
|
||||||
fn default() -> Self {
|
fn default() -> Self {
|
||||||
Playlist {
|
Playlist {
|
||||||
|
uuid: Uuid::new_v4(),
|
||||||
title: String::default(),
|
title: String::default(),
|
||||||
cover: None,
|
cover: None,
|
||||||
tracks: Vec::default(),
|
tracks: Vec::default(),
|
||||||
|
|
Loading…
Reference in a new issue