Module:ImageGallery

From Umamusume Wiki
Jump to navigation Jump to search

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

local p = {}
local cargo = mw.ext.cargo

function p.list(frame)
	local charName = frame.args[1]
	local header = frame.args[2]
	local imageType = frame.args[3]
	local query = {
		limit = '1000',
		where = 'Gallery_Image.characters HOLDS "' .. charName .. '"',
	}
	if imageType == 'other' then
		query.where = query.where .. ' AND (Gallery_Image.type = "other"'
			.. ' OR Gallery_Image.type IS NULL)'
	else
		query.where = query.where .. ' AND Gallery_Image.type = "'
			.. imageType .. '"'
	end
	local fields = {'file','caption'}
	local images = cargo.query('Gallery_Image',
		table.concat(fields, ','), query)
	
	if #images == 0 then
		return ''
	end
	
	local wikitext = '==' .. header .. '==\n'
	wikitext = wikitext .. '<gallery>\n'
	for _, img in ipairs(images) do
		wikitext = wikitext .. img.file .. '|' .. img.caption .. '\n'
	end
	wikitext = wikitext .. '</gallery>\n'
	return frame:preprocess(wikitext)
end

return p