icarus_models/src/song.rs

use std::io::Read;

use crate::constants;
use crate::types;

use rand::Rng;
use serde::{Deserialize, Serialize};

#[derive(Clone, Debug, Default, Deserialize, Serialize)]
pub struct Song {
    #[serde(skip_serializing_if = "is_zero")]
    #[serde(alias = "id")]
    pub id: i32,
    #[serde(skip_serializing_if = "String::is_empty")]
    pub title: String,
    #[serde(skip_serializing_if = "String::is_empty")]
    pub artist: String,
    #[serde(skip_serializing_if = "String::is_empty")]
    pub album: String,
    #[serde(skip_serializing_if = "String::is_empty")]
    pub album_artist: String,
    #[serde(skip_serializing_if = "String::is_empty")]
    pub genre: String,
    #[serde(skip_serializing_if = "is_zero")]
    pub year: i32,
    #[serde(skip_serializing_if = "is_dur_not_set")]
    pub duration: i32,
    #[serde(skip_serializing_if = "is_zero")]
    pub track: i32,
    #[serde(skip_serializing_if = "is_zero")]
    pub disc: i32,
    #[serde(skip_serializing_if = "is_zero")]
    pub disc_count: i32,
    #[serde(skip_serializing_if = "is_zero")]
    pub track_count: i32,
    #[serde(skip_serializing_if = "String::is_empty")]
    pub audio_type: String,
    #[serde(skip_serializing_if = "String::is_empty")]
    pub date_created: String,
    #[serde(skip_serializing_if = "String::is_empty")]
    pub filename: String,
    #[serde(skip_serializing_if = "is_zero")]
    pub user_id: i32,
    #[serde(skip)]
    pub data: Vec<u8>,
    #[serde(skip)]
    pub directory: String,
    // #[serde(skip)]
    // pub album_id: i32,
    // #[serde(skip)]
    // pub artist_id: i32,
    // #[serde(skip)]
    // pub genre_id: i32,
    // #[serde(skip)]
    // pub coverart_id: i32,
}

fn is_zero(num: &i32) -> bool {
    *num == 0
}

fn is_dur_not_set(num: &i32) -> bool {
    *num == 0
}

impl Song {
    pub fn to_metadata_json(&self, pretty: bool) -> Result<String, serde_json::Error> {
        if pretty {
            serde_json::to_string_pretty(&self)
        } else {
            serde_json::to_string(&self)
        }
    }

    pub fn song_path(&self) -> Result<String, std::io::Error> {
        if self.directory.is_empty() {
            return Err(std::io::Error::new(
                std::io::ErrorKind::Other,
                "Directory does not exist",
            ));
        }

        let directory = &self.directory;
        let mut buffer: String = directory.clone();
        let last_index = directory.len() - 1;

        if let Some(character) = directory.chars().nth(last_index) {
            if character != '/' {
                buffer += "/";
            }

            buffer += &self.filename.clone();

            Ok(buffer)
        } else {
            Err(std::io::Error::new(
                std::io::ErrorKind::Other,
                "Could not access last character of directory",
            ))
        }
    }

    pub fn to_data(&self) -> Result<Vec<u8>, std::io::Error> {
        let path_result = self.song_path();

        match path_result {
            Ok(path) => {
                let mut file = std::fs::File::open(path)?;
                let mut buffer: Vec<u8> = Vec::new();
                file.read_to_end(&mut buffer)?;

                if buffer.is_empty() {
                    Err(std::io::Error::new(
                        std::io::ErrorKind::Other,
                        "File is empty",
                    ))
                } else {
                    Ok(buffer)
                }
            }
            Err(er) => Err(er),
        }
    }

    pub fn generate_filename(&self, typ: types::MusicTypes, randomize: bool) -> String {
        let mut filename: String = String::new();
        let filename_len = 10;

        let file_extension = match typ {
            types::MusicTypes::DefaultMusicExtension => {
                String::from(constants::file_extensions::audio::DEFAULTMUSICEXTENSION)
            }

            types::MusicTypes::WavExtension => {
                String::from(constants::file_extensions::audio::WAVEXTENSION)
            }
            types::MusicTypes::FlacExtension => {
                String::from(constants::file_extensions::audio::FLACEXTENSION)
            }
            types::MusicTypes::MPThreeExtension => {
                String::from(constants::file_extensions::audio::MPTHREEEXTENSION)
            }
        };

        if randomize {
            let some_chars: String = String::from("abcdefghij0123456789");
            let mut rng = rand::rng();

            for _i in 0..filename_len {
                let random_number: i32 = rng.random_range(0..=19);
                let index = random_number as usize;
                let rando_char = some_chars.chars().nth(index);

                if let Some(c) = rando_char {
                    filename.push(c);
                }
            }
        } else {
            filename += "track-output";
        }

        filename += &file_extension;

        filename
    }
}

