tsk-83: Add token code prior to action execution (#91)
* tsk-83: Code cleanup * tsk-83: Added code to get token before an action begins * tsk-83: Small fix * tsk-83: Version bump * tsk-83: Cleanup
This commit was merged in pull request #91.
This commit is contained in:
Generated
+1
-1
@@ -451,7 +451,7 @@ dependencies = [
|
|||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "icarus-dm"
|
name = "icarus-dm"
|
||||||
version = "0.8.8"
|
version = "0.8.9"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"futures",
|
"futures",
|
||||||
"http",
|
"http",
|
||||||
|
|||||||
+1
-1
@@ -1,6 +1,6 @@
|
|||||||
[package]
|
[package]
|
||||||
name = "icarus-dm"
|
name = "icarus-dm"
|
||||||
version = "0.8.8"
|
version = "0.8.9"
|
||||||
rust-version = "1.90"
|
rust-version = "1.90"
|
||||||
edition = "2024"
|
edition = "2024"
|
||||||
|
|
||||||
|
|||||||
@@ -33,6 +33,7 @@ async fn main() {
|
|||||||
|
|
||||||
let mut cmt_mgr = managers::commit_manager::CommitManager {
|
let mut cmt_mgr = managers::commit_manager::CommitManager {
|
||||||
ica_action: chosen_act,
|
ica_action: chosen_act,
|
||||||
|
..Default::default()
|
||||||
};
|
};
|
||||||
|
|
||||||
cmt_mgr.commit_action().await;
|
cmt_mgr.commit_action().await;
|
||||||
|
|||||||
@@ -10,9 +10,11 @@ use crate::parsers;
|
|||||||
use crate::syncers;
|
use crate::syncers;
|
||||||
use crate::utilities;
|
use crate::utilities;
|
||||||
|
|
||||||
#[derive(Debug)]
|
#[derive(Debug, Default)]
|
||||||
pub struct CommitManager {
|
pub struct CommitManager {
|
||||||
pub ica_action: models::icarus_action::IcarusAction,
|
pub ica_action: models::icarus_action::IcarusAction,
|
||||||
|
pub api: models::api::Api,
|
||||||
|
pub auth_api: models::api::Api,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Clone, Debug)]
|
#[derive(Clone, Debug)]
|
||||||
@@ -84,27 +86,36 @@ pub fn retrieve_song(
|
|||||||
|
|
||||||
impl CommitManager {
|
impl CommitManager {
|
||||||
pub async fn commit_action(&mut self) {
|
pub async fn commit_action(&mut self) {
|
||||||
|
self.init_api().await;
|
||||||
|
|
||||||
let action = &self.ica_action.action;
|
let action = &self.ica_action.action;
|
||||||
println!("Committing {action} action");
|
println!("Committing {action} action");
|
||||||
|
|
||||||
let mapped_actions = &self.map_actions();
|
let mapped_actions = &self.map_actions();
|
||||||
let mapped_action = self.find_mapped_action(mapped_actions, action);
|
let token = self.parse_token(&self.auth_api).await;
|
||||||
|
|
||||||
println!("{mapped_action:?}");
|
match self.find_mapped_action(mapped_actions, action) {
|
||||||
|
ActionValues::DeleteAct => self.delete_song(&token).await,
|
||||||
// TODO: Move code to get token here and then pass it to the respective functions
|
ActionValues::DownloadAct => self.download_song(&token).await,
|
||||||
|
ActionValues::RetrieveAct => self.retrieve_object(&token).await,
|
||||||
match mapped_action {
|
ActionValues::UploadSongWithMetadata => self.upload_song_with_metadata(&token).await,
|
||||||
ActionValues::DeleteAct => self.delete_song().await,
|
|
||||||
ActionValues::DownloadAct => self.download_song().await,
|
|
||||||
ActionValues::RetrieveAct => self.retrieve_object().await,
|
|
||||||
ActionValues::UploadSongWithMetadata => self.upload_song_with_metadata().await,
|
|
||||||
_ => {
|
_ => {
|
||||||
println!("Nothing good here");
|
println!("Nothing good here");
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
async fn init_api(&mut self) {
|
||||||
|
let mut prsr = parsers::api_parser::APIParser {
|
||||||
|
apis: vec![models::api::Api::default(), models::api::Api::default()],
|
||||||
|
ica_act: self.ica_action.clone(),
|
||||||
|
};
|
||||||
|
prsr.parse_api(parsers::api_parser::APIType::Main);
|
||||||
|
prsr.parse_api(parsers::api_parser::APIType::Auth);
|
||||||
|
self.api = prsr.retrieve_api(parsers::api_parser::APIType::Main);
|
||||||
|
self.auth_api = prsr.retrieve_api(parsers::api_parser::APIType::Auth);
|
||||||
|
}
|
||||||
|
|
||||||
fn find_mapped_action(
|
fn find_mapped_action(
|
||||||
&self,
|
&self,
|
||||||
actions: &HashMap<String, ActionValues>,
|
actions: &HashMap<String, ActionValues>,
|
||||||
@@ -134,19 +145,12 @@ impl CommitManager {
|
|||||||
actions
|
actions
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn delete_song(&self) {
|
async fn delete_song(&self, token: &icarus_models::token::AccessToken) {
|
||||||
let mut prsr = parsers::api_parser::APIParser {
|
|
||||||
apis: vec![models::api::Api::default(), models::api::Api::default()],
|
|
||||||
ica_act: self.ica_action.clone(),
|
|
||||||
};
|
|
||||||
prsr.parse_api(parsers::api_parser::APIType::Main);
|
|
||||||
prsr.parse_api(parsers::api_parser::APIType::Auth);
|
|
||||||
let auth_api = prsr.retrieve_api(parsers::api_parser::APIType::Auth);
|
|
||||||
|
|
||||||
let token = self.parse_token(&auth_api).await;
|
|
||||||
|
|
||||||
println!("Deleting song");
|
println!("Deleting song");
|
||||||
|
|
||||||
|
let mut del = syncers::delete::Delete {
|
||||||
|
api: self.api.clone(),
|
||||||
|
};
|
||||||
let mut song = icarus_models::song::Song::default();
|
let mut song = icarus_models::song::Song::default();
|
||||||
|
|
||||||
for arg in &self.ica_action.flags {
|
for arg in &self.ica_action.flags {
|
||||||
@@ -158,12 +162,7 @@ impl CommitManager {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
let api = prsr.retrieve_api(parsers::api_parser::APIType::Main);
|
match del.delete_song(token, &song).await {
|
||||||
let mut del = syncers::delete::Delete { api: api.clone() };
|
|
||||||
|
|
||||||
println!("Deleting song..");
|
|
||||||
|
|
||||||
match del.delete_song(&token, &song).await {
|
|
||||||
Ok((song, _coverart)) => {
|
Ok((song, _coverart)) => {
|
||||||
println!("Song (Id {:?}) has been successfully deleted", song.id);
|
println!("Song (Id {:?}) has been successfully deleted", song.id);
|
||||||
}
|
}
|
||||||
@@ -173,33 +172,20 @@ impl CommitManager {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn download_song(&self) {
|
async fn download_song(&self, token: &icarus_models::token::AccessToken) {
|
||||||
println!("Deleting song");
|
println!("Downloading 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 mut dwn_loader = syncers::download::Download {
|
||||||
|
api: self.api.clone(),
|
||||||
let mut prsr = parsers::api_parser::APIParser {
|
|
||||||
apis: vec![models::api::Api::default(), models::api::Api::default()],
|
|
||||||
ica_act: self.ica_action.clone(),
|
|
||||||
};
|
};
|
||||||
prsr.parse_api(parsers::api_parser::APIType::Main);
|
|
||||||
prsr.parse_api(parsers::api_parser::APIType::Auth);
|
|
||||||
|
|
||||||
let auth_api = prsr.retrieve_api(parsers::api_parser::APIType::Auth);
|
|
||||||
let token = self.parse_token(&auth_api).await;
|
|
||||||
println!("Message: {}", token.message);
|
|
||||||
|
|
||||||
let api = prsr.retrieve_api(parsers::api_parser::APIType::Main);
|
|
||||||
|
|
||||||
let mut dwn_loader = syncers::download::Download { api: api.clone() };
|
|
||||||
let mut song = icarus_models::song::Song {
|
let mut song = icarus_models::song::Song {
|
||||||
id: song_id,
|
id: uuid::Uuid::from_str(dwn.as_str()).unwrap(),
|
||||||
..Default::default()
|
..Default::default()
|
||||||
};
|
};
|
||||||
|
|
||||||
match dwn_loader.download_song(&token, &song).await {
|
match dwn_loader.download_song(token, &song).await {
|
||||||
Ok(o) => {
|
Ok(o) => {
|
||||||
println!("Success");
|
println!("Song downloaded");
|
||||||
|
|
||||||
song.data = o.as_bytes().to_vec();
|
song.data = o.as_bytes().to_vec();
|
||||||
song.directory = String::from(".");
|
song.directory = String::from(".");
|
||||||
@@ -230,7 +216,7 @@ impl CommitManager {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn retrieve_object(&self) {
|
async fn retrieve_object(&self, token: &icarus_models::token::AccessToken) {
|
||||||
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"));
|
||||||
|
|
||||||
@@ -238,22 +224,11 @@ impl CommitManager {
|
|||||||
panic!("Unsupported -rt: {rt}");
|
panic!("Unsupported -rt: {rt}");
|
||||||
}
|
}
|
||||||
|
|
||||||
let mut prsr = parsers::api_parser::APIParser {
|
let mut repo = syncers::retrieve_records::RetrieveRecords {
|
||||||
apis: vec![models::api::Api::default(), models::api::Api::default()],
|
api: self.api.clone(),
|
||||||
ica_act: self.ica_action.clone(),
|
|
||||||
};
|
};
|
||||||
prsr.parse_api(parsers::api_parser::APIType::Main);
|
|
||||||
prsr.parse_api(parsers::api_parser::APIType::Auth);
|
|
||||||
|
|
||||||
let auth_api = prsr.retrieve_api(parsers::api_parser::APIType::Auth);
|
match repo.get_all_songs(token).await {
|
||||||
let token = self.parse_token(&auth_api).await;
|
|
||||||
println!("Token {token:?}");
|
|
||||||
|
|
||||||
let api = prsr.retrieve_api(parsers::api_parser::APIType::Main);
|
|
||||||
|
|
||||||
let mut repo = syncers::retrieve_records::RetrieveRecords { api: api.clone() };
|
|
||||||
|
|
||||||
match repo.get_all_songs(&token).await {
|
|
||||||
Ok(o) => {
|
Ok(o) => {
|
||||||
println!("Songs");
|
println!("Songs");
|
||||||
println!("=====");
|
println!("=====");
|
||||||
@@ -292,7 +267,7 @@ impl CommitManager {
|
|||||||
token.unwrap()
|
token.unwrap()
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn upload_song_with_metadata(&mut self) {
|
async fn upload_song_with_metadata(&mut self, token: &icarus_models::token::AccessToken) {
|
||||||
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"));
|
||||||
@@ -320,10 +295,10 @@ impl CommitManager {
|
|||||||
println!("cover art path: {coverpath}");
|
println!("cover art path: {coverpath}");
|
||||||
|
|
||||||
let _ = self
|
let _ = self
|
||||||
.sing_target_upload(&songpath, &track_id, &metadata_path, &coverpath)
|
.sing_target_upload(&songpath, &track_id, &metadata_path, &coverpath, token)
|
||||||
.await;
|
.await;
|
||||||
} else if multitarget {
|
} else if multitarget {
|
||||||
let _ = self.multi_target_upload(&uni).await;
|
let _ = self.multi_target_upload(&uni, token).await;
|
||||||
} else {
|
} else {
|
||||||
println!("Single or Multi target has not been chosen");
|
println!("Single or Multi target has not been chosen");
|
||||||
}
|
}
|
||||||
@@ -336,19 +311,8 @@ impl CommitManager {
|
|||||||
track_id: &str,
|
track_id: &str,
|
||||||
meta_path: &String,
|
meta_path: &String,
|
||||||
cover_path: &str,
|
cover_path: &str,
|
||||||
|
token: &icarus_models::token::AccessToken,
|
||||||
) -> Result<()> {
|
) -> Result<()> {
|
||||||
let mut prsr = parsers::api_parser::APIParser {
|
|
||||||
apis: vec![models::api::Api::default(), models::api::Api::default()],
|
|
||||||
ica_act: self.ica_action.clone(),
|
|
||||||
};
|
|
||||||
prsr.parse_api(parsers::api_parser::APIType::Main);
|
|
||||||
prsr.parse_api(parsers::api_parser::APIType::Auth);
|
|
||||||
|
|
||||||
let auth_api = prsr.retrieve_api(parsers::api_parser::APIType::Auth);
|
|
||||||
let token = self.parse_token(&auth_api).await;
|
|
||||||
|
|
||||||
println!("Token: {:?}", token.token);
|
|
||||||
|
|
||||||
let song_file = std::path::Path::new(&songpath);
|
let song_file = std::path::Path::new(&songpath);
|
||||||
|
|
||||||
if !song_file.exists() {
|
if !song_file.exists() {
|
||||||
@@ -381,14 +345,13 @@ impl CommitManager {
|
|||||||
println!("Filename: {:?}", s.filename);
|
println!("Filename: {:?}", s.filename);
|
||||||
println!("Path: {:?}", s.song_path());
|
println!("Path: {:?}", s.song_path());
|
||||||
s.data = icarus_models::song::io::to_data(&s).unwrap();
|
s.data = icarus_models::song::io::to_data(&s).unwrap();
|
||||||
|
|
||||||
cover_art.data =
|
cover_art.data =
|
||||||
icarus_models::coverart::io::to_data(&cover_art).unwrap();
|
icarus_models::coverart::io::to_data(&cover_art).unwrap();
|
||||||
|
|
||||||
let members = UploadSongMembers {
|
let members = UploadSongMembers {
|
||||||
song: s,
|
song: s,
|
||||||
coverart: cover_art,
|
coverart: cover_art,
|
||||||
token,
|
token: token.clone(),
|
||||||
album,
|
album,
|
||||||
};
|
};
|
||||||
|
|
||||||
@@ -527,16 +490,11 @@ impl CommitManager {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn multi_target_upload(&mut self, sourcepath: &String) -> std::io::Result<()> {
|
async fn multi_target_upload(
|
||||||
let mut prsr = parsers::api_parser::APIParser {
|
&mut self,
|
||||||
apis: vec![models::api::Api::default(), models::api::Api::default()],
|
sourcepath: &String,
|
||||||
ica_act: self.ica_action.clone(),
|
token: &icarus_models::token::AccessToken,
|
||||||
};
|
) -> std::io::Result<()> {
|
||||||
prsr.parse_api(parsers::api_parser::APIType::Main);
|
|
||||||
prsr.parse_api(parsers::api_parser::APIType::Auth);
|
|
||||||
let auth_api = prsr.retrieve_api(parsers::api_parser::APIType::Auth);
|
|
||||||
let token = self.parse_token(&auth_api).await;
|
|
||||||
|
|
||||||
let directory_path = std::path::Path::new(&sourcepath);
|
let directory_path = std::path::Path::new(&sourcepath);
|
||||||
|
|
||||||
if !directory_path.exists() {
|
if !directory_path.exists() {
|
||||||
|
|||||||
Reference in New Issue
Block a user