Module:Mod

From Path of Exile 2 Wiki
Revision as of 14:46, 7 August 2016 by >Illviljan
Jump to navigation Jump to search
Module documentation[view] [edit] [history] [purge]


Lua logo

This module depends on the following other modules:

Module for handling for modifiers with Cargo support.

List of currently implemented templates

--
-- Module for mod related templates
--

local util = require('Module:Util')
local getArgs = require('Module:Arguments').getArgs
local game = require('Module:Game')

local p = {}

-- ----------------------------------------------------------------------------
-- Utility / Helper functions
-- ----------------------------------------------------------------------------

local h = {}

-- Validate single value properties and set them

h.validate = {}

function h.validate.not_nil (args)
    return function (arg)
        if g_args[arg] == nil then
            error(string.format('%s must not be nil', arg))
        end
    end
end

function h.validate.number (args)
    return function (arg)
        g_args[arg] = util.cast.number(g_args[arg], args)
        return g_args[arg]
    end
end


function h.handle_mapped_property_args (map)
    local properties = {}
    
    for _, data in ipairs(map) do
        if data.func ~= nil then
            data.func(data.name)
        end
        
        if data.property ~= nil then
            properties[data.property] = g_args[data.name]
        end
    end
    
    g_frame:callParserFunction('#set:', properties)
end

function h.create_header(row)
    local stat = mw.html.create('span')
    local text, nsub = mw.ustring.gsub(row['Has stat text'], '%d+', '?')
    stat
        :attr('class', 'mod-table-header-stat')
        :wikitext(text)
        :done()
        
    local mgroup = mw.html.create('span')
    mgroup
        :attr('class', 'mod-table-header-modgroup')
        :wikitext(row['Has mod group'])
        :done()
        
    local tbl = mw.html.create('table')
    tbl
        :attr('class', 'wikitable mw-collapsible mw-collapsed mod-table') 
        :tag('tr')
            :tag('th')
                :attr('class', 'mod-table-header')
                :attr('colspan', g_args.colspan)
                :tag('span')
                    :attr('class', 'mod-table-header-container')
                    :wikitext(tostring(stat) .. tostring(mgroup))
                :done()
            :done()
    return tbl
end

function h.format_mod(tbl, row, tags)
    local tr = tbl:tag('tr')
    tr
        :tag('td')
            :wikitext(string.format('[[%s|%s]]', row[1], row['Has name']))
            :attr('class', 'mod-table-cell-name')
            :done()
        :tag('td')
            :wikitext(row['Has level requirement'])
            :attr('class', 'mod-table-cell-level')
            :done()
        :tag('td')
            :wikitext(row['Has stat text'])
            :attr('class', 'mod-table-cell-stat')
            :done()
        :tag('td')
            :wikitext(table.concat(tags, ', '))
            :attr('class', 'mod-table-cell-tags')
            :done()
end


-- ----------------------------------------------------------------------------
-- Templates
-- ----------------------------------------------------------------------------

--
-- Template: Mod
--

