Module:Infobox
Jump to navigation
Jump to search
Documentation for this module may be created at Module:Infobox/doc
local counter
local h = {}
local p = {}
function p.arraymap(frame)
-- a lua implementation of Page Forms' arraymap
local args = h.overwrite()
local items = h.split(args[1], args[2] or ',')
for i, item in ipairs(items) do
items[i] = args[4]:gsub(args[3], item)
end
return table.concat(items, args[6] or ',')
end
function p.preprocess(frame)
return frame:preprocess(frame.args[1] or frame:getParent().args[1])
end
function p.main(frame)
h.increment()
local args = h.overwrite()
local sep = args.sep or ','
h.castArgs(args, sep)
return h.makeInfobox(args, sep)
end
function h.increment()
counter = mw.getCurrentFrame():callParserFunction('#var', {'DRUID_INFOBOX_ID'}) or 1
mw.getCurrentFrame():callParserFunction('#vardefine', {'DRUID_INFOBOX_ID', counter})
end
function h.castArgs(args, sep)
if args.images and not args.image then
args.image = args.images
end
args.image_labels = h.split(args.image_labels, sep)
end
function h.makeInfobox(args, sep)
local out = mw.html.create('table')
:addClass('druid-container')
:attr('id', 'druid-' .. counter)
if args.kind then out:addClass('druid-container-' .. h.escape(args.kind)) end
if args.title then
out:tag('tr')
:tag('th')
:addClass('druid-title')
:attr('colspan', 2)
:wikitext(args.title)
end
h.printImages(out, h.split(args.image, sep), args)
for _, section in ipairs(h.split(args.sections, sep)) do
-- cannot begin tagging here because we don't know if any applicable args are present
local cols = args[section .. '_columns']
if cols then
out:node(h.makeGridSection(
section, h.split(args[section], sep), tonumber(cols), args
))
else
out:node(h.makeSection(section, h.split(args[section], sep), args))
end
end
return out
end
function h.printImages(out, images, args)
if #images == 0 then return end
local labels = args.image_labels
-- burden is on the user to format this as an image. this should be done in the infobox template,
-- with something like |image={{#if:{{{image|}}}|[[File:{{{image|}}}{{!}}300px{{!}}link=]]}}
local td = out:tag('tr')
:tag('td')
:attr('colspan', 2)
if #images == 1 then
td:addClass('druid-main-image')
:wikitext(images[1])
return
end
td:addClass('druid-main-images')
local labelsContainer = td:tag('div')
:addClass('druid-main-images-labels')
local imagesContainer = td:tag('div')
:addClass('druid-main-images-files')
for i, item in ipairs(images) do
local labelText = labels[i] or ('[[Category:Infoboxes missing image labels]]Image ' .. i)
local label = labelsContainer:tag('div')
:addClass('druid-main-images-label')
:attr('data-druid', counter .. '-' .. i)
:wikitext(labelText)
local container = imagesContainer:tag('div')
:addClass('druid-main-images-file')
:attr('data-druid', counter .. '-' .. i)
:wikitext(item)
if args[labelText .. '_caption'] then
container:tag('div')
:addClass('druid-main-images-caption')
:wikitext(args[labelText .. '_caption'])
end
if i == 1 then
label:addClass('focused')
container:addClass('focused')
end
end
end
function h.makeGridSection(section, sectionFields, cols, args)
local shouldPrint = false
local node = mw.html.create()
h.printSectionHeader(node, section, args)
local grid = node:tag('tr'):tag('td')
:attr('colspan', 2)
:addClass('druid-grid-section')
:addClass('druid-grid-section-' .. h.escape(section))
:tag('div')
:addClass('druid-grid')
:css('grid-template-columns', ('repeat(%s, 1fr)'):format(cols))
local row = 1
local col = 1
local itemContainer
for _, item in ipairs(sectionFields) do
if args[item] then
shouldPrint = true
itemContainer = grid:tag('div')
:addClass('druid-grid-item')
:addClass('druid-grid-item-' .. h.escape(item))
:css('grid-column', col)
:css('grid-row', row)
h.printLabel(itemContainer:tag('div'), item, args)
h.printData(itemContainer:tag('div'), item, args)
if col == cols then
row = row + 1
col = 1
else
col = col + 1
end
end
end
if not shouldPrint then return nil end
if cols ~= 1 then
-- this will overwrite the existing `grid-column` value for each
itemContainer:css('grid-column', ('%s / -1'):format(col - 1))
end
return node
end
function h.makeSection(section, sectionFields, args)
local shouldPrint = false
local node = mw.html.create()
h.printSectionHeader(node, section, args)
for _, item in ipairs(sectionFields) do
if h.shouldPrint(item, args) then
shouldPrint = true
local tr = node:tag('tr')
:addClass('druid-row')
:addClass('druid-row-' .. h.escape(item))
if args[item .. '_wide'] or args[item .. '_nolabel'] then
local td = h.printData(tr:tag('td'), item, args)
td
:attr('colspan', 2)
:addClass('druid-data-wide')
else
h.printLabel(tr:tag('th'), item, args)
h.printData(tr:tag('td'), item, args)
end
end
end
if not shouldPrint then return nil end
return node
end
function h.shouldPrint(item, args)
if args[item] then return true end
for _, key in ipairs(args.image_labels) do
if args[key .. '_' .. item] then
return true
end
end
return false
end
function h.printLabel(node, item, args)
return node
:addClass('druid-label')
:addClass('druid-label-' .. h.escape(item))
:wikitext(args[item .. '_display'] or item)
end
function h.printData(node, item, args)
if not args.image_labels or #args.image_labels == 0 then
h.printSimpleData(node, item, args)
return node
end
if not h.hasComplexData(item, args) then
h.printSimpleData(node, item, args)
return node
end
for i, v in ipairs(args.image_labels) do
local div = node:tag('div')
:addClass('druid-toggleable-data')
:attr('data-druid', counter .. '-' .. i)
:wikitext(args[v .. '_' .. item] or args[item])
if i == 1 then div:addClass('focused') end
end
return node
end
function h.printSimpleData(node, item, args)
node:addClass('druid-data')
:addClass('druid-data-' .. h.escape(item))
:wikitext(args[item])
end
function h.hasComplexData(item, args)
for _, v in ipairs(args.image_labels) do
if args[v .. '_' .. item] then return true end
end
return false
end
function h.printSectionHeader(node, section, args)
if args[section .. '_nolabel'] then return end
node:tag('tr')
:tag('th')
:attr('colspan', 2)
:wikitext(section)
:addClass('druid-section')
:addClass('druid-section-' .. h.escape(section))
end
function h.overwrite()
-- this is a generic utility function that collects args from the invoke call & the parent template.
-- normally, you merge args with parent template overwriting the invoke call, but
-- since we'll be putting markup/formatting into our invoke call,
-- we actually want to overwrite what the user sent.
local f = mw.getCurrentFrame()
local origArgs = f.args
local parentArgs = f:getParent().args
local args = {}
for k, v in pairs(parentArgs) do
v = mw.text.trim(v)
if v ~= '' then
args[k] = v
end
end
for k, v in pairs(origArgs) do
v = mw.text.trim(tostring(v))
if v ~= '' then
args[k] = v
end
end
return args
end
-- generic utility functions
-- these would normally be provided by other modules, but to make installation easy
-- I'm including everything here
function h.split(text, pattern, plain)
if not text then
return {}
end
local ret = {}
for m in h.gsplit(text, pattern, plain) do
ret[#ret+1] = m
end
return ret
end
function h.gsplit( text, pattern, plain )
if not pattern then pattern = ',' end
if not plain then
pattern = '%s*' .. pattern .. '%s*'
end
local s, l = 1, text:len()
return function ()
if s then
local e, n = text:find( pattern, s, plain )
local ret
if not e then
ret = text:sub( s )
s = nil
elseif n < e then
-- Empty separator!
ret = text:sub( s, e )
if e < l then
s = e + 1
else
s = nil
end
else
ret = e > s and text:sub( s, e - 1 ) or ''
s = n + 1
end
return ret
end
end, nil, nil
end
function h.escape(s)
s = s:gsub(' ', '')
:gsub('"', '')
:gsub("'", '')
:gsub("%?", '')
:gsub("%%", '')
:gsub("%[", '')
:gsub("%]", '')
:gsub("{", '')
:gsub("}", '')
:gsub("!", '')
return s
end
return p