Make testing framework more generic

This commit is contained in:
2025-09-25 20:00:49 -04:00
parent 057b930711
commit eb617a454f
4 changed files with 44 additions and 63 deletions
+6 -6
View File
@@ -1,4 +1,5 @@
local serde = require("@lune/serde")
local types = require("./types")
type CompressionModule = {
zlibDecompress: (contents: string, recursive: boolean?) -> string | boolean
@@ -10,19 +11,18 @@ local compressionModule = {} :: CompressionModule
-- take in file contents and zlib decompress them
-- if recursive param is true repeatedly zlib decompress the file until it fails
-- (modelscrape files are often double/triple zlib decompressed)
function compressionModule.zlibDecompress(contents: string, recursive: boolean?): string | boolean
function compressionModule.zlibDecompress(contents: string, opts: types.opts): string | boolean
assert(typeof(contents) == "string", "Expected contents to be of type 'string'")
if recursive == nil then recursive = false end
assert(typeof(recursive) == "boolean", "Expected recursive to be of type 'boolean'")
assert(typeof(opts) == "table", "Expected opts to be of type 'table'")
local success, decompressed: string = pcall(function() return serde.decompress("zlib", contents) end)
if not success then
return false
end
if recursive then
local nextDecompressed = compressionModule.zlibDecompress(decompressed, true)
if opts.zlibDecompressFilesRecursive then
local nextDecompressed = compressionModule.zlibDecompress(decompressed, opts)
while nextDecompressed do
decompressed = nextDecompressed
nextDecompressed = compressionModule.zlibDecompress(decompressed, true)
nextDecompressed = compressionModule.zlibDecompress(decompressed, opts)
end
end
return decompressed