Module:Uses material list

From Roat Pkz
Jump to navigation Jump to search

Documentation for this module may be created at Module:Uses material list/doc

local p = {}

local geprice = require('Module:Exchange')._price
local commas = require('Module:Addcommas')
local skillpic = require('Module:SCP')._main
local yesno = require('Module:Yesno')
local lang = mw.getContentLanguage()
local trim = mw.text.trim

-- Sorting function for two item objects, first by name, then by sub-text
function sortItemsByNameSubtext(item1, item2)
	if (item1.output.name < item2.output.name) then
		return true
	elseif (item1.output.name > item2.output.name) then
		return false
	else
		return (item1.output.subtxt or '') < (item2.output.subtxt or '')
	end
end

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

function p._main(args)
	local materials = {}
	if (args == nil) then
		table.insert(materials, mw.title.getCurrentTitle().text)
	else
		local splitMultiple = {}
		for type in string.gmatch(args[1], '([^,]+)') do
			table.insert(splitMultiple, type)
		end
		
		for _, v in ipairs(splitMultiple) do
			mw.log(v)
			table.insert(materials, trim(v))
		end
	end
	
	local materialsLowered = {}
	for index, material in pairs(materials) do
	    materialsLowered[material:lower()] = true
	end

    -- Fetch data
    local query = {
        '[[Uses material::'..table.concat(materials, '||')..']]',
        '[[Production JSON::+]]',
        '?=#-',
        '?Production JSON = json',
        limit = args.limit or 500,
    }
    local t1 = os.clock()
    local t2 = os.clock()

    if not smwdata then
        return 'Failed to find products with those materials - ensure they are spelled correctly. (ERR: no results from SMW)[[Category:Empty drop lists]]'
    end
    mw.log(string.format('SMW: entries %d, time elapsed: %.3f ms.', #smwdata, (t2 - t1) * 1000))

    -- Post-process
    data = {}
    for _, e in ipairs(smwdata) do
    	if type(e['json']) == 'string' then
            local j = mw.text.jsonDecode(e['json'])
            table.insert(data, j)
        elseif type(e['json']) == 'table' then
        	for _, f in ipairs(e['json']) do
        		local j = mw.text.jsonDecode(f)
                table.insert(data, j)
        	end
    	end
    end
    smwdata = nil
    
    -- Items with different variants (e.g. Divine battlemage potion has (1), (2), (3), (4) variants)
    -- or methods (e.g. "4-dose oil" vs "3-dose oil" for pyre logs) may not come in sorted, so now do a final alphabetical sort on name, sub-text
    table.sort(data, sortItemsByNameSubtext)

    -- Render page
    local t = mw.html.create('table')
    t:addClass('wikitable sortable products-list align-center-1 align-right-3 align-right-4')
    local ttlrow = t:tag('tr')
            :tag('th')
            	:attr('colspan', '2')
                :wikitext('Item')
            :done()
            :tag('th')
                :wikitext('Members')
               :done()
            :tag('th')
            	:attr('data-sort-type', 'number')
                :wikitext('Skills')
            :done()
            :tag('th')
            	:attr('data-sort-type', 'number')
            	:wikitext('XP')
            :done()
            :tag('th')
                :wikitext('Materials')
            :done()

    local rows = 0

    -- Render rows
    for _, item in ipairs(data) do
        local _found = false
        for _,mat_info in ipairs(item.materials) do
            if materialsLowered[mat_info.name:lower()] == true then
                _found = true
                break
            end
        end
        if _found then
            local namestr
            if (tonumber(item.output.quantity) or 0) > 1 then
                namestr = string.format('[[%s]] × %s', item.output.name, item.output.quantity)
            else
                namestr = string.format('[[%s]]', item.output.name)
            end
            
            if item.output.subtxt then
            	namestr = string.format('%s <br /><small>(%s)</small>', namestr, item.output.subtxt)	
            end
            
            local memberstr
            members = item.members
            if members == 'Yes' then
				memberstr = "[[File:Member icon.png|center|link=Members]]"	
			elseif members == 'No' then
				memberstr = "[[File:Free-to-play icon.png|center|link=Free-to-play]]"	
			end
		
            local experience_cell = mw.html.create('td')
            experience_cell:addClass('plainlist')
            local experience_ul = experience_cell:tag('ul')
            experience_ul:addClass('skills-list')
            local skills_cell = mw.html.create('td')
            skills_cell:addClass('plainlist')
            local skills_ul = skills_cell:tag('ul')
            skills_ul:addClass('skills-list')
            if #item.skills == 0 then
        		local skill_li = skills_ul:tag('li')
        		local experience_li = experience_ul:tag('li')
        		experience_cell:addClass('table-na'):css('text-align', 'center')
        		experience_li:wikitext(string.format('None'))
        		skills_cell:addClass('table-na'):css('text-align', 'center')
        		skill_li:wikitext(string.format('None'))
        	else
        		for index, v in ipairs(item.skills) do
            		local skill_li = skills_ul:tag('li')
            		local experience_li = experience_ul:tag('li')
            		local stripped_experience = string.gsub(v.experience, ',', '')
            		skill_li:attr('data-sort-value', index == 1 and v.level or ''):wikitext(skillpic(lang:ucfirst(v.name), v.level))
            		experience_li:attr('data-sort-value', index == 1 and v.experience or ''):wikitext(tonumber(stripped_experience) ~= nil and skillpic(lang:ucfirst(v.name), v.experience) or v.experience)
        		end
        	end
            local mats_ul = mw.html.create('ul')
            mats_ul:addClass('products-materials')
            local materialsortvalue = nil
            for _, mat_info in ipairs(item.materials) do
                local mat_li = mats_ul:tag('li')
                local qty = string.gsub(mat_info.quantity, '%-', '–')
                local matnm = mat_info.name
            	for _,mat in ipairs(materialsLowered) do
            		if mat_info.name:lower() == mat then
            			mat_li:addClass('production-selected')
            		end
            	end
                if materialsLowered[mat_info.name:lower()] == true then
                    mat_li:addClass('production-selected')
                end
                
                if materialsortvalue == nil then
                	materialsortvalue = qty
                end
                
                mat_li:wikitext(string.format('%s × [[%s]]', commas._add(qty), matnm))
            end

            local prow = t:tag('tr')
            	:tag('td')
            		:wikitext(item.output.image)
                :tag('td')
                	:attr('data-sort-value', item.product)
                    :wikitext(namestr)
                :done()
                :tag('td')
                	:wikitext(memberstr)
                :node(skills_cell)
                :node(experience_cell)
                :tag('td')
                	:attr('data-sort-value', materialsortvalue)
                	:addClass('plainlist')
                    :node(mats_ul)
                :done()
            rows = rows + 1
        end
    end

    if rows == 0 then
        return 'Failed to find products with those materials - ensure they are spelled correctly. (ERR: no mats found in results)[[Category:Empty drop lists]]'
    end

    return t
end

--[[ DEBUG COPYPASTA
= p._main({'Mithril bar'})
--]]

return p