Support for icarus v2 API #52

Merged
kdeng00 merged 12 commits from icarus_v2_support into master 2025-08-30 13:23:15 -04:00
7 changed files with 436 additions and 56 deletions
Showing only changes of commit 809f27cf01 - Show all commits
Generated
+1 -1
View File
@@ -479,7 +479,7 @@ dependencies = [
[[package]] [[package]]
name = "icarus-dm" name = "icarus-dm"
version = "0.6.1" version = "0.6.2"
dependencies = [ dependencies = [
"futures", "futures",
"http", "http",
+1 -1
View File
@@ -1,6 +1,6 @@
[package] [package]
name = "icarus-dm" name = "icarus-dm"
version = "0.6.1" version = "0.6.2"
rust-version = "1.88" rust-version = "1.88"
edition = "2024" edition = "2024"
+2 -2
View File
@@ -48,13 +48,13 @@ icarus-dm download -u spacecadet -p stellar40 -h https://icarus.com -b e8407fc6-
### Uploading Song with metadata ### Uploading Song with metadata
```BASH ```BASH
icarus-dm upload-meta -u spacecadet -p stellar40 -h https://icarus.com -s /path/of/song.flac -t 1 -m /path/to/metadata/config/collection.json -ca /path/to/cover/art/image.png icarus-dm upload-meta -u spacecadet -p stellar40 -h https://icarus.com -ha https://auth.icarus.com -s /path/of/song.flac -t 1 -m /path/to/metadata/config/collection.json -ca /path/to/cover/art/image.png
``` ```
### Uploading Song with metadata from directory ### Uploading Song with metadata from directory
```BASH ```BASH
icarus-dm upload-meta -u spacecadet -p stellar40 -h https://icarus.com -smca /path/where/songs/and/metadata/exists/ icarus-dm upload-meta -u spacecadet -p stellar40 -h https://icarus.com -ha https://auth.icarus.com -smca /path/where/songs/and/metadata/exists/
``` ```
### Retrieving Song in json ### Retrieving Song in json
+3 -2
View File
@@ -7,7 +7,8 @@ mod utilities;
use std::env; use std::env;
fn main() { #[tokio::main]
async fn main() {
let args: Vec<String> = env::args().collect(); let args: Vec<String> = env::args().collect();
let args_len = args.len() as i32; let args_len = args.len() as i32;
@@ -29,7 +30,7 @@ fn main() {
ica_action: chosen_act, ica_action: chosen_act,
}; };
cmt_mgr.commit_action(); cmt_mgr.commit_action().await;
} }
#[cfg(test)] #[cfg(test)]
+133 -44
View File
@@ -36,6 +36,14 @@ enum En {
Other, Other,
} }
#[derive(Debug)]
struct UploadSongMembers {
pub song: icarus_models::song::Song,
pub coverart: icarus_models::coverart::CoverArt,
pub token: icarus_models::token::AccessToken,
pub album: icarus_models::album::collection::Album,
}
pub fn retrieve_song( pub fn retrieve_song(
album: &icarus_models::album::collection::Album, album: &icarus_models::album::collection::Album,
track: i32, track: i32,
@@ -79,7 +87,7 @@ pub fn retrieve_song(
} }
impl CommitManager { impl CommitManager {
pub fn commit_action(&mut self) { pub async fn commit_action(&mut self) {
let action = &self.ica_action.action; let action = &self.ica_action.action;
println!("Committing {action} action"); println!("Committing {action} action");
@@ -91,11 +99,11 @@ impl CommitManager {
// TODO: Move code to get token here and then pass it to the respective functions // TODO: Move code to get token here and then pass it to the respective functions
match mapped_action { match mapped_action {
ActionValues::DeleteAct => self.delete_song(), ActionValues::DeleteAct => self.delete_song().await,
ActionValues::DownloadAct => self.download_song(), ActionValues::DownloadAct => self.download_song().await,
ActionValues::RetrieveAct => self.retrieve_object(), ActionValues::RetrieveAct => self.retrieve_object().await,
ActionValues::UploadAct => self.upload_song(), ActionValues::UploadAct => self.upload_song(),
ActionValues::UploadSongWithMetadata => self.upload_song_with_metadata(), ActionValues::UploadSongWithMetadata => self.upload_song_with_metadata().await,
_ => { _ => {
println!("Nothing good here"); println!("Nothing good here");
} }
@@ -132,7 +140,7 @@ impl CommitManager {
actions actions
} }
fn delete_song(&self) { async fn delete_song(&self) {
let mut prsr = parsers::api_parser::APIParser { let mut prsr = parsers::api_parser::APIParser {
apis: vec![models::api::Api::default(), models::api::Api::default()], apis: vec![models::api::Api::default(), models::api::Api::default()],
ica_act: self.ica_action.clone(), ica_act: self.ica_action.clone(),
@@ -141,7 +149,7 @@ impl CommitManager {
prsr.parse_api(parsers::api_parser::APIType::Auth); prsr.parse_api(parsers::api_parser::APIType::Auth);
let auth_api = prsr.retrieve_api(parsers::api_parser::APIType::Auth); let auth_api = prsr.retrieve_api(parsers::api_parser::APIType::Auth);
let token = self.parse_token(&auth_api); let token = self.parse_token(&auth_api).await;
println!("Deleting song"); println!("Deleting song");
@@ -173,7 +181,7 @@ impl CommitManager {
} }
} }
fn download_song(&self) { async fn download_song(&self) {
println!("Deleting song"); println!("Deleting song");
let dwn = self.ica_action.retrieve_flag_value(&String::from("-b")); let dwn = self.ica_action.retrieve_flag_value(&String::from("-b"));
let song_id = uuid::Uuid::from_str(dwn.as_str()).unwrap(); let song_id = uuid::Uuid::from_str(dwn.as_str()).unwrap();
@@ -186,7 +194,7 @@ impl CommitManager {
prsr.parse_api(parsers::api_parser::APIType::Auth); prsr.parse_api(parsers::api_parser::APIType::Auth);
let auth_api = prsr.retrieve_api(parsers::api_parser::APIType::Auth); let auth_api = prsr.retrieve_api(parsers::api_parser::APIType::Auth);
let token = self.parse_token(&auth_api); let token = self.parse_token(&auth_api).await;
println!("Message: {}", token.message); println!("Message: {}", token.message);
let api = prsr.retrieve_api(parsers::api_parser::APIType::Main); let api = prsr.retrieve_api(parsers::api_parser::APIType::Main);
@@ -220,7 +228,7 @@ impl CommitManager {
} }
} }
fn retrieve_object(&self) { async fn retrieve_object(&self) {
println!("Retrieving song"); println!("Retrieving song");
let rt = self.ica_action.retrieve_flag_value(&String::from("-rt")); let rt = self.ica_action.retrieve_flag_value(&String::from("-rt"));
@@ -236,7 +244,7 @@ impl CommitManager {
prsr.parse_api(parsers::api_parser::APIType::Auth); prsr.parse_api(parsers::api_parser::APIType::Auth);
let auth_api = prsr.retrieve_api(parsers::api_parser::APIType::Auth); let auth_api = prsr.retrieve_api(parsers::api_parser::APIType::Auth);
let token = self.parse_token(&auth_api); let token = self.parse_token(&auth_api).await;
println!("Token {token:?}"); println!("Token {token:?}");
let api = prsr.retrieve_api(parsers::api_parser::APIType::Main); let api = prsr.retrieve_api(parsers::api_parser::APIType::Main);
@@ -265,7 +273,7 @@ impl CommitManager {
panic!("Not supported"); panic!("Not supported");
} }
fn parse_token(&self, api: &models::api::Api) -> icarus_models::token::AccessToken { async fn parse_token(&self, api: &models::api::Api) -> icarus_models::token::AccessToken {
println!("Fetching token"); println!("Fetching token");
let mut usr_mgr: managers::user_manager::UserManager = let mut usr_mgr: managers::user_manager::UserManager =
@@ -282,12 +290,12 @@ impl CommitManager {
}; };
tok_mgr.init(); tok_mgr.init();
let token = Runtime::new().unwrap().block_on(tok_mgr.request_token()); let token = tok_mgr.request_token().await;
token.unwrap() token.unwrap()
} }
fn upload_song_with_metadata(&mut self) { async fn upload_song_with_metadata(&mut self) {
println!("Uplodaring song with metadara"); println!("Uplodaring song with metadara");
let songpath = self.ica_action.retrieve_flag_value(&String::from("-s")); let songpath = self.ica_action.retrieve_flag_value(&String::from("-s"));
@@ -314,15 +322,18 @@ impl CommitManager {
println!("metadata path: {metadata_path}"); println!("metadata path: {metadata_path}");
println!("cover art path: {coverpath}"); println!("cover art path: {coverpath}");
let _ = self.sing_target_upload(&songpath, &track_id, &metadata_path, &coverpath); let _ = self
.sing_target_upload(&songpath, &track_id, &metadata_path, &coverpath)
.await;
} else if multitarget { } else if multitarget {
let _ = self.multi_target_upload(&uni); let _ = self.multi_target_upload(&uni).await;
} else { } else {
println!("Single or Multi target has not been chosen"); println!("Single or Multi target has not been chosen");
} }
} }
fn sing_target_upload( /// Only uploading one song
async fn sing_target_upload(
&mut self, &mut self,
songpath: &String, songpath: &String,
track_id: &str, track_id: &str,
@@ -337,7 +348,7 @@ impl CommitManager {
prsr.parse_api(parsers::api_parser::APIType::Auth); prsr.parse_api(parsers::api_parser::APIType::Auth);
let auth_api = prsr.retrieve_api(parsers::api_parser::APIType::Auth); let auth_api = prsr.retrieve_api(parsers::api_parser::APIType::Auth);
let token = self.parse_token(&auth_api); let token = self.parse_token(&auth_api).await;
println!("Token: {:?}", token.token); println!("Token: {:?}", token.token);
@@ -376,21 +387,16 @@ impl CommitManager {
cover_art.data = cover_art.to_data().unwrap(); cover_art.data = cover_art.to_data().unwrap();
let mut up = syncers::upload::Upload::default(); let members = UploadSongMembers {
let host = self.ica_action.retrieve_flag_value(&String::from("-h")); song: s,
up.set_api(&host); coverart: cover_art,
token: token,
album: album,
};
let res = up.upload_song_with_metadata(&token, &s, &cover_art); match self.upload_song_process(&members).await {
Ok(_) => Ok(()),
match Runtime::new().unwrap().block_on(res) { Err(err) => Err(err),
Ok(o) => {
println!("Successfully sent {o:?}");
Ok(())
}
Err(er) => {
println!("Some error {er:?}");
Err(std::io::Error::other(er.to_string()))
}
} }
} }
Err(err) => { Err(err) => {
@@ -408,6 +414,77 @@ impl CommitManager {
} }
} }
/// Upload song to the queue to get processed
async fn upload_song_process(&self, data: &UploadSongMembers) -> Result<()> {
let mut up = syncers::upload::Upload::default();
let host = self.ica_action.retrieve_flag_value(&String::from("-h"));
up.set_api(&host);
let token = &data.token;
let song = &data.song;
let album = &data.album;
let coverart = &data.coverart;
println!("Queueing song");
let queued_song_id = match up.queue_song(token, song).await {
Ok(id) => id,
Err(err) => {
return Err(std::io::Error::other(err.to_string()));
}
};
println!("Queued song Id: {queued_song_id:?}");
match up.link_user_to_queued_song(token, &queued_song_id).await {
Ok(_) => {
println!("Queued song linked to user");
}
Err(err) => {
return Err(std::io::Error::other(err.to_string()));
}
}
let queued_metadata_id = match up.queue_metadata(token, album, song, &queued_song_id).await
{
Ok(id) => id,
Err(err) => {
return Err(std::io::Error::other(err.to_string()));
}
};
println!("Queued metadata Id: {queued_metadata_id:?}");
match up.queue_coverart(token, coverart).await {
Ok(id) => match up
.link_queued_song_to_queued_coverart(token, &queued_song_id, &id)
.await
{
Ok(_) => match up
.update_queued_song_status(token, &queued_song_id, "ready")
.await
{
Ok(_) => {
println!("Queued coverart Id: {id:?}");
println!("Linked queued song to queued coverart");
println!("Queued status updated");
}
Err(err) => {
return Err(std::io::Error::other(err.to_string()));
}
},
Err(err) => {
return Err(std::io::Error::other(err.to_string()));
}
},
Err(err) => {
return Err(std::io::Error::other(err.to_string()));
}
}
Ok(())
}
fn get_songs( fn get_songs(
&self, &self,
metadata_path: &String, metadata_path: &String,
@@ -455,7 +532,7 @@ impl CommitManager {
} }
} }
fn multi_target_upload(&mut self, sourcepath: &String) -> std::io::Result<()> { async fn multi_target_upload(&mut self, sourcepath: &String) -> std::io::Result<()> {
let mut prsr = parsers::api_parser::APIParser { let mut prsr = parsers::api_parser::APIParser {
apis: vec![models::api::Api::default(), models::api::Api::default()], apis: vec![models::api::Api::default(), models::api::Api::default()],
ica_act: self.ica_action.clone(), ica_act: self.ica_action.clone(),
@@ -463,7 +540,7 @@ impl CommitManager {
prsr.parse_api(parsers::api_parser::APIType::Main); prsr.parse_api(parsers::api_parser::APIType::Main);
prsr.parse_api(parsers::api_parser::APIType::Auth); prsr.parse_api(parsers::api_parser::APIType::Auth);
let auth_api = prsr.retrieve_api(parsers::api_parser::APIType::Auth); let auth_api = prsr.retrieve_api(parsers::api_parser::APIType::Auth);
let token = self.parse_token(&auth_api); let token = self.parse_token(&auth_api).await;
let directory_path = std::path::Path::new(&sourcepath); let directory_path = std::path::Path::new(&sourcepath);
@@ -483,28 +560,40 @@ impl CommitManager {
cover_art.data = cover_art.to_data().unwrap(); cover_art.data = cover_art.to_data().unwrap();
match self.get_songs(&metadatapath, sourcepath) { match self.get_songs(&metadatapath, sourcepath) {
Ok(sngs) => { Ok(sngs) => match icarus_models::album::collection::parse_album(&metadatapath) {
Ok(album) => {
for song in sngs { for song in sngs {
match Runtime::new() let members = UploadSongMembers {
.unwrap() song: song,
.block_on(up.upload_song_with_metadata(&token, &song, &cover_art)) coverart: cover_art.clone(),
{ token: token.clone(),
album: album.clone(),
};
match self.upload_song_process(&members).await {
Ok(o) => { Ok(o) => {
println!("Response: {o:?}"); println!("Response: {o:?}");
} }
Err(err) => { Err(err) => {
println!("Error: {err:?}"); println!("Error: {err:?}");
return Err(err);
} }
};
}
}
Err(error) => {
println!("Error: {error:?}");
} }
} }
Ok(()) Ok(())
} }
Err(err) => {
println!("Error: {err:?}");
Err(std::io::Error::other(err.to_string()))
}
},
Err(error) => {
println!("Error: {error:?}");
Err(std::io::Error::other(error.to_string()))
}
}
}
fn get_cover_art_path(&self, directory_path: &String) -> Result<String> { fn get_cover_art_path(&self, directory_path: &String) -> Result<String> {
for entry in read_dir(std::path::Path::new(directory_path))? { for entry in read_dir(std::path::Path::new(directory_path))? {
+9
View File
@@ -19,3 +19,12 @@ fn retrieve_url_with_id(api: &models::api::Api, id: &uuid::Uuid) -> String {
url url
} }
pub async fn auth_header(
token: &icarus_models::token::AccessToken,
) -> (http::HeaderName, http::HeaderValue) {
let auth = reqwest::header::AUTHORIZATION;
let auth_value = http::HeaderValue::from_str(&token.bearer_token()).unwrap();
(auth, auth_value)
}
+282 -1
View File
@@ -11,6 +11,32 @@ pub struct Upload {
pub api: models::api::Api, pub api: models::api::Api,
} }
mod response {
pub mod queue_song {
#[derive(Debug, serde::Deserialize)]
pub struct Response {
pub message: String,
pub data: Vec<uuid::Uuid>,
}
}
pub mod queue_metadata {
#[derive(Debug, serde::Deserialize)]
pub struct Response {
pub message: String,
pub data: Vec<uuid::Uuid>,
}
}
pub mod queue_coverart {
#[derive(Debug, serde::Deserialize)]
pub struct Response {
pub message: String,
pub data: Vec<uuid::Uuid>,
}
}
}
impl Upload { impl Upload {
pub async fn upload_song_with_metadata( pub async fn upload_song_with_metadata(
&mut self, &mut self,
@@ -51,6 +77,261 @@ impl Upload {
} }
} }
pub async fn queue_song(
&self,
token: &icarus_models::token::AccessToken,
song: &icarus_models::song::Song,
) -> Result<uuid::Uuid, reqwest::Error> {
let songpath = song.song_path().unwrap();
let file = tokio::fs::File::open(&songpath).await.unwrap();
let stream = tokio_util::codec::FramedRead::new(file, tokio_util::codec::BytesCodec::new());
let file_body = reqwest::Body::wrap_stream(stream);
let form = reqwest::multipart::Form::new().part(
"file",
reqwest::multipart::Part::stream(file_body)
.file_name(song.filename.clone())
.mime_str("application/octet-stream")
.unwrap(),
);
let endpoint = String::from("api/v2/song/queue");
let url = format!("{}/{endpoint}", self.api.url);
println!("Url: {url:?}");
let mut headers = reqwest::header::HeaderMap::new();
let (auth, auth_val) = syncers::common::auth_header(token).await;
headers.insert(auth, auth_val);
headers.insert("Accept", HeaderValue::from_str("*/*").unwrap());
headers.insert("Connection", HeaderValue::from_str("keep-alive").unwrap());
headers.insert("Cache-Control", HeaderValue::from_str("no-cache").unwrap());
let client = reqwest::Client::builder()
.timeout(std::time::Duration::from_secs(300))
.pool_idle_timeout(std::time::Duration::from_secs(30))
.build()
.unwrap();
match client
.post(url)
.headers(headers)
.multipart(form)
.send()
.await
{
Ok(response) => match response.json::<response::queue_song::Response>().await {
Ok(resp) => {
println!("Message: {:?}", resp.message);
Ok(resp.data[0])
}
Err(err) => Err(err),
},
Err(err) => Err(err),
}
}
pub async fn link_user_to_queued_song(
&self,
token: &icarus_models::token::AccessToken,
queued_song_id: &uuid::Uuid,
) -> Result<(), reqwest::Error> {
let endpoint = String::from("api/v2/song/queue/link");
let url = format!("{}/{endpoint}", self.api.url);
let mut headers = reqwest::header::HeaderMap::new();
let (auth, auth_val) = syncers::common::auth_header(token).await;
headers.insert(auth, auth_val);
let payload = serde_json::json!({
"song_queue_id": queued_song_id,
"user_id": token.user_id
});
let client = reqwest::Client::builder().build().unwrap();
match client
.patch(url)
.headers(headers)
.json(&payload)
.send()
.await
{
Ok(_) => Ok(()),
Err(err) => Err(err),
}
}
pub async fn queue_metadata(
&self,
token: &icarus_models::token::AccessToken,
album: &icarus_models::album::collection::Album,
song: &icarus_models::song::Song,
queued_song_id: &uuid::Uuid,
) -> Result<uuid::Uuid, reqwest::Error> {
let endpoint = String::from("api/v2/song/metadata/queue");
let url = format!("{}/{endpoint}", self.api.url);
let mut headers = reqwest::header::HeaderMap::new();
let (auth, auth_val) = syncers::common::auth_header(token).await;
headers.insert(auth, auth_val);
let payload = serde_json::json!({
"song_queue_id": queued_song_id,
"title": song.title,
"artist": song.artist,
"album_artist": album.artist,
"album": album.title,
"genre": song.genre,
"track": song.track,
"track_count": album.track_count,
"disc": song.disc,
"disc_count": album.disc_count,
"year": album.year,
"duration": song.duration,
});
let client = reqwest::Client::builder().build().unwrap();
match client
.post(url)
.headers(headers)
.json(&payload)
.send()
.await
{
Ok(response) => match response.json::<response::queue_metadata::Response>().await {
Ok(resp) => {
println!("Message: {:?}", resp.message);
Ok(resp.data[0])
}
Err(err) => Err(err),
},
Err(err) => Err(err),
}
}
pub async fn queue_coverart(
&self,
token: &icarus_models::token::AccessToken,
coverart: &icarus_models::coverart::CoverArt,
) -> Result<uuid::Uuid, reqwest::Error> {
let coverartpath = coverart.path.clone();
let file = tokio::fs::File::open(&coverartpath).await.unwrap();
let stream = tokio_util::codec::FramedRead::new(file, tokio_util::codec::BytesCodec::new());
let file_body = reqwest::Body::wrap_stream(stream);
let file_name = std::path::Path::new(&coverartpath)
.file_name()
.and_then(|name| name.to_str())
.map(String::from)
.unwrap_or("file".to_string());
let form = reqwest::multipart::Form::new().part(
"file",
reqwest::multipart::Part::stream(file_body)
.file_name(file_name)
.mime_str("application/octet-stream")
.unwrap(),
);
let endpoint = String::from("api/v2/coverart/queue");
let url = format!("{}/{endpoint}", self.api.url);
println!("Url: {url:?}");
let mut headers = reqwest::header::HeaderMap::new();
let (auth, auth_val) = syncers::common::auth_header(token).await;
headers.insert(auth, auth_val);
headers.insert("Accept", HeaderValue::from_str("*/*").unwrap());
headers.insert("Connection", HeaderValue::from_str("keep-alive").unwrap());
headers.insert("Cache-Control", HeaderValue::from_str("no-cache").unwrap());
let client = reqwest::Client::builder()
.timeout(std::time::Duration::from_secs(300))
.pool_idle_timeout(std::time::Duration::from_secs(30))
.build()
.unwrap();
match client
.post(url)
.headers(headers)
.multipart(form)
.send()
.await
{
Ok(response) => match response.json::<response::queue_coverart::Response>().await {
Ok(resp) => {
println!("Message: {:?}", resp.message);
Ok(resp.data[0])
}
Err(err) => Err(err),
},
Err(err) => Err(err),
}
}
pub async fn link_queued_song_to_queued_coverart(
&self,
token: &icarus_models::token::AccessToken,
queued_song_id: &uuid::Uuid,
queued_coverart_id: &uuid::Uuid,
) -> Result<(), reqwest::Error> {
let endpoint = String::from("api/v2/coverart/queue/link");
let url = format!("{}/{endpoint}", self.api.url);
let mut headers = reqwest::header::HeaderMap::new();
let (auth, auth_val) = syncers::common::auth_header(token).await;
headers.insert(auth, auth_val);
let payload = serde_json::json!({
"song_queue_id": queued_song_id,
"coverart_id": queued_coverart_id
});
let client = reqwest::Client::builder().build().unwrap();
match client
.patch(url)
.headers(headers)
.json(&payload)
.send()
.await
{
Ok(_) => Ok(()),
Err(err) => Err(err),
}
}
pub async fn update_queued_song_status(
&self,
token: &icarus_models::token::AccessToken,
queued_song_id: &uuid::Uuid,
status: &str,
) -> Result<(), reqwest::Error> {
let endpoint = String::from("api/v2/song/queue");
let url = format!("{}/{endpoint}", self.api.url);
let mut headers = reqwest::header::HeaderMap::new();
let (auth, auth_val) = syncers::common::auth_header(token).await;
headers.insert(auth, auth_val);
let payload = serde_json::json!({
"id": queued_song_id,
"status": status
});
let client = reqwest::Client::builder().build().unwrap();
match client
.patch(url)
.headers(headers)
.json(&payload)
.send()
.await
{
Ok(_) => Ok(()),
Err(err) => Err(err),
}
}
fn init_form( fn init_form(
&self, &self,
song: &icarus_models::song::Song, song: &icarus_models::song::Song,
@@ -85,7 +366,7 @@ impl Upload {
pub fn set_api(&mut self, host: &str) { pub fn set_api(&mut self, host: &str) {
let api = models::api::Api { let api = models::api::Api {
url: host.to_owned(), url: host.to_owned(),
version: String::from("v1"), version: String::from(crate::parsers::api_parser::API_VERSION),
endpoint: String::new(), endpoint: String::new(),
}; };
self.api = api; self.api = api;