Module:EnchantedChest

--[=[ Lua script replacement for wikitext template



Parameters: 1: Enchanted Chest name 2: Size in pixels optional, defaults to 25 If equal to "N", includes the chest Name Link also If equal to "L", includes the chest Name Link only 3: Enchanted Chest link if diffrent from name -- Unknown is returned if Enchanted Chest is not in data.

--]=]

local p = {}

local chestData = mw.loadData( 'Module:EnchantedChest/data' ) local cData = mw.loadData( 'Module:ItemCollection/data' )

local COLL = require("Module:ItemCollection").getCollection local util = require('Module:Utility')

local function getCollection(item) local collName = "" for _, data in pairs(cData.Collections) do		if (data[3] and data[3] == item) or item == data[1] then collName = data[1] end end if collName == "" then collName = COLL({item, "N"}) end return collName end

function p.getChest(frame) local tArgs = util.getArgs(frame)

local allChest = tArgs[1] or "" local size = tArgs[2] local plural = tArgs[3] local outType = ""

if not tonumber(size) then outType = size if outType == "P" then size = 35 else size = 25 end end

local tAll = {} local space = ""

for chest in mw.text.gsplit(allChest, "%s*$%s*") do		local tChest = {space, "", "", "", "", "",	   				"", "", "", "", ""	    				}		local cName, version, count = mw.ustring.match(chest, "^([^,]-)%s*-*%s*(%w+)%s*,%s*(.-)$") if not cName then cName, version = mw.ustring.match(chest, "^([^,]-)%s*-%s*(.-)$") if not cName then cName = chest end end if not cName then cName = chest end cName = mw.ustring.gsub(cName, " Chests", "") if outType == "S" then version = mw.title.getCurrentTitle.text end local cFile = "work_in_progress" local cLink = "Work In Progress" local cDisp = cName .. " Chest" if chestData[cName] then tChest[ 7] = "ec-" cFile = mw.ustring.lower(mw.ustring.gsub(cName, " ", "_")) cLink = cName .. " Chests" cDisp = cName .. " Chest" if plural or (count and count > "1") then cDisp = cDisp .. "s" end if cName == "Legendary" and version then if outType ~= "S" then cDisp = getCollection(version) .. " " .. cDisp if outType ~= "L" then local bSize = math.floor(size / 3.3) local lSize = math.floor(size / 2.6) local cSize = math.floor(size / 1.6) tChest[ 6] = "" .. COLL({getCollection(version), cSize}) .. " [[File:"					end   			end    			cLink = cLink .. "#" .. getCollection(version)    		elseif (cName == "Sapphire" and version == "Halloween") or (cName == "Red Ribbon" and version == "Christmas") then    				cLink = version .. " Chests"   					cDisp = cDisp .. " (" .. cLink .. ")"    		end	    end		if outType == "N" then			tChest[ 2] = "[["			tChest[ 3] = cLink			tChest[ 4] = "|" .. cDisp			tChest[ 5] = "]] ("			tChest[16] = ")"		elseif outType == "L" then			tChest[ 6] = ""			tChest[10] = cDisp			tChest[11] = ""			tChest[13] = ""			tChest[15] = ""		elseif outType == "P" then			tChest[16] = " "			tChest[19] = cDisp			tChest[20] = " "		elseif outType == "S" then			tChest[ 6] = "FREE ("			tChest[10] = cDisp			tChest[11] = ") "			tChest[13] = ""			tChest[15] = ""			space = " "		end		if outType ~= "L" and outType ~= "S" then			tChest[ 8] = cFile			tChest[12] = cDisp			tChest[14] = cLink			space = " "			if count and count > "1" then 				if outType == "N" then					tChest[ 5] = "]]" .. count .. " ("				else					tChest[15] = "]]" .. count 				end			end		end   	table.insert(tAll, table.concat(tChest))	end    return table.concat(tAll) end

local EC = require("Module:EC").getEC local RARITY = require("Module:Rarity").rarity local TOK = require("Module:CharacterToken").getToken

local function addItem(data) local dataCommon = {} if data["Common"] then table.insert(dataCommon, data["Common"]) end local dataUncommon = {} if data["Uncommon"] then table.insert(dataUncommon, data["Uncommon"]) end local dataRare = {} if data["Rare"] then table.insert(dataRare, data["Rare"]) end local dataEpic = {} if data["Epic"] then table.insert(dataEpic, data["Epic"]) end local dataLegendary = {} if data["Legendary"] then table.insert(dataLegendary, data["Legendary"]) end

local raTok = "" local rbTok = "" local daTok = "" local dbTok = ""

if data["Tokens"] and data["Tokens"][1] then for tCh in mw.text.gsplit(data["Tokens"][1], "%s*$%s*") do raTok = TOK({(tCh .. ",2"), "R"}) rbTok = TOK({(tCh .. ",3"), "R"}) daTok = string.upper(string.sub(raTok, 1, 1))..string.sub(raTok, 2) dbTok = string.upper(string.sub(rbTok, 1, 1))..string.sub(rbTok, 2)

