Module:NUMBEROFSECTIONS: Difference between revisions

Content deleted Content added
a few simple comments and preparedness for expansion
slight improvement for handling unexpected-and-wanted whitespace around the # delimiters; %s? --> %s* i.e. possible single whitespace char --> zero or more whitespace chars
 
(19 intermediate revisions by 6 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
local function count(haystack, needles)
local number = 0
local index = 1
local needle = needles[index]
-- While we have needles to look for
whilefor index, needle in ipairs(needles) do
-- find them all in our haystack
for m in string.gmatch(haystack, needle) do
number = number + 1
end
index = index + 1
needle = needles[index]
end
return number
end
 
-- Function acceptingtakes aany number of # delimited page namenames and section level numbers
function p.sectionsmain(frame)
local indextotal = 10
local needles = {}
local levelshaystack = frame.args[2]''
-- Separate page names from # delimited string into table
local title = mw.title.new(frame.args[1])
local pages = mw.text.split(unescape(frame.args[1]), '%s*#%s*')
-- For every section level number
-- Separate whitespace delimited section level numbers into table
for level in mw.text.gsplit(levels, "") do
local titlelevels = mw.titletext.newsplit(frame.args[1'level'], '%s*')
if level ~= " " then
-- Iterate through levels
-- add the needle to our table of needles
for level in mw.text.gsplit(table.concat(levels), ""'') do
needles[#needles + 1] = "\n" ..
-- and add the level needle to our table of needles
string.rep("=", tonumber(level)) .. "[^=]"
needles[#needles + 1] = "'\n"'..string.rep('=', tonumber(level))..'[^=]'
end
-- For each page name in pages
for index, page in ipairs(pages) do
-- create a haystack to search from the page content
haystack = mw.title.new(page):getContent()
-- If we've requested the content of a legitimate page
if level ~= " "haystack then
--[[ pass the raw markup and needles to count
and add the return to total ]]
total = total + count('\n' .. haystack, needles)
end
end
--[[ then return how many needlessections areof inthe ourrequired haystacklevel
are in all the pages passed ]]
return count(title:getContent(), needles)
return total
end