local MemoryStoreService = game:GetService("MemoryStoreService")

local testHashMap = MemoryStoreService:GetHashMap("HashMap1")

local EXPIRATION = 600
local NUM_TEST_ITEMS = 32

local function populateHashMap(hashMap: MemoryStoreHashMap, numItems: number): { [string]: any }
	print("Definindo os dados do HashMap...")
	local createdItems = {}

	for index = 1, numItems do
		local key = tostring(index) -- As chaves do HashMap devem ser strings
		local value = `{key}_valor_teste`
		local success, result = pcall(hashMap.SetAsync, hashMap, key, value, EXPIRATION)

		if success then
			createdItems[key] = value
		else
			warn(`Erro ao definir a chave {key}: {result}`)
		end
	end

	print("Concluído o envio dos dados do HashMap.")
	return createdItems
end

local function getItemsFromAllPages(pages: MemoryStoreHashMapPages): { [string]: any }
	-- Puramente para fins de registro, rastreamos em qual número de página estamos
	local currentPageNumber = 1
	local retrievedItems = {}

	while not pages.IsFinished do
		print(`Obtendo itens na página {currentPageNumber}...`)
		local items = pages:GetCurrentPage()

		for _, entry in pairs(items) do
			print(`	{entry.key}: {entry.value}`)
			retrievedItems[entry.key] = entry.value
		end

		-- Avançar páginas se houver mais páginas para ler
		if not pages.IsFinished then
			pages:AdvanceToNextPageAsync()
			currentPageNumber += 1
		end
	end

	print("Leitura de todas as páginas concluída")
	return retrievedItems
end

local function compareAllItems(retrievedItems: { [string]: any }, expectedItems: { [string]: any }): number
	print("Comparando itens recuperados com itens esperados...")
	local numMatchingItems = 0

	for key, expectedValue in pairs(expectedItems) do
		if retrievedItems[key] == expectedValue then
			numMatchingItems += 1
		else
			warn(`Valor recuperado não corresponde para a chave {key}: esperado {expectedValue}, recuperado {retrievedItems[key]}`)
		end
	end

	print("Comparação concluída!")
	return numMatchingItems
end

-- As chaves adicionadas ao hashmap também são adicionadas a esta tabela expectedItems.
-- Mais tarde, os itens do hashmap recuperados serão comparados com esta tabela de itens esperados.
local expectedItems = populateHashMap(testHashMap, NUM_TEST_ITEMS)

-- Obter páginas pode falhar. Neste caso, deixaremos falhar e parar a execução do programa,
-- mas você pode querer usar pcall e tratar de forma diferente.
print(`Obtendo páginas do HashMap com ListItemsAsync...`)
local pages = testHashMap:ListItemsAsync(NUM_TEST_ITEMS)

local retrievedItems = getItemsFromAllPages(pages)
local numMatchingItems = compareAllItems(retrievedItems, expectedItems)

-- Se não houve erros ao definir ou obter itens, todos os itens devem corresponder.
print(`Programa completo. {numMatchingItems}/{NUM_TEST_ITEMS} itens recuperados corresponderam aos valores esperados.`)