/*
mod embedded {
    use std::io::Read;

    use serde::{Deserialize, Serialize};

    impl Song {
        pub fn to_metadata_json(&self, pretty: bool) -> Result<String, serde_json::Error> {
            if pretty {
                serde_json::to_string_pretty(&self)
            } else {
                serde_json::to_string(&self)
            }
        }

        pub fn song_path(&self) -> Result<String, std::io::Error> {
            if self.directory.is_empty() {
                return Err(std::io::Error::new(
                    std::io::ErrorKind::Other,
                    "Directory does not exist",
                ));
            }

            let directory = &self.directory;
            let mut buffer: String = directory.clone();
            let last_index = directory.len() - 1;

            if let Some(character) = directory.chars().nth(last_index) {
                if character != '/' {
                    buffer += "/";
                }

                buffer += &self.filename.clone();

                Ok(buffer)
            } else {
                Err(std::io::Error::new(
                    std::io::ErrorKind::Other,
                    "Could not access last character of directory",
                ))
            }
        }

        pub fn to_data(&self) -> Result<Vec<u8>, std::io::Error> {
            let path_result = self.song_path();

            match path_result {
                Ok(path) => {
                    let mut file = std::fs::File::open(path)?;
                    let mut buffer: Vec<u8> = Vec::new();
                    file.read_to_end(&mut buffer)?;

                    if buffer.is_empty() {
                        Err(std::io::Error::new(
                            std::io::ErrorKind::Other,
                            "File is empty",
                        ))
                    } else {
                        Ok(buffer)
                    }
                }
                Err(er) => Err(er),
            }
        }
    }

    // The song's duration is a floating point in seconds
    #[derive(Clone, Debug, Deserialize, Serialize)]
    pub struct Song {
        #[serde(skip_serializing_if = "is_embed_zero")]
        #[serde(alias = "id")]
        pub id: i32,
        #[serde(skip_serializing_if = "String::is_empty")]
        pub title: String,
        #[serde(skip_serializing_if = "String::is_empty")]
        pub artist: String,
        #[serde(skip_serializing_if = "String::is_empty")]
        pub album: String,
        #[serde(skip_serializing_if = "String::is_empty")]
        pub album_artist: String,
        #[serde(skip_serializing_if = "String::is_empty")]
        pub genre: String,
        #[serde(skip_serializing_if = "is_embed_zero")]
        pub year: i32,
        #[serde(skip_serializing_if = "is_embed_dur_not_set")]
        pub duration: f64,
        #[serde(skip_serializing_if = "is_embed_zero")]
        pub track: i32,
        #[serde(skip_serializing_if = "is_embed_zero")]
        pub disc: i32,
        #[serde(skip_serializing_if = "is_embed_zero")]
        pub disc_count: i32,
        #[serde(skip_serializing_if = "is_embed_zero")]
        pub track_count: i32,
        #[serde(skip_serializing_if = "String::is_empty")]
        pub audio_type: String,
        #[serde(skip_serializing_if = "String::is_empty")]
        pub date_created: String,
        #[serde(skip_serializing_if = "String::is_empty")]
        pub filename: String,
        #[serde(skip_serializing_if = "is_embed_zero")]
        pub user_id: i32,
        #[serde(skip)]
        pub data: Vec<u8>,
        #[serde(skip)]
        pub directory: String,
        // #[serde(skip)]
        // pub album_id: i32,
        // #[serde(skip)]
        // pub artist_id: i32,
        // #[serde(skip)]
        // pub genre_id: i32,
        // #[serde(skip)]
        // pub coverart_id: i32,
    }

    fn is_embed_zero(num: &i32) -> bool {
        *num == 0
    }

    fn is_embed_dur_not_set(num: &f64) -> bool {
        *num == 0.0
    }

    impl Default for Song {
        fn default() -> Self {
            Song {
                id: 0,
                title: String::new(),
                artist: String::new(),
                album: String::new(),
                album_artist: String::new(),
                genre: String::new(),
                year: 0,
                duration: 0.0,
                track: 0,
                disc: 0,
                disc_count: 0,
                track_count: 0,
                audio_type: String::new(),
                date_created: String::new(),
                filename: String::new(),
                user_id: 0,
                data: Vec::new(),
                directory: String::new(),
                // album_id: 0,
                // artist_id: 0,
                // genre_id: 0,
                // coverart_id: 0,
            }
        }
    }
}
*/