feat: add --format argument to print command

Author: ankddev

README.md

@@ -140,6 +140,9 @@ Usage:
 
 Options:
 - `--help`/`-h` - show help message
+- `--format <FORMAT>`/`-f <FORMAT>` - set custom format.
+
+By default, it uses format `{name} - "{value}"`. In format string `{name}` is replaced with name of the variable and `{value}` is replaced with it's value.
 
 For example:
 ```shell
@@ -150,7 +153,15 @@ SystemDrive = "C:"
 SystemRoot = "C:\\Windows"
 ...
 ```
-It will print all environment variables in format `VAR = "VALUE"`.
+```shell
+$ envfetch print --format "{name}: {value};"
+SHELL: powershell;
+windir: C:\\Windows;
+SystemDrive: C:;
+SystemRoot: C:\\Windows;
+...
+```
+It will print all environment variables in given format.
 #### Get
 Get value of environment variable
 

src/commands.rs

@@ -33,7 +33,7 @@ pub fn run_command(command: &Commands) -> ExitCode {
                 return ExitCode::FAILURE;
             }
         }
-        Commands::Print => print_env(),
+        Commands::Print(opt) => print_env(opt),
         Commands::Load(opt) => {
             if let Err(error) = load(opt) {
                 error!("{}", error);
@@ -72,9 +72,10 @@ pub fn run_command(command: &Commands) -> ExitCode {
 }
 
 /// Print all environment variables
-pub fn print_env() {
+pub fn print_env(opt: &PrintArgs) {
+    let format = &opt.format.clone().unwrap_or("{name} = \"{value}\"".to_owned());
     // Print all environment variables
-    variables::print_env();
+    variables::print_env(format);
 }
 
 /// Load variables from dotenv-style file
@@ -237,7 +238,7 @@ mod tests {
     fn test_run_command_print() {
         unsafe { env::set_var("TEST_PRINT_RUN", "test_value") };
         with_captured_output(|| {
-            run_command(&Commands::Print);
+            run_command(&Commands::Print(PrintArgs { format: None }));
         });
         unsafe { env::remove_var("TEST_PRINT_RUN") };
     }
@@ -280,7 +281,7 @@ mod tests {
         unsafe { env::set_var("TEST_PRINT_VAR", "test_value") };
 
         // Call function - just verify it executes without panicking
-        print_env();
+        print_env(&PrintArgs { format: None });
 
         // Clean up
         unsafe { env::remove_var("TEST_PRINT_VAR") };
@@ -293,7 +294,7 @@ mod tests {
         unsafe { env::set_var("TEST_VAR_2", "value2") };
 
         // Call function - just verify it executes without panicking
-        print_env();
+        print_env(&PrintArgs { format: None });
 
         // Clean up
         unsafe { env::remove_var("TEST_VAR_1") };
@@ -728,7 +729,7 @@ mod tests {
     fn test_run_command_print_env() {
         unsafe { env::set_var("TEST_PRINT_ENV", "test_value") };
         with_captured_output(|| {
-            assert_eq!(run_command(&Commands::Print), ExitCode::SUCCESS);
+            assert_eq!(run_command(&Commands::Print(PrintArgs { format: None })), ExitCode::SUCCESS);
         });
         unsafe { env::remove_var("TEST_PRINT_ENV") };
     }

src/main.rs

@@ -63,7 +63,7 @@ mod tests {
     #[test]
     fn test_print_command() {
         let args = Cli::parse_from(["envfetch", "print"]);
-        assert_eq!(args.command, Commands::Print);
+        assert_eq!(args.command, Commands::Print(PrintArgs { format: None }));
     }
 
     #[test]

src/models.rs

@@ -37,7 +37,15 @@ pub enum Commands {
     /// Load environment variables from dotenv file and optionally run given process.
     Load(LoadArgs),
     /// Print all environment variables.
-    Print,
+    Print(PrintArgs),
+}
+
+/// Args for print command
+#[derive(Args, Debug, PartialEq, Eq)]
+pub struct PrintArgs {
+    /// Set custom format, by default {name} = "{value}" is used.
+    #[arg(long, short)]
+    pub format: Option<String>,
 }
 
 /// Args for get command

src/variables.rs

@@ -1,4 +1,3 @@
-use colored::Colorize;
 use std::env;
 
 use crate::{models::ErrorKind, utils::*};
@@ -7,9 +6,10 @@ use crate::{models::ErrorKind, utils::*};
 type VariablesList = Vec<(String, String)>;
 
 /// Print all environment variables
-pub fn print_env() {
+pub fn print_env(format: &str) {
     for (key, value) in get_variables() {
-        println!("{} = \"{}\"", key.blue(), value);
+        let entry = format.replace("{name}", &key).replace("{value}", &value);
+        println!("{}", entry);
     }
 }
 
@@ -88,7 +88,7 @@ mod tests {
     #[test]
     fn test_print_env() {
         unsafe { env::set_var("TEST_PRINT_VAR", "test_value") };
-        print_env();
+        print_env("{name} = \"{value}\"");
         unsafe { env::remove_var("TEST_PRINT_VAR") };
     }
 
@@ -134,7 +134,7 @@ mod tests {
         unsafe { env::set_var("TEST_VAR_1", "value1") };
         unsafe { env::set_var("TEST_VAR_2", "value2") };
 
-        print_env();
+        print_env("{name} = \"{value}\"");
 
         // Clean up
         unsafe { env::remove_var("TEST_VAR_1") };
@@ -145,7 +145,7 @@ mod tests {
     fn test_print_env_empty_value() {
         unsafe { env::set_var("TEST_EMPTY", "") };
 
-        print_env();
+        print_env("{name} = \"{value}\"");
 
         unsafe { env::remove_var("TEST_EMPTY") };
     }
@@ -154,7 +154,7 @@ mod tests {
     fn test_print_env_special_characters() {
         unsafe { env::set_var("TEST_SPECIAL", "value with spaces and $#@!") };
 
-        print_env();
+        print_env("{name} = \"{value}\"");
 
         unsafe { env::remove_var("TEST_SPECIAL") };
     }