模組:Autovirusbox/sandbox

require('strict')
local ItalicTitle = require('Module:Italic title')
local p = {} -- functions made public
local l = {} -- internal functions, kept separate

-- =============================================================================
-- main implements Template:Virusbox; see the documentation of that template
-- for details.
-- =============================================================================

function p.main(frame)
	local args
	if frame.args['direct'] == 'yes' then args = frame.args
	else args = frame:getParent().args end
	-- ---------------------------------------------------------------------
	-- pick up taxobox parameters from the caller that need to be processed;
	-- most are passed on unchanged
	-- ---------------------------------------------------------------------
	local name = args['name'] or ''
	local taxon = args['taxon'] or ''
	local parent = args['parent'] or ''
	local species = args['species'] or '' 
	local strain = args['strain'] or '' 
	local serotype = args['serotype'] or ''
	local virus = args['virus'] or ''
	local displayParents = args['display_parents'] or args['display parents'] or '1'
	local authority = args['authority'] or ''
	local parentAuthority = args['parent_authority'] or args['parent authority'] or ''
	local gparentAuthority = args['grandparent_authority'] or args['grandparent authority'] or ''
	local ggparentAuthority = args['greatgrandparent_authority'] or args['greatgrandparent authority'] or ''
	local gggparentAuthority = args['greatgreatgrandparent_authority'] or args['greatgreatgrandparent authority'] or ''
	local typeGenusAuthority = args['type_genus_authority'] or args['type genus authority'] or ''
	local subdivision = args['subdivision'] or ''
	local subdivisionRanks = args['subdivision_ranks'] or args['subdivision ranks'] or ''
	local specieslink = args['species link'] or args['species_link'] or frame:expandTemplate{ title = 'Infraspeciesbox/search wikidata', args = { 'Q7432' ,  args['from'] or '', }, } or ''  -- same as {{Infraspeciesbox/search wikidata|Q7432|{{{from|}}}}}
	local extinct = args['extinct'] or ''
	-- ------------------------------------------------------
	-- set the taxobox parameters determined by this function
	-- ------------------------------------------------------
	local autoTaxon, autoTaxonType, infraTaxon, infraTaxonRank, targetTaxon, targetTaxonRank = l.paramChk(frame, taxon, parent, species, strain, serotype, virus)
	-- set default taxobox name/title
	local italicsRequired = frame:expandTemplate{ title = 'Is italic taxon', args = {targetTaxonRank, virus='yes'} } == 'yes'
	-- the page name (title) should be italicized if it's the same as the target taxon and that is italicized
	local currentPage = mw.title.getCurrentTitle()
	local pagename = currentPage.text
	if pagename == targetTaxon then
		if italicsRequired then ItalicTitle._main({}) end
	end
	if name == '' then
		if autoTaxonType == 'ERROR' then
			name = '<span class="error">ERROR: parameter(s) specifying taxon are incorrect; see [[Template:Virusbox/doc#Usage|documentation]]</span>'
		else
			name = pagename
			if italicsRequired and pagename == targetTaxon then
				name = "''" .. targetTaxon .. "''"
			end
		end
	end
	-- is the auto-taxon name bold or linked (i.e. will it be the last row in the taxobox or not)?
	local boldFirst = 'bold' 
	if autoTaxonType == 'PARENT' then boldFirst = 'link' end
	-- italicize and link species name, or embolden if nothing below
	if species ~= '' then
		if infraTaxon ~= '' then
			if specieslink ~= '' then
				species = "[["..specieslink.."]] ''"..species.."''"
			else
				species = "''[["..species.."]]''"
			end	
		else
			species = "'''"..name.." ''"..species.."'''''" or "'''"..pagename.." ''"..species.."'''''"
			if (extinct == 'yes' or extinct == 'true') and not string.find(species, '†', 1, true) then
				species = '<span style="font-style:normal;font-weight:normal;">†</span>' .. species
			end
			if string.sub(taxon, -2) == '/?' and not string.find(link, '?', 1, true) then
				species = species .. '<span style="font-style:normal;font-weight:normal;"> (?)</span>'
			end
		end
		species = '</span>'..species..'<span class="nowrap">'
	end
	-- embolden lowest rank
	if infraTaxon ~= '' then
		infraTaxon = "'''"..name.." ''"..infraTaxon.."'''''" or "'''"..pagename.." ''"..infraTaxon.."'''''"
		if (extinct == 'yes' or extinct == 'true') and not string.find(infraTaxon, '†', 1, true) then
			infraTaxon = '<span style="font-style:normal;font-weight:normal;">†</span>' .. infraTaxon
		end
		if string.sub(taxon, -2) == '/?' and not string.find(link, '?', 1, true) then
			infraTaxon = infraTaxon .. '<span style="font-style:normal;font-weight:normal;"> (?)</span>'
		end
		infraTaxon = '</span>'..infraTaxon..'<span class="nowrap">'
	end
	-- set offset and fix display_parents if there are ranks below autoTaxon
	local offset = 0
	if infraTaxon ~= '' then offset = offset + 1 end
	if species ~= '' then offset = offset + 1 end
	if offset ~= 0 then
		displayParents = tostring(tonumber(displayParents) - offset)
	end
	-- fill in a missing subdivision_ranks parameter
	if subdivision ~= '' and subdivisionRanks == '' then
		subdivisionRanks =  frame:expandTemplate{ title = 'Children rank', args = {targetTaxonRank} }
	end
	-- ------------------------------------------------
	-- now call Taxobox/core with all of its parameters
	-- ------------------------------------------------
	local res = frame:expandTemplate{ title = 'Taxobox/core', args =
		{ ['edit link'] = 'e',
		  virus = 'yes',
		  colour = frame:expandTemplate{ title = 'Taxobox colour', args = { 'virus' } },
		  name = name,
		  parent = autoTaxon,
		  bold_first = boldFirst,
		  authority = authority,
		  parent_authority = parentAuthority,
		  grandparent_authority = gparentAuthority,
		  greatgrandparent_authority = ggparentAuthority,
		  greatgreatgrandparent_authority = gggparentAuthority,
		  offset = tostring(offset),		 
		  image = args['image'] or frame:expandTemplate{ title = 'Wikidata', args = { 'property' , 'raw' ,  args['from'] or '',  'P18',  }, },   -- equivalent to {{wikidata|property|raw|{{{from|}}}|P18}}
		  image_upright = args['image_upright'] or args['image upright'] or '',
		  image_alt = args['image_alt'] or args['image alt'] or '',
		  image_caption = args['image_caption'] or args['image caption'] or frame:expandTemplate{ title = 'Wikidata', args = { 'qualifier',  args['from'] or '',  'P18', 'P2096'  }, },  --equivalent to {{wikidata|qualifier|{{{from|}}}|P18|P2096}}
		  image2 = args['image2'] or '',
		  image2_upright = args['image2_upright'] or args['image2 upright'] or '',
		  image2_alt = args['image2_alt'] or args['image2 alt'] or '',
		  image2_caption = args['image2_caption'] or args['image2 caption'] or '',
		  species = species,
		  virus_infrasp = infraTaxon,
		  virus_infrasp_rank =  infraTaxonRank,
		  display_taxa = displayParents,
		  type_genus = args['type_genus'] or args['type genus'] or '',
		  type_genus_authority = typeGenusAuthority,
		  type_species = args['type_species'] or args['type species'] or '',
		  type_species_authority = args['type_species_authority'] or args['type species authority'] or '',
		  species_authority = args['species_authority'] or args['species authority'] or '',
		  subdivision_ranks = subdivisionRanks,
		  subdivision = subdivision,
		  type_strain = args['type_strain'] or args['type strain'] or '',
		  synonyms = args['synonyms'] or '',
		  synonyms_ref = args['synonyms_ref'] or args['synonyms ref'] or '',
		  range_map = args['range_map'] or args['range map'] or frame:expandTemplate{ title = 'Wikidata', args = { 'property' , 'raw' ,  args['from'] or '',  'P181',  }, },   -- equivalent to {{wikidata|property|raw|{{{from|}}}|P181}}
		  range_map_upright = args['range_map_upright'] or args['range map upright'] or '',
		  range_map_alt = args['range_map_alt'] or args['range map alt'] or '',
		  range_map_caption = args['range_map_caption'] or args['range map caption'] or frame:expandTemplate{ title = 'Wikidata', args = { 'qualifier',  args['from'] or '',  'P181', 'P2096'  }, },  --equivalent to {{wikidata|qualifier|{{{from|}}}|P181|P2096}}
		  defaultsort= args['taxon'] or args['serotype'] or args['strain'] or args['species'] or args['virus'] or frame:expandTemplate{ title = 'Wikidata', args = { 'property' , 'raw' ,  args['from'] or '',  'P225',  }, } or pagename, -- {{{taxon|{{wikidata|property|raw|{{{from|}}}|P225}}}}}
		} }
	-- put page in error-tracking category if required
	if autoTaxonType == 'ERROR' then
		res = res .. frame:expandTemplate{ title = 'Main other', args = {'[[Category:Virusboxes with incorrect parameters that specify taxon]]'} }
	end
	return res