-- p.mod{id = "LocalIncreasedPhysicalDamagePercentUniqueOneHandSword2", name = "", mod_group = "LocalPhysicalDamagePercent", domain = "1", generation_type = "3", required_level = "1", mod_type = "LocalPhysicalDamagePercent", stat_text = "150% increased Physical Damage", stat1_id = "local_physical_damage_+%", stat1_min = "150", stat1_max = "150"}
function p.mod(frame)
    -- Get args
    g_args = getArgs(frame, {
        parentFirst = true
    })
    g_frame = util.misc.get_frame(frame)
    
    --
    -- Validation & semantic properties
    --
    
    -- Validate single value properties and set them
    
    local map = {
        {
            name = 'id',
            property = 'Is mod',
            wikitext = 'Mod Id',
        },
        {
            name = 'name',
            property = 'Has name',
            wikitext = 'Name',
        },
        {
            name = 'mod_group',
            property = 'Has mod group',
            wikitext = 'Group',
        },
        {
            name = 'mod_type',
            property = 'Has mod type',
            wikitext = 'Mod type',
        },
        {
            name = 'domain',
            property = 'Has mod domain',
            func = h.validate.number{min=1, max=11},
            wikitext = 'Mod domain',
            display = function (value)
                return game.constants.mod.domains[value]['short_upper'] .. ' (Id: ' .. value .. ')'
            end,
        },
        {
            name = 'generation_type',
            property = 'Has mod generation type',
            func = h.validate.number{min=1, max=10},
            wikitext = 'Generation type',
            display = function (value)
                return game.constants.mod.generation_types[value]['short_upper'] .. ' (Id: ' .. value .. ')'
            end,
        },
        {
            name = 'required_level',
            property = 'Has level requirement',
            func = h.validate.number{min=0, max=100},
            wikitext = 'Req. level',
        },
        {
            name = 'stat_text',
            property = 'Has stat text',
            wikitext = 'Effect',
        },
        {
            name = 'granted_buff_id',
            property = 'Has granted buff id',
            wikitext = 'Granted Buff Id',
        },
        {
            name = 'granted_buff_value',
            property = 'Has granted buff value',
            wikitext = 'Granted Buff Value',
        },
        {
            name = 'granted_skill',
            property = 'Has granted skill id',
            wikitext = 'Granted Skill',
        },
    }
    
    h.handle_mapped_property_args(map)
    
    -- Validate & set multi value property
    
    if g_args['tags'] == nil then
        g_args['tags'] = {}
    else
        local tags = mw.text.split(g_args['tags'], ', ')
        
        g_args['tags'] = tags
    
        properties = {}
        properties['Has tag'] = table.concat(tags, ';')
        properties['+sep'] = ';'
        g_frame:callParserFunction('#set:', properties)
    end
    
    -- Validate % set the stat multi value property
    local i
    local id
    local value
    for i=1, 5 do
        local id = {
            id = 'stat' .. i .. '_id',
            min = 'stat' .. i .. '_min',
            max = 'stat' .. i .. '_max',
        }
        
        local value = {
            id = g_args[id.id],
            min = g_args[id.min],
            max = g_args[id.max],
        }
        
        if not (value.id == nil and value.min == nil and value.max == nil) then
            local onenil = nil
            for _, k in ipairs({'id', 'min', 'max'}) do
                if value[k] == nil then
                    onenil = k
                    break
                end
            end

            if onenil ~= nil then
                error('"' .. id[onenil] .. '" is not set')
            end
            
            properties = {}
            properties['Is stat number'] = i
            properties['Has stat id'] = value.id
            properties['Has minimum stat value'] = h.validate.number({})(id.min)
            properties['Has maximum stat value'] = h.validate.number({})(id.max)
            
            g_frame:callParserFunction(string.format('#subobject: stat %s', i), properties)
        end
    end
    
    -- Validate & set spawn weight multi value property
    i = 0
    id = nil
    value = nil

    repeat
        i = i + 1
        id = {
            tag = string.format('spawn_weight%s_tag', i),
            value = string.format('spawn_weight%s_value', i),
        }
    
        value = {
            tag = g_args[id.tag],
            value = g_args[id.value],
        }
        
        if value.tag ~= nil and value.value ~= nil then
            properties = {}
            properties['Is tag number'] = i
            properties['Has tag'] = value.tag
            properties['Has spawn weight'] = h.validate.number({min=0})(id.value)
            
            g_frame:callParserFunction(string.format('#subobject: spawn weight %s', i), properties)
            
        elseif not (value.tag == nil and value.value == nil) then
            error ('Both ' .. id.tag .. ' and ' .. id.value .. ' must be specified')
        end
    until value.tag == nil
    
    -- Validate & set generation weight multi value property
    i = 0
    id = nil
    value = nil

    repeat
        i = i + 1
        id = {
            tag = string.format('generation_weight%s_tag', i),
            value = string.format('generation_weight%s_value', i),
        }
    
        value = {
            tag = g_args[id.tag],
            value = g_args[id.value],
        }
        
        if value.tag ~= nil and value.value ~= nil then
            properties = {}
            properties['Is tag number'] = i
            properties['Has tag'] = value.tag
            properties['Has generation weight'] = h.validate.number({min=0})(id.value)
            
            g_frame:callParserFunction(string.format('#subobject: generation weight %s', i), properties)
            
        elseif not (value.tag == nil and value.value == nil) then
            error ('Both ' .. id.tag .. ' and ' .. id.value .. ' must be specified')
        end
    until value.tag == nil
    
    --
    -- Display
    --
    
    local container = mw.html.create('div')
    container
        :attr('class', 'modbox')
    
    -- core stats
    
    local tbl = container:tag('table')
    tbl
        :attr('class', 'wikitable')
    
    for _, data in ipairs(map) do
        local text
        if data.display == nil then
            text = g_args[data.name]
        else
            text = data.display(g_args[data.name])
        end
        
        tbl
            :tag('tr')
                :tag('th')
                    :wikitext(string.format('[[Property:%s|%s]]', data.property, data.wikitext))
                    :done()
                :tag('td')
                    :wikitext(text)
                    :done()
                :done()
            :done()
    end
    
    tbl
        :tag('tr')
            :tag('th')
                :wikitext('[[Property:Has tag|Tags]]')
                :done()
            :tag('td')
                :wikitext(table.concat(g_args['tags'], ', '))
                :done()
            :done()
        :done()
    
    -- stat table
    
    tbl = container:tag('table')
    tbl
        :attr('class', 'wikitable sortable')
        :tag('tr')
            :tag('th')
                :attr('colspan', 4)
                :wikitext('Stats')
                :done()
            :done()
        :tag('tr')
            :tag('th')
                :wikitext('[[Property:Is stat number|#]]')
                :done()
            :tag('th')
                :wikitext('[[Property:Has stat id|Stat Id]]')
                :done()
            :tag('th')
                :wikitext('[[Property:Has minimum stat value|Minimum]]')
                :done()
            :tag('th')
                :wikitext('[[Property:Has maximum stat value|Maximum]]')
                :done()
            :done()
        :done()
        
    for i=1, 5 do
        local value = {
            id = g_args['stat' .. i .. '_id'],
            min = g_args['stat' .. i .. '_min'],
            max = g_args['stat' .. i .. '_max'],
        }
        
        if value.id then
            tbl
                :tag('tr')
                    :tag('td')
                        :wikitext(i)
                        :done()
                    :tag('td')
                        :wikitext(value.id)
                        :done()
                    :tag('td')
                        :wikitext(value.min)
                        :done()
                    :tag('td')
                        :wikitext(value.max)
                        :done()
                    :done()
                :done()
        end
    end
    
    -- spawn weight table
    
    tbl = container:tag('table')
    tbl
        :attr('class', 'wikitable sortable')
        :tag('tr')
            :tag('th')
                :attr('colspan', 3)
                :wikitext('Spawn Weights')
                :done()
            :done()
        :tag('tr')
            :tag('th')
                :wikitext('[[Property:Is tag number|#]]')
                :done()
            :tag('th')
                :wikitext('[[Property:Has tag|Tag]]')
                :done()
            :tag('th')
                :wikitext('[[Property:Has spawn weight|Weight]]')
                :done()
            :done()
        :done()
        
    i = 0
    value = nil
    repeat
        i = i + 1
        value = {
            tag = g_args[string.format('spawn_weight%s_tag', i)],
            value = g_args[string.format('spawn_weight%s_value', i)],
        }
        
        if value.tag then
            tbl
                :tag('tr')
                    :tag('td')
                        :wikitext(i)
                        :done()
                    :tag('td')
                        :wikitext(value.tag)
                        :done()
                    :tag('td')
                        :wikitext(value.value)
                        :done()
                    :done()
                :done()
        end
    until value.tag == nil
    
    -- generation weight table
    
    tbl = container:tag('table')
    tbl
        :attr('class', 'wikitable sortable')
        :tag('tr')
            :tag('th')
                :attr('colspan', 3)
                :wikitext('Generation Weights')
                :done()
            :done()
        :tag('tr')
            :tag('th')
                :wikitext('[[Property:Is tag number|#]]')
                :done()
            :tag('th')
                :wikitext('[[Property:Has tag|Tag]]')
                :done()
            :tag('th')
                :wikitext('[[Property:Has generation weight|Weight]]')
                :done()
            :done()
        :done()
    
    i = 0
    value = nil
    repeat
        i = i + 1
        value = {
            tag = g_args[string.format('generation_weight%s_tag', i)],
            value = g_args[string.format('generation_weight%s_value', i)],
        }
        
        if value.tag then
            tbl
                :tag('tr')
                    :tag('td')
                        :wikitext(i)
                        :done()
                    :tag('td')
                        :wikitext(value.tag)
                        :done()
                    :tag('td')
                        :wikitext(value.value)
                        :done()
                    :done()
                :done()
        end
    until value.tag == nil
    
    -- Generic messages on the page
    
    out = {}
    
    if mw.ustring.find(g_args['id'], '_') then
        out[#out+1] = g_frame:expandTemplate{ title = 'Incorrect title', args = { title=g_args['id'] } } .. '\n\n\n'
    end
    
    if g_args['name'] then
        out[#out+1] = string.format("'''%s''' is the internal id of modifier '''%s'''.\n", g_args['id'], g_args['name'])
    else
        out[#out+1] = string.format("'''%s''' is the internal id of an unnamed modifier.\n", g_args['id'], g_args['name'])
    end
    
    -- Categories
    
    cats = {'Mods'}
    
    -- Done -> output
    
    return tostring(container) .. util.misc.add_category(cats) .. '\n' .. table.concat(out) 
end

--
-- Template: SMW query mods
-- 

function p.query_mods(frame)
    -- Args
    g_args = getArgs(frame, {
        parentFirst = true
    })
    g_frame = util.misc.get_frame(frame)
    
    g_args.colspan = 4
    
    local conditions = {}
    conditions[#conditions+1] = 'concept'
    if g_args.tag then
        conditions[#conditions+1] = string.format('[[Has subobject::<q>[[-Has subobject::+]] [[Has spawn weight::>>0]] [[Has tag::%s]]</q>]]', g_args.tag)
    end
    
    g_args.header_level = g_args.header_level or 2
    
    -- Fields
    local fields = {}
    fields[#fields+1] = '?Is mod'
    fields[#fields+1] = '?Has name'
    fields[#fields+1] = '?Has level requirement'
    fields[#fields+1] = '?Has mod group'
    fields[#fields+1] = '?Has stat text'
    -- parameters
    local parameters = {}
    parameters.sort = 'Has mod group, '
    parameters.limit = 1000 -- lets see
    
    local data = {}
    data.header = {
        prefix = 'Prefix',
        suffix = 'Suffix',
    }
    
    local out = {}
    for _, v in ipairs({'prefix', 'suffix'}) do
        out[#out+1] = string.format('<h%i>%s</h%i>', g_args.header_level, data.header[v], g_args.header_level)
        conditions[1] = string.format('[[Concept:Spawnable named %s item mods]]', v)
        
        local query
        local results
        --
        -- Query tags
        --
        query = {}
        query[#query+1] = string.format('[[-Has subobject::<q>%s</q>]]', table.concat(conditions, ' '))
        query[#query+1] = '[[Has tag::+]]'
        query[#query+1] = '[[Has spawn weight::+]]'
        --query[#query+1] = '[[Has spawn weight::>>0]]'
        query[#query+1] = '?Has tag'
        query[#query+1] = '?Has spawn weight#' -- need native number
        query.limit = 1000
        query.offset = 0
        -- Tag order is very important
        query.sort = ', Is tag number'
        
        local tags = {}
        -- this works because lua only considers nil to be false >_>
        while query.offset do
            results = util.smw.query(query, g_frame)
            query.offset = query.offset + #results
            -- terminates the while if enough reuslts have been fetched
            if query.offset % 1000 ~= 0 then
                query.offset = nil
            end
            
            for _, row in ipairs(results) do
                local page, _ = string.gsub(row[1], '#_[%x]+', '')
                if tags[page] == nil then
                    tags[page] = {}
                end
                
                local text
                if tonumber(row['Has spawn weight']) > 0 then
                    text = '[[File:Yes.png|yes|link=]]'
                else
                    text = '[[File:No.png|no|link=]]'
                end
                
                tags[page][#tags[page]+1] = string.format('%s %s', row['Has tag'], text)
            end
        end
        
        --
        -- Query mods
        --
        query = {}
        for _, v in ipairs(conditions) do
            query[#query+1] = v
        end
        for _, v in ipairs(fields) do
            query[#query+1] = v
        end
        for k, v in pairs(parameters) do
            query[k] = v
        end
        
        results = util.smw.query(query, g_frame)
        
        local last = ''
        local tbl = ''
        for _, row in ipairs(results) do
            local current = string.gsub(row['Is mod'], '%d+.*', '%%d.*')
            if string.match(last, current) then
                h.format_mod(tbl, row, tags[row[1]])
            else
                out[#out+1] = tostring(tbl)
                tbl = h.create_header(row)
                h.format_mod(tbl, row, tags[row[1]])
            end
            last = row['Is mod']
        end
        
        -- add the last table
        out[#out+1] = tostring(tbl)
    end
    
    return table.concat(out, '')
end

--
-- Template: SMW mod table
-- 

-- =p.mod_list{'Strength1', 'shitty name', 'test', 'test2', userparam='extra_rows=2, show_jewels=1'}
-- =p.mod_list{'ColdCritMultiplier', 'shitty name', 'test', 'test2', userparam='extra_rows=2, show_jewels=1'}
-- =p.mod_list{'MapMultipleExilesMap2Tier', 'asdasda', 'Area yields 15% more Items<br>8% increased Rarity of Items found in this Area<br>Area is inhabited by 2 additional Rogue Exiles<br>Extra monsters ignore rarity bias (Hidden)<br>+14% Monster pack size', userparam='extra_rows=1, type=map, effect_rowid=2'}

function p.mod_list(frame)
    local types = {'map', 'jewel'}
    
    -- Args
    g_args = getArgs(frame, {
        parentFirst = true
    })
    g_frame = util.misc.get_frame(frame)
    
    --
    local args = util.string.split_args(g_args.userparam, {sep=', '})
    g_args.userparam = args
    
    args.extra_rows = (tonumber(args.extra_rows) or 0)
    if args.show_tags == nil then
        args.show_tags = true
    else
        args.show_tags = util.cast.boolean(args.show_tags)
    end
    args.effect_rowid = (tonumber(args.effect_rowid) or 0) + 1

    tr = mw.html.create('tr')
    tr
        :tag('td')
            :attr('data-sort-value', g_args[2] or g_args[1])
            :wikitext(string.format('[[%s|%s]]', g_args[1], g_args[2] or g_args[1]))
            :done()
    
    local i = 2
    local row_max = i + args.extra_rows
    local text
    while i < row_max do
        i = i + 1
        text = g_args[i] or ''
        text = table.concat(mw.text.split(text, ';', true), '<br>')
        
        if args.type == 'map' and i == args.effect_rowid then
            text = mw.text.split(text, '<br>', true)
            
            local map = {
                '%d+%% increased Quantity of Items found in this Area',
                '%d+%% increased Rarity of Items found in this Area',
                '%+%d+%% Monster pack size',
            }
            out = {}
            
            local valid
            for k, v in pairs(text) do
                valid = true
                for _, pattern in ipairs(map) do
                    if mw.ustring.find(v, pattern) ~= nil then
                        valid = false
                        break
                    end
                end
                
                if valid then
                    table.insert(out, v)
                end
            end
            
            text = table.concat(out, '<br>')
        end
            
        tr
            :tag('td')
                :wikitext(text)
                :done()
    end
    
    
    local query
    local result
    
    if args.type == 'map' then
        query = {
            string.format('[[-Has subobject::%s]]', g_args[1]),
            '[[Has stat id::+]]',
            '?Has stat id',
            '?Has minimum stat value',
            '?Has maximum stat value',
        }
        
        result = util.smw.query(query, g_frame)
        
        local stat_map = {
            ['map_item_drop_quantity_+%'] = {disp=0, sort=0},
            ['map_item_drop_rarity_+%'] = {disp=0, sort=0},
            ['map_pack_size_+%'] = {disp=0, sort=0},
        }
        
        local stat
        for _, row in ipairs(result) do
            stat = stat_map[row['Has stat id']]
            if stat ~= nil then
                stat.sort = (row['Has minimum stat value'] + row['Has maximum stat value']) / 2
                if row['Has minimum stat value'] ~= row['Has minimum stat value'] then
                    stat.disp = string.format('(%s to %s)', row['Has minimum stat value'], row['Has maximum stat value'])
                else
                    stat.disp = row['Has minimum stat value']
                end
            end
        end
        
        for _, k in ipairs({'map_item_drop_quantity_+%', 'map_item_drop_rarity_+%', 'map_pack_size_+%'}) do
            stat = stat_map[k]
            tr
                :tag('td')
                    :attr('data-sort-value', stat.sort)
                    :wikitext(stat.disp)
                    :done()
                :done()
        end
    end
    
    local tags
    if args.show_tags or args.type == 'jewel' then
        query = {
            string.format('[[-Has subobject::%s]]', g_args[1]),
            '[[Has tag::+]]',
            '?Has tag',
            '?Has spawn weight',
            sort='Is tag number',
        }

        tags = {}
        result = util.smw.query(query, g_frame)
    end
    
    if args.type == 'jewel' then
        local jewels = {
            dex=0,
            str=0,
            int=0,
            pris=0,
        }
        
        local cast_tbl = {
            not_dex={'str', 'int'},
            not_int={'str', 'dex'},
            not_str={'int', 'dex'},
            default={'str','int','dex','pris'},
        }
        
        local i = #result
        local row
        local cast
        while i > 0 do
            row = result[i]
            cast = cast_tbl[row['Has tag']]
            if cast ~= nil then
                for _, k in ipairs(cast) do
                    jewels[k] = row['Has spawn weight']
                end
            end
            
            i = i - 1
        end
        
        tr
            :tag('td')
                :attr('class', 'table-cell-dex')
                :wikitext(jewels.dex)
                :done()
            :tag('td')
                :attr('class', 'table-cell-int')
                :wikitext(jewels.int)
                :done()
            :tag('td')
                :attr('class', 'table-cell-str')
                :wikitext(jewels.str)
                :done()
            :tag('td')
                :attr('class', 'table-cell-prismatic')
                :wikitext(jewels.pris)
                :done()
            :done()
    end
    
    if args.show_tags then
        for _, row in ipairs(result) do
            tags[#tags+1] = string.format('%s %s', row['Has tag'], row['Has spawn weight'])
        end
        
        tr
            :tag('td')
                :wikitext(table.concat(tags, '<br>'))
                :done()
            :done()
    end
    
    return tostring(tr)
end

--
-- Template: 
-- 

function p.item_affix_table(frame)
	-- Args
	g_args = getArgs(frame, {
		parentFirst = true
	})
	g_frame = util.misc.get_frame(frame)
	
	-- This function adds specific tags first then tags that follows a pattern and lastly the generic default tag.
	function add_tags(frame)
	    -- Args
		g_args = getArgs(frame, {
			parentFirst = true
		})
		g_frame = util.misc.get_frame(frame)

		local tag_spec = g_args.tag_prefix
		local pattern = g_args.pattern		
		local tags = game.constants.tags
		local header_prefix = g_args.header_prefix or ''
		local tags_filtered = {}
		local out = {}
		local ignored = {}
		if g_args.ignore ~= nil then 
			ignored = mw.text.split(g_args.ignore, ', ')
		end 

		function titleCase( first, rest )
			return first:upper()..rest:lower()
		end

		if pattern ~= nil then 
			for i = 1, #tags do
				-- Add only the tags matching the string.
				if string.match(tags[i]['full'], pattern) then
					tags_filtered[#tags_filtered+1] = tags[i]['full']
				end
				
				-- Check that the latest added tag isn't on the ignore list.
				for _, ignore in ipairs(ignored) do 
					if tags_filtered[#tags_filtered] == ignore  then
						table.remove(tags_filtered, #tags_filtered)
					end
				end
			end
			
			for _,tag_i in ipairs(tags_filtered) do
				if tag_i ~= pattern then
					out[#out+1] = {
						tags = table.concat({tag_spec, tag_i, pattern, 'default'}, ', '),
						name = string.format('%s %s', 
							header_prefix or '', 
							tag_i:gsub('_', ' '):gsub(pattern, tag_spec)
						):gsub('%s+', ' '):gsub("(%a)([%w_']*)", titleCase)
					}
				end
			end
		else
			out[#out+1] = {
				tags = table.concat({tag_spec, 'default'}, ', '),
				name = header_prefix:gsub("(%a)([%w_']*)", titleCase)
			}
		end
		
		return out
	end
	
	local out = {}
	
	-- 
	local input = {
		-- Weapons
		add_tags{tag_prefix = 'axe',		pattern = 'weapon', ignore = 'specific_weapon'},
		add_tags{tag_prefix = 'bow',		pattern = 'weapon', ignore = 'specific_weapon, one_hand_weapon'},
		add_tags{tag_prefix = 'claw',		pattern = 'weapon', ignore = 'specific_weapon, two_hand_weapon'},
		add_tags{tag_prefix = 'dagger',		pattern = 'weapon', ignore = 'specific_weapon, two_hand_weapon'},
		add_tags{tag_prefix = 'mace', 		pattern = 'weapon', ignore = 'specific_weapon'},
		add_tags{tag_prefix = 'sceptre',	pattern = 'weapon', ignore = 'specific_weapon, two_hand_weapon'},
		add_tags{tag_prefix = 'staff',		pattern = 'weapon', ignore = 'specific_weapon, one_hand_weapon'},
		add_tags{tag_prefix = 'sword', 		pattern = 'weapon', ignore = 'specific_weapon'},
		add_tags{tag_prefix = 'sword', 		pattern = 'weapon', ignore = 'specific_weapon, two_hand_weapon', header_prefix = 'Thrusting'},
		add_tags{tag_prefix = 'wand', 		pattern = 'weapon', ignore = 'specific_weapon, two_hand_weapon'},
		{{tags = 'fishing_rod', name = 'Fishing Rod'}},
		
		-- Accessories
		add_tags{tag_prefix = 'amulet'},
		add_tags{tag_prefix = 'belt'},
		add_tags{tag_prefix = 'quiver'},
		add_tags{tag_prefix = 'ring'},
		
		-- Armour
		add_tags{tag_prefix = 'body_armour',pattern = 'armour', ignore = 'body_armour'},
		add_tags{tag_prefix = 'boots',		pattern = 'armour', ignore = 'body_armour'},
		add_tags{tag_prefix = 'gloves',		pattern = 'armour', ignore = 'body_armour'},
		add_tags{tag_prefix = 'helmet',		pattern = 'armour', ignore = 'body_armour'},
	}

	-- local input = {
		-- add_tags{tag_prefix = 'axe',		pattern = 'weapon', ignore = 'specific_weapon'},
		-- add_tags{tag_prefix = 'bow',		pattern = 'weapon', ignore = 'specific_weapon, one_hand_weapon'},
	-- }
	
	--
	for _, tag_tbl in ipairs(input) do
		for _, v in ipairs(tag_tbl) do
			out[#out+1] = p.drop_down_table(v)
		end
	end

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



function p.drop_down_table(frame)
	-- Args
    g_args = getArgs(frame, {
        parentFirst = true
    })
    g_frame = util.misc.get_frame(frame)
    
	-- Items can have several tags affecting them. 
	-- Misses forsaken masters currently
	-- p.drop_down_table{tags = 'dagger, one_hand_weapon, weapon, default', name = 'Daggers'} -- name could be taken from module:game?
	-- p.drop_down_table{tags = 'sword, one_hand_weapon, weapon, default', name = 'One Hand Swords'}
	-- p.drop_down_table{tags = 'sword, two_hand_weapon, weapon, default', name = 'Two Hand Swords'}
	
	-- p.drop_down_table{tags = 'amulet, default', name = 'Amulets'}
	-- p.drop_down_table{tags = 'ring, default', name = 'Rings'}
	-- p.drop_down_table{tags = 'belt, default', name = 'Belts'}
	-- p.drop_down_table{tags = 'quiver, default', name = 'Quivers'}
	
	-- p.drop_down_table{tags = 'body_armour, int_armour, armour, default', name = 'Intelligence body armours'}
	-- p.drop_down_table{tags = 'helmet, int_armour, armour, default', name = 'Intelligence helmets'}
	-- p.drop_down_table{tags = 'gloves, int_armour, armour, default', name = 'Intelligence gloves'}
	-- p.drop_down_table{tags = 'boots', int_armour, armour, default', name = 'Intelligence boots'}
	
	g_args.tag = mw.text.split(g_args['tags'], ', ')

	-- local tags_condition = table.concat(g_args.tag, ' || ')
	
	-- Conditions
    local conditions = {}
    conditions[#conditions+1] = 'concept'
    if g_args.tag then
        conditions[#conditions+1] = string.format('[[Has subobject::<q> [[Has spawn weight::>>0]] <q> [[Has tag::%s]] </q> </q>]]', table.concat(g_args.tag, ' || '))
    end
 
    -- Fields
    local fields = {
		'?Is mod',
		'?Has name',
		'?Has level requirement',
		'?Has mod group',
		'?Has mod type',
		'?Has stat text'
	}

	-- Parameters
    local parameters = {
		sort = 'Has mod group, Has mod type, Has level requirement',
		limit = 1000 -- lets see
	}
    
    local data = {}
    data.header = {
        prefix = 'Prefix',
        suffix = 'Suffix',
    }
	
	local header_item
    if g_args.name ~= nil then
		header_item = g_args.name
	else
		header_item = table.concat(g_args.tag, ', ')
	end
	
	-- Define the output.
	local out = {}
	out[#out+1] = '<div class="mw-collapsible-collapse-all" style="cursor:pointer;">Collapse All</div> | <div class="mw-collapsible-expand-all" style="cursor:pointer;">Expand All</div> \n'

	out[#out+1] = string.format('==%s== \n', header_item:gsub("^%l", string.upper):gsub("_", " "))
	
	local results ={}
    for _, val in ipairs({'prefix', 'suffix'}) do -- Loop through prefixes and suffixes.
		out[#out+1] = '<div style="vertical-align:top; display:inline-block;">'
		out[#out+1] = string.format('\n===%s===\n',data.header[val])
        conditions[1] = string.format('[[Concept:Spawnable named %s item mods]]', val)
       
        --
        -- Query mods
        --
        query = {}
        for _, v in ipairs(conditions) do
            query[#query+1] = v
        end
        for _, v in ipairs(fields) do
            query[#query+1] = v
        end
        for k, v in pairs(parameters) do
            query[k] = v
        end
        
        results[val] = util.smw.query(query, g_frame)
        
		local results_tags = {}
		local tags_weighting = {}
		local tbl = {}
		local last
		 
		for i, v in ipairs(results[val]) do -- Loop through all the results from the smw query.
			-- Query tags
			local mod_id = results[val][i]['Is mod']:gsub('_','~') -- mod id and mod pagename can sometimes be different. This tag query method is similar to other functions, so this bug might be noticeable on other places as well?
			query_tags = {
				string.format('[[-Has subobject::%s]]', mod_id),
				'?Has tag',
				'?Has spawn weight',
				'?Is tag number',
				sort='Is tag number',
			}
			results_tags[i] = util.smw.query(query_tags, g_frame) 
			tags_weighting[i] = {}
			tags_weighting[i][1] = results_tags[i][1]

			for _, row in ipairs(results_tags[i]) do
				local spawn_weight = row['Has spawn weight']:gsub(',', '') -- Remove thousand separator.
				tags_weighting[i][row['Has tag']] = tonumber(spawn_weight)
			end
			
			local j = 0 
			
			repeat  
				j = j+1
				
				if (tags_weighting[i][g_args.tag[j]] or -1) > 0 then					
					-- Display in a list, similar to poedb. Drop down table, with the varius tiers of mods in that mod type inside. 
					if results[val][i]['Has mod type'] ~= last then -- If the mod type is different to the last, create a new table.
						if tbl[#tbl] ~= nil then --i ~= 1 then
							tbl[#tbl+1] = '</table>'
						end
						
						tbl[#tbl+1] = '<table class="wikitable mw-collapsible mw-collapsed" style="text-align:left; line-height: 1.75em; width:800px; ">'
						
						-- Search for "Local" in the mod id. IF it doesn't exist assume that the mod applies globally.
						local mod_appl
						if results[val][i]['Is mod']:find('Local') ~= nil then
							mod_appl = 'Local'
						else 
							mod_appl = 'Global'
						end
						
						local mod_header = results[val][i]['Has stat text']:gsub('%d', '#'):gsub('%[%[(.-)%]%]', '%1'):gsub('(%a*)%|', ''):gsub('<br>', ', ') -- Removes the links since they're annoying when expanding and collapsing a lot.
						
						tbl[#tbl+1] = string.format('<caption style="text-align:left; line-height: 1.0em;> %s (%s) &nbsp;</caption>', g_frame:expandTemplate{ title = 'c', args = { 'mod', mod_header} }, mod_appl)
						
						last = results[val][i]['Has mod type']
					end

					--Display the bulk of the mods.
					tbl[#tbl+1] = string.format('<tr><td> %s %s, %s ([[%s|%s]]) </td></tr>', game.level_requirement['short_upper'], results[val][i]['Has level requirement'], g_frame:expandTemplate{ title = 'c', args = { 'mod', results[val][i]['Has stat text']:gsub('<br>', ', ') } }, results[val][i][1], results[val][i]['Has name'])
				end
			until (tags_weighting[i][g_args.tag[j]] or -1) >= 0  or j > #g_args.tag
		end
		tbl[#tbl+1] = '</table>' -- outro

		out[#out+1] = table.concat(tbl, '')
		out[#out+1] = '</div>'
    end
	
	return table.concat(out, '')
end

return p