Refactoring (#24)

Reviewed-on: #24
Co-authored-by: kdeng00 <kundeng00@pm.me>
Co-committed-by: kdeng00 <kundeng00@pm.me>

.env.docker.sample

@@ -0,0 +1,2 @@
+ROOT_DIRECTORY=/home/songparser/mydata
+ICARUS_BASE_API_URL=http://localhost:3000

.env.sample

@@ -0,0 +1,2 @@
+ROOT_DIRECTORY=/home/songparser/mydata
+ICARUS_BASE_API_URL=http://localhost:3000

.gitignore

@@ -1,2 +1,2 @@
 /target
-Cargo.lock
+.env

Cargo.toml

@@ -6,4 +6,11 @@ rust-version = "1.86"
 
 [dependencies]
 tokio = { version = "1.44.1", features = ["full"] }
+futures = { version = "0.3.31" }
+reqwest = { version = "0.12.19", features = ["json", "stream"] }
+serde = { version = "1.0.218", features = ["derive"] }
+serde_json = { version = "1.0.139" }
+time = { version = "0.3.41", features = ["macros", "serde"] }
+uuid = { version = "1.16.0", features = ["v4", "serde"] }
 icarus_models = { git = "ssh://git@git.kundeng.us/phoenix/icarus_models.git", tag = "v0.4.3" }
+icarus_envy = { git = "ssh://git@git.kundeng.us/phoenix/icarus_envy.git", tag = "v0.2.2-devel-84ea6e4c22-006" }

src/main.rs

@@ -1,38 +1,249 @@
-use std::error::Error;
-use tokio::io::AsyncReadExt;
-use tokio::net::{TcpListener, TcpStream};
-use tokio::spawn;
+use std::io::Write;
+
+pub const SECONDS_TO_SLEEP: u64 = 5;
 
 #[tokio::main]
-async fn main() -> Result<(), Box<dyn Error>> {
-    let listener = TcpListener::bind("127.0.0.1:8080").await?;
-    println!("API calling service listening on 127.0.0.1:8080");
+async fn main() -> Result<(), Box<dyn std::error::Error>> {
+    let app_base_url = icarus_envy::environment::get_icarus_base_api_url().await;
 
     loop {
-        let (stream, addr) = listener.accept().await?;
-        println!("Accepted connection from: {}", addr);
+        println!("Base URL: {}", app_base_url);
 
-        spawn(async move {
-            if let Err(e) = handle_connection(stream).await {
-                eprintln!("Error handling connection from {}: {}", addr, e);
+        match is_queue_empty(&app_base_url).await {
+            Ok((empty, song_queue_item)) => {
+                if !empty {
+                    println!("Queue is not empty");
+                    println!("SongQueueItem: {:?}", song_queue_item);
+                    let song_queue_id = song_queue_item.data[0].id;
+
+                    // TODO: Do something with the result later
+                    let _ = process_song(&app_base_url, &song_queue_id).await;
+                } else {
+                    println!("Queue is empty");
+                }
+            }
+            Err(err) => {
+                eprintln!("Error checking if queue is empty: {:?}", err);
             }
-        });
-    }
-}
-
-async fn handle_connection(mut stream: TcpStream) -> Result<(), Box<dyn Error + Send + Sync>> {
-    let mut buffer = [0; 1024];
-
-    loop {
-        let n = stream.read(&mut buffer).await?;
-
-        if n == 0 {
-            break; // Connection closed
         }
 
-        let request_data = String::from_utf8_lossy(&buffer[..n]).trim().to_string();
-        println!("Received request: {}", request_data);
+        println!("Sleeping");
+        tokio::time::sleep(tokio::time::Duration::from_secs(SECONDS_TO_SLEEP)).await;
+    }
+}
+
+async fn is_queue_empty(
+    api_url: &String,
+) -> Result<(bool, responses::fetch_next_queue_item::SongQueueItem), reqwest::Error> {
+    match api::fetch_next_queue_item(api_url).await {
+        Ok(response) => {
+            match response
+                .json::<responses::fetch_next_queue_item::SongQueueItem>()
+                .await
+            {
+                Ok(response) => {
+                    if response.data.is_empty() {
+                        Ok((true, response))
+                    } else {
+                        Ok((false, response))
+                    }
+                }
+                Err(err) => Err(err),
+            }
+        }
+        Err(err) => Err(err),
+    }
+}
+
+async fn process_song(api_url: &String, song_queue_id: &uuid::Uuid) -> Result<(), reqwest::Error> {
+    match api::fetch_song_queue_data::get_data(api_url, song_queue_id).await {
+        Ok(response) => {
+            // Process data here...
+            match api::fetch_song_queue_data::response::parse_response(response).await {
+                Ok(all_bytes) => {
+                    let (directory, filename) = generate_song_queue_dir_and_filename().await;
+                    let save_path = save_song_to_fs(&directory, &filename, &all_bytes).await;
+
+                    println!("Saved at: {:?}", save_path);
+
+                    match api::get_metadata_queue::get(api_url, song_queue_id).await {
+                        Ok(response) => {
+                            match response
+                                .json::<api::get_metadata_queue::response::Response>()
+                                .await
+                            {
+                                Ok(response) => {
+                                    let id = &response.data[0].id;
+                                    let metadata = &response.data[0].metadata;
+                                    let created_at = &response.data[0].created_at;
+                                    println!("Id: {:?}", id);
+                                    println!("Metadata: {:?}", metadata);
+                                    println!("Created at: {:?}", created_at);
+                                    // TODO: Get queued coverart
+                                    // TODO: Get queued coverart's data
+                                    // TODO: Apply metadata to the queued song (modifying file)
+                                    // TODO: Update the queued song with the updated queued song
+                                    // TODO: Create song
+                                    // TODO: Create coverart
+                                    // TODO: Wipe data from queued song
+                                    // TODO: Wipe data from queued coverart
+                                    Ok(())
+                                }
+                                Err(err) => {
+                                    eprintln!("Error: {:?}", err);
+                                    Err(err)
+                                }
+                            }
+                        }
+                        Err(err) => {
+                            eprintln!("Error: {:?}", err);
+                            Err(err)
+                        }
+                    }
+                }
+                Err(err) => Err(err),
+            }
+        }
+        Err(err) => Err(err),
+    }
+}
+
+// TODO: Consider having something like this in icarus_models
+pub async fn generate_song_queue_dir_and_filename() -> (String, String) {
+    let mut song = icarus_models::song::Song::default();
+    song.filename = song.generate_filename(icarus_models::types::MusicTypes::FlacExtension, true);
+
+    song.directory = icarus_envy::environment::get_root_directory().await;
+
+    (song.directory, song.filename)
+}
+
+// TODO: Check to see if this is available in icarus_models
+pub async fn save_song_to_fs(
+    directory: &String,
+    filename: &String,
+    data: &[u8],
+) -> std::path::PathBuf {
+    // TODO: Add function to save bytes to a file in icarus_models
+    // repo
+    let dir = std::path::Path::new(directory);
+    let save_path = dir.join(filename);
+
+    let mut file = std::fs::File::create(&save_path).unwrap();
+    file.write_all(data).unwrap();
+
+    save_path
+}
+
+mod responses {
+    pub mod fetch_next_queue_item {
+        use serde::{Deserialize, Serialize};
+
+        #[derive(Debug, Deserialize, Serialize)]
+        pub struct QueueItem {
+            pub id: uuid::Uuid,
+            pub filename: String,
+            pub status: String,
+        }
+
+        #[derive(Debug, Deserialize, Serialize)]
+        pub struct SongQueueItem {
+            pub message: String,
+            pub data: Vec<QueueItem>,
+        }
+    }
+}
+
+mod api {
+    pub async fn fetch_next_queue_item(
+        base_url: &String,
+    ) -> Result<reqwest::Response, reqwest::Error> {
+        let client = reqwest::Client::new();
+        let fetch_endpoint = String::from("api/v2/song/queue/next");
+        let api_url = format!("{}/{}", base_url, fetch_endpoint);
+        client.get(api_url).send().await
     }
 
-    Ok(())
+    pub mod fetch_song_queue_data {
+        pub async fn get_data(
+            base_url: &String,
+            id: &uuid::Uuid,
+        ) -> Result<reqwest::Response, reqwest::Error> {
+            let client = reqwest::Client::new();
+            let endpoint = String::from("api/v2/song/queue");
+            let api_url = format!("{}/{}/{}", base_url, endpoint, id);
+            client.get(api_url).send().await
+        }
+
+        pub mod response {
+            use futures::StreamExt;
+
+            pub async fn parse_response(
+                response: reqwest::Response,
+            ) -> Result<Vec<u8>, reqwest::Error> {
+                // TODO: At some point, handle the flow if the size is small or
+                // large
+                let mut byte_stream = response.bytes_stream();
+                let mut all_bytes = Vec::new();
+
+                while let Some(chunk) = byte_stream.next().await {
+                    let chunk = chunk?;
+                    all_bytes.extend_from_slice(&chunk);
+                }
+
+                Ok(all_bytes)
+            }
+        }
+    }
+
+    pub mod get_metadata_queue {
+        pub async fn get(
+            base_url: &String,
+            song_queue_id: &uuid::Uuid,
+        ) -> Result<reqwest::Response, reqwest::Error> {
+            let client = reqwest::Client::new();
+            let endpoint = String::from("api/v2/song/metadata/queue");
+            let api_url = format!("{}/{}", base_url, endpoint);
+            client
+                .get(api_url)
+                .query(&[("song_queue_id", song_queue_id)])
+                .send()
+                .await
+        }
+
+        pub mod response {
+            use serde::{Deserialize, Serialize};
+
+            #[derive(Debug, Deserialize, Serialize)]
+            pub struct Metadata {
+                pub id: uuid::Uuid,
+                pub album: String,
+                pub album_artist: String,
+                pub artist: String,
+                pub disc: i32,
+                pub disc_count: i32,
+                pub duration: i64,
+                pub genre: String,
+                pub title: String,
+                pub track: i32,
+                pub track_count: i32,
+                pub year: i32,
+            }
+
+            #[derive(Debug, Deserialize, Serialize)]
+            pub struct QueueItem {
+                pub id: uuid::Uuid,
+                pub metadata: Metadata,
+                #[serde(with = "time::serde::rfc3339")]
+                pub created_at: time::OffsetDateTime,
+                pub song_queue_id: uuid::Uuid,
+            }
+
+            #[derive(Debug, Deserialize, Serialize)]
+            pub struct Response {
+                pub message: String,
+                pub data: Vec<QueueItem>,
+            }
+        }
+    }
 }