optimize wasi-sdk detect

3 files changed, 107 insertions, 0 deletions

diff --git a/xmake/modules/detect/sdks/find_wasisdk.lua b/xmake/modules/detect/sdks/find_wasisdk.lua
new file mode 100644
index 000000000..fee876716
--- /dev/null
+++ b/xmake/modules/detect/sdks/find_wasisdk.lua
@@ -0,0 +1,94 @@
+--
+-- @author      vkensou
+-- @file        find_wasisdk.lua
+--
+
+-- imports
+import("core.base.semver")
+import("core.base.option")
+import("core.base.global")
+import("core.project.config")
+import("core.cache.detectcache")
+import("lib.detect.find_file")
+
+-- find wasm-ld directory
+function _find_wasm_ld(sdkdir)
+    local subdirs = {}
+    table.insert(subdirs, "bin")
+
+    local paths = {}
+    if sdkdir then
+        table.insert(paths, sdkdir)
+    end
+    table.insert(paths, "$(env WASI_SDK_PATH)")
+
+    local wasm_ldname = (is_host("windows") and "wasm-ld.exe" or "wasm-ld")
+    local wasm_ld = find_file(wasm_ldname, paths, {suffixes = subdirs})
+    if wasm_ld then
+        return path.directory(wasm_ld)
+    end
+end
+
+-- find wasi-sdk
+function _find_wasisdk(sdkdir)
+
+    -- find bin directory
+    local bindir = _find_wasm_ld(sdkdir)
+    if not bindir or path.filename(bindir) ~= "bin" then
+        return {}
+    end
+
+    -- find sdk root directory
+    sdkdir = path.directory(bindir)
+    if not sdkdir then
+        return {}
+    end
+
+    return {sdkdir = sdkdir, bindir = bindir}
+end
+
+-- find wasi-sdk directory
+--
+-- @param sdkdir    the wasi-sdk directory
+-- @param opt       the argument options, e.g. {force = true}
+--
+-- @return          the sdk toolchains. e.g. {sdkdir = ..}
+--
+-- @code
+--
+-- local sdk = find_wasisdk("~/wasi-sdk")
+--
+-- @endcode
+--
+function main(sdkdir, opt)
+
+    -- init arguments
+    opt = opt or {}
+
+    -- attempt to load cache first
+    local key = "detect.sdks.find_wasisdk"
+    local cacheinfo = detectcache:get(key) or {}
+    if not opt.force and cacheinfo.sdk and cacheinfo.sdk.sdkdir and os.isdir(cacheinfo.sdk.sdkdir) then
+        return cacheinfo.sdk
+    end
+
+    -- find sdk
+    local sdk = _find_wasisdk(sdkdir or config.get("wasi_sdk") or global.get("wasi_sdk"))
+    if sdk and sdk.sdkdir and sdk.bindir then
+        config.set("wasi_sdk", sdk.sdkdir, {force = true, readonly = true})
+        if opt.verbose or option.get("verbose") then
+            cprint("checking for wasi-sdk directory ... ${color.success}%s", sdk.sdkdir)
+        end
+
+    else
+        if opt.verbose or option.get("verbose") then
+            cprint("checking for wasi-sdk directory ... ${color.nothing}${text.nothing}")
+        end
+    end
+
+    -- save to cache
+    cacheinfo.sdk = sdk or false
+    detectcache:set(key, cacheinfo)
+    detectcache:save()
+    return sdk
+end
diff --git a/xmake/platforms/wasm/xmake.lua b/xmake/platforms/wasm/xmake.lua
index 885233cad..840d6f5d4 100644
--- a/xmake/platforms/wasm/xmake.lua
+++ b/xmake/platforms/wasm/xmake.lua
@@ -36,11 +36,15 @@ platform("wasm")
         {
             {category = "Emscripten Configuration"          }
         ,   {nil, "emsdk", "kv", nil, "The emsdk directory" }
+        ,   {category = "WASI-SDK Configuration"            }
+        ,   {nil, "wasi_sdk", "kv", nil, "The wasi-sdk directory" }
         }
     ,   global =
         {
             {category = "Emscripten Configuration"          }
         ,   {nil, "emsdk", "kv", nil, "The emsdk directory" }
+        ,   {category = "WASI-SDK Configuration"            }
+        ,   {nil, "wasi_sdk", "kv", nil, "The wasi-sdk directory" }
         }
     }
 
diff --git a/xmake/toolchains/wasi/xmake.lua b/xmake/toolchains/wasi/xmake.lua
index 00ad3465a..758579b6e 100644
--- a/xmake/toolchains/wasi/xmake.lua
+++ b/xmake/toolchains/wasi/xmake.lua
@@ -41,6 +41,15 @@ toolchain("wasi")
 
     -- check toolchain
     on_check(function (toolchain)
+        import("lib.detect.find_tool")
+        import("detect.sdks.find_wasisdk")
+        local wasisdk = find_wasisdk(toolchain:sdkdir())
+        if wasisdk then
+            toolchain:config_set("bindir", wasisdk.bindir)
+            toolchain:config_set("sdkdir", wasisdk.sdkdir)
+            toolchain:configs_save()
+            return wasisdk
+        end
         return import("lib.detect.find_tool")("clang", {paths = toolchain:bindir()})
     end)