refactor: use for loop to generate completions file (#1869)

sxyazi · Oct 31, 2024 · a510673 · a510673
1 parent e3c9111
commit a510673
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 10 deletions.
diff --git a/yazi-boot/build.rs b/yazi-boot/build.rs
@@ -22,11 +22,10 @@ fn main() -> Result<(), Box<dyn Error>> {
 	let out = "completions";
 
 	std::fs::create_dir_all(out)?;
-	generate_to(Shell::Bash, cmd, bin, out)?;
-	generate_to(Shell::Fish, cmd, bin, out)?;
-	generate_to(Shell::Zsh, cmd, bin, out)?;
-	generate_to(Shell::Elvish, cmd, bin, out)?;
-	generate_to(Shell::PowerShell, cmd, bin, out)?;
+	for sh in [Shell::Bash, Shell::Fish, Shell::Zsh, Shell::Elvish, Shell::PowerShell] {
+		generate_to(sh, cmd, bin, out)?;
+	}
+
 	generate_to(clap_complete_nushell::Nushell, cmd, bin, out)?;
 	generate_to(clap_complete_fig::Fig, cmd, bin, out)?;
 

diff --git a/yazi-cli/build.rs b/yazi-cli/build.rs
@@ -22,11 +22,10 @@ fn main() -> Result<(), Box<dyn Error>> {
 	let out = "completions";
 
 	std::fs::create_dir_all(out)?;
-	generate_to(Shell::Bash, cmd, bin, out)?;
-	generate_to(Shell::Fish, cmd, bin, out)?;
-	generate_to(Shell::Zsh, cmd, bin, out)?;
-	generate_to(Shell::Elvish, cmd, bin, out)?;
-	generate_to(Shell::PowerShell, cmd, bin, out)?;
+	for sh in [Shell::Bash, Shell::Fish, Shell::Zsh, Shell::Elvish, Shell::PowerShell] {
+		generate_to(sh, cmd, bin, out)?;
+	}
+
 	generate_to(clap_complete_nushell::Nushell, cmd, bin, out)?;
 	generate_to(clap_complete_fig::Fig, cmd, bin, out)?;