Merge pull request #42 from afh/export-envs

Add --envs option to export
envio-cli · Feb 26, 2024 · 95878ee · 95878ee
2 parents 9126e60 + fcba8b0
commit 95878ee
Show file tree

Hide file tree

Showing 8 changed files with 77 additions and 8 deletions.
diff --git a/Cargo.toml b/Cargo.toml
@@ -1,7 +1,7 @@
 [package]
 name = "envio"
 version = "0.5.1"
-rust-version = "1.76.0"
+rust-version = "1.75.0"
 description = "A Modern And Secure CLI Tool For Managing Environment Variables"
 edition = "2021"
 authors = ["Humble Penguin <humblepenguinn@gmail.com>"]

diff --git a/completions/_envio b/completions/_envio
@@ -105,6 +105,8 @@ _arguments "${_arguments_options[@]}" \
 _arguments "${_arguments_options[@]}" \
 '-f+[]:FILE: ' \
 '--file-to-export-to=[]:FILE: ' \
+'*-e+[]:ENVS: ' \
+'*--envs=[]:ENVS: ' \
 '-h[Print help]' \
 '--help[Print help]' \
 ':profile_name:' \

diff --git a/completions/_envio.ps1 b/completions/_envio.ps1
@@ -100,6 +100,8 @@ Register-ArgumentCompleter -Native -CommandName 'envio' -ScriptBlock {
         'envio;export' {
             [CompletionResult]::new('-f', 'f', [CompletionResultType]::ParameterName, 'f')
             [CompletionResult]::new('--file-to-export-to', 'file-to-export-to', [CompletionResultType]::ParameterName, 'file-to-export-to')
+            [CompletionResult]::new('-e', 'e', [CompletionResultType]::ParameterName, 'e')
+            [CompletionResult]::new('--envs', 'envs', [CompletionResultType]::ParameterName, 'envs')
             [CompletionResult]::new('-h', 'h', [CompletionResultType]::ParameterName, 'Print help')
             [CompletionResult]::new('--help', 'help', [CompletionResultType]::ParameterName, 'Print help')
             break

diff --git a/completions/envio.bash b/completions/envio.bash
@@ -165,7 +165,7 @@ _envio() {
             return 0
             ;;
         envio__export)
-            opts="-f -h --file-to-export-to --help <PROFILE_NAME>"
+            opts="-f -e -h --file-to-export-to --envs --help <PROFILE_NAME>"
             if [[ ${cur} == -* || ${COMP_CWORD} -eq 2 ]] ; then
                 COMPREPLY=( $(compgen -W "${opts}" -- "${cur}") )
                 return 0
@@ -179,6 +179,14 @@ _envio() {
                     COMPREPLY=($(compgen -f "${cur}"))
                     return 0
                     ;;
+                --envs)
+                    COMPREPLY=($(compgen -f "${cur}"))
+                    return 0
+                    ;;
+                -e)
+                    COMPREPLY=($(compgen -f "${cur}"))
+                    return 0
+                    ;;
                 *)
                     COMPREPLY=()
                     ;;

diff --git a/completions/envio.fish b/completions/envio.fish
@@ -30,6 +30,7 @@ complete -c envio -n "__fish_seen_subcommand_from list" -s h -l help -d 'Print h
 complete -c envio -n "__fish_seen_subcommand_from update" -s e -l envs -r
 complete -c envio -n "__fish_seen_subcommand_from update" -s h -l help -d 'Print help'
 complete -c envio -n "__fish_seen_subcommand_from export" -s f -l file-to-export-to -r
+complete -c envio -n "__fish_seen_subcommand_from export" -s e -l envs -r
 complete -c envio -n "__fish_seen_subcommand_from export" -s h -l help -d 'Print help'
 complete -c envio -n "__fish_seen_subcommand_from import" -s f -l file-to-import-from -r
 complete -c envio -n "__fish_seen_subcommand_from import" -s u -l url -r

diff --git a/src/bin/envio/cli.rs b/src/bin/envio/cli.rs
@@ -135,6 +135,14 @@ pub enum Command {
         profile_name: String,
         #[arg(required = false, long = "file-to-export-to", short = 'f')]
         file: Option<String>,
+        #[arg(
+            required = false,
+            long = "envs",
+            short = 'e',
+            value_delimiter = ' ',
+            num_args = 1..,
+        )]
+        envs: Option<Vec<String>>,
     },
     #[command(
         name = "import",

diff --git a/src/bin/envio/commands.rs b/src/bin/envio/commands.rs
@@ -192,7 +192,7 @@ impl Command {
 
                     let prompt = MultiSelect::new("Select the environment variables you want to keep in your new profile:", options.clone())
                         .with_default(&default_options)
-                        .with_help_message("↑↓ to move, space to select one, → to all, ← to none, type to filter, enter to confirm")
+                        .with_help_message("↑↓ to move, space to select/unselect one, → to all, ← to none, type to filter, enter to confirm")
                         .prompt();
 
                     if let Err(e) = prompt {
@@ -591,7 +591,11 @@ impl Command {
                 profile.push_changes();
             }
 
-            Command::Export { profile_name, file } => {
+            Command::Export {
+                profile_name,
+                file,
+                envs,
+            } => {
                 if !check_profile(profile_name) {
                     println!("{}: Profile does not exist", "Error".red());
                     return;
@@ -615,7 +619,33 @@ impl Command {
                         return;
                     };
 
-                profile.export_envs(file_name);
+                if envs.is_none() {
+                    let prompt = MultiSelect::new("Select the environment variables you want to export:", profile.envs.keys().collect())
+                        .with_default(&(0..profile.envs.len()).collect::<Vec<usize>>())
+                        .with_help_message("↑↓ to move, space to select/unselect one, → to all, ← to none, type to filter, enter to confirm")
+                        .prompt();
+
+                    if let Err(e) = prompt {
+                        println!("{}: {}", "Error".red(), e);
+                        std::process::exit(1);
+                    }
+
+                    profile.export_envs(
+                        file_name,
+                        &Some(
+                            prompt
+                                .unwrap()
+                                .iter()
+                                .cloned()
+                                .map(|s| s.to_owned())
+                                .collect(),
+                        ),
+                    );
+
+                    return;
+                }
+
+                profile.export_envs(file_name, envs);
             }
 
             Command::Import {

diff --git a/src/lib.rs b/src/lib.rs
@@ -155,7 +155,7 @@ impl Profile {
 
     @param file_name &str
     */
-    pub fn export_envs(&self, file_name: &str) {
+    pub fn export_envs(&self, file_name: &str, envs: &Option<Vec<String>>) {
         let mut file = std::fs::OpenOptions::new()
             .create(true)
             .write(true)
@@ -170,8 +170,26 @@ impl Profile {
             return;
         }
 
-        for key in self.envs.keys() {
-            buffer = buffer + key + "=" + self.envs.get(key).unwrap() + "\n";
+        let mut keys: Vec<_> = self.envs.keys().cloned().collect::<Vec<String>>();
+        if let Some(envs) = envs {
+            if !envs.is_empty() {
+                keys = self
+                    .envs
+                    .keys()
+                    .into_iter()
+                    .filter(|item| envs.contains(item))
+                    .cloned()
+                    .collect::<Vec<String>>();
+            }
+
+            if keys.is_empty() {
+                println!("{}: No envs to export", "Error".red());
+                return;
+            }
+        }
+
+        for key in keys {
+            buffer = buffer + key.as_str() + "=" + self.envs.get(key.as_str()).unwrap() + "\n";
         }
 
         if let Err(e) = writeln!(file, "{}", buffer) {