Module:Infobox team

From TwogPedia
Revision as of 20:51, 22 October 2023 by Couchor (talk | contribs)

Documentation for this module may be created at Module:Infobox team/doc

local getArgs = require('Module:Arguments').getArgs
local Links = require('Module:Links')
local Header = require('Module:Infobox/Widget/Header')
local Subheader = require('Module:Infobox/Widget/Subheader')
local LinksWidget = require('Module:Infobox/Widget/Links')
local SponsorsWidget = require('Module:Infobox/Widget/Sponsors')
local LocationWidget = require('Module:Infobox/Widget/Location')
local RowWidget = require('Module:Infobox/Widget/Row')
local stringifyDate = require('Module:Functions').stringifyDate
local cargo = mw.ext.cargo
VariablesLua = mw.ext.VariablesLua

local Flags = require('Module:Flags')
local Infobox = {}

function Infobox.main(frame)
	local args = getArgs(frame)
	local currentTitle = mw.title.getCurrentTitle().text
	local currentTitleSplit = mw.text.split(currentTitle, '/')
	
	local headerNode
	
	-- Just have to access arg.logos for VariablesLua values to get initialized
	if args.logos then end
	
	local logoLight = VariablesLua.varexists('logoLight') and VariablesLua.var('logoLight') or nil
	local logoDark = VariablesLua.varexists('logoDark') and VariablesLua.var('logoDark') or nil
	local logoAll = VariablesLua.varexists('logoAll') and VariablesLua.var('logoAll') or nil

	if logoAll then
		headerNode = Header.make(args.title or currentTitle, nil, nil, logoAll)
	elseif logoLight or logoDark then
		headerNode = Header.make(args.title or currentTitle, logoLight, logoDark)
	elseif args.org then
		local currentDate = os.date('%Y-%m-%d')
		local tables = 'Logos'
		local fields = '_pageName, logoLight, logoDark, logoAll, start, end'
		local cargoArgs = {
			where = '_pageName="Companies/' .. args.org .. '" AND  (start is NULL OR start < "' .. currentDate .. '") AND (end is NULL OR end > "' .. currentDate .. '")'
		}
		local results = cargo.query(tables, fields, cargoArgs)

		if #results > 0 then
			if ( results[1].logoAll ) then
				headerNode = Header.make(args.title or currentTitle, nil, nil, results[1].logoAll)
			else 
				headerNode = Header.make(args.title or currentTitle, results[1].logoLight, results[1].logoDark)
			end
		end
	end
	
	local container = mw.html.create('div')
		:addClass('ib')
		
	container:node(headerNode):node(Subheader.make('Team Information'))
	
	if args.roster then
		container:node(RowWidget.make('Roster', LocationWidget.make(args.roster, args.city, currentTitleSplit[1] .. '/Teams')))
	end
	
	if args.region then
		container:node(RowWidget.make('Region', LocationWidget.make(args.region, nil, currentTitleSplit[1] .. '/Teams')))
	end
	
	-- Look up Cargo table for captain, coach and manager
	local tables = 'Squads'
	local fields = '_pageName, captain, coach, manager, analyst'
	local cargoArgs = {
		where = '_pageName = "' .. mw.title.getCurrentTitle().text .. '"'
	}
	local results = cargo.query(tables, fields, cargoArgs)
	
	if #results > 0 then
		-- For each position, check if the player is in database and has flag info
		tables = 'People'
		fields = '_pageName, nationality, name, romanized'
		if ( results[1].captain ) then
            container:node(Infobox.personRow(results[1].captain, 'Captain', tables, fields, cargoArgs))
		end
		if ( results[1].coach ) then
            container:node(Infobox.personRow(results[1].coach, 'Coach', tables, fields, cargoArgs))
		end
		if ( results[1].manager ) then
			container:node(Infobox.personRow(results[1].manager, 'Manager', tables, fields, cargoArgs))
		end
        if ( results[1].analyst) then
			container:node(Infobox.personRow(results[1].analyst, 'Analyst', tables, fields, cargoArgs))
		end
	end
	
	if args.founded then
		container:node(RowWidget.make('Founded', stringifyDate(args.founded)))
	end
	
	if args.org then
		container:node(RowWidget.make('Organization', 'Companies/' .. args.org, frame))
		local tables = 'Sponsors'
		local fields = 'id, startRef, startRefName, endRef, endRefName'
		local cargoArgs = {
			where = '_pageName="Companies/' ..args.org .. '"'
		}
		local results = cargo.query(tables, fields, cargoArgs)
		if #results > 0 then
	    	container:node(RowWidget.make('Sponsors', SponsorsWidget.make(results, frame)))
		end
		
	end
	
	if args.sponsors and args.org == nil then
		container:node(RowWidget.make('Sponsors', SponsorsWidget.make(args.sponsors, frame) ))
	end
	
	if args.gamesponsors then
		container:node(RowWidget.make('Sponsors', SponsorsWidget.make(args.gamesponsors, frame) ))
	end
	
	if args.coach then
		container:node(RowWidget.make('Coach', args.coach, frame))
	end
	
	if args.manager then
		container:node(RowWidget.make('Manager', args.manager, frame))
	end
	
	if args.captain then
		container:node(RowWidget.make('Captain', args.captain, frame))
	end
	
	-- Check if there is a "links" argument and then do magic to add them all to arguments
	if args.links then
		local splitLinks = mw.text.split(args.links, ',')
		for i = 1, #splitLinks do
			local split = mw.text.split(splitLinks[i], '=')
			args[split[1]] = split[2]
		end
	end

    -- Add dota2 api links if api id added
    if args.api_id then
        args.datdota = args.api_id
        args.dotabuff = args.api_id
        args.stratz = args.api_id
    end
	
	local transformedLinks = Links.transform(args)
	local fullLinks = LinksWidget.make(transformedLinks, 'team')
	
	if fullLinks then
		container:node(Subheader.make('Links'))
		container:node(fullLinks)
	end
	
	return container
end

function Infobox.personRow(resultStr, rowName, tables, fields, cargoArgs)
	local stringSplit = mw.text.split(resultStr, ',')
	local personDiv = mw.html.create('div')
	local flag = nil
	for j = 1, #stringSplit do
		local singleDiv = mw.html.create('div'):attr('style', 'display: flex; gap: 0.5rem;')
		cargoArgs.where = 	'_pageName="' .. stringSplit[j] .. '"'
		local personResults = cargo.query(tables, fields, cargoArgs)
		if #personResults > 0 and personResults[1].nationality then
			flag = Flags.icon(personResults[1].nationality)
		end
		personDiv:node(singleDiv:wikitext(flag):wikitext('[[' .. stringSplit[j] .. '|' .. string.gsub( stringSplit[j], 'People/' , '') .. ']]'))
	end   
    return RowWidget.make(rowName, personDiv)      
end

return Infobox