local fs = require("@lune/fs")
local process = require("@lune/process")

local frktest = require("../lune_packages/frktest")
local check = frktest.assert.check

local ZipReader = require("../lib")

-- Reuse the same ZIP files from metadata tests
local ZIPS = fs.readDir("tests/data")
local FALLIBLES = {
	"invalid_cde_number_of_files_allocation_greater_offset.zip",
	-- FIXME: Incorrectly handled, file tree is empty and walk silently errors
	-- "invalid_cde_number_of_files_allocation_smaller_offset.zip",
	"invalid_offset.zip",
	"invalid_offset2.zip",
	-- FIXME: Does not error when it should
	-- "comment_garbage.zip",
	"chinese.zip",
    "non_utf8.zip", -- FIXME: Lune breaks for non utf8 data in process stdout
	"pandoc_soft_links.zip", -- FIXME: Soft links are not handled correctly
	-- FIXME: Files with a misaligned comments are not correctly located
	-- "misaligned_comment.zip",
}

return function(test: typeof(frktest.test))
	test.suite("ZIP extraction tests", function()
		for _, file in ZIPS do
			if not string.match(file, "%.zip$") then
				continue
			end

			local checkErr: ((...any) -> any?) -> nil = if table.find(FALLIBLES, file)
				then check.should_error
				else check.should_not_error

			test.case(`Extracts files correctly - {file}`, function()
				checkErr(function()
					local zipPath = "tests/data/" .. file
					local data = fs.readFile(zipPath)
					local zip = ZipReader.load(buffer.fromstring(data))

					-- Test both string and buffer extraction
					local stringOptions = { isString = true, decompress = true }
					local bufferOptions = { isString = false, decompress = true }

					-- Extract and verify each file
					zip:walk(function(entry)
						if entry.isDirectory then
							return
						end

						-- Extract with unzip for comparison
						local unzipOutput = process.spawn(`unzip`, {  "-p", zipPath, entry:getPath() })
						
						-- NOTE: We use assert since we don't know whether to expect true or false
						assert(unzipOutput.ok)

						-- Test string extraction
						local contentString = zip:extract(entry, stringOptions) :: string
						check.equal(#contentString, entry.size)
                        check.equal(contentString, unzipOutput.stdout)

						-- Test buffer extraction
						local contentBuffer = zip:extract(entry, bufferOptions) :: buffer
						check.equal(buffer.len(contentBuffer), entry.size)
						check.equal(buffer.tostring(contentBuffer), unzipOutput.stdout)

						-- Test directory extraction
						local parentPath = entry:getPath():match("(.+)/[^/]*$") or "/"
						local dirContents = zip:extractDirectory(parentPath, stringOptions)
						check.not_nil(dirContents[entry:getPath()])
						check.equal(dirContents[entry:getPath()], unzipOutput.stdout)
					end)

					return
				end)
			end)
		end
	end)
end