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_one, page)
	if story_one == true then
		return string.format('[[%s/Side_Story_1|Side Story 1]]', page)
	end
	return nil
end

function p.main(frame)
	local args = getArgs(frame)
	
	local columns = args['columns']
	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,
			MAX(IF(card_side_stories.story = 1, true, false))=has_side_story_one = story_one
			MAX(IF(card_side_stories.story = 2, true, false))=has_side_story_two = story_two
		]])
		:addWhere('card_side_stories._pageID IS NOT NULL')
		:setJoinOn('cards.card_id = card_side_stories.card_id')
		:setGroupBy('cards.card_id')
	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',
				dataFields = { 'cardPageName', 'cards.card_name' },
				fn = formatLink
			},
			{
				name = 'side story 1',
				header = 'Side Story 1',
				dataFields = { 'story_one', 'cardPageName' },
				fn = story1,
				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