From StrategyWiki, the video game walkthrough and strategy guide wiki
Jump to navigation Jump to search

Documentation for this module may be created at Module:Game/Documentation

local getArgs = require('Module:Arguments').getArgs

local p = {}

local args = {}
local root
local infobox

local function getArgNums(prefix)
	-- Returns a table containing the numbers of the arguments that exist
	-- for the specified prefix. For example, if the prefix was 'data', and
	-- 'data1', 'data2', and 'data5' exist, it would return {1, 2, 5}.
	local nums = {}
	for k, v in pairs(args) do
		local num = tostring(k):match('^' .. prefix .. '([1-9]%d*)$')
		if num then table.insert(nums, tonumber(num)) end
	end
	table.sort(nums)
	return nums
end

local function addRow( rowArgs )
    -- Adds a row to the infobox, with either a header cell
    -- or a label/data cell combination.
    if rowArgs.header then
        root
            :tag( 'tr' )
                :addClass( rowArgs.rowclass )
                :tag( 'th' )
                    :attr( 'colspan', 2 )
                    :css( 'text-align', 'center' )
                    :cssText( rowArgs.headerstyle )
                    :wikitext( rowArgs.header )
    else
        local row = root:tag( 'tr' ):addClass( rowArgs.rowclass )

        if rowArgs.label then
            row
                :tag( 'th' )
                    :cssText( rowArgs.labelstyle )
                    :wikitext( rowArgs.label )
                    :done()
        end

        local dataCell = row:tag( 'td' )
        if not rowArgs.label then
            dataCell
                :attr( 'colspan', 2 )
                :cssText( rowArgs.rowstyles or 'text-align: center;' )
        end
        dataCell
            :wikitext( rowArgs.data )
    end
end

local function renderTitle()
    args.title = args.title or mw.title.getCurrentTitle().text

    root
        :tag( 'tr' )
            :tag( 'th' )
                :attr( 'colspan', 2 )
                :addClass( 'infobox_heading' )
                :wikitext( args.title )
end

local function renderImage()
    if not args.image then
        root:wikitext( '[[Category:Pages needing box artwork]]' )
        return
    end

    local data = mw.html.create():wikitext( '[[File:' .. args.image .. '|250px|Box artwork for ' .. args.title .. '.]]' )

    addRow( {
        data = tostring( data ),
        rowclass = 'mw-row-no-collapse',
    } )
end

local function renderRows()
    if args.japanese then addRow( {label = 'Japanese title', data = args.japanese} ) end
    addRow( { label = 'Developer(s)', data = args.developer, rowclass = 'mw-row-no-collapse' } )
    addRow( { label = 'Publisher(s)', data = args.publisher } )
    if args.distributor then addRow( {label = 'Distributor(s)', data = args.distributor} ) end
    if args.designer then addRow( {label = 'Designer(s)', data = args.designer} ) end
    if args.version then addRow( {label = 'Latest version', data = args.version} ) end
    addRow( { label = 'Release date(s)', labelstyle = 'white-space: nowrap;', data = args.released } )
    addRow( { label = 'Genre(s)', data = args.genre, rowclass = 'mw-row-no-collapse' } )
    addRow( { label = 'System(s)', data = args.systems, rowclass = 'mw-row-no-collapse' } )
    if args.players then addRow( {label = 'Players', data = args.players} ) end
    if args.modes then addRow( {label = 'Modes', data = args.modes} ) end
    if args.ratings then addRow( { label = 'Rating(s)', data = args.ratings, rowclass = 'mw-row-no-collapse' } ) end
    if args.input then addRow( {label = 'Input', data = args.input} ) end
    if args.expansion then addRow( { label = 'Expansion pack(s)', labelstyle = 'white-space: nowrap;', data = args.expansion, rowclass = 'mw-row-no-collapse' } ) end
    if args['preceded by'] then addRow( { label = 'Preceded by', labelstyle = 'white-space: nowrap;', data = "''" .. args['preceded by'] .. "''" } ) end
    if args['followed by'] then addRow( { label = 'Followed by', labelstyle = 'white-space: nowrap;', data = "''" .. args['followed by'] .. "''" } ) end
    if args.series then
        local data2 = ''
    	if args.series2 then data2 = '<br>and<br>[[:Category:' .. args.series2 .. '|' .. args.series2 .. ']]' end
        addRow( { label = 'Series', data = '[[:Category:' .. args.series .. '|' .. args.series .. ']]' .. data2, rowclass = 'mw-row-no-collapse' } )
    end
    if args.website then addRow( {label = 'Website', data = args.website} ) end
