Module:Redirect: Difference between revisions
From Test Wiki
Wikipedia >Johnuniq (restore p.getTargetFromText which is used by Module:RfD which is causing "Lua error in Module:RfD at line 87: attempt to call upvalue 'getTargetFromText' (a nil value)") |
m (1 revision imported) Tags: Mobile edit Mobile web edit |
||
(One intermediate revision by one other user not shown) | |||
Line 12: | Line 12: | ||
return nil | return nil | ||
end | end | ||
end | |||
local function case_insensitive_pattern(pattern) | |||
-- find an optional '%' (group 1) followed by any character (group 2) | |||
local p = mw.ustring.gsub(pattern, "(%%?)(.)", function(percent, letter) | |||
if percent ~= "" or not mw.ustring.match(letter, "%a") then | |||
-- if the '%' matched, or `letter` is not a letter, return "as is" | |||
return percent .. letter | |||
else | |||
-- else, return a case-insensitive character class of the matched letter | |||
return string.format("[%s%s]", mw.ustring.lower(letter), mw.ustring.upper(letter)) | |||
end | |||
end) | |||
return p | |||
end | end | ||
Line 17: | Line 31: | ||
-- redirect. | -- redirect. | ||
function p.getTargetFromText(text) | function p.getTargetFromText(text) | ||
local | local en_redirect = case_insensitive_pattern("redirect"); | ||
local ru_redirect = case_insensitive_pattern("перенаправление"); | |||
local result = string.match( | |||
text, | text, | ||
"^%s*# | "^%s*#".. en_redirect .. "%s*:?%s*%[%[([^%[%]|]-)%]%]" | ||
) or string.match( | ) or string.match( | ||
text, | text, | ||
"^%s*#[ | "^%s*#".. en_redirect .. "%s*:?%s*%[%[([^%[%]|]-)|[^%[%]]-%]%]" | ||
) or mw.ustring.match( | |||
text, | |||
"^%s*#" .. ru_redirect .. "%s*:?%s*%[%[([^%[%]|]-)%]%]" | |||
) or mw.ustring.match( | |||
text, | |||
"^%s*#" .. ru_redirect .. " %s*:?%s*%[%[([^%[%]|]-)|[^%[%]]-%]%]" | |||
) | ) | ||
return | return result and mw.uri.decode(result, 'WIKI') | ||
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) | ||
-- 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 44: | Line 66: | ||
), 2) | ), 2) | ||
end | end | ||
if not titleObj then | if not titleObj or not titleObj.isRedirect then | ||
return nil | return nil | ||
end | end | ||
local | |||
if targetTitle | -- Find the target by using string matching on the page content. | ||
if | local target = p.getTargetFromText(titleObj:getContent() or "") | ||
return targetTitle. | if target then | ||
local targetTitle = getTitle(target) | |||
if targetTitle then | |||
return targetTitle.prefixedText | |||
else | else | ||
return | return nil | ||
end | end | ||
else | else | ||
-- 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 | ||
end | end | ||
Line 63: | Line 93: | ||
-- target page name, or the passed page name when not a redirect. The passed | -- 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. | -- 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 | -- 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 | -- page link. Returns an error message when page does not exist or the redirect | ||
-- target cannot be determined for some reason. | -- target cannot be determined for some reason. | ||
--]] | --]] | ||
function p.luaMain(rname, bracket | function p.luaMain(rname, bracket) | ||
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 74: | Line 104: | ||
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) | ||
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 | |||
return bracket:format(ret) | return bracket:format(ret) | ||
else | else | ||
Line 92: | Line 118: | ||
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 | return p.luaMain(args[1], args.bracket) or '' | ||
end | end | ||
Latest revision as of 06:56, 11 April 2024
Documentation for this module may be created at Module:Redirect/doc
-- This module provides functions for getting the target of a redirect page.
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, ...)
if success then
return titleObj
else
return nil
end
end
local function case_insensitive_pattern(pattern)
-- find an optional '%' (group 1) followed by any character (group 2)
local p = mw.ustring.gsub(pattern, "(%%?)(.)", function(percent, letter)
if percent ~= "" or not mw.ustring.match(letter, "%a") then
-- if the '%' matched, or `letter` is not a letter, return "as is"
return percent .. letter
else
-- else, return a case-insensitive character class of the matched letter
return string.format("[%s%s]", mw.ustring.lower(letter), mw.ustring.upper(letter))
end
end)
return p
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 en_redirect = case_insensitive_pattern("redirect");
local ru_redirect = case_insensitive_pattern("перенаправление");
local result = string.match(
text,
"^%s*#".. en_redirect .. "%s*:?%s*%[%[([^%[%]|]-)%]%]"
) or string.match(
text,
"^%s*#".. en_redirect .. "%s*:?%s*%[%[([^%[%]|]-)|[^%[%]]-%]%]"
) or mw.ustring.match(
text,
"^%s*#" .. ru_redirect .. "%s*:?%s*%[%[([^%[%]|]-)%]%]"
) or mw.ustring.match(
text,
"^%s*#" .. ru_redirect .. " %s*:?%s*%[%[([^%[%]|]-)|[^%[%]]-%]%]"
)
return result and mw.uri.decode(result, 'WIKI')
end
-- Gets the target of a redirect. If the page specified is not a redirect,
-- returns nil.
function p.getTarget(page)
-- Get the title object. Both page names and title objects are allowed
-- as input.
local titleObj
if type(page) == 'string' or type(page) == 'number' then
titleObj = getTitle(page)
elseif type(page) == 'table' and type(page.getContent) == 'function' then
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 or not titleObj.isRedirect then
return nil
end
-- Find the target by using string matching on the page content.
local target = p.getTargetFromText(titleObj:getContent() or "")
if target then
local targetTitle = getTitle(target)
if targetTitle then
return targetTitle.prefixedText
else
return nil
end
else
-- 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
--[[
-- 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.
--]]
function p.luaMain(rname, bracket)
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)
local ret = target or rname
ret = getTitle(ret)
if ret then
ret = ret.prefixedText
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) 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
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'
else
return ''
end
end
return p