Modulo:StagioniTV/sandbox

require('Modulo:No globals')

local wikidata = require('Modulo:Wikidata')
local navbox = require('Modulo:Navbox')
local mNumeroRomano = require('Modulo:Numero romano')
local getArgs = require('Modulo:Arguments').getArgs
local errorCategory = '[[Categoria:Errori di compilazione del template Stagioni televisive]]'
local p = {}

local function ordinalefemminile(num)
	return mw.getCurrentFrame():expandTemplate {
		title = 'Ordinale femminile',
		args = { num }
	}
end

function p.prova(frame)
	local args = getArgs(frame)
	return p._prova(args)
end

function p._prova(args)
	local template, sfondo, categoria, target, stagioneoedizione, title, preposizione, links, errore
	local id, localID, tipo, titolo, etichetta, sitelink, stagioni
	local pageTitle = args[1] or args.titolo

-- Rileva l'identificativo della scheda wikidata collegata alla fiction o al programma TV,
-- se nessun titolo è specificato prova a recuperlo dalla voce della stagione/edizione, se questa ha una sua scheda su wikidata	
	if not args[1] or args.titolo then
		if string.find(mw.title.getCurrentTitle().text, '^Episodi d[ei]') then
		localID = wikidata._getProperty({ 'P361', n = 1, from = mw.wikibase.getEntityIdForCurrentPage(), formatting = 'raw' })
		end
			if localID ~= nil then
			id = localID
			end
	else id = mw.wikibase.getEntityIdForTitle( pageTitle )
	end

-- Identifica il tipo di programma (serie televisiva, serie animata o programma TV generico)
-- ed estrae gli altri dati fondamentali da Wikidata (etichetta in italiano, numero stagioni, sitelink)
-- se uno di essi non è presente è generato un messaggio di errore
	if wikidata._instanceOf({ 'Q581714', from = id }) then tipo = 'animata'
	elseif wikidata._instanceOf({ 'Q5398426', from = id }) then tipo = 'serie TV'
	else tipo = 'programma'
	end
	if id then
	titolo = wikidata._getLink({ id })
	etichetta = wikidata._getLabel({ id, 'it'})
	sitelink = mw.wikibase.sitelink( id )
	stagioni = wikidata._getProperty({ 'P2437', n = 1, from = id, formatting = 'raw' })
	else errore = 'Template non compilato correttamente, controlla le [[Template:Stagioni televisive|istruzioni]]'
	end

if not etichetta or not sitelink or not stagioni then
	errore = 'Template non compilato correttamente, controlla le [[Template:Stagioni televisive|istruzioni]]'
else

-- Questa parte del codice genere i collegamenti alle singole edizioni/stagioni e la relativa categoria
	local tbl = {}
	local lang = mw.language.getContentLanguage()
	local sep = args.debug and '<br />' or '&nbsp;<b>&middot;</b>&#32;'
	
	if tipo == 'animata' then sfondo = 'background:#ab110c;'
	elseif tipo == 'serie TV' then sfondo = 'background:#115806;'
	elseif tipo == 'programma' then sfondo = 'background:#161e8e;'
	end
	
	if tipo == 'programma' then
	title = mw.title.new(string.format('Categoria:Edizioni de %s', etichetta))
	preposizione = (title and title.exists) and 'de' or 'di'
	else title = mw.title.new(string.format('Episodi de %s (prima stagione)', etichetta))
	preposizione = (title and title.exists) and 'de' or 'di'
	end
	
	for i = 1, stagioni do
		local ordinaleF = ordinalefemminile(i)
		
		if tipo == 'programma' then 
		target = string.format('%s (%s edizione)', etichetta, ordinaleF) else
		target = string.format('Episodi %s %s (%s stagione)', preposizione, etichetta, ordinaleF)
		end
		if tipo == 'programma' then stagioneoedizione = 'edizione' else stagioneoedizione = 'stagione' end
		local label = args.formato == 'numeri romani' and mNumeroRomano.main({ i }) or
					  (args.formato == 'forma breve' and string.format('%dª', i) or
					  string.format('%s %s', lang:ucfirst(ordinaleF), stagioneoedizione))
		table.insert(tbl, string.format('[[%s|%s]]', target, label))
		
		if (target == mw.title.getCurrentTitle().text and mw.title.getCurrentTitle().namespace == 0) or args.debug then
			if stagioni == '1' or stagioni == '1±0' then
				if tipo == 'programma' then
				categoria = string.format('[[Categoria:Edizioni di programmi televisivi|%s]]', etichetta) else
				categoria = string.format('[[Categoria:Liste di episodi di serie televisive|%s]]', etichetta)
				end
		else
				if tipo == 'programma' then
				categoria = string.format('[[Categoria:Edizioni %s %s|%02d]]', preposizione, etichetta, i) else
				categoria = string.format('[[Categoria:Stagioni %s %s|%02d]]', preposizione, etichetta, i)
				end
			end
		end
	end
	links = mw.text.listToText(tbl, sep, sep) .. (categoria or '')
end

-- Infine è generato il navbox con i dati su estratti e formattati
	if not id or not etichetta or not stagioni then
		template = mw.title.getCurrentTitle().namespace == 0 and errorCategory or nil
	elseif args[2] == 'elenco'  then template = links
		elseif args[3] == 'elenco'  then template = links
		elseif args[4] == 'elenco'  then template = links
	elseif stagioni == '1' or stagioni == '1±0' then template = categoria
	else template = navbox._navbox({
			name = 'Stagioni televisive',
			title = string.format("[[%s|<span style=\"color:white;\">''%s''</span>]]", sitelink, etichetta),
			titlestyle = sfondo,
			list1 = links
		})
	end

	return template, errore
end


return p