Line 1: |
Line 1: |
| + | local makeUrl = require('Module:URL')._url |
| + | |
| local p = {} | | local p = {} |
| + | |
| + | -- Wrapper for pcall which returns nil on failure. |
| + | local function quickPcall(func) |
| + | local success, result = pcall(func) |
| + | if success then |
| + | return result |
| + | end |
| + | end |
| + | |
| + | -- Gets the rank for a Wikidata property table. Returns 1, 0 or -1, in |
| + | -- order of rank. |
| + | local function getRank(prop) |
| + | local rank = prop.rank |
| + | if rank == 'preferred' then |
| + | return 1 |
| + | elseif rank == 'normal' then |
| + | return 0 |
| + | elseif rank == 'deprecated' then |
| + | return -1 |
| + | else |
| + | -- No rank or undefined rank is treated as "normal". |
| + | return 0 |
| + | end |
| + | end |
| + | |
| + | -- Finds whether a Wikidata property is qualified as being in English. |
| + | local function isEnglish(prop) |
| + | local ret = quickPcall(function () |
| + | for i, lang in ipairs(prop.qualifiers.P407) do |
| + | if lang.datavalue.value['numeric-id'] == 1860 then |
| + | return true |
| + | end |
| + | end |
| + | return false |
| + | end) |
| + | return ret == true |
| + | end |
| + | |
| + | -- Fetches the official website URL from Wikidata. |
| + | local fetchWikidataUrl |
| + | fetchWikidataUrl = function() |
| + | -- Get objects for all official sites on Wikidata. |
| + | local websites = quickPcall(function () |
| + | return mw.wikibase.getEntityObject().claims.P856 |
| + | end) |
| + | |
| + | -- Clone the objects in case other code needs them in their original order. |
| + | websites = websites and mw.clone(websites) or {} |
| + | |
| + | -- Add the table index to the objects in case it is needed in the sort. |
| + | for i, website in ipairs(websites) do |
| + | website._index = i |
| + | end |
| + | |
| + | -- Sort the websites, first by highest rank, and then by websites in the |
| + | -- English language, then by the website's original position in the |
| + | -- property list. When we are done, get the URL from the highest-sorted |
| + | -- object. |
| + | table.sort(websites, function(ws1, ws2) |
| + | local r1 = getRank(ws1) |
| + | local r2 = getRank(ws2) |
| + | if r1 ~= r2 then |
| + | return r1 > r2 |
| + | end |
| + | local e1 = isEnglish(ws1) |
| + | local e2 = isEnglish(ws2) |
| + | if e1 ~= e2 then |
| + | return e1 |
| + | end |
| + | return ws1._index < ws2._index |
| + | end) |
| + | local url = quickPcall(function () |
| + | return websites[1].mainsnak.datavalue.value |
| + | end) |
| + | |
| + | -- Cache the result so that we only do the heavy lifting once per #invoke. |
| + | fetchWikidataUrl = function () |
| + | return url |
| + | end |
| + | |
| + | return url |
| + | end |
| + | |
| + | -- Render the URL link, plus other visible output. |
| + | local function renderUrl(options) |
| + | if not options.url and not options.wikidataurl then |
| + | local entity = mw.wikibase.getEntityObject() or {} |
| + | local qid = entity.id |
| + | local result = '<strong class="error">' .. |
| + | 'No URL found. Please specify a URL here or add one to Wikidata.' .. |
| + | '</strong>' |
| + | if qid then |
| + | result = result.. ' [[File:OOjs UI icon edit-ltr-progressive.svg |frameless |text-top |10px |alt=Edit this at Wikidata |link=https://www.wikidata.org/wiki/' .. qid .. '#P856|Edit this at Wikidata]]' |
| + | end |
| + | return result |
| + | end |
| + | local ret = {} |
| + | ret[#ret + 1] = string.format( |
| + | '<span class="official-website">%s</span>', |
| + | makeUrl(options.url or options.wikidataurl, options.display) |
| + | ) |
| + | if options.wikidataurl and not options.url then |
| + | local entity = mw.wikibase.getEntityObject() or {} |
| + | local qid = entity.id |
| + | if qid then |
| + | ret[#ret + 1] = '[[File:OOjs UI icon edit-ltr-progressive.svg |frameless |text-top |10px |alt=Edit this at Wikidata |link=https://www.wikidata.org/wiki/' .. qid .. '#P856|Edit this at Wikidata]]' |
| + | end |
| + | end |
| + | if options.format == 'flash' then |
| + | ret[#ret + 1] = mw.getCurrentFrame():expandTemplate{ |
| + | title = 'Color', |
| + | args = {'#505050', '(Requires [[Adobe Flash Player]])'} |
| + | } |
| + | end |
| + | if options.mobile then |
| + | ret[#ret + 1] = '(' .. makeUrl(options.mobile, 'Mobile') .. ')' |
| + | end |
| + | return table.concat(ret, ' ') |
| + | end |
| + | |
| + | -- Render the tracking category. |
| + | local function renderTrackingCategory(url, wikidataurl) |
| + | if mw.title.getCurrentTitle().namespace ~= 0 then |
| + | return '' |
| + | end |
| + | local category |
| + | if not url and not wikidataurl then |
| + | category = 'Official website missing URL' |
| + | elseif not url and wikidataurl then |
| + | return '' |
| + | elseif url and wikidataurl then |
| + | if url:gsub('/%s*$', '') ~= wikidataurl:gsub('/%s*$', '') then |
| + | category = 'Official website different in Wikidata and Wikipedia' |
| + | end |
| + | else |
| + | category = 'Official website not in Wikidata' |
| + | end |
| + | return category and string.format('[[Category:%s]]', category) or '' |
| + | end |
| + | |
| + | function p._main(args) |
| + | local url = args[1] or args.URL or args.url |
| + | local wikidataurl = fetchWikidataUrl() |
| + | local formattedUrl = renderUrl{ |
| + | url = url, |
| + | wikidataurl = wikidataurl, |
| + | display = args[2] or args.name or 'Official website', |
| + | format = args.format, |
| + | mobile = args.mobile |
| + | } |
| + | return formattedUrl .. renderTrackingCategory(url, wikidataurl) |
| + | end |
| | | |
| function p.main(frame) | | function p.main(frame) |
− | return '' | + | local args = require('Module:Arguments').getArgs(frame, { |
| + | wrappers = 'Template:Official website' |
| + | }) |
| + | return p._main(args) |
| end | | end |
| | | |
| return p | | return p |