Merge branch 'song_changes' into 'dev'

Song changes

See merge request kdeng00/icarus-models!37

Cargo.toml

@@ -1,10 +1,11 @@
 [package]
 name = "icarus-models"
-version = "0.1.7"
+version = "0.1.10"
 edition = "2024"
 description = "models used for the icarus project"
 license = "MIT"
 
 [dependencies]
 serde = { version = "1.0.218", features = ["derive"] }
-serde_json = "1.0.139"
+serde_json = { version = "1.0.139" }
+rand = { version = "0.9" }

src/constants.rs

@@ -1,5 +1,5 @@
-pub const DEFAULT_MUSIC_EXTENSION: &str = FLAC_EXTENSION;
-pub const FLAC_EXTENSION: &str = ".flac";
-pub const WAV_EXTENSION: &str = ".wav";
-pub const MPTHREE_EXTENSION: &str = ".mp3";
-pub const JPG_EXTENSION: &str = ".jpg";
+pub const DEFAULTMUSICEXTENSION: &str = FLACEXTENSION;
+pub const FLACEXTENSION: &str = ".flac";
+pub const WAVEXTENSION: &str = ".wav";
+pub const MPTHREEEXTENSION: &str = ".mp3";
+pub const JPGEXTENSION: &str = ".jpg";

src/song.rs

@@ -1,6 +1,10 @@
 use std::default::Default;
 use std::io::Read;
 
+use crate::constants;
+use crate::types;
+
+use rand::Rng;
 use serde::{Deserialize, Serialize};
 
 #[derive(Clone, Debug, Deserialize, Serialize)]
@@ -42,14 +46,14 @@ pub struct Song {
     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,
+    // #[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 {
@@ -81,10 +85,10 @@ impl Default for Song {
             user_id: 0,
             data: Vec::new(),
             directory: String::new(),
-            album_id: 0,
-            artist_id: 0,
-            genre_id: 0,
-            coverart_id: 0,
+            // album_id: 0,
+            // artist_id: 0,
+            // genre_id: 0,
+            // coverart_id: 0,
         }
     }
 }
@@ -98,32 +102,175 @@ impl Song {
         }
     }
 
-    // TODO: Implement
-    pub fn song_path(&self) -> String {
-        return String::new();
+    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 = String::from(directory.clone());
+        let last_index = directory.len() - 1;
+
+        if let Some(character) = directory.chars().nth(last_index) {
+            if character != '/' {
+                buffer += "/";
+            }
+
+            buffer += &self.filename.clone();
+
+            return Ok(buffer);
+        } else {
+            return 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 = self.song_path();
+        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.len() == 0 {
-            Err(std::io::Error::new(
+                    return Err(std::io::Error::new(
                         std::io::ErrorKind::Other,
                         "File is empty",
-            ))
+                    ));
                 } else {
-            Ok(buffer)
+                    return Ok(buffer);
                 }
             }
+            Err(er) => {
+                return Err(er);
+            }
+        }
+    }
+
+    pub fn generate_filename(&self, typ: types::types::Types, randomize: bool) -> String {
+        let mut filename: String = String::new();
+        let filename_len = 10;
+
+        let file_extension = match typ {
+            types::types::Types::DefaultMusicExtension =>
+            {
+                String::from(constants::DEFAULTMUSICEXTENSION)
+            }
+
+            types::types::Types::WavExtension =>
+            {
+                String::from(constants::WAVEXTENSION)
+            }
+            types::types::Types::FlacExtension =>
+            {
+                String::from(constants::FLACEXTENSION)
+            }
+            types::types::Types::MPThreeExtension =>
+            {
+                String::from(constants::MPTHREEEXTENSION)
+            }
+        };
+
+        if randomize {
+            let some_chars: String = String::from("abcdefghij0123456789");
+            let mut rng = rand::thread_rng();
+
+            for _i in 0..filename_len {
+                let random_number: i32 = rng.gen_range(0..=19);
+                let index = random_number as usize;
+                let rando_char = some_chars.chars().nth(index);
+
+                match rando_char {
+                    Some(c) => {
+                        filename.push(c);
+                    }
+                    None => {}
+                };
+            }
+        } else {
+            filename += "track-output";
+        }
+
+        filename += &file_extension;
+
+        return 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 {
+                return serde_json::to_string_pretty(&self);
+            } else {
+                return 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 = String::from(directory.clone());
+            let last_index = directory.len() - 1;
+
+            if let Some(character) = directory.chars().nth(last_index) {
+                if character != '/' {
+                    buffer += "/";
+                }
+
+                buffer += &self.filename.clone();
+
+                return Ok(buffer);
+            } else {
+                return 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.len() == 0 {
+                        return Err(std::io::Error::new(
+                            std::io::ErrorKind::Other,
+                            "File is empty",
+                        ));
+                    } else {
+                        return Ok(buffer);
+                    }
+                }
+                Err(er) => {
+                    return Err(er);
+                }
+            }
+        }
+    }
+
     // The song's duration is a floating point in seconds
     #[derive(Clone, Debug, Deserialize, Serialize)]
     pub struct Song {
@@ -164,14 +311,14 @@ mod embedded {
         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,
+        // #[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 {
@@ -203,10 +350,10 @@ mod embedded {
                 user_id: 0,
                 data: Vec::new(),
                 directory: String::new(),
-                album_id: 0,
-                artist_id: 0,
-                genre_id: 0,
-                coverart_id: 0,
+                // album_id: 0,
+                // artist_id: 0,
+                // genre_id: 0,
+                // coverart_id: 0,
             }
         }
     }

src/token.rs

@@ -58,7 +58,13 @@ impl Token {
     }
 
     // TODO: Implement
-    pub fn contains_scope(&self, des_scope: String) -> bool {
+    pub fn contains_scope(&self, des_scope: &String) -> bool {
+        let extracted_token: String = String::from("Token");
+
+        if extracted_token == *des_scope {
+            return true;
+        }
+
         return false;
     }
 }

src/types.rs

@@ -1,9 +1,8 @@
-use std::default::Default;
-
-mod types {
+pub mod types {
     pub enum Types {
-        WAV_EXTENSION,
-        FLAC_EXTENSION,
-        MP_EXTENSION,
+        DefaultMusicExtension,
+        WavExtension,
+        FlacExtension,
+        MPThreeExtension,
     }
 }