Compare commits
5 Commits
v0.4.8
...
39ee3a6dfb
| Author | SHA1 | Date | |
|---|---|---|---|
|
39ee3a6dfb
|
|||
|
6727d44496
|
|||
|
9096d02e79
|
|||
| 9b6ccb032f | |||
| 8a8f3c1075 |
6
Cargo.lock
generated
6
Cargo.lock
generated
@@ -520,8 +520,8 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "icarus_models"
|
||||
version = "0.8.0"
|
||||
source = "git+ssh://git@git.kundeng.us/phoenix/icarus_models.git?tag=v0.8.0#860d684e7b340b05d43d1fc3e6bd36b7cc45d81b"
|
||||
version = "0.9.2"
|
||||
source = "git+ssh://git@git.kundeng.us/phoenix/icarus_models.git?tag=v0.9.2-91-d121dbb06f-111#d121dbb06fd2605e8789b37245f2e1af083edd00"
|
||||
dependencies = [
|
||||
"josekit",
|
||||
"rand",
|
||||
@@ -1322,7 +1322,7 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "songparser"
|
||||
version = "0.4.8"
|
||||
version = "0.4.11"
|
||||
dependencies = [
|
||||
"futures",
|
||||
"icarus_envy",
|
||||
|
||||
@@ -1,6 +1,6 @@
|
||||
[package]
|
||||
name = "songparser"
|
||||
version = "0.4.8"
|
||||
version = "0.4.11"
|
||||
edition = "2024"
|
||||
rust-version = "1.90"
|
||||
|
||||
@@ -14,5 +14,5 @@ time = { version = "0.3.44", features = ["macros", "serde"] }
|
||||
uuid = { version = "1.18.1", features = ["v4", "serde"] }
|
||||
rand = { version = "0.9.2" }
|
||||
icarus_meta = { git = "ssh://git@git.kundeng.us/phoenix/icarus_meta.git", tag = "v0.4.3" }
|
||||
icarus_models = { git = "ssh://git@git.kundeng.us/phoenix/icarus_models.git", tag = "v0.8.0" }
|
||||
icarus_models = { git = "ssh://git@git.kundeng.us/phoenix/icarus_models.git", tag = "v0.9.2-91-d121dbb06f-111" }
|
||||
icarus_envy = { git = "ssh://git@git.kundeng.us/phoenix/icarus_envy.git", tag = "v0.5.0" }
|
||||
|
||||
@@ -1,5 +1,4 @@
|
||||
pub mod fetch_next_queue_item {
|
||||
|
||||
pub async fn fetch_next_queue_item(
|
||||
app: &crate::config::App,
|
||||
) -> Result<reqwest::Response, reqwest::Error> {
|
||||
@@ -12,9 +11,7 @@ pub mod fetch_next_queue_item {
|
||||
}
|
||||
|
||||
pub mod response {
|
||||
use serde::{Deserialize, Serialize};
|
||||
|
||||
#[derive(Debug, Deserialize, Serialize)]
|
||||
#[derive(Debug, serde::Deserialize, serde::Serialize)]
|
||||
pub struct QueueItem {
|
||||
pub id: uuid::Uuid,
|
||||
pub filename: String,
|
||||
@@ -22,7 +19,7 @@ pub mod fetch_next_queue_item {
|
||||
pub user_id: uuid::Uuid,
|
||||
}
|
||||
|
||||
#[derive(Debug, Deserialize, Serialize)]
|
||||
#[derive(Debug, serde::Deserialize, serde::Serialize)]
|
||||
pub struct SongQueueItem {
|
||||
pub message: String,
|
||||
pub data: Vec<QueueItem>,
|
||||
|
||||
@@ -3,4 +3,25 @@ pub struct App {
|
||||
pub uri: String,
|
||||
pub auth_uri: String,
|
||||
pub token: icarus_models::login_result::LoginResult,
|
||||
pub root_directory: String,
|
||||
}
|
||||
|
||||
impl App {
|
||||
pub fn does_root_directory_exists(&self) -> bool {
|
||||
let path = std::path::Path::new(&self.root_directory);
|
||||
if path.exists() { path.is_dir() } else { false }
|
||||
}
|
||||
}
|
||||
|
||||
pub async fn initialize_app_config() -> App {
|
||||
App {
|
||||
uri: icarus_envy::environment::get_icarus_base_api_url()
|
||||
.await
|
||||
.value,
|
||||
auth_uri: icarus_envy::environment::get_icarus_auth_base_api_url()
|
||||
.await
|
||||
.value,
|
||||
root_directory: icarus_envy::environment::get_root_directory().await.value,
|
||||
..Default::default()
|
||||
}
|
||||
}
|
||||
|
||||
29
src/main.rs
29
src/main.rs
@@ -11,22 +11,27 @@ pub const SECONDS_TO_SLEEP: u64 = 5;
|
||||
|
||||
#[tokio::main]
|
||||
async fn main() -> Result<(), Box<dyn std::error::Error>> {
|
||||
let mut app = config::App {
|
||||
uri: icarus_envy::environment::get_icarus_base_api_url()
|
||||
.await
|
||||
.value,
|
||||
auth_uri: icarus_envy::environment::get_icarus_auth_base_api_url()
|
||||
.await
|
||||
.value,
|
||||
..Default::default()
|
||||
};
|
||||
let mut app = config::initialize_app_config().await;
|
||||
println!("Base URL: {:?}", app.uri);
|
||||
println!("Auth URL: {:?}", app.auth_uri);
|
||||
|
||||
match auth::get_token(&app).await {
|
||||
Ok(login_result) => {
|
||||
app.token = login_result;
|
||||
if !app.does_root_directory_exists() {
|
||||
eprintln!("Root directory does not exist");
|
||||
println!("Attempting to create directory");
|
||||
let path = std::path::Path::new(&app.root_directory);
|
||||
match std::fs::create_dir(path) {
|
||||
Ok(_) => {
|
||||
println!("Directory created");
|
||||
}
|
||||
Err(err) => {
|
||||
eprintln!("Error creating directory: {err:?}");
|
||||
std::process::exit(-1);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
app.token = match auth::get_token(&app).await {
|
||||
Ok(login_result) => login_result,
|
||||
Err(err) => {
|
||||
eprintln!("Error: {err:?}");
|
||||
std::process::exit(-1);
|
||||
|
||||
@@ -107,9 +107,10 @@ pub async fn prep_song(
|
||||
let song = icarus_models::song::Song {
|
||||
directory: icarus_envy::environment::get_root_directory().await.value,
|
||||
filename: icarus_models::song::generate_filename(
|
||||
icarus_models::types::MusicTypes::FlacExtension,
|
||||
icarus_models::types::MusicType::FlacExtension,
|
||||
true,
|
||||
),
|
||||
)
|
||||
.unwrap(),
|
||||
data: song_bytes,
|
||||
..Default::default()
|
||||
};
|
||||
@@ -139,80 +140,112 @@ pub async fn prep_song(
|
||||
.await
|
||||
{
|
||||
Ok(response) => {
|
||||
let id = &response.data[0].id;
|
||||
let created_at = &response.data[0].created_at;
|
||||
let metadata = &response.data[0].metadata;
|
||||
let bod = &response.data[0];
|
||||
match process_coverart(app, &queued_song.id, bod).await {
|
||||
Ok(qc) => Ok((queued_song, qc, bod.metadata.clone())),
|
||||
Err(err) => Err(err),
|
||||
}
|
||||
}
|
||||
Err(err) => Err(err),
|
||||
}
|
||||
}
|
||||
Err(err) => Err(err),
|
||||
}
|
||||
}
|
||||
Err(err) => Err(err),
|
||||
}
|
||||
}
|
||||
Err(err) => Err(err),
|
||||
}
|
||||
}
|
||||
|
||||
pub async fn process_coverart(
|
||||
app: &crate::config::App,
|
||||
queued_song_id: &uuid::Uuid,
|
||||
queued_item: &crate::api::get_metadata_queue::response::QueueItem,
|
||||
) -> Result<crate::queued_item::QueuedCoverArt, reqwest::Error> {
|
||||
let id = queued_item.id;
|
||||
let created_at = queued_item.created_at;
|
||||
let metadata = &queued_item.metadata;
|
||||
println!("Id: {id:?}");
|
||||
println!("Metadata: {metadata:?}");
|
||||
println!("Created at: {created_at:?}");
|
||||
|
||||
println!("Getting coverart queue");
|
||||
match crate::api::get_coverart_queue::get(app, &queued_song.id).await {
|
||||
match crate::api::get_coverart_queue::get(app, queued_song_id).await {
|
||||
Ok(response) => {
|
||||
match response.json::<crate::api::get_coverart_queue::response::Response>().await {
|
||||
match response
|
||||
.json::<crate::api::get_coverart_queue::response::Response>()
|
||||
.await
|
||||
{
|
||||
Ok(response) => {
|
||||
let coverart_queue = &response.data[0];
|
||||
let coverart_queue_id = coverart_queue.id;
|
||||
println!("Coverart queue Id: {coverart_queue_id:?}");
|
||||
|
||||
match crate::api::get_coverart_queue::get_data(app, &coverart_queue_id).await {
|
||||
Ok(response) => match crate::api::parsing::parse_response_into_bytes(response).await {
|
||||
Ok(response) => {
|
||||
match crate::api::parsing::parse_response_into_bytes(response).await {
|
||||
Ok(coverart_queue_bytes) => {
|
||||
let (directory, filename) = crate::util::generate_coverart_queue_dir_and_filename(&coverart_queue.file_type).await;
|
||||
let queued_coverart = init_queued_coverart(
|
||||
&coverart_queue_id,
|
||||
&coverart_queue.file_type,
|
||||
coverart_queue_bytes,
|
||||
)
|
||||
.await;
|
||||
println!(
|
||||
"Saved coverart queue file at: {:?}",
|
||||
queued_coverart.path
|
||||
);
|
||||
|
||||
Ok(queued_coverart)
|
||||
}
|
||||
Err(err) => Err(err),
|
||||
}
|
||||
}
|
||||
Err(err) => Err(err),
|
||||
}
|
||||
}
|
||||
Err(err) => Err(err),
|
||||
}
|
||||
}
|
||||
Err(err) => Err(err),
|
||||
}
|
||||
}
|
||||
|
||||
async fn init_queued_coverart(
|
||||
coverart_queue_id: &uuid::Uuid,
|
||||
file_type: &str,
|
||||
bytes: Vec<u8>,
|
||||
) -> crate::queued_item::QueuedCoverArt {
|
||||
// TODO: Consider separating song and coverart when saving to the filesystem
|
||||
let covart_type = if file_type == icarus_meta::detection::coverart::constants::PNG_TYPE {
|
||||
icarus_models::types::CoverArtType::PngExtension
|
||||
} else if file_type == icarus_meta::detection::coverart::constants::JPEG_TYPE {
|
||||
icarus_models::types::CoverArtType::JpegExtension
|
||||
} else if file_type == icarus_meta::detection::coverart::constants::JPG_TYPE {
|
||||
icarus_models::types::CoverArtType::JpgExtension
|
||||
} else {
|
||||
icarus_models::types::CoverArtType::None
|
||||
};
|
||||
let coverart = icarus_models::coverart::CoverArt {
|
||||
directory,
|
||||
filename,
|
||||
data: coverart_queue_bytes,
|
||||
directory: icarus_envy::environment::get_root_directory().await.value,
|
||||
filename: match icarus_models::coverart::generate_filename(covart_type, true) {
|
||||
Ok(filename) => filename,
|
||||
Err(err) => {
|
||||
eprintln!("Error generating CoverArt filename: {err:?}");
|
||||
panic!("Error initializing queued CoverArt");
|
||||
}
|
||||
},
|
||||
data: bytes,
|
||||
..Default::default()
|
||||
};
|
||||
coverart.save_to_filesystem().unwrap();
|
||||
let coverart_queue_fs_path = match coverart.get_path() {
|
||||
Ok(path) => {
|
||||
path
|
||||
}
|
||||
Err(err) => {
|
||||
eprintln!("Error: {err:?}");
|
||||
std::process::exit(-1);
|
||||
}
|
||||
};
|
||||
|
||||
let queued_coverart = crate::queued_item::QueuedCoverArt {
|
||||
id: coverart_queue_id,
|
||||
let coverart_queue_fs_path = coverart.get_path().unwrap();
|
||||
crate::queued_item::QueuedCoverArt {
|
||||
id: *coverart_queue_id,
|
||||
coverart,
|
||||
path: coverart_queue_fs_path
|
||||
};
|
||||
|
||||
println!("Saved coverart queue file at: {:?}", queued_coverart.path);
|
||||
|
||||
Ok((queued_song, queued_coverart, metadata.clone()))
|
||||
}
|
||||
Err(err) => {
|
||||
Err(err)
|
||||
}
|
||||
}
|
||||
Err(err) => {
|
||||
Err(err)
|
||||
}
|
||||
}
|
||||
}
|
||||
Err(err) => {
|
||||
Err(err)
|
||||
}
|
||||
}
|
||||
}
|
||||
Err(err) => Err(err),
|
||||
}
|
||||
}
|
||||
Err(err) => Err(err),
|
||||
}
|
||||
}
|
||||
Err(err) => Err(err),
|
||||
}
|
||||
}
|
||||
Err(err) => Err(err),
|
||||
}
|
||||
}
|
||||
Err(err) => Err(err),
|
||||
path: coverart_queue_fs_path,
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
36
src/util.rs
36
src/util.rs
@@ -4,39 +4,3 @@ pub fn path_buf_to_string(path: &std::path::Path) -> String {
|
||||
None => String::new(),
|
||||
}
|
||||
}
|
||||
|
||||
// TODO: Consider having something like this in icarus_models
|
||||
pub async fn generate_coverart_queue_dir_and_filename(file_type: &str) -> (String, String) {
|
||||
use rand::Rng;
|
||||
|
||||
let mut filename: String = String::new();
|
||||
let filename_len = 10;
|
||||
|
||||
let some_chars: String = String::from("abcdefghij0123456789");
|
||||
let mut rng = rand::rng();
|
||||
|
||||
for _ 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);
|
||||
}
|
||||
}
|
||||
|
||||
filename += if file_type == icarus_meta::detection::coverart::constants::JPEG_TYPE
|
||||
|| file_type == icarus_meta::detection::coverart::constants::JPG_TYPE
|
||||
{
|
||||
icarus_models::constants::file_extensions::image::JPEGEXTENSION
|
||||
} else if file_type == icarus_meta::detection::coverart::constants::PNG_TYPE {
|
||||
icarus_models::constants::file_extensions::image::PNGEXTENSION
|
||||
} else {
|
||||
""
|
||||
};
|
||||
|
||||
// TODO: Consider separating song and coverart when saving to the filesystem
|
||||
let directory = icarus_envy::environment::get_root_directory().await.value;
|
||||
|
||||
(directory, filename)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user