end

local function renderRowsNew()
    if args.japanese then addRow( {label = 'Japanese title', data = args.japanese} ) end
    if args.designer then addRow( {label = 'Designer(s)', data = args.designer} ) end
    addRow( { label = 'Genre(s)', data = args.genre, rowclass = 'mw-row-no-collapse' } )
    if args.players then addRow( {label = 'Players', data = args.players} ) end
    if args.modes then addRow( {label = 'Modes', data = args.modes} ) end
    if args.input then addRow( {label = 'Input', data = args.input} ) end
    if args.expansion then addRow( { label = 'Expansion pack(s)', labelstyle = 'white-space: nowrap;', data = args.expansion, rowclass = 'mw-row-no-collapse' } ) end
    if args['preceded by'] then addRow( { label = 'Preceded by', labelstyle = 'white-space: nowrap;', data = "''" .. args['preceded by'] .. "''" } ) end
    if args['followed by'] then addRow( { label = 'Followed by', labelstyle = 'white-space: nowrap;', data = "''" .. args['followed by'] .. "''" } ) end
    if args.series then
        local data2 = ''
    	if args.series2 then data2 = '<br>and<br>[[:Category:' .. args.series2 .. '|' .. args.series2 .. ']]' end
        addRow( { label = 'Series', data = '[[:Category:' .. args.series .. '|' .. args.series .. ']]' .. data2, rowclass = 'mw-row-no-collapse' } )
    end
    if args.website then addRow( {label = 'Website', data = args.website} ) end
end

local function renderRelease( listnum )
    if args[ 'title' .. listnum ] then 
        root
            :tag( 'tr' )
                :tag( 'th' )
                    :attr( 'colspan', 2 )
                    :addClass( 'infobox_heading' )
                    :wikitext( args[ 'title' .. listnum ] )
    end

    if args[ 'japanese' .. listnum ] then addRow( {label = 'Japanese title', data = args[ 'japanese' .. listnum ]} ) end
    addRow( { label = 'Developer(s)', data = args[ 'developer' .. listnum ], rowclass = 'mw-row-no-collapse' } )
    addRow( { label = 'Publisher(s)', data = args[ 'publisher' .. listnum ] } )
    if args[ 'version' .. listnum ] then addRow( {label = 'Latest version', data = args[ 'version' .. listnum ]} ) end
    addRow( { label = 'Year released', data = '[[:Category:' .. args[ 'year' .. listnum ] .. '|' .. args[ 'year' .. listnum ] .. ']]', rowclass = 'mw-row-no-collapse' } )
    root:wikitext( '[[Category:' .. args[ 'year' .. listnum ] .. ']]' )
    addRow( { label = 'System(s)', data = args[ 'systems' .. listnum ] } )
    if args[ 'ratings' .. listnum ] then addRow( { label = 'Rating(s)', data = args[ 'ratings' .. listnum ] } ) end
    if args[ 'input' .. listnum ] then addRow( {label = 'Input', data = args[ 'input' .. listnum ]} ) end
end

