Module:ProjectCounts: Difference between revisions

Newer edit →

@@ Line 1: / Line 1: @@
--- Counts data rows in the wikitable(s) inside one named section of a page.
+-- Module:ProjectCounts
--- Usage: {{#invoke:ProjectCounts|count|Active BOINC Projects|page=BOINC projects}}
 local p = {}
 local function esc(s) return (s:gsub('(%W)', '%%%1')) end
+local function getContent(page)
+	local t = mw.title.new(page)
+	if not t then return nil end
+	return t:getContent()
+end
+-- Extract the wikitext of one section: from its heading (any level)
+-- to the next heading of the same or higher level.
+local function getSection(content, section)
+	local hStart, hEnd, marks = content:find('\n(=+)%s*' .. esc(section) .. '%s*=*')
+	if not hStart then return nil end
+	local depth = #marks
+	local secEnd = #content + 1
+	local pos = hEnd
+	while true do
+		local nStart, nEnd, nMarks = content:find('\n(=+)', pos)
+		if not nStart then break end
+		if #nMarks <= depth then secEnd = nStart break end
+		pos = nEnd
+	end
+	return '\n' .. content:sub(hEnd + 1, secEnd - 1)
+end
+-- Data rows = row separators minus one header separator per table.
+local function countRows(body)
+	local _, tables = body:gsub('\n{|', '')
+	local _, seps   = body:gsub('\n|%-', '')
+	return math.max(seps - tables, 0)
+end
+local function countSection(page, section)
+	local content = getContent(page)
+	if not content then return 0 end
+	local body = getSection(content, mw.text.trim(section))
+	if not body then return 0 end
+	return countRows(body)
+end
+-- {{#invoke:ProjectCounts|count|Section name|page=Page title}}
 function p.count(frame)
-    local section = mw.text.trim(frame.args[1] or '')
+	return countSection(frame.args.page or 'BOINC projects', frame.args[1] or '')
-    local title = mw.title.new(frame.args.page or 'BOINC projects')
+end
-    if not title then return 0 end
-    local content = title:getContent()
-    if not content then return 0 end
-    -- Find the heading line, any level: ==== Active BOINC Projects ====
-    local hStart, hEnd, marks = content:find('\n(=+)%s*' .. esc(section) .. '%s*=*')
-    if not hStart then return 0 end
-    local depth = #marks
-    -- Section ends at the next heading of the same or higher level
+-- {{#invoke:ProjectCounts|total|Section A|Section B|...|page=Page title}}
-    local secEnd = #content + 1
+function p.total(frame)
-    local pos = hEnd
+	local page = frame.args.page or 'BOINC projects'
-    while true do
+	local content = getContent(page)
-        local nStart, nEnd, nMarks = content:find('\n(=+)', pos)
+	if not content then return 0 end
-        if not nStart then break end
+	local sum = 0
-        if #nMarks <= depth then secEnd = nStart break end
+	for _, s in ipairs(frame.args) do
-        pos = nEnd
+		local body = getSection(content, mw.text.trim(s))
-    end
+		if body then sum = sum + countRows(body) end
+	end
+	return sum
+end
-    local body = '\n' .. content:sub(hEnd + 1, secEnd - 1)
+-- {{#invoke:ProjectCounts|documented|Local BOINC project wiki pages|page=Main Page}}
-    local _, tables = body:gsub('\n{|', '')
+-- Counts distinct mainspace article links in a section, skipping redlinks.
-    local _, seps   = body:gsub('\n|%-', '')
+function p.documented(frame)
-    -- each table has one header-row separator; the rest are data rows
+	local page = frame.args.page or 'Main Page'
-    return math.max(seps - tables, 0)
+	local section = mw.text.trim(frame.args[1] or 'Local BOINC project wiki pages')
+	local content = getContent(page)
+	if not content then return 0 end
+	local body = getSection(content, section)
+	if not body then return 0 end
+	local seen, n = {}, 0
+	for link in body:gmatch('%[%[([^|%]%[#:]+)') do
+		link = mw.text.trim(link)
+		if link ~= '' and not seen[link] then
+			seen[link] = true
+			local t = mw.title.new(link)
+			if t and t.namespace == 0 and t.exists then
+				n = n + 1
+			end
+		end
+	end
+	return n
 end
 return p