Module:NUMBEROFSECTIONS: Difference between revisions

Content deleted Content added
slight improvement for handling unexpected-and-wanted whitespace around the # delimiters; %s? --> %s* i.e. possible single whitespace char --> zero or more whitespace chars
 
(10 intermediate revisions by 5 users not shown)
Line 1:
local p = {}
 
-- Unescape functionality grabbed from https://stackoverflow.com/a/14899740/1832568
local function unescape(str)
str = string.gsub(str, '&#(%d+);', string.char)
str = string.gsub(str, '&#x(%d+);', function(n) return string.char(tonumber(n, 16)) end)
return str
end
 
-- Counting function accepting a string haystack and table of needles
Line 15 ⟶ 22:
 
-- Function takes any number of # delimited page names and section level numbers
function p.sectionsmain(frame)
local total = 0
local needles = {}
local haystack = ''
-- Separate page names from # delimited string into table
local titlespages = mw.text.split(unescape(frame.args[1]), '%s*#%s*')
titles = titles:gsub("'", "'") -- workaround astonishing bug: {{FULLPAGENAME}} gives "'" for apostrophe!
local pages = mw.text.split(titles, '%s?#%s?')
-- Separate whitespace delimited section level numbers into table
local levels = mw.text.split(frame.args['level'], '%s*')
Line 38 ⟶ 43:
--[[ pass the raw markup and needles to count
and add the return to total ]]
total = total + count('\n' .. haystack, needles)
end
end