|
-- 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
function warOnGsub(text, repl)
-- are over the expensive function count limit (among other possible causes).
if repl then
local function getTitle(...)
text = mw.ustring.gsub(text, "%%", "%%%%")
local success, titleObj = pcall(mw.title.new, ...)
if success then
return titleObj
else
return nil
text = mw.ustring.gsub(text, "([%?%^%$%(%)%%%.%[%]%*%+%-%]])", "%%%1")
end
return text
end
-- Gets the name of a page that a redirect leads to, or nil if it isn't a
function getArgs(frame)
-- redirect.
-- If called via #invoke, use the args passed into the invoking
function p.getTargetFromText(text)
-- template, or the args passed to #invoke if any exist. Otherwise
local target = string.match(
-- assume args are being passed directly in from the debug console
text,
-- or from another Lua module.
"^%s*#[Rr][Ee][Dd][Ii][Rr][Ee][Cc][Tt]%s*:?%s*%[%[([^%[%]|]-)%]%]"
local origArgs
) or string.match(
if frame == mw.getCurrentFrame() then
text,
origArgs = frame:getParent().args
"^%s*#[Rr][Ee][Dd][Ii][Rr][Ee][Cc][Tt]%s*:?%s*%[%[([^%[%]|]-)|[^%[%]]-%]%]"
for k, v in pairs( frame.args ) do
)
origArgs = frame.args
return target and mw.uri.decode(target, 'PATH')
break
end
else
origArgs = frame
end
-- Trim whitespace and remove blank arguments.
local args = {}
for k, v in pairs( origArgs ) do
v = mw.text.trim( v )
if v ~= '' then
args[k] = v
end
end
return args
end
-- Gets the target of a redirect. If the page specified is not a redirect,
function getRedirect(rname)
-- returns nil.
-- Get the title object, passing the function through pcall
function p.getTarget(page, fulltext)
-- in case we are over the expensive function count limit.
-- Get the title object. Both page names and title objects are allowed
-- returning nil indicates the redirect did not parse, but nonexistent or non-redirect pages are not nil.
-- as input.
local noError, rpage = pcall(mw.title.new, rname)
local titleObj
if not noError or noError and not rpage or not rpage.isRedirect then
if type(page) == 'string' or type(page) == 'number' then
-- mw.title.new failed, or the page is not a redirect, so use the passed page name.
titleObj = getTitle(page)
return rname
elseif type(page) == 'table' and type(page.getContent) == 'function' then
end
titleObj = page
else
error(string.format(
"bad argument #1 to 'getTarget'"
.. " (string, number, or title object expected, got %s)",
type(page)
), 2)
end
if not titleObj then
return nil
end
local targetTitle = titleObj.redirectTarget
if targetTitle then
if fulltext then
return targetTitle.fullText
else
return targetTitle.prefixedText
end
else
return nil
end
end
--[[
local redirect = mw.ustring.match(rpage:getContent() or "", "^%s*#[Rr][Ee][Dd][Ii][Rr][Ee][Cc][Tt]%s*:?%s*%[%[([^%[%]]-)%]%]" )
-- Given a single page name determines what page it redirects to and returns the
if redirect then
-- target page name, or the passed page name when not a redirect. The passed
-- Decode html entities and percent encodings.
-- page name can be given as plain text or as a page link.
redirect = mw.text.decode(redirect, true)
--
redirect = mw.uri.decode(redirect, 'WIKI')
-- Returns page name as plain text, or when the bracket parameter is given, as a
end
-- page link. Returns an error message when page does not exist or the redirect
return redirect
-- target cannot be determined for some reason.
--]]
function p.luaMain(rname, bracket, fulltext)
if type(rname) ~= "string" or not rname:find("%S") then
return nil
end
bracket = bracket and "[[%s]]" or "%s"
rname = rname:match("%[%[(.+)%]%]") or rname
local target = p.getTarget(rname, fulltext)
local ret = target or rname
ret = getTitle(ret)
if ret then
if fulltext then
ret = ret.fullText
else
ret = ret.prefixedText
end
return bracket:format(ret)
else
return nil
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.luaMain(args[1], args.bracket, args.fulltext) or ''
local args = getArgs(frame)
local rname, bracket = args[1], args.bracket
bracket = bracket and "[[%s]]" or "%s"
if type(rname) ~= "string" or not mw.ustring.match(rname, "%S") then return end
rname = mw.ustring.match(rname, "%[%[(.+)%]%]") or rname
local redirect = getRedirect(rname)
if redirect then
return mw.ustring.format(bracket, redirect)
else
return mw.ustring.format('<span class="error">[[Module:redirect]] error: could not parse redirect - [[%s]]</span>', rname)
end
end
-- Returns true if the specified page is a redirect, and false otherwise.
function getPage(sourceName)
function p.luaIsRedirect(page)
local titleObj = getTitle(page)
if type(sourceName) ~= "string" or not mw.ustring.match(sourceName, "%S") then return end
if not titleObj then
-- the following delink doesn't make as much sense for this usage but is done for consistency.
return false
sourceName = mw.ustring.match(sourceName, "%[%[(.+)%]%]") or sourceName
local noError, title = pcall(mw.title.new, sourceName)
if not noError then
text = nil -- should be anyway; nil means error
else
text = title:getContent()
end
return text
end
function nowikize(frame, text, nowiki, default)
if (default or nowiki) and (nowiki ~= 'no') then
text = frame:preprocess('<pre><nowiki>'..text..'</nowiki></pre>')
end
return texttitleObj.isRedirect
end
-- Provides access to the luaIsRedirect function from wikitext, returning 'yes'
function p.block(frame)
-- if the specified page is a redirect, and the blank string otherwise.
-- this feature takes an initial page as an argument. It obtains all the links from that page
function p.isRedirect(frame)
-- starting from the first instance of some link named as start=, and replaces every link after
local args = require('Module:Arguments').getArgs(frame, {frameOnly = true})
-- that which is a redirect with the non-redirected canonical name, up until pcall throws an error
if p.luaIsRedirect(args[1]) then
-- due to expense. Finally, it returns the substituted page.
return 'yes'
else
local args = getArgs(frame)
return ''
local sourceName, start, text, nowiki = args[1], args[2], args.text, args.nowiki
end
if type(sourceName) == "string" then sourceName = mw.text.trim(sourceName) end
if type(start) == "string" then start = mw.text.trim(start) end
if type(text) ~= "string" then
text = getPage(sourceName)-- we're getting text = the contents of the page at args[1]
end
if not(text) then -- nothing to work from
return -- optional error here
end
local originalText = text -- I don't want to even think about conflicts...
local nextLink = mw.ustring.gmatch(originalText, "%[%[([^%]|]*)|?([^%]]-)%]%]")
local link = " " -- true
local display = "" -- false
while link do
if not(start) and link ~= " " then
newLink = getRedirect(link)
if newLink and (newLink ~= link) then
link = warOnGsub(link, nil)
newLink = warOnGsub(newLink, true)
text = mw.ustring.gsub(text, link, newLink, 1) or text -- no sense looking past 1, we'll be back...
end
end
link, display = nextLink()
if link then link = mw.text.trim(link) or link end
if start and link then
if start == link then
start = nil
else
link = " " -- discard all links before the one to start with
end
end
end
return nowikize(frame, text, nowiki, true)
end
| 