Modul:Official links

Fra Wikipedia, den frie encyklopedi
Moduldokumentasjon

require('strict')

local contLang = mw.language.getContentLanguage()
local conf = require 'Module:Official links/conf'(contLang:getCode())

local labels = {}

local function buildList( pid, t  )
	if #t == 0 then
		return ''
	elseif #t == 1 then
		return t[1]
	end
	local last = table.remove(t)
	local str = table.concat(t, mw.message.newRawMessage(conf:g('initial-' .. pid .. '-combiner', 'initial-default-combiner')):plain())
	return mw.message.newRawMessage(conf:g('final-' .. pid .. '-combiner', 'final-default-combiner'), str, last):plain()
end

-- todo: messy solution, refactor
local function wrap( content, marker )
	if content == '' or content == nil then
		return nil
	end
	return '<span class="mw-collapsible mw-collapsed" id="mw-customcollapsible-'..marker..'"><span class="mw-collapsible-content">' .. content .. '</span>'
		.. ' <span class="mw-customtoggle-'..marker..'" data-expandtext="{{int:show}}" data-collapsetext="{{int:hide}}" /></span>'
end

local function getEntityLabel( id )
	local entity = mw.wikibase.getEntity( id )
	if entity then
		return entity:getLabel()
	end
end

local function findEntityLabel( id )
	if not labels[id] then
		local label = getEntityLabel( id )
		if label then
			labels[id] = label -- labels is an outer structure
		end
	end
	return labels[id]
end

local mainFormatter = {}
mainFormatter['string'] = function( pid, datavalue )
	if datavalue['type'] ~= 'string' then
		return nil
	end
	return datavalue.value
end

local qualFormatter = {}
qualFormatter['wikibase-entityid'] = function( pid, datavalue )
	if datavalue['type'] ~= 'wikibase-entityid' then
		return nil
	end
	if datavalue.value['entity-type'] ~= 'item' then
		return nil
	end
	return findEntityLabel( 'Q'..datavalue.value["numeric-id"] )
end

local qualPrefer = {}
qualPrefer['wikibase-entityid'] = function( pid, datavalue )
	if datavalue['type'] ~= 'wikibase-entityid' then
		return nil
	end
	if datavalue.value['entity-type'] ~= 'item' then
		return nil
	end
	return conf:p( 'Q'..datavalue.value["numeric-id"] ) or false
end

local main = {}
main.P856 = {
	types = {
		snaktype = 'value',
		datatype = 'url',
	},
	formatter = mainFormatter['string']
}
main.P1019 = {
	types = {
		snaktype = 'value',
		datatype = 'url',
	},
	formatter = mainFormatter['string']
}
main.P1581 = {
	types = {
		snaktype = 'value',
		datatype = 'url',
	},
	formatter = mainFormatter['string']
}


local qual = {}
qual.P407 = {
	types = {
		snaktype = 'value',
		datatype = 'wikibase-item',
	},
	formatter = qualFormatter['wikibase-entityid'],
	preferred = qualPrefer['wikibase-entityid']
}

local qorder = {'P407'}

local p = {}

