Editing Module: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, category)
	return {
		'[[Has subobject::<q>[[Located in::' .. location .. ']]</q>]]',
		'category=' .. category,
		'?#-',
		'?Name',
		'?Image#-',
	}
end

function p._main(args)
	local ret = ''
	local location = args[1] or mw.title.getCurrentTitle()
	
	for _,cat in ipairs(categories) do
		ret = ret .. '===' .. cat .. '===\n'
		local query = mw.smw.ask(base_query(location, cat))
		
		local gallery = '<gallery>\n'
		for i,v in ipairs(query or {}) do
			gallery = gallery .. (v.Image or 'File:Section anchor light.svg') .. '|' .. '[[' .. v[1] .. '|' .. v.Name .. ']]\n'
		end
		gallery = gallery .. '</gallery>\n'
		
		ret = ret .. frame:preprocess(gallery)
	end
end

return p
@@ Line 6: / Line 6: @@
 end
-local infobox_cat_map = {
+local categories = {
+	'Non-player characters',
-	['Monster'] = 'Monsters',
+	'Monsters',
-	['NPC'] = 'Non-player characters',
+	'Skill nodes',
-	['Scenery'] = 'Scenery',
+	'Scenery'
-	['Skill_node'] = 'Skill nodes'
 }
+function base_query(location, category)
-local header_text_override = {
+	return {
-	['Non-player characters'] = 'Personalities'
+		'[[Has subobject::<q>[[Located in::' .. location .. ']]</q>]]',
-}
+		'category=' .. category,
-function base_query(location)
-	local ret = {
-		'[[Located in::' .. location .. ']]',
-		'?Location object#-',
-		'?Location object.Name=name_sub',
-		'?Location object.Image#-=image_sub',
-		'?Location object.Infobox=infobox_sub',
-		'?Location quantity',
 		'?#-',
-		'?Name=name',
+		'?Name',
-		'?Image#-=image',
+		'?Image#-',
-		'?Infobox=infobox'
 	}
-	return ret
 end
 function p._main(args)
-	local frame = mw.getCurrentFrame()
+	local ret = ''
+	local location = args[1] or mw.title.getCurrentTitle()
-	local ret = {}
-	local location = args[1] or mw.title.getCurrentTitle().fullText
+	for _,cat in ipairs(categories) do
+		ret = ret .. '===' .. cat .. '===\n'
-	local query = mw.smw.ask(base_query(location))
+		local query = mw.smw.ask(base_query(location, cat))
-	if query ~= nil then
-		local results_by_type = {}
+		local gallery = '<gallery>\n'
-		for i,v in pairs(infobox_cat_map) do
+		for i,v in ipairs(query or {}) do
-			results_by_type[v] = {}
+			gallery = gallery .. (v.Image or 'File:Section anchor light.svg') .. '|' .. '[[' .. v[1] .. '|' .. v.Name .. ']]\n'
-		end
-		for i,v in ipairs(query) do
-			local key = infobox_cat_map[v.infobox_sub or v.infobox]
-			-- Can be nil - See East Beach for example
-			if key ~= nil then
-				table.insert(results_by_type[key], v)
-			end
-		end
-		for cat,results in pairs(results_by_type) do
-			if #results > 0 then
-				table.insert(ret, tostring(mw.html.create('h3'):wikitext(header_text_override[cat] or cat)))
-				local gallery_lines = {}
-				for i,v in ipairs(results) do
-					local image = v.image_sub or v.image or 'File:Unknown episode infobox icon.png'
-					local quantity = ''
-					if v['Location quantity'] and (v['Location quantity'] ~= 'Unknown' and v['Location quantity'] ~= '1') then
-						quantity = string.format('%s × ', v['Location quantity'])
-					end
-					local obj_name = v['Location object'] or v[1]
-					local version_name = ''
-					local version = string.match(obj_name, "[^#]+%#([^#]+)")
-					if version then
-						version_name = string.format(" <sub>''%s''</sub>", version)
-					end
-					local display_name = v.name_sub or v.name
-					table.insert(gallery_lines, string.format('%s|link=%s|%s[[%s|%s%s]]', image, obj_name, quantity, obj_name, display_name, version_name))
-				end
-				table.insert(ret, tostring(frame:extensionTag{ name = 'gallery', content = table.concat(gallery_lines, '\n'), args = {} }))
-			end
 		end
+		gallery = gallery .. '</gallery>\n'
+		ret = ret .. frame:preprocess(gallery)
 	end
-	return table.concat(ret, '')
 end