embed.lua

--------------------------------------------------------------------------------
local function exec(str)
    -- Premake5 bug (see docs/premake5.lua)
    local x = path.normalize
    path.normalize = function (y) return y end
    os.execute(str)
    path.normalize = x
end

--------------------------------------------------------------------------------
local function spairs(t, order)
    -- collect the keys
    local keys = {}
    for k in pairs(t) do keys[#keys+1] = k end

    -- if order function given, sort by it by passing the table and keys a, b,
    -- otherwise just sort the keys
    if order then
        table.sort(keys, function(a,b) return order(t, a, b) end)
    else
        table.sort(keys)
    end

    -- return the iterator function
    local i = 0
    return function()
        i = i + 1
        if keys[i] then
            return keys[i], t[keys[i]]
        end
    end
end

--------------------------------------------------------------------------------
local function compare_values(t, a, b)
    return t[a] < t[b]
end

--------------------------------------------------------------------------------
local function do_embed(debug_info)
    -- Find the Lua compilers.
    local archs = {
        ["64"] = { luac = os.matchfiles(".build/*/bin/final/luac_x64.exe")[1] },
        ["86"] = { luac = os.matchfiles(".build/*/bin/final/luac_x86.exe")[1] },
    }

    debug_info = debug_info and "" or " -s"

    for name, arch in spairs(archs) do
        if not arch.luac then
            error("Unable to find Lua compiler binary (x"..name.." final).")
        end

        arch.luac = arch.luac:gsub("/", "\\")
    end

    local manifests = os.matchfiles("clink/**/_manifest.lua")
    for _, manifest in spairs(manifests, compare_values) do
        local root = path.getdirectory(manifest)

        out = path.join(root, "lua_scripts.cpp")
        print("\n"..out)
        out = io.open(out, "w")
        out:write("#include \"pch.h\"\n")
        out:write("#include <core/embedded_scripts.h>\n")
        out:write("#if defined(CLINK_USE_EMBEDDED_SCRIPTS)\n")

        -- Write each sanitised script to 'out' as a global variable.
        local symbols = {}
        local manifest = dofile(manifest)
        for _, file in spairs(manifest.files, compare_values) do
            local name = path.getname(file)
            local symbol = manifest.name .. "_" .. name:gsub("%.", "_") .. "_script"
            table.insert(symbols, symbol)

            file = path.join(root, file)
            print("   "..file)

            for name, arch in spairs(archs) do
                out:write("#if ARCHITECTURE == "..name.."\n")

                -- Compile the input Lua script to binary.
                exec(arch.luac..debug_info.." -o .build/embed_temp "..file)
                local bin_in = io.open(".build/embed_temp", "rb")
                local bin_data = bin_in:read("*a")
                bin_in:close()
                os.remove(".build/embed_temp")

                local crlf_counter = 0
                out:write("const unsigned char " .. symbol .. "_[] = {\n")
                for byte in string.gmatch(bin_data, ".") do
                    out:write(string.format("0x%02x, ", byte:byte()))
                    crlf_counter = crlf_counter + 1
                    if crlf_counter > 16 then
                        out:write("\n")
                        crlf_counter = 0
                    end
                end
                out:write("};\n")
                out:write("unsigned char const* "..symbol.." = "..symbol.."_;\n")
                out:write("int "..symbol.."_len = sizeof("..symbol.."_);\n")

                out:write("#endif // ARCHITECTURE == "..name.."\n")

                print("      x"..name.." : "..tostring(#bin_data).." bytes")
            end
        end

        -- Some debug stuff so loose files can be loaded in debug builds.
        symbols = {}
        out:write("#else\n")
        for _, file in spairs(manifest.files, compare_values) do
            local symbol = path.getname(file):gsub("%.", "_")
            symbol = manifest.name .. "_" .. symbol .. "_file"
            table.insert(symbols, symbol)

            file = file:gsub("\\", "/")
            out:write("const char* " .. symbol .. " = CLINK_BUILD_ROOT \"/../../" .. root .. "/" .. file .. "\";\n")
        end

        out:write("#endif\n")
        out:close()
    end
end

--------------------------------------------------------------------------------
newaction {
    trigger = "embed",
    description = "Clink: Update embedded scripts for Clink",
    execute = function ()
        do_embed()
    end
}

--------------------------------------------------------------------------------
newaction {
    trigger = "embed_debug",
    description = "Clink: Update embedded scripts for Clink with debugging info",
    execute = function ()
        do_embed(true--[[debug_info]])
    end
}