Module:Data Table

local p = {} local h = {}

local File = require("Module:File") local Franchise = require("Module:Franchise") local Sequences = require("Module:Sequences") local Term = require("Module:Term") local TermList = require("Module:Term List") local utilsArg = require("Module:UtilsArg") local utilsCargo = require("Module:UtilsCargo") local utilsLayout = require("Module:UtilsLayout") local utilsMarkup = require("Module:UtilsMarkup") local utilsString = require("Module:UtilsString") local utilsTable = require("Module:UtilsTable")

local CATEGORY_INVALID_ARGS = require("Module:Constants/category/invalidArgs")

-- Other modules can extend this one by using storeFn to store data in a custom table function p.Main(frame, storeFn) local args, err = utilsArg.parse(frame:getParent.args, p.Templates["Data Table"]) local categories = err and err.categoryText or "" local columnHeaders, columnTags = h.extractColumnTags(args.columns) local name = args.name or columnHeaders[1] local tableId = utilsString.kebabCase(name) local unsortableColumns = {} local dataRows = {} local currentRow = { cells = {} }	local columnIndex = 1 for i, cellText in ipairs(args.cells or {}) do		if cellText == "-" and i ~= 1 then currentRow.id = currentRow.id or tableId and tableId.."-"..(#dataRows+1) table.insert(dataRows, currentRow) currentRow = { cells = {} }			columnIndex = 1 elseif cellText ~= "-" then local cell = { class = "data-table__cell", content = mw.html.create("div"):addClass("data-table__cell-content"), raw = cellText, unsortable = false, }			local tags = columnTags[columnIndex] or {} for _, tag in ipairs(tags) do				h.formatCell(cell, args.game, tag) end if #tags == 0 then cell.content = cell.content:wikitext(cellText) end cell.content = tostring(cell.content) cell.columnHeader = columnHeaders[columnIndex] table.insert(currentRow.cells, cell) if cell.unsortable or args.sortable == "false" then unsortableColumns[columnIndex] = true end if cell.rowId then currentRow.id = cell.rowId end columnIndex = columnIndex + 1 end end currentRow.id = currentRow.id or tableId and tableId.."-"..(#dataRows+1) if #currentRow.cells > 0 then table.insert(dataRows, currentRow) end

h.store(args, dataRows, unsortableColumns) if storeFn then storeFn(args, dataRows, columnHeaders) end local dataTable = p.printTable(columnHeaders, dataRows, unsortableColumns) return dataTable, categories end

function p.Copy(frame) local args, err = utilsArg.parse(frame:getParent.args, p.Templates["Data Table Copy"]) local errCategories = err and err.categoryText or "" if not args.fromPage or not args.name then return "", errCategories end local dataTable, categories = p.copy(args) return dataTable, categories..errCategories end

function p.copy(args, filterFn) local categories = ""

local queryResults = utilsCargo.query("DataTables", "name, rowIndex, rowId, columnIndex, columnHeader, columnSortable, cellContent, cellSortValue, cellRaw", {		where = utilsCargo.allOf({ _pageName = args.fromPage, name = args.name, game = args.game, }),		orderBy= "game, name, rowIndex, columnIndex",		limit= 1000	}) if #queryResults == 0 then h.warn(string.format("No table  exists on page %s", args.name, args.fromPage)) return "", categories.."" end local dataRows = {} local currentRow = {} for i, result in ipairs(queryResults) do		if i ~= 1 and result.columnIndex == "1" then table.insert(dataRows, currentRow) currentRow = {} end table.insert(currentRow, result) end table.insert(dataRows, currentRow)

dataRows = utilsTable.map(dataRows, function(row)		return {			id = row[1].rowId,			cells = utilsTable.map(row, function(cell) return { class = "data-table__cell", columnHeader = cell.columnHeader, content = cell.cellContent, sortValue = cell.cellSortValue, raw = cell.cellRaw, }			end)		}	end) if args.rowsWith then dataRows = utilsTable.filter(dataRows, function(row)			local includeRow = false			for i, cell in ipairs(row.cells) do				if string.find(cell.raw, args.rowsWith) then					includeRow = true				end			end			return includeRow		end) end if filterFn then dataRows = utilsTable.filter(dataRows, filterFn) end local columns = utilsTable.uniqueBy(dataRows[1] and dataRows[1].cells or {}, function(cell)		return cell.columnHeader	end) local columnHeaders = {} local columnLookup = {} for i, column in ipairs(columns) do		columnHeaders[i] = column.columnHeader columnLookup[column.columnHeader] = column end local hasInvalidColumns = false local columnsToCopy if args.columns or args.excludeColumns then if args.columns then columnsToCopy = {} else columnsToCopy = utilsTable.invert(columnHeaders) end local validColumnsList = utilsMarkup.bulletList(columnHeaders) for i, column in ipairs(args.columns or args.excludeColumns) do			if not columnLookup[column] then hasInvalidColumns = true h.warn(string.format("Column  does not exist in table   on page %s. The columns defined for this table are as follows: %s", column, args.name, args.fromPage, validColumnsList)) categories = categories.."" elseif args.columns then columnsToCopy[column] = true elseif args.excludeColumns then columnsToCopy[column] = false end end end if columnsToCopy and not hasInvalidColumns then for i, row in ipairs(dataRows) do			local cells = {} for j, cell in ipairs(row.cells) do				if columnsToCopy[cell.columnHeader] then table.insert(cells, cell) end end row.cells = cells end columnHeaders = utilsTable.filter(columnHeaders, function(header)			return columnsToCopy[header]		end) end local unsortableColumns = {} for i, column in ipairs(columnHeaders) do		if columnLookup[column] and columnLookup[column].columnSortable ~= "1" then unsortableColumns[i] = true end end local dataTable = p.printTable(columnHeaders, dataRows, unsortableColumns) return dataTable, categories end

function p.printTable(columnHeaders, dataRows, unsortableColumns) -- We re-apply the data-table__cell class as other modules may re-use this function but not add that class for i, row in ipairs(dataRows) do		for j, cell in ipairs(row.cells) do local currentClass = cell.class and (cell.class.." ") or "" cell.class = currentClass.."data-table__cell" end end local styles = mw.getCurrentFrame:extensionTag({		name = "templatestyles",		args = { src = "Template:Data Table/Styles.css" }	}) local desktopTable = h.desktopTable(columnHeaders, dataRows, unsortableColumns) local mobileTable = h.mobileTable(columnHeaders, dataRows) return styles..desktopTable..mobileTable end

function h.extractColumnTags(columns) local columnHeaders = {} local columnTags = {} for i, column in ipairs(columns) do		columnHeaders[i] = column columnTags[i] = {} local matches repeat columnHeaders[i], matches = string.gsub(columnHeaders[i], "%s*%[([%w:%s]+)%]$", function(tag)				tag = utilsString.split(tag, ":")				table.insert(columnTags[i], { name = tag[1], args = utilsTable.tail(tag), })				return ""			end) until matches == 0 columnTags[i] = utilsTable.reverse(columnTags[i]) end return columnHeaders, columnTags end

local playerName function h.playerName return playerName or mw.getCurrentFrame:expandTemplate({		title = "Player Name"	}) end

local NUMBER_VARIANT_REGEX = "%s*%[([%d]+)%]%s*$" function h.formatCell(cell, game, tag) local baseClass = "data-table__cell-content" local textNode = mw.html.create("div"):addClass("data-table__cell-content-text") local frame = mw.getCurrentFrame if cell.raw == "" then cell.content:addClass(baseClass.."--empty") elseif cell.raw == "N/A" then local tooltip = utilsMarkup.tooltip("N/A", "Not Applicable") textNode:wikitext(tooltip) cell.content :addClass(baseClass.."--not-applicable") :node(textNode) elseif tag.name == "Amount" or tag.name == "Amounts" then cell.unsortable = true local amounts = utilsString.split(cell.raw, '%s*,[%D+|%s*]') -- %D ensures that we don't split large numbers (e.g. 1,500) amounts = utilsTable.map(amounts, function(itemAmount)			local amountStart, amountEnd, amount = string.find(itemAmount, "^(%d+)%s")			if not amount then				h.warn(string.format("Invalid entry . Amount columns must start with a number.", itemAmount))				textNode:wikitext("")				return itemAmount			else				local item = string.sub(itemAmount, amountEnd+1)				return frame:expandTemplate({ title = "Item Amount", args = {game, amount, item} })			end		end) textNode:wikitext(h.list(amounts)) cell.content :addClass(baseClass.."--amounts") :node(textNode) elseif tag.name == "Description" then local text = string.gsub(cell.raw, "%[Player Name%]", h.playerName) cell.unsortable = true textNode:wikitext(text) cell.content :addClass(baseClass.."--description") :node(textNode) elseif tag.name == "ID" then local text = string.gsub(cell.raw, "%[Player Name%]", h.playerName) text = string.gsub(text, NUMBER_VARIANT_REGEX, "") textNode:wikitext(text) cell.content :addClass("baseClass".."--id") :node(textNode) cell.rowId = cell.raw elseif tag.name == "Image" then local fileType = tag.args[1] or Franchise.graphics(game) == "2D" and "Sprite" or "Icon" local text = string.gsub(cell.raw, "%[Player Name%]", "Link") local number = string.match(text, NUMBER_VARIANT_REGEX) if number then text = string.gsub(text, NUMBER_VARIANT_REGEX, "") fileType = fileType.." "..number end cell.sortValue = text local image = File.gameImage(game, text, fileType, {			size = tag.args[2] or "64x64px",		}) textNode:wikitext(image) cell.content :addClass(baseClass.."--image") :node(textNode) elseif tag.name == "Rupees" then local amounts = utilsString.split(cell.raw, '%s*,[%D+|%s*]') if #amounts == 1 then cell.sortValue = amounts[1] end amounts = utilsTable.map(amounts, function(amount)			return frame:expandTemplate({ title = "Rupee", args = {game, amount} })		end) if #amounts > 1 then cell.unsortable = true end textNode:wikitext(h.list(amounts)) cell.content :addClass(baseClass.."--amounts") :node(textNode) elseif tag.name == "SectionLink" then cell.rowId = string.gsub(cell.raw, "%[Player Name%]", "Link") local text = string.gsub(cell.raw, "%[Player Name%]", h.playerName) text = string.gsub(text, NUMBER_VARIANT_REGEX, "") local page = mw.title.getCurrentTitle.fullText local link = string.format("%s", page, text) textNode:wikitext(link) cell.content :addClass(baseClass.."--section-link") :node(textNode) elseif tag.name == "SortValue" then local text = string.gsub(cell.raw, "%[Player Name%]", "Link") cell.sortValue = Sequences.sortValue(game, nil, text) local termLink = Term.link(text, game) textNode:wikitext(termLink) cell.content :addClass(baseClass.."--terms") :node(textNode) elseif tag.name == "Term" or tag.name == "Terms" then local text = string.gsub(cell.raw, "%[Player Name%]", "Link") text = string.gsub(text, NUMBER_VARIANT_REGEX, "") local pages = utilsString.split(text) local termLinks = TermList.termList(game, pages, { link = true }) local list = h.list(termLinks) if #termLinks > 1 then cell.unsortable = true end textNode:wikitext(list) cell.content :addClass(baseClass.."--terms") :node(textNode) else local text = string.gsub(cell.raw, "%[Player Name%]", h.playerName) h.warn(string.format("Unrecognized tag ", tag.name)) textNode:wikitext(text.."") cell.content:node(textNode) end return cell end

function h.list(items) return #items > 1 and utilsMarkup.list(items) or items[1] end

local frame = mw.getCurrentFrame function h.store(args, dataRows, unsortableColumns) for i, row in ipairs(dataRows) do		for j, cell in ipairs(row.cells) do			frame:expandTemplate({				title = "Data Table/Store",				args = {					game = args.game,					name = args.name,					rowIndex = i,					rowId = row.id,					columnIndex = j,					columnHeader = cell.columnHeader,					columnSortable = unsortableColumns[j] and "0" or "1",					cellContent= cell.content,					cellRaw = cell.raw,					cellSortValue = cell.sortValue,				}			}) end end end

function h.desktopTable(headerRows, dataRows, unsortableColumns) local sortableColumns = {} for i = 1, #headerRows do		if not unsortableColumns[i] then table.insert(sortableColumns, i)		end end local wikitable = utilsLayout.table({		sortable = sortableColumns,		headers = headerRows,		rows = dataRows,	}) local html = mw.html.create("div") :addClass("data-table data-table--desktop size-large-up") :wikitext(wikitable) return tostring(html) end

function h.mobileTable(columnHeaders, dataRows) local html = mw.html.create("div") :addClass("data-table data-table--mobile size-medium-down") :tag("div") :addClass("data-table__list") for i, row in ipairs(dataRows) do		local item = html:tag("table"):addClass("wikitable data-table__list-item") for j, header in ipairs(columnHeaders) do			local cell = row.cells[j] or {} item:tag("tr") :tag("th") :addClass("data-table__row-header") :wikitext(header) :done :tag("td") :addClass(cell.class) :wikitext(cell.content) :done end end return tostring(html:allDone) end

function h.warn(msg) local utilsError = require("Module:UtilsError") utilsError.warn(msg) end

p.Templates = { ["Data Table/Store"] = { purpose = "Stores data into the DataTables Cargo table, for use by .", storesData = true, usage = "This template is transcluded by Module:Data Table.", },	["Data Table"] = { format = "block", purpose = " Displays tabular data in a way that is mobile-friendly. Minimizes the amount of boilerplate wikitext that most tables require. Automatically center-aligns columns and applies templates such as, , , etc.  The table data is stored in Cargo so that the table (or a subset of its rows) can be displayed on other relevant pages using .", boilerplate = { separateRequiredParams = false, },		paramOrder = {"name", "game", "sortable", "columns", "..."}, params = { name = { type = "string", required = true, desc = "Assigns an internal name to the table so that other pages can retrieve it using .", },			game = { type = "string", suggested = true, enum = Franchise.enum, desc = "A game code. Used to automatically apply game-based templates (e.g., , ) to table cells.", },			sortable = { type = "boolean", desc = "If set to, the data table will not be sortable.", trim = true, },			columns = { type = "content", required = true, desc = "Comma-separated list of column headers. One or more tag can be appended to each column to indicate how the template should handle data in that column. See below.", trim = true, nilIfEmpty= true, split = true, },			["..."] = {				name = "cells", placeholder = "cell", required = true, type = "content", desc = "Cell values. Type  to separate rows, as shown in the examples below.", trim = true, }		}	},	["Data Table Copy"] = { purpose = "Creates a copy of a located on another page.", format = "block", paramOrder = {"fromPage", "name", "columns", "excludeColumns", "rowsWith"}, params = { fromPage = { required = true, type = "wiki-page-name", desc = "The name of the wiki page containing the data table to be copied.", trim = true, nilIfEmpty = true, },			name = { required = true, type = "string", desc = "The internal name of the table to be copied, as specified by the  parameter of Template:Data Table.", trim = true, nilIfEmpty = true, },			columns = { type = "string", desc = "A comma-separated list of columns to copy. If absent, all columns are copied.", trim = true, nilIfEmpty = true, split = true, },			excludeColumns = { type = "string", desc = "A comma-separated list of columns not to copy. If absent, all columns are copied. Ignored if  is present.", trim = true, nilIfEmpty = true, split = true, },			rowsWith = { type = "string", desc = "If specified, only rows containing the given string will be copied. If absent, all rows are copied.", trim = true, nilIfEmpty = true, },		},	}, }

return p