From e95f60971b81920b52b24f5efd29c1b6c6aa3c43 Mon Sep 17 00:00:00 2001
From: Tilen Gimpelj <66419530+Tiggax@users.noreply.github.com>
Date: Mon, 13 Mar 2023 15:21:21 +0100
Subject: [PATCH] Updated conda script to default to base if no arg given
 (#409)

---
 virtual_environments/conda.nu | 15 ++++++++++-----
 1 file changed, 10 insertions(+), 5 deletions(-)

diff --git a/virtual_environments/conda.nu b/virtual_environments/conda.nu
index 21f99f6..61dd730 100644
--- a/virtual_environments/conda.nu
+++ b/virtual_environments/conda.nu
@@ -1,14 +1,19 @@
 # Activate conda environment
 export def-env activate [
-    env_name: string@'nu-complete conda envs' # name of the environment
-    --no-prompt                               # do not update the prompt
+    env_name?: string@'nu-complete conda envs' # name of the environment
+    --no-prompt                                # do not update the prompt
 ] {
     let conda_info = (conda info --envs --json | from json)
 
-    mut env_dir = ($conda_info.envs_dirs | each {|it| $it | path join $env_name })
-
+    mut $env_name = $env_name
+    if $env_name == null {$env_name = "base"}
+    mut env_dir = ""
+    mut env_dirs = []
+    for i in $conda_info.envs_dirs {
+        $env_dirs = ($env_dirs | append ($i | path join $env_name))
+    }
     if $env_name != "base" {
-        $env_dir = (check-if-env-exists $env_name $env_dir)
+        $env_dir = ((check-if-env-exists $env_name $env_dirs) | into string)
     } else {
         $env_dir = $conda_info.root_prefix
         }