Working nushell implementation

libmamba/CMakeLists.txt

@@ -100,7 +100,7 @@ set(SHELL_SCRIPTS
     Mamba.psm1
     mamba.xsh
     mamba.fish
-    mamba.nu
+    micromamba.nu
     compile_pyc.py
     mamba_completion.posix
     )

libmamba/data/micromamba.nu

@@ -0,0 +1,31 @@
+#!/usr/bin/env nu
+def-env "micromamba activate"  [name: string] {
+  # get envs and update current env to path
+  let envs = (
+     (micromamba env list --json | from json).envs | reduce -f {} {|it, acc|
+     $acc | upsert ($it | path basename) $it
+    }
+  )
+
+  # fetch path of current environment and update if present
+  let env_path = ($envs | get $name)
+  let p = ([$env_path "bin"] | path join)
+  if not $p in $env.PATH {
+    $env.PATH = ($env.PATH | prepend $p)
+    $env.MICROMAMBA_CURRENT_ENV = $name
+  }
+}
+
+def-env "micromamba deactivate" [] {
+  # get envs and unset current env
+  let envs = (
+     (micromamba env list --json | from json).envs | reduce -f {} {|it, acc|
+     $acc | upsert ($it | path basename) $it
+    }
+  )
+  if $env.MICROMAMBA_CURRENT_ENV in $envs {
+    let path = ([($envs | get $env.MICROMAMBA_CURRENT_ENV) "bin"] | path join)
+    $env.PATH = ($env.PATH | filter {|it| $it != $path})
+    $env.MICROMAMBA_CURRENT_ENV = ""
+  }
+}

libmamba/include/mamba/core/shell_init.hpp

@@ -24,7 +24,6 @@ extern const char data_Mamba_psm1[];
 extern const char data_mamba_xsh[];
 extern const char data_mamba_fish[];
 extern const char data_mamba_completion_posix[];
-extern const char data_mamba_nu[];
 
 namespace mamba
 {

libmamba/src/core/activation.cpp

@@ -1270,7 +1270,7 @@ namespace mamba
     // TODO: check this
     fs::u8path NuActivator::hook_source_path()
     {
-        return Context::instance().prefix_params.root_prefix / "etc" / "profile.d" / "mamba.nu";
+        return Context::instance().prefix_params.root_prefix / "etc" / "profile.d" / "micromamba.nu";
     }
 
     std::pair<std::string, std::string>

libmamba/src/core/shell_init.cpp

@@ -407,10 +407,11 @@ namespace mamba
 
         content << "\n# >>> mamba initialize >>>\n";
         content << "# !! Contents within this block are managed by 'mamba init' !!\n";
-        content << "$env.MAMBA_EXE = " << mamba_exe << "\n";
-        content << "$env.MAMBA_ROOT_PREFIX = " << env_prefix << "\n";
-        content << "run-external $env.MAMBA_EXE 'shell' 'hook' '--shell' 'nu' '--root-prefix' $env.MAMBA_ROOT_PREFIX --redirect-stdout \n";
-        content << "$env.PATH = ($env.PATH | append ( [$env.MAMBA_ROOT_PREFIX bin] | path join)) \n";
+        content << "$env.MICROMAMBA_MAMBA_EXE = " << mamba_exe << "\n";
+        content << "$env.MICROMAMBA_ROOT_PREFIX = " << env_prefix << "\n";
+        content << "$env.MICROMAMBA_CURRENT_ENV = \"\"\n";
+        content << "$env.PATH = ($env.PATH | append ([$env.MICROMAMBA_ROOT_PREFIX bin] | path join) | uniq)\n";
+        content << "source " << env_prefix.string() << "/etc/profile.d/micromamba.nu\n";
         content << "# <<< mamba initialize <<<\n";
         return content.str();
     }
@@ -612,7 +613,7 @@ namespace mamba
         // deprecated according to https://github.com/mamba-org/mamba/pull/2693
         else if (shell == "nu")
         {
-            std::string contents = data_mamba_nu;
+            std::string contents = data_micromamba_nu;
             replace_all(contents, "$MAMBA_EXE", exe.string());
             return contents;
         }
@@ -809,7 +810,7 @@ namespace mamba
             {
             }
             std::ofstream sh_file = open_ofstream(sh_source_path);
-            sh_file << data_mamba_nu;
+            sh_file << data_micromamba_nu;
         }
         else if (shell == "cmd.exe")
         {

mamba/mamba/mamba_shell_init.py

@@ -49,7 +49,7 @@ def check_init_block(lines, start_i, prefix):
     elif file.endswith(".nu"):
         snippet = MAMBA_SNIPPET_NU
         mamba_source_path = native_path_to_unix(
-            join(conda_prefix, "etc", "profile.d", "mamba.nu")
+            join(conda_prefix, "etc", "profile.d", "micromamba.nu")
         )
     else:
         snippet = MAMBA_SNIPPET_SH

mamba/setup.py

@@ -22,7 +22,7 @@
 data_files = [
     (
         "etc/profile.d",
-        ["mamba/shell_templates/mamba.sh", "mamba/shell_templates/mamba.nu"],
+        ["mamba/shell_templates/mamba.sh", "mamba/shell_templates/micromamba.nu"],
     ),
     ("etc/fish/conf.d", ["mamba/shell_templates/mamba.fish"]),
 ]

micromamba/tests/test_activation.py

@@ -450,7 +450,7 @@ def test_shell_init_deinit_root_prefix_files(
         elif interpreter in ["csh", "tcsh"]:
             files = [tmp_root_prefix / "etc" / "profile.d" / "micromamba.csh"]
         elif interpreter == "nu":
-            files = [tmp_root_prefix / "etc" / "profile.d" / "mamba.nu"]
+            files = [tmp_root_prefix / "etc" / "profile.d" / "micromamba.nu"]
         else:
             raise ValueError(f"Unknown shell {interpreter}")