Documentation for this module may be created at Մոդուլ:Wikidata/Places2/doc

local categorizeByPlace = true;

local WDS = require( 'Module:WikidataSelectors' );
local Flags = require( 'Module:Wikidata/Flags' );
local p = {
	config = {
		hideSameLabels = false,
		hidePartOfLabels = false,
		hideUnitsForCapitals = true,
		reverseOrder = false,
		catAmbiguousGeoChains = '[[Ստորոգութիւն:Ուիքիփետիա:Սխալ աշխշղթա]]',
		catWikibaseError = '[[Ստորոգութիւն:Ուիքիփետիա:Սխալներ Ուիքիտվեալների]]'
	}
};

local function min( prev, next )
	if ( prev == nil ) then return next;
	elseif ( prev > next ) then return next;
	else return prev; end
end

local function max( prev, next )
	if ( prev == nil ) then return next;
	elseif ( prev < next ) then return next;
	else return prev; end
end

local function getTimeBoundariesFromProperty( context, propertyId )
	-- mw.log( 'Get time boundaries for ' .. propertyId .. '...');

	local dateClaims = WDS.filter( context.entity.claims, propertyId );
	if ( not dateClaims or #dateClaims == 0 ) then return nil; end
	-- mw.log( 'Get time boundaries for ' .. propertyId .. '... Got ' .. #dateClaims .. ' date claim(s)');

	-- only support exact date so far, but need improvment
	local left = nil;
	local right = nil;
	for _, claim in pairs( dateClaims ) do
		if ( not claim.mainsnak ) then return nil; end
		local boundaries = context.parseTimeBoundariesFromSnak( claim.mainsnak );
		if ( not boundaries ) then return nil; end
		left = min( left, boundaries[ 1 ] );
		right = max( right, boundaries[ 2 ] );
	end

	if ( not left or not right ) then return nil; end

	-- mw.log( 'Time boundaries for ' .. propertyId .. ' are ' .. left .. ' and ' .. right );
	return { left, right };
end

local function getTimeBoundariesFromProperties( context, propertyIds )
	for _, propertyId in ipairs( propertyIds ) do
		local result = getTimeBoundariesFromProperty( context, propertyId );
		if result then
			return result;
		end
	end

	return nil;
end

local function getTimeBoundariesFromQualifiers( context, statement, qualifierId )
	-- only support exact date so far, but need improvment
	local left = nil;
	local right = nil;
	if ( statement.qualifiers and statement.qualifiers[qualifierId] ) then
		for _, qualifier in pairs( statement.qualifiers[qualifierId] ) do
			local boundaries = context.parseTimeBoundariesFromSnak( qualifier );
			if ( not boundaries ) then return nil; end
			left = min( left, boundaries[1] );
			right = max( right, boundaries[2] );
		end
	end

	if ( not left or not right ) then
		return nil;
	end

	return { left, right };
end

local function getParentsInBoundariesSnakImpl( context, entity, boundaries, propertyIds )
	local results = {};

	if not propertyIds or #propertyIds == 0 then
		return results;
	end

	if entity.claims then
		for _, propertyId in ipairs( propertyIds ) do
			local filteredClaims = WDS.filter( entity.claims, propertyId .. '[rank:preferred, rank:normal]' );
			if filteredClaims then
				for _, claim in pairs( filteredClaims ) do
					if not boundaries or not propertyIds or #propertyIds == 0 then
						table.insert( results, claim.mainsnak );
					else
						local startBoundaries = getTimeBoundariesFromQualifiers( context, claim, 'P580' );
						local endBoundaries = getTimeBoundariesFromQualifiers( context, claim, 'P582' );
			
						if ( (startBoundaries == nil or startBoundaries[2] <= boundaries[1] )
								and ( endBoundaries == nil or endBoundaries[1] >= boundaries[2] ) ) then
							table.insert( results, claim.mainsnak );
						end 
					end
				end
			end

			if #results > 0 then
				break;
			end
		end
	end

	return results;
end

local function getParentsInBoundariesSnak( context, entity, boundaries )
	if ( not entity ) then error('entity must be specified'); end
	if ( type(entity) ~= 'table' ) then error('entity must be table'); end
	if ( not boundaries ) then error('boundaries must be specified'); end
	if ( type(boundaries) ~= 'table' ) then error('boundaries must be table'); end

	local results = getParentsInBoundariesSnakImpl( context, entity, boundaries, {'P131'} ) -- located in
	if not results or #results == 0 then
		results = getParentsInBoundariesSnakImpl( context, entity, boundaries, {'P17'} ) -- country
	end

	for r, result in pairs( results ) do
		if result.snaktype ~= 'value' then
			return nil;
		end
		local resultId = result.datavalue.value.id;
		if ( resultId == entity.id ) then
			return nil;
		end
	end
	return results;
end

local unions = { 
	Q1140229 = true, -- քաղաքաքան միութիւն
	Q3623811 = true, -- տնտեսական մմիութիւն
	Q4120211 = true -- ռեգիոնալ կազմակերպութիւն
}

local countries = {
	Q6256 = true, -- երկիր
	Q7275 = true, -- պետութիւն
	Q3624078 = true -- անկախ պետութիւն
}

local function isSkipTopLevel( entity )
	local isCountry = false;
	local isUnion = false;
	if ( entity
			and entity.claims
			and entity.claims.P31 ) then
		for c, claim in pairs( entity.claims.P31 ) do
			if ( claim
					and claim.mainsnak
					and claim.mainsnak.datavalue
					and claim.mainsnak.datavalue.value
					and claim.mainsnak.datavalue.value.id ) then
				local typeId = claim.mainsnak.datavalue.value.id;
				isCountry = isCountry or countries[ typeId ];
				isUnion = isUnion or unions[ typeId ];
			end
		end
	end
	return isUnion and not isCountry;
end

local function isPartOfNext( prevLabel, nextLabel )
	return ( mw.ustring.len( prevLabel ) > mw.ustring.len( nextLabel ) )
		and ( mw.ustring.sub( prevLabel, mw.ustring.len( prevLabel ) - mw.ustring.len( nextLabel ) + 1 ) == nextLabel );
end

--Property:P19, Property:P20, Property:P119
function p.formatPlaceWithQualifiers( context, options, statement )
	local property = mw.ustring.upper( options.property );
	-- mw.log( 'formatPlaceWithQualifiers(..., ' .. property .. ')');

	local actualDateBoundariesProperties = nil;
	if ( property == 'P19' ) then actualDateBoundariesProperties = {'P569','P570'}; end
	if ( property == 'P20' ) then actualDateBoundariesProperties = {'P570','P569'}; end
	if ( property == 'P119' ) then actualDateBoundariesProperties = {'P570','P569'}; end

	local boundaries = nil;
	if ( actualDateBoundariesProperties ~= nil ) then
		boundaries = getTimeBoundariesFromProperties( context, actualDateBoundariesProperties );
	end

	local entriesToLookupCategory = {};

	local circumstances = context.getSourcingCircumstances( statement );
	local result = '';
	local baseResult = context.formatSnak( options, statement.mainsnak, circumstances );
	insertFromSnak( statement.mainsnak, entriesToLookupCategory )

	local hasAdditionalQualifiers = false;
	if ( statement.qualifiers ) then
		--parent divisions
		if ( statement.qualifiers.P131 ) then
			for i, qualifier in ipairs( statement.qualifiers.P131 ) do
				local parentOptions = context.cloneOptions( options );
				local qualifierEntityId = qualifier.datavalue.value.id;
				local wbStatus, qualifierEntity = pcall( mw.wikibase.getEntity, qualifierEntityId )
				if wbStatus ~= true then
					return '[[:d:' .. qualifierEntityId .. '|' .. qualifierEntityId .. ']]<span style="color:#b32424; border-bottom: 1px dotted #b32424; cursor: help; white-space: nowrap" title="Ошибка получения элемента из Викиданных.">×</span>' .. categoryLinksToEntitiesWithWikibaseError;
				end
				parentOptions['text'] = getLabel( context, qualifierEntity, boundaries );
				local link = context.formatSnak( parentOptions, qualifier );

				if p.config.reverseOrder then
					result = link .. ', ' .. result;
				else
					result = result .. ', ' .. link;
				end
				insertFromSnak( qualifier, entriesToLookupCategory )
				hasAdditionalQualifiers = true;
			end
		end

		--country
		if ( statement.qualifiers.P17 ) then
			for i, qualifier in ipairs( statement.qualifiers.P17 ) do
				local parentOptions = context.cloneOptions( options );
				local qualifierEntityId = qualifier.datavalue.value.id;
				local wbStatus, qualifierEntity = pcall( mw.wikibase.getEntity, qualifierEntityId )
				if wbStatus ~= true then
					return '[[:d:' .. qualifierEntityId .. '|' .. qualifierEntityId .. ']]'
						.. '<span style="color:#b32424; border-bottom:1px dotted #b32424; cursor:help; white-space:nowrap"'
						.. ' title="Ошибка получения элемента из Викиданных.">×</span>'
						.. categoryLinksToEntitiesWithWikibaseError;
				end
				parentOptions[ 'text' ] = getLabel( context, qualifierEntity, boundaries );
				local link = context.formatSnak( parentOptions, qualifier );

				if p.config.reverseOrder then
					result = link .. ', ' .. result;
				else
					result = result .. ', ' .. link;
				end
				insertFromSnak( qualifier, entriesToLookupCategory )
				hasAdditionalQualifiers = true;
			end
		end
	end

	if ( statement.mainsnak
			and statement.mainsnak.datavalue
			and statement.mainsnak.datavalue.value
			and statement.mainsnak.datavalue.value.id ) then
		local entityId = statement.mainsnak.datavalue.value.id;
		local wbStatus, entity = pcall( mw.wikibase.getEntity, entityId );
		local parentSnaks = { statement.mainsnak };
		local parentEntities = { entity };

		if ( wbStatus == true and actualDateBoundariesProperties ~= nil ) then
			local filterCapitalOf = {
				[ entityId ] = getParentsInBoundariesSnakImpl( context, entity, boundaries, {'P1376'} )
			};
			if ( boundaries ) then
				local entityOptions = context.cloneOptions( options );
				entityOptions['text'] = getLabel( context, entity, boundaries );
				baseResult = context.formatSnak( entityOptions, statement.mainsnak, circumstances );

				local parent = entity;
				local wbStatus = true;
				while ( wbStatus == true and parent ~= nil ) do
					-- get parent
					local newParentSnaks = getParentsInBoundariesSnak( context, parent, boundaries );
					if ( not newParentSnaks or #newParentSnaks == 0 ) then
						parent = nil;
					elseif ( #newParentSnaks == 1 ) then
						local parentSnak = newParentSnaks[ 1 ];
						local parentId = parentSnak.datavalue.value.id;
						wbStatus, parent = pcall( mw.wikibase.getEntity, parentId );
						if wbStatus == true then
							table.insert( parentSnaks, parentSnak );
							table.insert( parentEntities, parent );
							filterCapitalOf[ parentId ] = getParentsInBoundariesSnakImpl( context, parent, boundaries, { 'P1376' } );
						end
					else
						parent = nil;
						if p.config and p.config.catAmbiguousGeoChains then
							result = result .. p.config.catAmbiguousGeoChains;
						end
					end
				end
				local ussr=false;
				local nado=false;

				if ( not hasAdditionalQualifiers ) then
					for i = 2, #parentSnaks, 1 do
						local parentSnak = parentSnaks[ i ];
						insertFromSnak( parentSnak, entriesToLookupCategory )
					end
				end

				do
					local i = #parentSnaks;
                                        if (i == 2) then
                                          nado=true;
                                        end
                    
					while ( i > 1 ) do
						if (ussr) then
							if (isInstanceOf( parentEntities[ i ], 'Q236036' ) ) or (isInstanceOf( parentEntities[ i ], 'Q3336843' ) ) or (isInstanceOf( parentEntities[ i ], 'Q12959600' ) ) or (isInstanceOf( parentEntities[ i ], 'Q56219758' ) ) then
								for _capitalId, capitalSnaks in pairs( filterCapitalOf ) do
									if #capitalSnaks > 0 then
										for __, capitalSnak in pairs( capitalSnaks ) do
											if parentSnaks[ i ] ~= nil then
												if parentSnaks[ i ].datavalue.value.id == capitalSnak.datavalue.value.id then
													nado=true;
													break;
												end
											end
										end
									end
								end
								table.remove( parentSnaks, i );
								table.remove( parentEntities, i );
							end
							ussr=false;
						end
						
						ussr=true;
						local prevEntity = parentEntities[ i - 1 ];
						-- TODO: use English labels, if there is no current language labels
						local prevLabel = getLabel( context, prevEntity, boundaries ) or '';
						local nextEntity = parentEntities[ i ];
						local nextLabel = getLabel( context, nextEntity, boundaries ) or '';
						if ( p.config and p.config.hideSameLabels == true and prevLabel == nextLabel ) then
							-- do not output same label twice (NY, NY, USA)
							table.remove( parentSnaks, i );
							table.remove( parentEntities, i );
						elseif ( p.config and p.config.hidePartOfLabels == true and isPartOfNext( prevLabel, ' ' .. nextLabel ) ) then
							-- do not output same label if it's part of previos
							table.remove( parentSnaks, i - 1 );
							table.remove( parentEntities, i - 1 );
						elseif ( p.config and p.config.hideUnitsForCapitals == true ) then
							-- do not ouput items whose capital is the first item
							local capitalId = nil;
							for _capitalId, capitalSnaks in pairs( filterCapitalOf ) do
							    if #capitalSnaks > 0 then
							        for __, capitalSnak in pairs( capitalSnaks ) do
								        if parentSnaks[ i ] ~= nil then
											if parentSnaks[ i ].datavalue.value.id == capitalSnak.datavalue.value.id then
												capitalId = _capitalId;
												break;
											end
										end
									end
								end
							end

							if (capitalId ~= nil or i ~= #parentSnaks) then
								table.remove( parentSnaks, 1 );
								table.remove( parentEntities, 1 );
							end

							if (capitalId ~= nil and nado3) then
								if ( i == #parentSnaks ) then
									i = i - 1;
								end

								while ( i > 1 and parentEntities[ i ].id ~= capitalId ) do
									table.remove( parentSnaks, i );
									table.remove( parentEntities, i );
									i = i - 1;
								end

								nado3=false;
							end
						end

						i = i - 1;

						if (i == 2) then
							table.remove( parentSnaks, i+1);
							table.remove( parentEntities, i+1);
						end
						if (i == 1) and (nado == true) then
							table.remove( parentSnaks, 1);
							table.remove( parentEntities, 1);
						end
					end
				end

				if ( isSkipTopLevel( parentEntities[ #parentEntities ] ) ) then
					table.remove( parentSnaks, #parentEntities );
					table.remove( parentEntities, #parentEntities );
				end

				if ( not hasAdditionalQualifiers ) then
					for i = 2, #parentSnaks, 1 do
						local parentSnak = parentSnaks[ i ];
						local parentOptions = context.cloneOptions( options );
						parentOptions['text'] = getLabel( context, parentEntities[ i ], boundaries );

						if p.config.reverseOrder then
							result = context.formatSnak( parentOptions, parentSnak ) .. ', ' .. result;
						else
							result = result .. ', ' .. context.formatSnak( parentOptions, parentSnak );
						end
					end
				end
			end
		end

		if wbStatus ~= true and p.config and p.config.catWikibaseError then
				result = result .. p.config.catWikibaseError;
		end
	end

	if ( options[ 'thisLocationOnly' ] ) then
		result = baseResult  .. context.formatRefs( options, statement );
	elseif p.config.reverseOrder then
		result = result .. baseResult .. context.formatRefs( options, statement );
	else
		result = baseResult .. result .. context.formatRefs( options, statement );
	end
	if ( categorizeByPlace ) then
		if ( property == 'P19' ) then result = result .. getCategory( 'P1464', entriesToLookupCategory ); end
		if ( property == 'P20' ) then result = result .. getCategory( 'P1465', entriesToLookupCategory ); end
		if ( property == 'P119' ) then result = result .. getCategory( 'P1791', entriesToLookupCategory ); end
	end
	return result;
end

-- append entity id from snak to result
function insertFromSnak( snak, result )
	if ( not categorizeByPlace ) then
		return;
	end
	if ( snak 
			and snak.datavalue
			and snak.datavalue.type == 'wikibase-entityid'
			and snak.datavalue.value
			and snak.datavalue.value[ 'entity-type' ] == 'item' ) then
		table.insert( result, snak.datavalue.value.id );
	end
end

function getCategory( propertyToSearch, entriesToLookupCategoryFor )
	for _, placeId in pairs( entriesToLookupCategoryFor ) do
		local wbStatus, placeEntity = pcall( mw.wikibase.getEntity, placeId );
		if wbStatus == true and placeEntity then
			local claims = WDS.filter( placeEntity.claims, propertyToSearch );
	
			if ( claims ) then
				for _, claim in pairs( claims ) do
					if ( claim.mainsnak
							and claim.mainsnak
							and claim.mainsnak.datavalue
							and claim.mainsnak.datavalue.type == 'wikibase-entityid' ) then
						local catEntityId = claim.mainsnak.datavalue.value.id;
						local wbStatus, catEntity = pcall( mw.wikibase.getEntity, catEntityId );
						if ( wbStatus == true and catEntity and catEntity:getSitelink() ) then
							return '[[' .. catEntity:getSitelink() .. ']]';
						end
					end
				end
			end
		end
	end
	return '';
end

-- get current of historic name of place
function getLabel( context, entity, boundaries )
	if not entity then
		return nil;
	end

	local lang = mw.language.getContentLanguage();
	local langCode = lang:getCode();

	-- name from label
	-- TODO: lang:getFallbackLanguages()
	local label = nil;
	if entity.labels then
		if entity.labels[ langCode ] and entity.labels[ langCode ].value then
			label = entity.labels[ langCode ].value;
		elseif entity.labels.en and entity.labels.en.value then
			label = entity.labels.en.value;
		end
	end

	-- name from properties
	local results = getParentsInBoundariesSnakImpl( context, entity, boundaries, {
		'P1813[language:' .. langCode .. ']',
		'P1448[language:' .. langCode .. ']',
		'P1705[language:' .. langCode .. ']'
	} );

	for r, result in pairs( results ) do
		if result.datavalue
				and result.datavalue.value
				and result.datavalue.value.text then
			label = result.datavalue.value.text;
			break;
		end
	end

	return label;
end

p.getLabel = getLabel;

local function calculateEndDateTimestamp( context, options, statement )
	if ( not context ) then error( 'context not specified' ) end;
	if ( not options ) then error( 'options not specified' ) end;
	if ( not options.entity ) then error( 'options.entity missing' ) end;
	if ( not statement ) then error( 'statement not specified' ) end;

	if ( statement.qualifiers and statement.qualifiers.P582 ) then
		for i, qualifier in ipairs(statement.qualifiers.P582 ) do
			local parsedTime = context.parseTimeFromSnak( qualifier );
			if ( parsedTime ) then
				return parsedTime;
			end
		end
	end

	-- check death day... do we have it at all?
	for h, propertyId in pairs( { "P570", "P577", "P571" } ) do
		local dateClaims = context.selectClaims( options, propertyId );
		if ( dateClaims ) then
			for i, statement in ipairs( dateClaims ) do
				local parsedTime = context.parseTimeFromSnak( statement.mainsnak );
				if ( parsedTime ) then
					return parsedTime;
				end
			end
		end
	end

	-- TODO: check other "end" properties

	-- no death day
	return os.time() * 1000;
end

function p.formatCountryClaimWithFlag( context, options, statement )
	if (not context) then error('context not specified') end;
	if (not options) then error('options not specified') end;
	if (not options.entity) then error('options.entity is missing') end;
	if (not statement) then error('statement not specified') end;

	local countryEntityId = nil;
	local countryEntity = nil;
	local wbStatus = false;
	if ( statement.mainsnak and statement.mainsnak.datavalue and statement.mainsnak.datavalue.value and statement.mainsnak.datavalue.value.id ) then
		countryEntityId = statement.mainsnak.datavalue.value.id;
		wbStatus, countryEntity = pcall( mw.wikibase.getEntity, countryEntityId );
	end
	
	if wbStatus ~= true or not countryEntity then
		local result = '<span class="country-name">' .. context.formatStatementDefault( context, options, statement ) .. '</span>';
		if wbStatus ~= true and p.config and p.config.catWikibaseError then
			result = result .. p.config.catWikibaseError;
		end
		return result;
	end

	local endDateTimestamp = calculateEndDateTimestamp( context, options, statement );
	local boundaries = getTimeBoundariesFromProperties( context, {'P570', 'P577', 'P571'} );

	local countryOptions = context.cloneOptions( options );
	if not countryOptions['text'] or countryOptions['text'] == '' then
		countryOptions['text'] = getLabel( context, countryEntity, boundaries );
	end

	local flag = Flags.getFlag( context, countryEntityId, endDateTimestamp );	
	if ( flag ) then
		return flag .. '&nbsp;<span class="country-name">' .. context.formatStatementDefault( context, countryOptions, statement ) .. '</span>';
	end

	return '<span class="country-name">' .. context.formatStatementDefault( context, countryOptions, statement ) .. '</span>';
end

return p;