Module:AutosortTable/sandbox

 --[[
 AutosortTable: Creates a table which is automatically sorted

 Usage: (Remove the hidden comments before use)

 {{#invoke: AutosortTable|create

 | class = wikitable <!-- Class for the entire table -->
 | style = width: 50%; <!-- CSS for the entire table -->
 | separator = -- <!-- Separator string used to separate cells; pipe (|) invalid -->
 | order = 2, 1 <!-- Order for sorting preference, takes a comma-separated list of column numbers -->
 | numeric = 2 <!-- Columns which use numeric sorting. Takes comma-separated list of column numbers -->
 | descending = 1 <!-- Columns for which sort order should be descending. Takes comma-separated list of col numbers -->
 | hidden = 2 <!-- Columns which are not to be displayed. Takes comma-separated list of col numbers -->
 | rowheader = 1 <!-- Cell(s) in each non-header row to be emitted as row header, per WP:ACCESS#Data tables. Usually just 1, but accepts comma-separated list of col numbers -->
 | caption = Notable people by age <!-- Table caption per WP:ACCESS -->
 | header = -- Name -- Age <!-- Table header -->
 | footer = <!-- Table footer, typically a totals row or duplication of header -->
 | -- Bob -- 20 <!-- Row 1 -->
 | -- Peter -- 35 <!-- Row 2 -->
 | -- John -- 35 <!-- Row 3 -->
 | -- James -- 50 <!-- Row 4 -->
 | background-color: #FFDDDD -- Henry -- 45 <!-- Row 5, with CSS -->
 | colstyle = -- text-align:left; -- text-align:right; -- -- --
 											 <!-- CSS to be used on content of respective columns, here 1st & 2nd -->

 }}
 ]]

 local_module={}

 _module.create=function(frame)

 localargs=frame.args

 -- Named parameters

 localclass=args.class
 localstyle=args.style
 localsep=args.separator
 localorder=args.order
 localdesc=args.descendingor""
 localnsort=args.numericor""
 localhidden=args.hiddenor""
 localheader=args.header
 localfooter=args.footer
 localcolstyle=args.colstyle
 localrowheader=args.rowheaderor""
 localcaption=args.caption

 -- Frequently-used functions

 localstrIndexOf=mw.ustring.find
 localstrSplit=mw.text.split
 localstrSub=mw.ustring.sub
 localstrTrim=mw.text.trim

 localseplen=#sep
 localnsortLookup,descLookup,hiddenLookup,rowHeading={},{},{},{}

 -- Create the table

 localhtml=mw.html.create()
 localhtable=html:tag('table')
 ifclassthenhtable:attr('class',class)end
 ifstylethenhtable:attr('style',style)end
 ifcaptionthen
 localhcaption=htable:tag('caption')
 hcaption:wikitext(caption)
 end

 -- Parses a row string. The 'key' parameter is used to assign a unique key to the result so that equal rows do not cause sort errors.
 localparse=function(s,key)
 localcss
 localfirstSep=strIndexOf(s,sep,1,true)
 iffirstSep==1then-- no CSS
 css=nil
 s=strSub(s,seplen+1,-1)
 else-- CSS before first separator
 css=strSub(s,1,firstSep-1)
 s=strSub(s,firstSep+seplen,-1)
 end
 return{key=key,css=css,data=strSplit(s,sep,true)}
 end

 --[[
  Writes a row to the table.
  css: CSS to apply to the row
  data: The data (cells) of the row
  _type: Can be 'header', 'footer' or nil.
  ]]
 localwriteHtml=function(css,data,_type)
 localrow=htable:tag('tr')
 ifcssthenrow:attr('style',strTrim(css))end

 fori,vinipairs(data)do
 ifnothiddenLookup[i]then
 localcell
 if_type=='header'then
 -- Header: use the 'th' tag with scope="col"
 cell=row:tag('th')
 cell:attr('scope','col')
 elseif_type=='footer'then
 -- Footer: Mark as 'sortbottom' so that it does not sort when the table is made user-sortable
 -- with the 'wikitable sortable' class
 cell=row:tag('td')
 cell:attr('class','sortbottom')
 else
 ifrowHeading[i]then
 -- Cell is a row heading
 cell=row:tag('th')
 cell:attr('scope','row')
 else
 -- Ordinary cell
 cell=row:tag('td')
 end
 localcellCss=colstyleandcolstyle[i]
 ifcellCssthencell:attr('style',strTrim(cellCss))end-- Apply the column styling, if necessary
 end
 cell:wikitext(strTrim(v))
 end
 end
 returnrow
 end

 -- Parse the column styles
 ifcolstylethencolstyle=parse(colstyle,-1).dataend

 -- Write the header first
 ifheaderthen
 localheaderData=parse(header)
 writeHtml(headerData.css,headerData.data,'header')
 end

 -- Parse the data
 localdata={}
 fori,vinipairs(frame.args)dodata[i]=parse(v,i)end

 order=strSplit(order,'%s*,%s*')
 nsort=strSplit(nsort,'%s*,%s*')
 desc=strSplit(desc,'%s*,%s*')
 hidden=strSplit(hidden,'%s*,%s*')
 rowheader=strSplit(rowheader,'%s*,%s*')

 fori,vinipairs(order)doorder[i]=tonumber(v)end
 fori,vinipairs(nsort)donsortLookup[tonumber(v)or-1]=trueend
 fori,vinipairs(desc)dodescLookup[tonumber(v)or-1]=trueend
 fori,vinipairs(hidden)dohiddenLookup[tonumber(v)or-1]=trueend
 fori,vinipairs(rowheader)dorowHeading[tonumber(v)or-1]=trueend

 --Sorting comparator function.
 localsortFunc=function(a,b)
 localad,bd=a.data,b.data
 fori=1,#orderdo
 localindex=order[i]
 localai,bi=ad[index],bd[index]
 ifnsortLookup[index]then
 -- Numeric sort. Find the first occurrence of a number and use it. Decimal points are allowed. Scientific notation not supported.
 ai=tonumber((ai:find('.',1,true)andai:match('[+-]?%d*%.%d+')orai:match('[+-]?%d+'))or0)
 bi=tonumber((bi:find('.',1,true)andbi:match('[+-]?%d*%.%d+')orbi:match('[+-]?%d+'))or0)
 end
 ifai~=bithen
 ifdescLookup[index]thenreturnai>bielsereturnai<biend
 end
 end
 returna.key<b.key
 end
 table.sort(data,sortFunc)

 -- Write the sorted data to the HTML output
 fori,vinipairs(data)dowriteHtml(v.css,v.data,nil)end

 -- Write the footer
 iffooterthen
 localfooterData=parse(footer)
 writeHtml(footerData.css,footerData.data,'footer')
 end

 returntostring(html)
 end

 return_module