function p.findMainLinks(pid, qid)
	local head = {}
	local tail = {}
	local entity = mw.wikibase.getEntityObject( qid )
	-- to avoid deep tests
	if not entity or not entity.claims then
		return head, tail
	end
	local statements = entity.claims[pid]
	-- to avoid deep tests
	if not statements then
		return head, tail
	end
	for _, claim in ipairs( statements ) do
		-- to avoid deep tests
		if not claim then
			claim = {}
		end
		local valid = true
		if claim['type'] ~= 'statement' then
			valid = valid and false
		end
		if claim['rank'] == 'deprecated' then
			valid = valid and false
		end
		local mainsnak = claim.mainsnak or {}
		if not mainsnak or not main[pid] then
			valid = valid and false
		end
		if ((main[pid] and mainsnak.snaktype ~= main[pid].types.snaktype)
			or (main[pid] and mainsnak.datatype ~= main[pid].types.datatype))
		then
			valid = valid and false
		end
		if valid then
			local preferred = claim['rank'] == 'preferred'
			local mainStr = main[pid].formatter(pid, mainsnak.datavalue)
			local optionals = {}
			local qualifiers = claim.qualifiers or {}
			for _, qualid in ipairs( qorder ) do
				if qualifiers[qualid] then
					local items = {}
					for _, qualsnak in ipairs( qualifiers[qualid] ) do
						if qualsnak and qual[qualid] then
							if not (qualsnak.snaktype ~= qual[qualid].types.snaktype
								or qualsnak.datatype ~= qual[qualid].types.datatype)
							then
								items[1+#items] = qual[qualid].formatter(qualsnak.property, qualsnak.datavalue)
								preferred = preferred or qual[qualid].preferred(qualsnak.property, qualsnak.datavalue)
							end
						end
					end
					local text = buildList(qualid, items)
					if qualid == 'P407' then
						if not text or text == '' then
							-- this should never happen, unless it will be possible to add empty qualifiers
							text = mw.message.newRawMessage(conf:g('qualifier-'.. qualid..'-empty', 'qualifier-default-empty')):plain()
						end
						text = mw.message.newRawMessage(conf:g('ext-link-' .. claim['rank']), mainStr, text):plain()
					end
					optionals[1+#optionals] = text
				else
					local text = nil
					if qualid == 'P407' then
						text = conf:guess(mainStr)
					end
					if not text then
						text = mw.message.newRawMessage(
							conf:g('qualifier-'.. pid..'-missing-'..qualid,
								'qualifier-default-missing-'..qualid,
								'qualifier-default-missing')):plain()
					end
					optionals[1+#optionals] = mw.message.newRawMessage(conf:g('ext-link-' .. claim['rank']), mainStr, text):plain()
				end
			end
			if #optionals > 0 then
				mainStr = buildList(pid, optionals)
			end
			if preferred then
				head[1+#head] = mainStr
			else
				tail[1+#tail] = mainStr
			end
		end
	end
	return head, tail
end

function p.links( frame )
	local items = {}
	local counts = {}
	for _,v in ipairs(frame.args) do
		local _, _, ch, num= v:find("([pP])(%d+)")
		if ch then
			local pid = ch:upper()..num
			local label = findEntityLabel( pid )
			if not counts[pid] then
				counts[pid] = 0
			end
			if label then
				local head, tail = p.findMainLinks(pid)
				if (head and #head > 0) or (tail and #tail > 0) then
					counts[pid] = counts[pid] + 1
				end
				if head and #head > 0 then
					items[1+#items] = { contLang:ucfirst( label ), buildList('', head), wrap(buildList('', tail), pid) }
				elseif tail and #tail > 0 then
					items[1+#items] = { contLang:ucfirst( label ), buildList('', tail) }
				end
			end
		end
	end
	for i,v in ipairs(items) do
		items[i] = mw.message.newRawMessage(
			((#v >=3 and #v[3])
				and conf:g(i==1 and 'first-list-item-with-additional' or 'rest-list-item-with-additional')
				or conf:g(i==1 and 'first-list-item' or 'rest-list-item')),
			unpack(v))
		:plain()
	end
	local cats = {}
	for k,v in pairs(counts) do
		cats[1+#cats] = '[[Category:' .. mw.message.newRawMessage(conf:g(counts[k]==0 and 'cat-exclusion' or 'cat-inclusion'), findEntityLabel( k )):plain() .. ']]'
	end
	if #items > 0 then
		return table.concat(items, "\n") .. table.concat(cats, "")
	end
	return "''" .. mw.message.newRawMessage(conf:g('no-links-available')):plain() .. "''" .. table.concat(cats, "")
end

return p