Module:PrevalenceData

-- This module is specifically the Wikidata property "prevalence" (P1193), due
-- to its particular need for ranges and area-based qualifiers, and the lack of
-- support for these in the main Wikidata module.

-- Completely untested.
local p = {}

p.main = function ( frame )
	local entity = mw.wikibase.getEntity( frame.args.qId or nil )
	if entity then
		local prevalenceClaims = entity.claims.P1193
		-- TODO: Get best claim, not always just the first one.
		local prevalenceClaim = prevalenceClaims and prevalenceClaims[ 1 ]
		mw.logObject( prevalenceClaims[ 1 ] )
		if prevalenceClaim then
			local prevalenceValue = prevalenceClaim.mainsnak.datavalue.value
			mw.logObject( prevalenceValue )
			if prevalenceValue then
				mw.log( prevalenceValue.upperBound * 100 )
				local lowerBound = prevalenceValue.lowerBound * 100
				local upperBound = prevalenceValue.upperBound * 100
				local pRange = lowerBound
				if lowerBound ~= upperBound then
					pRange = pRange .. ' to ' .. upperBound
				end
				
				pRange = pRange .. '%'
				if prevalenceClaim.qualifiers then
					-- Qualifiers for prevalence are currently unstandardized.
					-- Keep guessing until the right one is found.
					local quals = prevalenceClaim.qualifiers.P276 or -- location
						prevalenceClaim.qualifiers.P1001 or          -- applies to jurisdiction
						prevalenceClaim.qualifiers.P17               -- country
					if quals then
						pRange = pRange .. ' ('
						for k, qual in pairs(quals) do
							mw.log( k, 123 )
							
							if k > 1 then
								pRange = pRange .. ', '
							end
							local qualId = qual.datavalue.value[ 'numeric-id' ]
							local link = mw.wikibase.sitelink( 'Q' .. qualId )
							local label = ({
								-- Certain geographic locales might need a
								-- manual-ish override for labels. 
								[ 132453 ] = 'developed world'
							})[ qualId ] or mw.wikibase.label( 'Q' .. qualId )
							if link then
								label = '[[' .. link .. '|' .. label .. ']]'
							end
							pRange = pRange .. label
							
						end
						pRange = pRange .. ')'
						mw.log( pRange )
					end
				end
			end
		end
	end
end

return p