mecomp_daemon/services/
mod.rs

use log::warn;
use mecomp_storage::{
    db::schemas::{
        album::{Album, TABLE_NAME as ALBUM_TABLE_NAME},
        artist::{Artist, TABLE_NAME as ARTIST_TABLE_NAME},
        collection::{Collection, TABLE_NAME as COLLECTION_TABLE_NAME},
        playlist::{Playlist, TABLE_NAME as PLAYLIST_TABLE_NAME},
        song::{Song, TABLE_NAME as SONG_TABLE_NAME},
        Thing,
    },
    errors::{Error, StorageResult},
};
use one_or_many::OneOrMany;
use surrealdb::{Connection, Surreal};

pub mod library;
#[cfg(feature = "analysis")]
pub mod radio;

/// Get the songs associated with every thing in the list.
/// This function will go through the list of things and get the songs associated with each thing.
/// It will then remove duplicates from the list of songs.
///
/// # Errors
///
/// This function will return an error if there is an issue reading the songs from the database.
pub async fn get_songs_from_things<C: Connection>(
    db: &Surreal<C>,
    things: Vec<Thing>,
) -> StorageResult<OneOrMany<Song>> {
    // go through the list, and get songs for each thing (depending on what it is)
    let mut songs: OneOrMany<Song> = OneOrMany::None;
    for thing in things {
        match thing.tb.as_str() {
            ALBUM_TABLE_NAME => {
                for song in Album::read_songs(db, thing.clone().into()).await? {
                    songs.push(song);
                }
            }
            ARTIST_TABLE_NAME => {
                for song in Artist::read_songs(db, thing.clone().into()).await? {
                    songs.push(song);
                }
            }
            COLLECTION_TABLE_NAME => {
                for song in Collection::read_songs(db, thing.clone().into()).await? {
                    songs.push(song);
                }
            }
            PLAYLIST_TABLE_NAME => {
                for song in Playlist::read_songs(db, thing.clone().into()).await? {
                    songs.push(song);
                }
            }
            SONG_TABLE_NAME => songs.push(
                Song::read(db, thing.clone().into())
                    .await?
                    .ok_or(Error::NotFound)?,
            ),
            _ => {
                warn!("Unknown thing type: {}", thing.tb);
            }
        }
    }

    // remove duplicates
    songs.dedup_by_key(|song| song.id.clone());

    Ok(songs)
}
mecomp_daemon/services/mod.rs

mecomp_daemon/services/
mod.rs