end

-- =============================================================================
-- paramChk checks the taxon-specifying parameters for consistency, selecting
-- the target taxon (the taxon that is the target of the taxobox), the
-- infra-taxon (the taxon below species level), if any, and the 'auto-taxon',
-- the taxon that is the entry point into the automated taxobox system.
-- =============================================================================

function l.paramChk(frame, taxon, parent, species, strain, serotype, virus)
	-- set target taxon and infra-taxon
	local infraTaxon = ''
	local infraTaxonRank = ''
	local targetTaxon
	local targetTaxonRank
	if strain ~= '' then
		infraTaxon = strain
		infraTaxonRank = 'strain'
		targetTaxon = infraTaxon
		targetTaxonRank = infraTaxonRank
	elseif serotype ~= '' then
		infraTaxon = serotype
		infraTaxonRank = 'serotype'
		targetTaxon = infraTaxon
		targetTaxonRank = infraTaxonRank
	elseif virus ~= '' then
		infraTaxon = virus
		infraTaxonRank = 'virus'
		targetTaxon = infraTaxon
		targetTaxonRank = infraTaxonRank
	elseif species ~= '' then
		targetTaxon = species
		targetTaxonRank = 'species'
	else
		targetTaxon = taxon
		targetTaxonRank = frame:expandTemplate{ title = 'Taxon info', args = {targetTaxon, 'rank' } }
	end
	-- set the autotaxon (entry into the automated taxobox system) if the
	-- parameters are valid; the default is invalid
	local autoTaxon = ''
	local autoTaxonType = 'ERROR'
	if taxon ~= '' then
		if parent..species..infraTaxon  == '' then
			autoTaxon = taxon
			autoTaxonType = 'TAXON'
		end
	elseif parent ~= '' and  (species ~='' or infraTaxon ~= '') then
		autoTaxon = parent
		autoTaxonType = 'PARENT'
	end
	-- check for multiple infra-taxa
	local count = 0
	if strain ~= '' then count = count + 1 end
	if serotype ~= '' then count = count + 1 end
	if virus ~= '' then count = count + 1 end
	if count > 1 then autoTaxonType = 'ERROR' end
	return autoTaxon, autoTaxonType, infraTaxon, infraTaxonRank, targetTaxon, targetTaxonRank
end

return p