feat: completions

feat: completions

Author: load1n9

Cargo.lock

@@ -15,7 +15,8 @@ andromeda-runtime = { path = "runtime", features = ["canvas", "crypto"] }
 anyhow = "1.0.98"
 anymap = "0.12.1"
 base64-simd = "0.8.0"
-clap = { version = "4.5.39", features = ["derive"] }
+clap = { version = "4.5.40", features = ["derive"] }
+clap_complete = "4.5.54"
 console = "0.15.8"
 dprint-core = "0.67.4"
 dprint-plugin-typescript = "0.95.5"

Cargo.toml

@@ -10,6 +10,7 @@ readme = "../README.md"
 
 [dependencies]
 clap.workspace = true
+clap_complete.workspace = true
 console.workspace = true
 dprint-core.workspace = true
 dprint-plugin-typescript.workspace = true

cli/Cargo.toml

@@ -3,8 +3,10 @@
 // file, You can obtain one at https://mozilla.org/MPL/2.0/.
 
 use andromeda_core::RuntimeFile;
-use clap::{Parser as ClapParser, Subcommand};
+use clap::{CommandFactory, Parser as ClapParser, Subcommand};
+use clap_complete::{Shell, generate};
 use libsui::find_section;
+use std::io;
 use std::path::PathBuf;
 
 mod compile;
@@ -82,6 +84,12 @@ enum Command {
         #[arg(required = false)]
         paths: Vec<PathBuf>,
     },
+    /// Generate shell completion scripts
+    Completions {
+        /// The shell to generate completions for
+        #[arg(value_enum)]
+        shell: Option<Shell>,
+    },
 }
 
 fn main() {
@@ -169,9 +177,12 @@ fn run_main() -> Result<()> {
                 }
                 Ok(())
             }
+            Command::Completions { shell } => {
+                generate_completions(shell);
+                Ok(())
+            }
         }
     });
-
     match rt.block_on(nova_thread) {
         Ok(result) => result,
         Err(e) => Err(AndromedaError::config_error(
@@ -181,3 +192,46 @@ fn run_main() -> Result<()> {
         )),
     }
 }
+
+fn generate_completions(shell: Option<Shell>) {
+    let mut cmd = Cli::command();
+    let bin_name = "andromeda";
+
+    match shell {
+        Some(shell) => {
+            generate(shell, &mut cmd, bin_name, &mut io::stdout());
+        }
+        None => {
+            // If no shell is specified, try to detect it from environment
+            // This mimics Deno's behavior
+            if let Some(detected_shell) = detect_shell() {
+                generate(detected_shell, &mut cmd, bin_name, &mut io::stdout());
+            } else {
+                eprintln!(
+                    "Error: Could not detect shell. Please specify one of: bash, zsh, fish, powershell"
+                );
+                std::process::exit(1);
+            }
+        }
+    }
+}
+
+fn detect_shell() -> Option<Shell> {
+    // Try to detect shell from environment variables
+    if let Ok(shell) = std::env::var("SHELL") {
+        if shell.contains("bash") {
+            return Some(Shell::Bash);
+        } else if shell.contains("zsh") {
+            return Some(Shell::Zsh);
+        } else if shell.contains("fish") {
+            return Some(Shell::Fish);
+        }
+    }
+
+    // On Windows, check for PowerShell
+    if cfg!(windows) && std::env::var("PSModulePath").is_ok() {
+        return Some(Shell::PowerShell);
+    }
+
+    None
+}