Module:Sandbox/Ita140188/Section sizes

]]

local function size (frame)

local A = {}; -- table to hold section names and sizes

local section_name_list = {} -- an interim list that holds just the section names

local section_content; -- section content used for counting

local section = '_LEAD_'; -- lead section doen't have a heading

local count = {}; -- number of bytes in a section including the header text

local cumcount = {};

local lastlevel;

local maxlevels;

local levelcounts = {};

local upperlevel;

local highlight;

local highlightc;

local total; -- sum of all byte counts

local max; -- largest section so far encountered

local cummax; -- largest section so far encountered (cumulative)

local _; -- dummy for using gsub to count bytes

local lang = mw.language.getContentLanguage(); -- language object for number formatting appropriate to local language

local s; -- start position of found heading (returned from string.find())

local e = 1; -- end position of found heading (returned from string.find())

local section_name; -- captured heading name (returned from string.find())

local level = {}; -- number of leading '=' in heading markup; used for indenting subsections in the rendered list

local wl_name; -- anchor and display portion for wikilinks in rendered list

local content = mw.title.new (frame.args[1]):getContent(); -- get unparsed wikitext from the article

if not content then

return 'error: no article:' .. frame.args[1] .. '';

end

if content:find ('#REDIRECT') then -- redirects don't have sections

return 'error: ' .. frame.args[1] .. ' is a redirect';

end

section_content = content:match ('(.-)===*'); -- get the lead section

if section_content then

_, count[0] = section_content:gsub ('.', '%1'); -- count the size of the lead section

else

return 'error: no sections found in: ' .. frame.args[1] .. '';

end

total = count[0];

max = count[0];

table.insert (A, make_wikilink (frame.args[1], section) .. '|| style="text-align:right"|' .. lang:formatNum (count[0]) .. '|| style="text-align:right"|' .. lang:formatNum (count[0]));

while (1) do -- done this way because some articles reuse section names

s, e, section_name = string.find (content, '\n==+ *(.-) *==+', e); -- get start, end, and section name beginning a end of last find; newline must precede '==' heading markup

if s then

table.insert (section_name_list, {section_name, s}); -- save section name and start location of this find

else

break;

end

end

for i, section_name in ipairs (section_name_list) do

local escaped_section_name = string.gsub (section_name[1], '([%(%)%.%%%+%-%*%?%[%^%$%]])', '%%%1'); -- escape lua patterns in section name

local pattern = '(==+ *' .. escaped_section_name .. ' *==+.-)==+'; -- make a pattern to get the content of a section

section_content = string.match (content, pattern, section_name[2]); -- get the content beginning at the string.find() start location

if section_content then

_, count[i] = section_content:gsub ('.', '%1'); -- count the bytes in the section

total = total + count[i];

max = max < count[i] and count[i] or max; -- keep track of largest count

else -- probably the last section (no proper header follows this section name)

pattern = '(==+ *' .. escaped_section_name .. ' *==+.+)'; -- make a new pattern

section_content = string.match (content, pattern, section_name[2]); -- try to get content

if section_content then

_, count[i] = section_content:gsub ('.', '%1'); -- count the bytes in the section

total = total + count[i];

max = max < count[i] and count[i] or max; -- keep track of largest count

else

count[i] = '—'; -- no content so show that

end

end

_, level[i] = section_content:find ('^=+'); -- should always be the first n characters of section content

end

cummax=0;

lastlevel=0;

maxlevels=7;

for j=1,maxlevels do

levelcounts[j]=0;

end

for i=#count,1,-1 do

--cumcount[i]=level[i];

if level[i]

cumcount[i]=levelcounts[level[i]]+count[i];

for j=level[i],maxlevels do

levelcounts[j]=0;

end

end

if level[i]>=lastlevel then

cumcount[i]=count[i];

end

if level[i]>0 then

upperlevel=level[i]-1;

levelcounts[upperlevel]=levelcounts[upperlevel]+cumcount[i];

end

lastlevel=level[i];

if cumcount[i]>cummax then

cummax=cumcount[i];

end

end

for i, section_name in ipairs (section_name_list) do

if count[i]==max then

highlight='color:red;"|';

else

highlight='"|';

end

highlightc=''; -- start the style declaration

if level[i]==2 then

highlightc=highlightc .. 'font-weight:bold;'; -- if main section, make it bold

elseif cumcount[i]==count[i] then

highlightc='color:#f8f9fa;'; -- same color as background for wikitable class tables for subsections with no subsubsections

end

if cumcount[i]==cummax then

highlightc=highlightc .. 'color:red;'; -- if the largest size, make it red

end

highlightc=highlightc .. '"|'; -- close the style declaration

level[i] = (2 < level[i]) and ((level[i]-2) * 1.6) or nil; -- remove offset and mult by 1.6em (same indent as ':' markup which doesn't work in a table)

wl_name = remove_wiki_link (section_name[1]):gsub ('%b{}', ''); -- remove wikilinks and templates from section headings so that we can link to the section

wl_name = wl_name:gsub ('[%[%]]', {['[']='[', [']']=']'}); -- replace '[' and ']' characters with html entities so that wikilinked section names work

wl_name = mw.text.trim (wl_name); -- trim leading/trailing white space if any because white space buggers up url anchor links

table.insert (A, table.concat ({ -- build most of a table row here because here we have heading information that we won't have later

level[i] and '' or ''; -- indent per heading level (number of '=' in heading markup)

make_wikilink (frame.args[1] .. '#' .. wl_name, wl_name), -- section link

level[i] and '' or '', -- close the span if opened

'||', -- table column separator

'style="text-align:right;', -- the byte count column is right aligned

highlight ,

lang:formatNum (count[i]), -- commafied byte count for section

'||', -- table column separator

'style="text-align:right;', -- the byte count column is right aligned

highlightc ,

lang:formatNum (cumcount[i]), -- cumulative count!!

}));

end

local out = {}; -- make a sortable wikitable for output

table.insert (out, string.format ('

--max = lang:formatNum (max); -- commafy so that the commafied value in the table can be found

--local result = table.concat (out, ''):gsub (max, '' .. max .. ''); -- make a big string, make largest count(s) red, and done

local result = table.concat (out, '');

return result; -- because gsub returns string and number of replacements

end

--[[--------------------------< E X P O R T E D F U N C T I O N S >------------------------------------------

]]

return

{

size = size,

}