fix: error when no envVars

Chen1Plus · Sep 12, 2024 · 6bb6f25 · 6bb6f25
1 parent ac7cc61
commit 6bb6f25
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/wmod.psm1 b/wmod.psm1
@@ -33,8 +33,10 @@ function Wmod-Load {
         $env:PATH = $Path + $env:PATH
 
         # add environment variables
-        $EnvVars.Keys | ForEach-Object {
-            New-Item -Path "Env:\$_" -Value $EnvVars[$_]
+        if ($EnvVars) {
+            $EnvVars.Keys | ForEach-Object {
+                New-Item -Path "Env:\$_" -Value $EnvVars[$_]
+            }
         }
 
         $script:Wmod_Loaded += $modulePath
@@ -82,8 +84,10 @@ function Unload-ByPath {
         $env:PATH = $env:PATH.Replace($Path, '')
 
         # remove environment variables
-        $EnvVars.Keys | ForEach-Object {
-            Remove-Item -Path "Env:\$_"
+        if ($EnvVars) {
+            $EnvVars.Keys | ForEach-Object {
+                Remove-Item -Path "Env:\$_"
+            }
         }
 
         $script:Wmod_Loaded = $script:Wmod_Loaded | Where-Object { $_ -ne $modulePath }