feat: Add ps1 overwrite to the completion script.

src/cli/mod.rs

@@ -1,10 +1,12 @@
 use super::util::IndicatifWriter;
+use crate::progress;
 use clap::{CommandFactory, Parser};
-use clap_complete::Shell;
+use clap_complete;
 use clap_verbosity_flag::Verbosity;
 use miette::IntoDiagnostic;
-
-use crate::progress;
+use rattler_shell::shell::{Shell, ShellEnum};
+use std::io::Write;
+use std::str::FromStr;
 use tracing_subscriber::{filter::LevelFilter, util::SubscriberInitExt, EnvFilter};
 
 pub mod add;
@@ -35,7 +37,7 @@ struct Args {
 pub struct CompletionCommand {
     /// The shell to generate a completion script for (defaults to 'bash').
     #[arg(short, long)]
-    shell: Option<Shell>,
+    shell: Option<clap_complete::Shell>,
 }
 
 #[derive(Parser, Debug)]
@@ -58,13 +60,39 @@ pub enum Command {
 }
 
 fn completion(args: CompletionCommand) -> miette::Result<()> {
+    let clap_shell = args
+        .shell
+        .or(clap_complete::Shell::from_env())
+        .unwrap_or(clap_complete::Shell::Bash);
     clap_complete::generate(
-        args.shell.or(Shell::from_env()).unwrap_or(Shell::Bash),
+        clap_shell,
         &mut Args::command(),
         "pixi",
         &mut std::io::stdout(),
     );
 
+    // Create PS1 overwrite command
+    // TODO: Also make this work for non bourne shells.
+    let mut script = String::new();
+    let shell = ShellEnum::from_str(clap_shell.to_string().as_str()).into_diagnostic()?;
+    // Generate a shell agnostic command to add the PIXI_PROMPT to the PS1 variable.
+    shell
+        .set_env_var(
+            &mut script,
+            "PS1",
+            format!(
+                "{}{}",
+                shell.format_env_var("PIXI_PROMPT"),
+                shell.format_env_var("PS1")
+            )
+            .as_str(),
+        )
+        .unwrap();
+    // Just like the clap autocompletion code write directly to the stdout
+    std::io::stdout()
+        .write_all(script.as_bytes())
+        .into_diagnostic()?;
+
     Ok(())
 }
 

src/project/environment.rs

@@ -28,6 +28,7 @@ pub fn get_metadata_env(project: &Project) -> HashMap<String, String> {
             format!("{ENV_PREFIX}ROOT"),
             project.root().to_string_lossy().into_owned(),
         ),
+        (format!("{ENV_PREFIX}NAME"), project.name().to_string()),
         (
             format!("{ENV_PREFIX}MANIFEST"),
             project.manifest_path().to_string_lossy().into_owned(),
@@ -44,5 +45,6 @@ pub fn get_metadata_env(project: &Project) -> HashMap<String, String> {
             format!("{ENV_PREFIX}VERSION"),
             project.version().to_string(),
         ),
+        ("PIXI_PROMPT".to_string(), format!("({}) ", project.name())),
     ])
 }