Make testing framework more generic
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user