feat: add add command

Author: ankddev

Cargo.lock

@@ -1,6 +1,6 @@
 [package]
 name = "envfetch"
-version = "1.2.1"
+version = "1.3.0"
 edition = "2021"
 authors = ["ANKDDEV"]
 description = "Lightweight cross-platform CLI tool for working with environment variables"

Cargo.toml

@@ -19,7 +19,7 @@
 - [x] Set variable (temporary and permanent)
 - [x] Delete variable (temporary and permanent)
 - [x] Load variables from dotenv-style file (temporary and permanent)
-- [ ] Add string to the end of variable
+- [x] Add string to the end of variable
 - [ ] Set and delete multiple variables at once
 # Get started
 ## Installing
@@ -64,7 +64,7 @@ You can run `envfetch help` to see help message or `envfetch --version` to see p
 
 ### Command list
 #### Set
-Set environment variable and run process.
+Set environment variable and optionally run process.
 
 Usage:
 `envfetch set <KEY> <VALUE> [PROCESS]`, where:
@@ -83,7 +83,26 @@ For example:
 $ envfetch set MY_VAR "Hello" "npm run"  # temporary for process
 $ envfetch set MY_VAR "Hello" --global   # permanent system-wide
 ```
+#### Add
+Add value to the end of environment variable and optionally run the process
 
+Usage:
+`envfetch add <KEY> <VALUE> [PROCESS]`, where:
+- `KEY` - name of environment variable
+- `VALUE` - value of environment variable to add
+- `PROCESS` - name of process which you want to run (optional if --global is used)
+
+Options:
+- `--help`/`-h` - show help message
+- `--global`/`-g` - update variable permanently in system environment
+  - On Windows: stores in registry
+  - On Unix: stores in shell config (.bashrc, .zshrc, or config.fish)
+
+For example:
+```shell
+$ envfetch add PATH "../hello.exe" "crago run"  # temporary for process
+$ envfetch add MY_VAR "Hello" --global   # permanent system-wide
+```
 #### Print
 Print all environment variables
 
@@ -121,7 +140,7 @@ $ envfetch get MY_VAR
 ```
 It will print value of specified variable.
 #### Delete
-Delete variable and start process.
+Delete variable and optionally start process.
 
 Usage:
 `envfetch delete <KEY> [PROCESS]`, where:
