Jump to content

Module:ResolveEntityId/sandbox

From Wikipedia, the free encyclopedia
This is an old revision of this page, as edited by Johnuniq (talk | contribs) at 03:50, 26 February 2022 (couple of fixes). The present address (URL) is a permanent link to this revision, which may differ significantly from the current revision.
local p = {}

function p._id(id, alt)
	local function resolveId(s)
		if mw.wikibase.entityExists(s) then
			local link = mw.wikibase.getSitelink(s)
			if link then
				local eid = mw.wikibase.getEntityIdForTitle(link)
				if eid then
					return eid
				end
			end
			return mw.wikibase.getEntity(s).id
		else
			return alt
		end
	end

	if type(id) == 'string' then
		id = mw.ustring.upper(mw.ustring.sub(id, 1, 1)) .. mw.ustring.sub(id, 2)
		if mw.ustring.match(id, '^Q%d+$') then
			-- id is in the proper format for a Wikidata entity
			if mw.wikibase.isValidEntityId(id) then
				-- id is valid
				return resolveId(id)
			end
		else
			local eid = mw.wikibase.getEntityIdForTitle(id)
			if eid then
				-- entity id is a title that matches a Wikidata entity
				local instanceOf = mw.wikibase.getBestStatements(eid, 'P31')[1] --instance of
				if not instanceOf or instanceOf.mainsnak.datavalue.value.id ~= 'Q4167410' then
					-- instance-of value is missing or is not disambiguation
					return resolveId(eid)
				end
			else
				-- no wikidata item exists
				local rtarget = mw.title.new(id).redirectTarget
				if rtarget then
					-- title is a Wikipedia redirect
					return p._id(rtarget.fullText, alt)
				end
			end
		end
	end
	return alt
end

function p.entityid(frame)
	return p._id(frame.args[1], frame.args[2])
end

return p