if daTok == "Common" then table.insert(dataCommon, (" $ " .. tCh .. ",2 #" .. (data["Tokens"][2] or "1"))) elseif daTok == "Uncommon" then table.insert(dataUncommon, (" $ " .. tCh .. ",2 #" .. (data["Tokens"][3] or "1"))) elseif daTok == "Rare" then table.insert(dataRare, (" $ " .. tCh .. ",2 #" .. (data["Tokens"][4] or "1"))) elseif daTok == "Epic" then table.insert(dataEpic, (" $ " .. tCh .. ",2 #" .. (data["Tokens"][5] or "1"))) elseif daTok == "Legendary" then table.insert(dataLegendary, (" $ " .. tCh .. ",2 #" .. (data["Tokens"][6] or "1"))) end

if dbTok == "Common" then table.insert(dataCommon, (" $ " .. tCh .. ",3 #" .. (data["Tokens"][2] or "1"))) elseif dbTok == "Uncommon" then table.insert(dataUncommon, (" $ " .. tCh .. ",3 #" .. (data["Tokens"][3] or "1"))) elseif dbTok == "Rare" then table.insert(dataRare, (" $ " .. tCh .. ",3 #" .. (data["Tokens"][4] or "1"))) elseif dbTok == "Epic" then table.insert(dataEpic, (" $ " .. tCh .. ",3 #" .. (data["Tokens"][5] or "1"))) elseif dbTok == "Legendary" then table.insert(dataLegendary, (" $ " .. tCh .. ",3 #" .. (data["Tokens"][6] or "1"))) end end end

local tRarity = {} table.insert(tRarity, {"Common", table.concat(dataCommon)}) table.insert(tRarity, {"Uncommon", table.concat(dataUncommon)}) table.insert(tRarity, {"Rare", table.concat(dataRare)}) table.insert(tRarity, {"Epic", table.concat(dataEpic)}) table.insert(tRarity, {"Legendary", table.concat(dataLegendary)})

return tRarity end

local function chItems(inStr) local tIn = mw.text.split(inStr, "%s*%$%s*") local tOut = {} for idx = 1, #tIn do		local tt = mw.text.split(tIn[idx], "%s*#%s*") if tt[1] ~= "" then if not tt[2] or tt[2] == "" then tt[2] = 1 end table.insert(tOut, tt) end end return tOut end

local function oneItem(item, count, withName, rarity) nCount = count or 0 if nCount == 0 then return "" end

local conv = { ["ch"] = "cp", ["cos"] = "cp", }

local tItem = {"", " " .. nCount, (withName and " " or ""), "", "",			(withName and " " or ""), }

local iName = item local iData = cData.Items[iName]

if iData and iData[2] then local fname = iData.I or iName:lower:gsub("[':%,%.]", ""):gsub(" ", "_") if iData[2] == "pf" then fname = fname:gsub("_float$", "") else fname = fname:gsub("^([^%(]-)_*%(([^%d]+%d*)%)$", "%1-%2")		end

tItem[1] = table.concat({								""							}) tItem[2] = "" if withName then tItem[4] = table.concat({"", iName, ""}) end elseif item:match("^EC") then tItem[1] = EC({item, nil, nil, nil, 30}) if withName then tItem[4] = EC({item, "L", S=(nCount == 1)}) end elseif item:match(" Decorations") then tItem[1] = "" tItem[2] = "" if withName then tItem[4] = "Some " .. rarity .. " Decorations" end elseif item:match(" Concessions") then tItem[1] = "" tItem[2] = "" if withName then tItem[4] = "Some " .. rarity .. " Concessions" end else tItem[1] = TOK({item, 40}) if withName then tItem[4] = TOK({item, "L"}) if nCount ~= 1 then tItem[5] = "s" end end end

return table.concat(tItem) end

local function colItem(iNumb, tNumb, aNumb, item) local tOut = {}

if iNumb == 1 then table.insert(tOut, '\n| colspan="2"|') table.insert(tOut, '\n| colspan="2"|') table.insert(tOut, '\n| colspan="2"|') table.insert(tOut, item) table.insert(tOut, '\n| colspan="2"|') table.insert(tOut, '\n| colspan="2"|') elseif iNumb == 2 then if aNumb == (tNumb - 1) then table.insert(tOut, '\n|\n| colspan="2"|') table.insert(tOut, '\n| colspan="2"|') table.insert(tOut, item) elseif aNumb == tNumb then table.insert(tOut, '\n| colspan="2"|') table.insert(tOut, item) table.insert(tOut, '\n| colspan="2"|\n|') end elseif iNumb == 3 then if aNumb == (tNumb - 2) then table.insert(tOut, '\n| colspan="2"|') table.insert(tOut, '\n| colspan="2"|') table.insert(tOut, item) elseif aNumb == (tNumb - 1) then table.insert(tOut, '\n| colspan="2"|') table.insert(tOut, item) elseif aNumb == tNumb then table.insert(tOut, '\n| colspan="2"|') table.insert(tOut, item) table.insert(tOut, '\n| colspan="2"|') end elseif iNumb == 4 then if aNumb == (tNumb - 3) then table.insert(tOut, '\n|\n| colspan="2"|') table.insert(tOut, item) elseif aNumb == (tNumb - 2) then table.insert(tOut, '\n| colspan="2"|') table.insert(tOut, item) elseif aNumb == (tNumb - 1) then table.insert(tOut, '\n| colspan="2"|') table.insert(tOut, item) elseif aNumb == tNumb then table.insert(tOut, '\n| colspan="2"|') table.insert(tOut, item) table.insert(tOut, '\n|') end else table.insert(tOut, '\n| colspan="2"|') table.insert(tOut, item) end