@@ -139,7 +158,7 @@ $ envfetch delete MY_VAR --global   # permanent system-wide
 ```
 
 #### Load
-Load environment variables from dotenv-style file and run process.
+Load environment variables from dotenv-style file and optionally run process.
 
 Usage:
 `envfetch load [PROCESS]`, where:

README.md

@@ -83,6 +83,25 @@ pub fn set(args: &SetArgs) {
     }
 }
 
+/// Add value to environment variable
+pub fn add(args: &AddArgs) {
+    if let Err(err) = validate_var_name(&args.key) {
+        error(&err);
+        process::exit(1);
+    }
+
+    let current_value = if let Ok(value) = env::var(&args.key) {
+        value
+    } else {
+        "".to_string()
+    };
+
+    if let Err(err) = variables::set_variable(&args.key, &format!("{}{}", current_value, args.value), args.global, args.process.clone()) {
+        error(&err);
+        process::exit(1);
+    }
+}
+
 /// Delete environment variable
 pub fn delete(args: &DeleteArgs, exit_on_warning: bool) {
     if let Err(err) = validate_var_name(&args.key) {

src/commands.rs

@@ -34,6 +34,10 @@ fn main() {
         Commands::Set(ref opt) => {
             set(opt);
         }
+        // Add command handler
+        Commands::Add(ref opt) => {
+            add(opt);
+        }
         // Delete command handler
         Commands::Delete(ref opt) => {
             delete(opt, cli.exit_on_warning);

src/main.rs

@@ -26,15 +26,17 @@ pub struct Cli {
 /// All tool's commands
 #[derive(Subcommand)]
 pub enum Commands {
-    /// Prints value of environment variable
+    /// Print value of environment variable.
     Get(GetArgs),
-    /// Set environment variable and run given process.
+    /// Set environment variable and optionally run given process.
     Set(SetArgs),
-    /// Delete environment variable and run given process.
+    /// Add value to the end of environment variable and optionally run given process.
+    Add(AddArgs),
+    /// Delete environment variable and optionally run given process.
     Delete(DeleteArgs),
-    /// Load environment variables from dotenv file
+    /// Load environment variables from dotenv file and optionally run given process.
     Load(LoadArgs),
-    /// Prints all environment variables
+    /// Print all environment variables.
     Print,
 }
 
@@ -81,6 +83,23 @@ pub struct SetArgs {
     pub process: Option<String>,
 }
 
+/// Args for add command
+#[derive(Args, Debug)]
+pub struct AddArgs {
+    /// Environment variable name
+    #[arg(required = true)]
+    pub key: String,
+    /// Value for add to the end of environment variable
+    #[arg(required = true)]
+    pub value: String,
+    /// Globally set variable
+    #[arg(required = false, long, short)]
+    pub global: bool,
+    /// Process to start, not required if --global flag is set
+    #[arg(required_unless_present = "global")]
+    pub process: Option<String>,
+}
+
 /// Args for delete command
 #[derive(Args, Debug)]
 pub struct DeleteArgs {

src/models.rs

@@ -194,4 +194,95 @@ fn load_custom_file_doesnt_exists() -> Result<(), Box<dyn std::error::Error>> {
     Ok(())
 }
 
+#[test]
+fn test_add_local_variable() -> Result<(), Box<dyn std::error::Error>> {
+    let envfetch = Command::cargo_bin("envfetch")?.get_program().to_string_lossy().to_string();
+    let mut cmd = Command::cargo_bin("envfetch")?;
+    
+    cmd.args([
+        "add", 
+        "TEST_VAR", 
+        "test_value", 
+        &format!("{} get TEST_VAR", envfetch)
+    ])
+    .assert()
+    .success()
+    .stdout(predicate::str::contains("test_value"));
+    
+    Ok(())
+}
+
+#[test]
+fn test_add_variable_with_special_characters() -> Result<(), Box<dyn std::error::Error>> {
+    let envfetch = Command::cargo_bin("envfetch")?.get_program().to_string_lossy().to_string();
+    let mut cmd = Command::cargo_bin("envfetch")?;
+    
+    cmd.args([
+        "add", 
+        "SPECIAL_VAR", 
+        "test@#$%^&*", 
+        &format!("{} get SPECIAL_VAR", envfetch)
+    ])
+    .assert()
+    .success()
+    .stdout(predicate::str::contains("test@#$%^&*"));
+    
+    Ok(())
+}
+
+#[test]
+fn test_add_empty_value() -> Result<(), Box<dyn std::error::Error>> {
+    let envfetch = Command::cargo_bin("envfetch")?.get_program().to_string_lossy().to_string();
+    let mut cmd = Command::cargo_bin("envfetch")?;
+    
+    cmd.args([
+        "add", 
+        "EMPTY_VAR", 
+        "", 
+        &format!("{} get EMPTY_VAR", envfetch)
+    ])
+    .assert()
+    .success()
+    .stdout(predicate::str::contains("\"\"\n")); // Expect empty string in quotes with newline
+    
+    Ok(())
+}
+
+#[test]
+fn test_add_invalid_variable_name() -> Result<(), Box<dyn std::error::Error>> {
+    let mut cmd = Command::cargo_bin("envfetch")?;
+    cmd.args(["add", "INVALID NAME", "test_value", "echo test"])
+        .assert()
+        .failure()
+        .stderr(predicates::str::contains("Variable name cannot contain spaces"));
+    Ok(())
+}
+
+#[test]
+fn test_add_missing_value() -> Result<(), Box<dyn std::error::Error>> {
+    let mut cmd = Command::cargo_bin("envfetch")?;
+    cmd.args(["add", "TEST_VAR"])
+        .assert()
+        .failure();
+    Ok(())
+}
+
+#[test]
+fn test_add_with_process() -> Result<(), Box<dyn std::error::Error>> {
+    let envfetch = Command::cargo_bin("envfetch")?.get_program().to_string_lossy().to_string();
+    let mut cmd = Command::cargo_bin("envfetch")?;
+    
+    cmd.args([
+        "add", 
+        "PROCESS_VAR", 
+        "test_value", 
+        &format!("{} get PROCESS_VAR", envfetch)
+    ])
+    .assert()
+    .success()
+    .stdout(predicate::str::contains("test_value"));
+    
+    Ok(())
+}
+
 // TODO: add tests for commands with --global flag