Module:Redirect/sandbox: Difference between revisions

Content deleted Content added
split decoding into a separate function
Trimmed redundancy
 
(21 intermediate revisions by 7 users not shown)
Line 1:
-- This module provides functions for getting the target of a redirect page.
-- Given a single page name determines what page it redirects to and returns the target page name, or the
-- passed page name when not a redirect. The passed page name can be given as plain text or as a page link.
-- Returns page name as plain text, or when the bracket parameter is given, as a page link. Returns an
-- error message when page does not exist or the redirect target cannot be determined for some reason.
 
-- Thus these are roughly the same:
-- [[{{#invoke:redirect|main|redirect-page-name}}]] and {{#invoke:redirect|main|redirect-page-name|bracket=yes}}
 
local p = {}
 
-- Gets a mw.title object, using pcall to avoid generating script errors if we
-- are over the expensive function count limit (among other possible causes).
local function getTitle(...)
local success, titleObj = pcall(mw.title.new, ...)
Line 18 ⟶ 14:
end
 
-- Gets the name of a page that a redirect leads to, or nil if it isn't a
local function decode(s)
-- redirect.
-- Decodes html entities and percent encodings.
function p.getTargetFromText(text)
s = mw.text.decode(s, true)
local target = string.match(
return mw.uri.decode(s, 'WIKI')
text,
"^%s*#[Rr][Ee][Dd][Ii][Rr][Ee][Cc][Tt]%s*:?%s*%[%[([^%[%]|]-)%]%]"
) or string.match(
text,
"^%s*#[Rr][Ee][Dd][Ii][Rr][Ee][Cc][Tt]%s*:?%s*%[%[([^%[%]|]-)|[^%[%]]-%]%]"
)
return target and mw.uri.decode(starget, 'WIKIPATH')
end
 
-- Gets the target of a redirect. If the page specified is not a redirect,
function p.getTarget(page)
-- returns nil.
function p.getTarget(page, fulltext)
-- Get the title object. Both page names and title objects are allowed
-- as input.
Line 39 ⟶ 44:
), 2)
end
if not titleObj or not titleObj.isRedirect then
return nil
end
local targetTitle = titleObj.redirectTarget
if targettargetTitle then
-- Find the target by using string matching on the page content.
if fulltext then
local target = string.match(
return targetTitle.fullText
titleObj:getContent() or "",
else
"^%s*#[Rr][Ee][Dd][Ii][Rr][Ee][Cc][Tt]%s*:?%s*%[%[([^%[%]]-)%]%]"
titleObjreturn targetTitle.prefixedText
)
end
if target then
return decode(target)
else
return nil
-- The page is a redirect, but matching failed. This indicates a bug in
-- the redirect matching pattern, so throw an error.
error(string.format(
'could not parse redirect on page "%s"',
titleObj.prefixedText
))
end
end
 
--[[
function p._main(rname, bracket)
-- Given a single page name determines what page it redirects to and returns the target page name, or the
-- target page name, or the passed page name when not a redirect. The passed
-- passed page name when not a redirect. The passed page name can be given as plain text or as a page link.
--
-- Returns page name as plain text, or when the bracket parameter is given, as a page link. Returns an
-- page link. Returns an error message when page does not exist or the redirect
-- error message when page does not exist or the redirect target cannot be determined for some reason.
--]]
function p._mainluaMain(rname, bracket, fulltext)
if type(rname) ~= "string" or not rname:find("%S") then
return nil
Line 66 ⟶ 74:
bracket = bracket and "[[%s]]" or "%s"
rname = rname:match("%[%[(.+)%]%]") or rname
local target = p.getTarget(rname, fulltext)
iflocal ret = target thenor rname
ret = getTitle(ret)
return bracket:format(target)
if ret then
if fulltext then
ret = ret.fullText
else
ret = ret.prefixedText
end
return bracket:format(targetret)
else
return nil
rname = decode(rname)
return bracket:format(rname)
end
end
 
-- Provides access to the luaMain function from wikitext.
function p.main(frame)
local args = require('Module:Arguments').getArgs(frame, {frameOnly = true})
return p._mainluaMain(args[1], args.bracket, args.fulltext) or ''
end
 
-- Returns true if the specified page is a redirect, and false otherwise.
function p.luaIsRedirect(page)
local titleObj = getTitle(page)
if not titleObj then
return false
end
return titleObj.isRedirect
end
 
-- Provides access to the luaIsRedirect function from wikitext, returning 'yes'
-- if the specified page is a redirect, and the blank string otherwise.
function p.isRedirect(frame)
local args = require('Module:Arguments').getArgs(frame, {frameOnly = true})
if p.luaIsRedirect(args[1]) then
return 'yes'
else
return ''
end
end