Module:Room features

Module documentation

This documentation is transcluded from Module:Room features/doc. [edit] [history] [purge]

This module does not have any documentation. Please consider adding documentation at Module:Room features/doc. [edit]

Module:Room features's function main is invoked by Template:Room features.

---

local p = {}

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

local categories = {
	'Non-player characters',
	'Monsters',
	'Skill nodes',
	'Scenery'
}

function base_query(location)
	local ret = {
		'[[Has subobject::<q>[[Located in::' .. location .. ']]</q>]]',
		'?#-',
		'?Name',
		'?Image#-'
	}
	for _,cat in ipairs(categories) do
		table.insert(ret, string.format('?Category:%s=%s', cat, cat))
	end
	return ret
end

function p._main(args)
	local frame = mw.getCurrentFrame()
	local ret = ''
	local location = args[1] or mw.title.getCurrentTitle()
	
	local query = mw.smw.ask(base_query(location))
	if query ~= nil then
		local results_by_type = {}
		for i,v in ipairs(query) do
			for _,cat in ipairs(categories) do
				if i == 1 then
					results_by_type[cat] = {}
				end
				if v[cat] == true then
					table.insert(results_by_type[cat], v)
				end
			end
		end
		for cat,results in pairs(results_by_type) do
			if #results > 0 then
				ret = ret .. '===' .. cat .. '===\n'
				
				local gallery = '<gallery>\n'
				for i,v in ipairs(results) do
					gallery = gallery .. (v.Image or 'File:Section anchor light.svg') .. '|' .. '[[' .. v[1] .. '|' .. v.Name .. ']]\n'
				end
				gallery = gallery .. '</gallery>\n\n'
				mw.log(gallery)
				
				ret = ret .. gallery
			end
		end
	end
	
	return frame:preprocess(ret)
end

return p