fix: init script

1 files changed, 6 insertions, 6 deletions

diff --git a/init.nu b/init.nu
index def503f..7039947 100755
--- a/init.nu
+++ b/init.nu
@@ -5,11 +5,9 @@ def main [ name: string idx?: int  ] {
   let idx = if ($idx | is-empty) {
     (
     glob "*_*"
-    | split row '/'
-    | get (($in | length) - 1)
-    | split row '_'
-    | get 0
-    | into int
+    | each {|l| $l | split row '/' | last | split row '_' | first | into int }
+    | sort
+    | last
     | $in + 1
     | (if $in < 10 { $"0($in)" } else { $in } )
     )
@@ -31,7 +29,9 @@ def main [ name: string idx?: int  ] {
   cargo new $problem_name --name $name --vcs none
 
   if $env.LAST_EXIT_CODE == 0  {
-    rm -r $tmp_dir
+    if ($existing | is-not-empty) {
+      rm -r $tmp_dir
+    }
   } else {
     mv $tmp_dir $problem_name
   }