improve to check msvc

1 files changed, 9 insertions, 29 deletions

diff --git a/xmake/toolchains/msvc/check.lua b/xmake/toolchains/msvc/check.lua
index 37f38b53c..c6b6dabd4 100644
--- a/xmake/toolchains/msvc/check.lua
+++ b/xmake/toolchains/msvc/check.lua
@@ -23,6 +23,7 @@ import("core.base.option")
 import("core.project.config")
 import("detect.sdks.find_vstudio")
 import("lib.detect.find_tool")
+imports("private.utils.toolchain", {alias = "toolchain_utils"})
 
 -- attempt to check vs environment
 function _check_vsenv(toolchain)
@@ -96,33 +97,6 @@ function _check_vstudio(toolchain)
     return vs
 end
 
-function _check_vc_build_tools(toolchain, sdkdir)
-    local opt = {}
-    opt.sdkdir = sdkdir
-    opt.vs_toolset = toolchain:config("vs_toolset") or config.get("vs_toolset")
-    opt.vs_sdkver = toolchain:config("vs_sdkver") or config.get("vs_sdkver")
-
-    local vcvarsall = find_vstudio.find_build_tools(opt)
-    if not vcvarsall then
-        return
-    end
-
-    local vcvars = vcvarsall[toolchain:arch()]
-    if vcvars and vcvars.PATH and vcvars.INCLUDE and vcvars.LIB then
-        toolchain:config_set("vcvars", vcvars)
-        toolchain:config_set("vcarchs", table.orderkeys(vcvarsall))
-        toolchain:config_set("vs_toolset", vcvars.VCToolsVersion)
-        toolchain:config_set("vs_sdkver", vcvars.WindowsSDKVersion)
-
-        -- check compiler
-        local cl = find_tool("cl.exe", {force = true, envs = vcvars})
-        if cl then
-            cprint("checking for Microsoft C/C++ Compiler (%s) ... ${color.success}", toolchain:arch())
-        end
-        return vcvars
-    end
-end
-
 function main(toolchain)
 
     -- only for windows or linux (msvc-wine)
@@ -135,16 +109,22 @@ function main(toolchain)
     local cxx = path.basename(config.get("cxx") or "cl"):lower()
     local mrc = path.basename(config.get("mrc") or "rc"):lower()
     if cc == "cl" or cxx == "cl" or mrc == "rc" then
+        local check = function(vcvars)
+            local cl = find_tool("cl.exe", {force = true, envs = vcvars})
+            if cl then
+                cprint("checking for Microsoft C/C++ Compiler (%s) ... ${color.success}", toolchain:arch())
+            end
+        end
         local sdkdir = toolchain:sdkdir()
         if sdkdir then
-            sdkdir = _check_vc_build_tools(toolchain, sdkdir)
+            sdkdir = toolchain_utils.check_vc_build_tools(toolchain, sdkdir, check)
         end
         if not sdkdir then
             -- find it from packages
             for _, package in ipairs(toolchain:packages()) do
                 local installdir = package:installdir()
                 if installdir and os.isdir(installdir) then
-                    local result = _check_vc_build_tools(toolchain, installdir)
+                    local result = toolchain_utils.check_vc_build_tools(toolchain, installdir, check)
                     if result then
                         return result
                     end