Difference between revisions of "Module:MultiReplace"
Jump to navigation
Jump to search
blackwiki>Ahecht (rm Superfluous input sanitization) |
blackwiki>Petr Matas (Better readable p.main) |
||
| Line 47: | Line 47: | ||
function p.main(frame) | function p.main(frame) | ||
local args | local args | ||
| − | if type(frame.args) | + | if type(frame.args) ~= 'table' then |
| − | + | args = frame | |
| − | + | elseif frame.args[1] then | |
| − | + | args = frame.args | |
| − | |||
| − | |||
else | else | ||
| − | args = frame | + | args = frame:getParent().args |
end | end | ||
return MultiReplace(args) | return MultiReplace(args) | ||
Revision as of 18:55, 25 April 2018
p = {}
local function MultiReplace(args)
local input = args[1] or "{{{1}}}"
local plain = args.plain == "yes"
local i = 1
local changeList = {}
while args[i * 2] do
local change = {pattern = args[i * 2], repl = args[i * 2 + 1]}
if not change.repl then
return require('Module:Error').error{
'MultiReplace: Unpaired argument: <code>' .. (i * 2) .. ' = ' .. change.pattern .. '</code>'
}
end
changeList[i] = change
i = i + 1
end
local matchList = {}
local pos = 1
local len = mw.ustring.len(input)
local result = ""
while pos <= len do
local bestStart = len + 1
local bestStop = len
local bestChange
for _, change in ipairs(changeList) do
local start, stop = mw.ustring.find(input, change.pattern, pos, plain)
if start and (start < bestStart) then
bestStart = start
bestStop = stop
bestChange = change
end
end
result = result .. mw.ustring.sub(input, pos, bestStart - 1)
if bestChange then
local fragment = mw.ustring.sub(input, bestStart, bestStop)
result = result .. (plain and bestChange.repl or
mw.ustring.gsub(fragment, bestChange.pattern, bestChange.repl, 1))
end
pos = bestStop + 1
end
return result
end
function p.main(frame)
local args
if type(frame.args) ~= 'table' then
args = frame
elseif frame.args[1] then
args = frame.args
else
args = frame:getParent().args
end
return MultiReplace(args)
end
return p