feat: add initialization command and enhance app setup with async file handling

Author: Kremilly

.env-example

@@ -0,0 +1,9 @@
+DS_DB_NAME=""
+DS_DB_PORT=""
+DS_DB_DSN=""
+DS_DB_USER=""
+DS_DB_PASSWORD=""
+DS_DB_HOST=""
+
+DS_DUMP_INTERVAL=""
+DS_DUMP_PATH=""

Cargo.toml

@@ -26,6 +26,7 @@ reqwest = { version = "0.12.9", features = ["blocking"] }
 serde = { version = "1.0.215", features = ["derive"] }
 serde_json = "1.0.133"
 serde_yaml = "0.9.34"
+tokio = { version = "1.41.1", features = ["full"] }
 
 [profile.release]
 lto = true

dumpsync1.yml

@@ -0,0 +1,10 @@
+exports:
+  dump_data: true
+  compress_data: true
+  insert_ignore_into: false
+  drop_table_if_exists: true
+  database_if_not_exists: true
+
+connection:
+  max_retries: 3
+  retry_connection_interval: 5

src/args_cli.rs

@@ -1,10 +1,10 @@
 use clap_cargo::style;
-use clap::builder::styling::Styles;
 
 use clap::{
     Parser,
     Subcommand,
     ColorChoice,
+    builder::styling::Styles,
 };
 
 pub const CLAP_STYLING: Styles = Styles::styled()
@@ -32,6 +32,9 @@ pub enum Commands {
 
     /// Import the database dump
     Import(ImportOptions),
+    
+    /// Initialize the new dump sync project
+    Init,
 }
 
 #[derive(Parser)]

src/dump_sync.rs

@@ -1,9 +1,17 @@
 use clap::Parser;
 
+use reqwest;
+use tokio::fs::File;
+use tokio::io::AsyncWriteExt;
+
+use std::error::Error;
+
 use crate::{
     ui::ui_base::UI,
     helpers::env::Env,
     engine::dump::Dump,
+    consts::global::Global,
+    ui::success_alerts::SuccessAlerts,
 
     args_cli::{
         Cli,
@@ -17,6 +25,17 @@ pub struct DumpSync;
 
 impl DumpSync {
 
+    async fn initialize(&self) -> Result<(), Box<dyn Error>> {
+        let response = reqwest::get(Global::APP_CONFIGS).await?;
+        let content = response.bytes().await?;
+        
+        let mut file = File::create(Global::app_config()).await?;
+        file.write_all(&content).await?;
+        
+        SuccessAlerts::settings();
+        Ok(())
+    }
+
     fn import(&self, options: ImportOptions) {
         Env::new();
         UI::header();
@@ -70,7 +89,7 @@ impl DumpSync {
         Dump::new(&host, port, &user, &password, &dbname, &backup_path, Some(interval), &backup_path).export();
     }
 
-    pub fn init(&self) {
+    pub async fn init(&self) -> Result<(), Box<dyn Error>> {
         let cli = Cli::parse();
 
         match cli.command {
@@ -81,7 +100,13 @@ impl DumpSync {
             Commands::Import(options) => {
                 self.import(options);
             },
+
+            Commands::Init => {
+                self.initialize().await?;
+            },
         }
+
+        Ok(())
     }
 
 }

src/main.rs

@@ -8,6 +8,12 @@ mod dump_sync;
 
 use crate::dump_sync::DumpSync;
 
-fn main() {
-    DumpSync.init();
+#[tokio::main]
+async fn main() -> Result<(), Box<dyn std::error::Error>> {
+    if let Err(e) = DumpSync.init().await {
+        eprintln!("Error initializing app: {}", e);
+        return Err(e);
+    }
+
+    Ok(())
 }

src/ui/success_alerts.rs

@@ -49,4 +49,13 @@ impl SuccessAlerts {
         );
     }
 
+    pub fn settings() {
+        let current_datetime = Date::date_time();
+    
+        println!(
+            "\r{} The settings file was successfully created", 
+            current_datetime.green().bold(),
+        );
+    }
+
 }