Module:Side story datatable

Documentation for this module may be created at Module:Side story datatable/doc

local getArgs          = require('Module:Arguments').getArgs
local DatatableBuilder = require('Module:DatatableBuilder')
local QueryBuilder     = require('Module:QueryBuilder')
local Yesno            = require('Module:Yesno')

local DisplayFns    = require('Module:DisplayFunctions')
local ProcessingFns = require('Module:ProcessingFunctions')

local VariablesLua = mw.ext.VariablesLua

local p = {}

local DisplayType = {
	text = 'TEXT',
	icon = 'ICON',
	both = 'BOTH'
}

local function formatLink(page, name)
	if page and name then
		return string.format('[[%s|%s]]', page, name)
	end
	return nil
end

local function story1(story, page)
	if story == 1 then
		return string.format('[[%s/Side_Story_1|Side Story 1]]')
	end
	return nil
end

local function formatIfPretty(pretty, fn)
	if pretty == true then return fn end
	return nil
end

function p.main(frame)
	local args = getArgs(frame)
	
	local columns = args['columns']	
	local isPretty = Yesno(args['pretty'])
	local displayType = DisplayType[args['display type']] or DisplayType.text
	
	local qb = QueryBuilder.new()
	qb
		:setTables('cards', 'card_side_stories')
		:setFields([[
			cards._pageName = cardPageName,
			cards.card_id,
			cards.thumbnail,
			cards.card_name,
			cards.card_character,
			cards.rarity,
			card_side_stories.story
		]])
		:addWhere('card_id', '>', 0)
		:setOrderBy('cards.card_id ASC')
		:setGroupBy('cards._pageID')
		:setLimit(args['limit'] or 500)
		:setOffset(args['offset'])
	qb
		:addWhereList(
			'cards.card_id',
			'=',
			args['card ids'],
			',',
			'OR'
		)
		:addWhere('rarity', '=', args['rarity'])
		:addWhereList(
			'cards.card_character',
			'=',
			args['character'],
			',',
			'OR'
		)

	local datatable = DatatableBuilder.new()
	datatable
		:setColumns{
			{
				name = 'id',
				header = 'ID',
				dataFields = { 'cards.card_id' },
				visible = false
			},
			{
				name = 'thumbnail',
				dataFields = { 'cards.thumbnail' },
				fn = DisplayFns.image('64px'),
				sortable = false
			},
			{
				name = 'card',
				header 'Card',
				dataFields { 'cardPageName', 'cards.card_name' },
				fn = formatLink
			},
			{
				name = 'stories',
				header = 'Side stories',
				dataFields = { 'cardPageName' },
				fn = formatStory
			},
			{
				name = 'character',
				header = 'Character',
				dataFields = { 'cards.card_character' },
				fn = formatIfPretty(isPretty, formatCharacter),
				visible = false
			},
			{
				name = 'side story 1',
				dataFields = { 'card_side_stories.story', 'cardPageName' },
				fn = story1,
				visible = false
			},
			{
				name = 'rarity',
				header = 'Rarity',
				dataFields = { 'cards.rarity' },
				fn = formatIfPretty(isPretty, formatRarity),
				visible = false
			}
		}
		:setData(qb:query())
	
	columns = ProcessingFns.stringToArray(',')(columns)
	if columns and #columns > 0 then
		for _, column in ipairs(columns) do
			datatable:setColumnVisibility(column, true)
		end
	end
	
	return datatable:tostring()
end

return p