--!A cross-platform build utility based on Lua
--
-- Licensed under the Apache License, Version 2.0 (the "License");
-- you may not use this file except in compliance with the License.
-- You may obtain a copy of the License at
--
--     http://www.apache.org/licenses/LICENSE-2.0
--
-- Unless required by applicable law or agreed to in writing, software
-- distributed under the License is distributed on an "AS IS" BASIS,
-- WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-- See the License for the specific language governing permissions and
-- limitations under the License.
--
-- Copyright (C) 2015-present, Xmake Open Source Community.
--
-- @author      jacklin
-- @file        find_matlab.lua
--

-- imports
import("detect.sdks.matlab")

-- find matlab sdk toolchains
--
-- @return          the matlab sdk toolchains. e.g. {sdkdir = ..., includedirs = ..., linkdirs = ..., .. }
--
-- @code
--
-- local toolchains = find_matlab(opt)
--
-- @endcode
--
function main(opt)
    opt = opt or {}
    local version = opt.require_version and tostring(opt.require_version) or nil
    local result = {sdkdir = "", includedirs = {}, linkdirs = {}, links = {}}
    if is_host("windows") then
        local matlabkey = "HKEY_LOCAL_MACHINE\\SOFTWARE\\MathWorks\\MATLAB"
        local valuekeys = winos.registry_keys(matlabkey)
        if #valuekeys == 0 then
            return
        end

        local itemkey
        if version == nil then
            itemkey = valuekeys[1] .. ";MATLABROOT"
        else
            local versionname = matlab.versions()[version]
            if versionname ~= nil then
                itemkey = matlabkey .. "\\" .. version .. ";MATLABROOT"
            else
                local versionvalue = matlab.versions_names()[version:lower()]
                if versionvalue ~= nil then
                    itemkey = matlabkey .. "\\" .. versionvalue .. ";MATLABROOT"
                else
                    print("allowed values are:")
                    for k, v in pairs(matlab.versions()) do
                        print("    ", k, v)
                    end
                    raise("MATLAB Runtime version does not exist: " .. version)
                end
            end
        end

        local sdkdir = try {function () return winos.registry_query(itemkey) end}
        if not sdkdir then
            return
        end
        result.sdkdir = sdkdir
        result.includedirs = path.join(sdkdir, "extern", "include")
        for _, value in ipairs(os.dirs(path.join(sdkdir, "extern", "lib", "**"))) do
            local dirbasename = path.basename(value)
            if not dirbasename:startswith("win") then
                result.linkdirs[dirbasename] = value
            end
        end
        for _, value in pairs(result.linkdirs) do
            local dirbasename = path.basename(value)
            result.links[dirbasename] = {}
            for _, filepath in ipairs(os.files(value.."/*.lib")) do
                table.insert(result.links[dirbasename], path.basename(filepath))
            end
            result.links[dirbasename] = table.unique(result.links[dirbasename])
        end
    end
    return result
end