local function renderSocialRows()
    local twitterData, facebookData, twitchData
    local pagename = mw.uri.encode( mw.title.getCurrentTitle().text, "PATH" )

    if args.twitch then
        twitchData = '[https://www.twitch.tv/directory/game/' .. mw.uri.encode( args.twitch, "PATH" ) .. ' ' .. args.twitch .. ' Channel]'
    else
        twitchData = '[https://www.twitch.tv/directory/game/' .. pagename .. ' ' .. mw.title.getCurrentTitle().text .. ' Channel]'
    end

    if args.twitter then
        twitterData = '[https://twitter.com/' .. args.twitter .. ' ' .. args.twitter .. ']'
    else
        twitterData = '[https://twitter.com/search?f=user&q=' .. pagename .. ' Search]'
    end

    if args.facebook then
        facebookData = '[https://www.facebook.com/' .. args.facebook .. ' ' .. args.facebook .. ']'
    else
        facebookData = '[https://www.facebook.com/search/pages/?q=' .. pagename .. ' Search]'
    end

    addRow( { rowclass = 'mw-row-no-collapse', label = 'Twitch', data = twitchData } )
    addRow( { rowclass = 'mw-row-no-collapse', label = 'Twitter', data = twitterData } )
    addRow( { rowclass = 'mw-row-no-collapse', label = 'Facebook', data = facebookData } )
end

local function renderTrackingCategories()
    root:wikitext( '[[Category:Games]]' )

    if not args.developer or not args.publisher or not args.released or not args.genre or not args.systems then
        root:wikitext( '[[Category:Pages needing an infobox]]' )
    end

    if args.series then
        root:wikitext( '[[Category:' .. args.series .. ']]' )
    end
    if args.series2 then
        root:wikitext( '[[Category:' .. args.series2 .. ']]' )
    end
end

local function _game()
    root = mw.html.create( 'table' )

    root
        :addClass( 'infobox' )
        :addClass( 'bordered' )
        :addClass( 'mw-collapsible' )

    renderTitle()
    renderImage()

    if args.year then 
        renderRowsNew()

        infobox = root

        root = mw.html.create( 'table' )
    
        root
            :addClass( 'infobox' )
            :addClass( 'bordered' )
            :addClass( 'mw-collapsible' )
            :addClass( 'mw-collapsed' )
    	    :css( 'margin-top', '-0.5em' )

        renderRelease( "" )
        infobox:node( root )
    else
        root:addClass( 'mw-collapsed' )

        renderRows()
        infobox = root
    end

    local releasenums = getArgNums( 'title' )
    for k, num in ipairs( releasenums ) do
        root = mw.html.create( 'table' )
    
        root
            :addClass( 'infobox' )
            :addClass( 'bordered' )
            :addClass( 'mw-collapsible' )
            :addClass( 'mw-collapsed' )
    	    :css( 'margin-top', '-0.5em' )

        renderRelease( num )
        infobox:node( root )
    end

    root = mw.html.create( 'table' )
    
    root
        :addClass( 'infobox' )
        :addClass( 'bordered' )
    	:css( 'margin-top', '-0.5em' )

    if args.pcgamingwiki then
        local data1 = '<ul><li>[[pcgamingwiki:' .. args.pcgamingwiki .. '|' .. mw.title.getCurrentTitle().text .. ' at PCGamingWiki]]</li></ul>'
        addRow( { header = 'PCGamingWiki', headerstyle = 'background-color: #c8c8c8;', rowclass = 'mw-row-no-collapse' } )
        addRow( { data = data1, rowclass = 'mw-row-no-collapse', rowstyles = 'text-align: left;' } )
    end
    root:wikitext( mw.getCurrentFrame():preprocess( '<NeoLink />' ) )
    root:wikitext( mw.getCurrentFrame():preprocess( '<SwitcherLink />' ) )

    renderSocialRows()
    renderTrackingCategories()

    infobox:node( root )
    root = infobox

    return tostring(root)
end

function p.game( frame )
    args = getArgs( frame, { wrappers = 'Template:Game' } )

    return _game()
end

return p