return table.concat(tOut) end

function p.getTable(frame) local tArgs = util.getArgs(frame)

local cName = tArgs[1] or mw.title.getCurrentTitle.baseText cName = mw.ustring.gsub(cName, " Chests", "") local conv = { ["Common"] = "1", ["Uncommon"] = "2", ["Rare"] = "3", ["Epic"] = "4", ["Legendary"] = "5", }	local tOut = {} local cSection = {} if chestData[cName] then for k, v in pairs(chestData[cName]) do			if v["R"] then table.insert(cSection, {(v[1] or 999), k, v}) end end table.sort(cSection, function(a,b) return a[1] > b[1] end) for _, cItem in pairs(cSection) do			local ecData = cItem[3] table.insert(tOut, '|-|') table.insert(tOut, cItem[2]) table.insert(tOut, '=\n') if ecData["Duration"] or ecData["Cost"] then table.insert(tOut, ' About  ') if ecData["Duration"] then table.insert(tOut, ' ') end if ecData["Cost"] then table.insert(tOut, ' ') end table.insert(tOut, ' \n') end table.insert(tOut, '{| class="article-table dmk-chests" style="width: ')			if ecData["Duration"] or ecData["Cost"] then table.insert(tOut, '55') else table.insert(tOut, '100') end		   	table.insert(tOut, '%;"\n! colspan="10"|') table.insert(tOut, cItem[2]) table.insert(tOut, ' ') table.insert(tOut, cName) table.insert(tOut, ' Chests Rewards') table.insert(tOut, ' ')

local tRarity = addItem(ecData["R"])

for i, item in ipairs(tRarity) do			   if item[2] ~= "" then table.insert(tOut, '\n|-\n! colspan="10"|') table.insert(tOut, RARITY({item[1], "ECR"})) if i == 1 and (ecData["Duration"] or ecData["Cost"]) then table.insert(tOut, '\n|} \n{| class="article-table dmk-chests" style="width: 100%;"\n|||||||||||||||||||\n|-') elseif i == 1 then table.insert(tOut, '\n|-\n|||||||||||||||||||\n|-') else table.insert(tOut, '\n|-') end item[2] = chItems(item[2])

local tItems = {} local tNo = "" local rNo = "" for No, rData in ipairs(item[2]) do					table.insert(tItems, oneItem(rData[1], rData[2], true, item[1])) tNo = No					rNo = No				end local iNo = 0 if tNo > 5 then repeat rNo = rNo - 5 iNo = iNo + 1 until rNo < 5 end for i, iName in ipairs(tItems) do					if i == 6 or i == 11 or i == 16 or i == 21 then table.insert(tOut, '\n|-') end if i <= (iNo * 5) then table.insert(tOut, '\n| colspan="2"|') table.insert(tOut, iName) else table.insert(tOut, colItem(rNo, tNo, i, iName)) end end end end if ecData["S"] then table.insert(tOut, "\n|-\n! colspan=\"10\"| Only at ''") table.insert(tOut, (ecData["S"][1] or "10")) table.insert(tOut, "% Special Chance'': ") local tRarity= addItem(ecData["S"])

for _, item in pairs(tRarity) do				   if item[2] ~= "" then table.insert(tOut, '\n|-\n! colspan="10"|') table.insert(tOut, RARITY({item[1], "ECR"})) table.insert(tOut, '\n|-') item[2] = chItems(item[2])

local tItems = {} local tNo = "" local rNo = "" for No, rData in ipairs(item[2]) do						table.insert(tItems, oneItem(rData[1], rData[2], true, item[1])) tNo = No						rNo = No					end local iNo = 0 if tNo > 5 then repeat rNo = rNo - 5 iNo = iNo + 1 until rNo < 5 end for i, item in ipairs(tItems) do						if i == 6 or i == 11 then table.insert(tOut, '\n|-') end if i <= (iNo * 5) then table.insert(tOut, '\n| colspan="2"|') table.insert(tOut, item) else table.insert(tOut, colItem(rNo, tNo, i, item)) end end end end end table.insert(tOut, '\n|}\n') if tArgs[cItem[2]] then table.insert(tOut, tArgs[cItem[2]]) end end end return table.concat(tOut) end

return p