Module:Table/IOTS

Documentation for this module may be created at Module:Table/IOTS/doc

local p = {}
local months = {
	January = 1,
	February = 2,
	March = 3,
	April = 4,
	May = 5,
	June = 6,
	July = 7,
	August = 8,
	September = 9,
	October = 10,
	November = 11,
	December = 12,
	['-'] = 13
}


function debug_result(var, debug_flag)
   if debug_flag == '1' then 
       return '<pre>' .. var .. '</pre>'
   else
       return mw.getCurrentFrame():preprocess(var)
   end
end
	
function p.build_table(frame) 
	local args = frame:getParent().args
	local debug_flag = args['debug'] or frame.args['debug'] or 0
	local data = args['data'] or ''
	local lines = mw.text.split(data, "\n")
	
	local results = {}
	for _, line in ipairs(lines) do
        line = mw.text.trim(line)
        if line ~= "" then
            local parts = mw.text.split(line, "|")
            local month = mw.text.trim(parts[1] or "-")
            local item  = mw.text.trim(parts[2] or "")
            if item ~= '' then
            	local key = months[month]
            	if  results[key] == nil then 
            		results[key] = {
            			month = month
            		}
            	end
            	if item:lower():find("royal") then
            		results[key].royal = results[key].royal or item
            	else
            		results[key].normal = results[key].normal or item
            	end
            end
        end
	end
	local keys = {}
	for k in pairs(results) do
		table.insert(keys, k)
	end
	table.sort(keys)
	
	local sanitized = {}
    for _, key in ipairs(keys) do
        local entry = results[key]
        local line = '|-\n|colspan="1" rowspan="2"|' .. entry.month .. '\n|'
        if entry.normal then
        	line = line .. '{{ItemLink|' .. entry.normal .. "}}"
        	line = line .. "\n|''{{Item/Description|" .. entry.normal .. "}}''"
        else 
        	line = line .. '\n|'
        end
        local line = line .. '\n|-\n|'
        if entry.royal then
        	line = line .. '{{ItemLink|' .. entry.royal .. "}}"
        	line = line .. "\n|''{{Item/Description|" .. entry.royal .. "}}''"
        else 
        	line = line .. '\n|'
        end
        table.insert(sanitized, line)
    end
    
    return debug_result(table.concat(sanitized, "\n"), debug_flag)
end

return p