54 lines
2.1 KiB
Lua
54 lines
2.1 KiB
Lua
-- handles core detection logic for workspace in models
|
|
local roblox = require("@lune/roblox")
|
|
local stdio = require("@lune/stdio")
|
|
local serde = require("@lune/serde")
|
|
local types = require("./types")
|
|
|
|
local core = {}
|
|
|
|
-- recursively searches a model for a workspace instance
|
|
function core.scanForWorkspace(model: types.model, printInstanceNames: boolean): boolean
|
|
for _, child in pairs(model) do
|
|
if printInstanceNames then
|
|
print(child:GetFullName())
|
|
end
|
|
if child:IsA("Workspace") or core.scanForWorkspace(child:GetChildren(), printInstanceNames) then
|
|
return true
|
|
end
|
|
end
|
|
return false
|
|
end
|
|
|
|
-- checks if file has valid extension
|
|
function core.isValidModelFile(fileName: string): boolean
|
|
local ext = string.match(fileName, "%.([^%.]+)$")
|
|
return ext == "rbxm" or ext == "rbxmx"
|
|
end
|
|
|
|
-- takes in fileContents as a string and deserializes them returning the results of scanForWorkspace() on the deserialized model. if it can't be deserialized, it will return the results of a naive search through the xml
|
|
function core.fileContainsWorkspace(fileContents: string, opts: types.opts): boolean
|
|
if opts.zlibDecompressFiles then
|
|
local success = pcall(function() fileContents = serde.decompress("zlib", fileContents) end)
|
|
if not success then
|
|
stdio.write(stdio.color("yellow"))
|
|
stdio.write("Warning: Failed to decompress file with zlib. Proceeding with original contents.\n")
|
|
stdio.write(stdio.color("reset"))
|
|
end
|
|
end
|
|
local success, instances = pcall(function() return roblox.deserializeModel(fileContents) end)
|
|
if not success then
|
|
return string.find(fileContents, "Item class=\"Workspace\"") and true or false
|
|
end
|
|
return core.scanForWorkspace(instances, opts.printInstanceNames)
|
|
end
|
|
|
|
function core.formatResult(result: boolean, fileName: string): string
|
|
if result then
|
|
return stdio.color("green") .. `File {fileName} contains a Workspace instance.` .. stdio.color("reset")
|
|
else
|
|
return stdio.color("yellow") .. `File {fileName} does not contain a Workspace instance.` .. stdio.color("reset")
|
|
end
|
|
end
|
|
|
|
return core
|