385
edits
(add getTarget and isRedirect functions, format comments) |
m (45 revisions imported from wikipedia:Module:Redirect: see Topic:Vtixlm0q28eo6jtf) |
||
(7 intermediate revisions by 5 users not shown) | |||
Line 12: | Line 12: | ||
return nil | return nil | ||
end | end | ||
end | |||
-- Gets the name of a page that a redirect leads to, or nil if it isn't a | |||
-- redirect. | |||
function p.getTargetFromText(text) | |||
local target = string.match( | |||
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(target, 'PATH') | |||
end | end | ||
-- Gets the target of a redirect. If the page specified is not a redirect, | -- Gets the target of a redirect. If the page specified is not a redirect, | ||
-- returns nil. | -- returns nil. | ||
function p.getTarget(page) | function p.getTarget(page, fulltext) | ||
-- Get the title object. Both page names and title objects are allowed | -- Get the title object. Both page names and title objects are allowed | ||
-- as input. | -- as input. | ||
Line 36: | Line 49: | ||
-- Find the target by using string matching on the page content. | -- Find the target by using string matching on the page content. | ||
local target = | local target = p.getTargetFromText(titleObj:getContent() or "") | ||
if target then | if target then | ||
local targetTitle = getTitle(target) | local targetTitle = getTitle(target) | ||
if targetTitle then | if targetTitle then | ||
return targetTitle.prefixedText | if fulltext then | ||
return targetTitle.fullText | |||
else | |||
return targetTitle.prefixedText | |||
end | |||
else | else | ||
return nil | return nil | ||
Line 52: | Line 66: | ||
error(string.format( | error(string.format( | ||
'could not parse redirect on page "%s"', | 'could not parse redirect on page "%s"', | ||
titleObj.prefixedText | fulltext and titleObj.fullText or titleObj.prefixedText | ||
)) | )) | ||
end | end | ||
Line 66: | Line 80: | ||
-- target cannot be determined for some reason. | -- target cannot be determined for some reason. | ||
--]] | --]] | ||
function p.luaMain(rname, bracket) | function p.luaMain(rname, bracket, fulltext) | ||
if type(rname) ~= "string" or not rname:find("%S") then | if type(rname) ~= "string" or not rname:find("%S") then | ||
return nil | return nil | ||
Line 72: | Line 86: | ||
bracket = bracket and "[[%s]]" or "%s" | bracket = bracket and "[[%s]]" or "%s" | ||
rname = rname:match("%[%[(.+)%]%]") or rname | rname = rname:match("%[%[(.+)%]%]") or rname | ||
local target = p.getTarget(rname) | local target = p.getTarget(rname, fulltext) | ||
local ret = target or rname | local ret = target or rname | ||
ret = getTitle(ret) | ret = getTitle(ret) | ||
if ret then | if ret then | ||
ret = ret.prefixedText | if fulltext then | ||
ret = ret.fullText | |||
else | |||
ret = ret.prefixedText | |||
end | |||
return bracket:format(ret) | return bracket:format(ret) | ||
else | else | ||
Line 86: | Line 104: | ||
function p.main(frame) | function p.main(frame) | ||
local args = require('Module:Arguments').getArgs(frame, {frameOnly = true}) | local args = require('Module:Arguments').getArgs(frame, {frameOnly = true}) | ||
return p.luaMain(args[1], args.bracket) or '' | return p.luaMain(args[1], args.bracket, args.fulltext) or '' | ||
end | end | ||
-- Returns | -- Returns true if the specified page is a redirect, and false otherwise. | ||
function p.luaIsRedirect(page) | |||
function p. | local titleObj = getTitle(page) | ||
local titleObj = getTitle( | |||
if not titleObj then | if not titleObj then | ||
return | return false | ||
end | end | ||
if titleObj.isRedirect then | if titleObj.isRedirect then | ||
return true | |||
else | |||
return false | |||
end | |||
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' | return 'yes' | ||
else | else |