improve envs

1 files changed, 17 insertions, 0 deletions

diff --git a/xmake/core/base/os.lua b/xmake/core/base/os.lua
index f9444d4b3..44b373a66 100644
--- a/xmake/core/base/os.lua
+++ b/xmake/core/base/os.lua
@@ -998,6 +998,23 @@ function os.getenvs()
     return envs
 end
 
+-- set all current environment variables
+function os.setenvs(envs)
+    if envs then
+        -- remove new added values
+        local curenvs = os.getenvs()
+        for name, _ in pairs(curenvs) do
+            if not envs[name] then
+                os.setenv(name, nil)
+            end
+        end
+        -- change values
+        for name, values in pairs(envs) do
+            os.setenv(name, values)
+        end
+    end
+end
+
 -- set values to environment variable
 function os.setenv(name, ...)
     local values = {...}