Module:Products

From Brighter Shores Wiki
Revision as of 12:12, 28 November 2024 by Artoire (talk | contribs) (Spaces to tabs)
Jump to navigation Jump to search
Module documentation
This documentation is transcluded from Module:Products/doc. [edit] [history] [purge]
This module does not have any documentation. Please consider adding documentation at Module:Products/doc. [edit]
Module:Products's function main is invoked by Template:Products.
Module:Products requires Module:Currency.

local currency = require('Module:Currency').parse

local p = {}

function p.main(frame)
	return p._main(frame:getParent().args)
end

function p._main(args)
	args = args or {}
	args.item = args.item or mw.title.getCurrentTitle().text
	local showPrices = args.showPrices or false
	local showValues = args.showValues or false
	local products = p.getProducts(args)
	if products ~= 0 then
		local recipeNames = p.extractRecipeNames(products)
		local prodprices = p.getShopBuyPrices(products)
		-- Generate and return a table containing the product information
		local output = p.displayProductTable(prodprices, showPrices, showValues)
		return output
	else
		return "There are no known products for item '''args.item'''"
	end
end

function p.extractRecipeNames(products)
	local recipeNames = {}
	for _, recipeInfo in ipairs(products) do
		local recipeLink = recipeInfo["Recipe"]
		if recipeLink then
			local displayName = recipeLink:match("%[%[.-|(.+)%]%]")
			if displayName then
				table.insert(recipeNames, displayName)
			end
		end
	end
	return recipeNames
end

function p.getProducts(args)
	local item = mw.smw.ask('[[:+]][[Uses item::' .. args.item .. ']]|?Uses item |?Uses item and quantity |?Profession A |?Profession Level A|?Value |mainlabel=Recipe') or 0
	return item
end

function p.getShopBuyPrices(products)
	for _, product in ipairs(products) do
		local usesItems = product["Uses item and quantity"]
		if type(usesItems) == "string" then
			-- Single value
			usesItems = { usesItems }
		end
		for _, item in ipairs(usesItems) do
			-- Extract item name and quantity from the "item,#" format
			local itemName, quantity = item:match("^([^,]+),([0-9.]+)$")
			quantity = tonumber(quantity) or 1 -- Default to 1 if quantity is missing

			if itemName then
				-- Query for the shop buy price of the item
				local shopPriceQuery = '[[:+]][[Sold item::' .. itemName .. ']]|?Shop buy price|mainlabel=' .. itemName
				local shopPriceResult = mw.smw.ask(shopPriceQuery) or {}
				local shopPrice = 0

				if shopPriceResult[1] and shopPriceResult[1]["Shop buy price"] then
					shopPrice = tonumber(shopPriceResult[1]["Shop buy price"]) or 0
				end

				-- Multiply the price by the quantity
				local totalPrice = shopPrice * quantity

				-- Update the product with the formatted string and total price
				product[itemName .. "_Shop_buy_price"] = totalPrice > 0 and totalPrice or "N/A"
				product[itemName .. "_Formatted"] = tostring(quantity) .. " × " .. itemName
			end
		end
	end
	return products
end

function p.displayProductTable(products, showPrices, showValues)
	local out = {}
	table.insert(out, '{| class="wikitable"')

	local headerRow = {}
	table.insert(headerRow, '!colspan="2" | Recipe')
	table.insert(headerRow, '!! Level')

	if showValues then
		table.insert(headerRow, '!! Value')
	end

	table.insert(headerRow, '!! Ingredients')

	if showPrices then
		table.insert(headerRow, '!! Price')
	end

	table.insert(out, table.concat(headerRow, ' '))

	for _, product in ipairs(products) do
		local recipeRawText = product["Recipe"]:match("%[%[.-|(.+)%]%]")
		local recipeImage = "[[File:" .. recipeRawText .. ".png|30px]]"
		local recipe = product["Recipe"] or "Unknown"
		local professionName = product["Profession A"]:match("%[%[.-|(.+)%]%]") or product["Profession A"]:match("%[%[(.-)%]%]") or product["Profession A"] or "Unknown"
		local professionLevel = "[[File:" .. professionName .. " small icon.png|15px]]  " .. (product["Profession Level A"] or "Unknown")
		local usesItems = product["Uses item"] or {}
		local value = currency(product["Value"]) or "Unknown"

		-- Concatenate uses items and their shop buy prices in unordered lists
		local usesItemStr = "<ul style='list-style:none; margin:0; padding-left:0;'>"
		local shopBuyPriceStr = "<ul style='list-style:none; margin:0; padding-left:0; text-align:right;'>"

		if type(usesItems) == "table" then
			for _, item in ipairs(usesItems) do
				local itemName = item:match("%[%[.-|(.+)%]%]") or item:match("%[%[(.-)%]%]") or item
				usesItemStr = usesItemStr .. "<li>" .. product[itemName .. "_Formatted"] .. "</li>"
				local shopBuyPrice = product[itemName .. "_Shop_buy_price"] or 0
				if shopBuyPrice == "N/A" then shopBuyPrice = 0 end
				shopBuyPriceStr = shopBuyPriceStr .. "<li>" .. currency(shopBuyPrice) .. "</li>"
			end
		elseif type(usesItems) == "string" then
			local itemName = usesItems:match("%[%[.-|(.+)%]%]") or usesItems:match("%[%[(.-)%]%]") or usesItems
			usesItemStr = usesItemStr .. "<li>" .. product[itemName .. "_Formatted"] .. "</li>"
			local shopBuyPrice = product[itemName .. "_Shop_buy_price"] or 0
			if shopBuyPrice == "N/A" then shopBuyPrice = 0 end
			shopBuyPriceStr = shopBuyPriceStr .. "<li>" .. currency(shopBuyPrice) .. "</li>"
		end

		usesItemStr = usesItemStr .. "</ul>"
		shopBuyPriceStr = shopBuyPriceStr .. "</ul>"

		table.insert(out, '|-')

		local valuesRow = {}

		table.insert(valuesRow, '| ' .. recipeImage)
		table.insert(valuesRow, '|| ' .. recipe)
		table.insert(valuesRow, '|| ' .. professionLevel)

		if showValues then
			table.insert(valuesRow, '|| ' .. value)
		end

		table.insert(valuesRow, '|| ' .. usesItemStr)

		if showPrices then
			table.insert(valuesRow, '|| ' .. shopBuyPriceStr)
		end

		table.insert(out, table.concat(valuesRow, ' '))


		--table.insert(out, '| ' .. recipe .. ' || ' .. professionLevel .. ' || ' .. value .. ' || ' .. usesItemStr .. ' || ' .. shopBuyPriceStr)
	end

	table.insert(out, '|}')
	return table.concat(out, '\n')
end

return p