Difference between revisions of "Module:Hatnote list"

From Eat Every Plant
Jump to navigation Jump to search
m (1 revision)
 
m (1 revision: 'Acacia')
(One intermediate revision by the same user not shown)
Line 45: Line 45:
 
-- Set the separator; if any item contains it, use the alternate separator
 
-- Set the separator; if any item contains it, use the alternate separator
 
local separator = options.separator
 
local separator = options.separator
 +
--searches display text only
 +
function searchDisp(t, f)
 +
return string.find(string.sub(t, (string.find(t, '|') or 0) + 1), f)
 +
end
 
for k, v in pairs(list) do
 
for k, v in pairs(list) do
if string.find(v, separator) then
+
if searchDisp(v, separator) then
 
separator = options.altSeparator
 
separator = options.altSeparator
 
break
 
break
Line 53: Line 57:
 
-- Set the conjunction, apply Oxford comma, and force a comma if #1 has "§"
 
-- Set the conjunction, apply Oxford comma, and force a comma if #1 has "§"
 
local conjunction = s .. options.conjunction .. s
 
local conjunction = s .. options.conjunction .. s
if #list == 2 and string.find(list[1], "§") or #list > 2 then
+
if #list == 2 and searchDisp(list[1], "§") or #list > 2 then
 
conjunction = separator .. conjunction
 
conjunction = separator .. conjunction
 
end
 
end
Line 60: Line 64:
 
end
 
end
  
-- Stringifies a list with "and"
+
--DRY function
function p.andList (list, formatted)
+
function conjList (conj, list, fmt)
return stringifyList(list, {conjunction = "and", formatted = formatted})
+
return stringifyList(list, {conjunction = conj, formatted = fmt})
 
end
 
end
  
-- Stringifies a list with "or"
+
-- Stringifies lists with "and" or "or"
function p.orList (list, formatted)
+
function p.andList (...) return conjList("and", ...) end
return stringifyList(list, {conjunction = "or", formatted = formatted})
+
function p.orList (...) return conjList("or", ...) end
end
 
  
 
--------------------------------------------------------------------------------
 
--------------------------------------------------------------------------------
Line 79: Line 82:
 
--default options table used across the forSee family of functions
 
--default options table used across the forSee family of functions
 
local forSeeDefaultOptions = {
 
local forSeeDefaultOptions = {
 +
andKeyword = 'and',
 
title = mw.title.getCurrentTitle().text,
 
title = mw.title.getCurrentTitle().text,
 
otherText = 'other uses',
 
otherText = 'other uses',
forseeForm = 'For %s, see %s.',
+
forSeeForm = 'For %s, see %s.'
punctuationCollapseReplacements = {
+
}
 +
 
 +
--Collapses duplicate punctuation
 +
function punctuationCollapse (text)
 +
local replacements = {
 
["%.%.$"] = ".",
 
["%.%.$"] = ".",
 
["%?%.$"] = "?",
 
["%?%.$"] = "?",
Line 90: Line 98:
 
["%!%]%]%.$"] = "!]]"
 
["%!%]%]%.$"] = "!]]"
 
}
 
}
}
+
for k, v in pairs(replacements) do text = string.gsub(text, k, v) end
 +
return text
 +
end
  
 
-- Structures arguments into a table for stringification, & options
 
-- Structures arguments into a table for stringification, & options
Line 108: Line 118:
 
if type(k) == 'number' and k > maxArg then maxArg = k end
 
if type(k) == 'number' and k > maxArg then maxArg = k end
 
end
 
end
-- Structure the data out from the parameter list
+
-- Structure the data out from the parameter list:
-- forTable is the wrapper table, with forRow rows
+
-- * forTable is the wrapper table, with forRow rows
-- Rows are tables of a "use" string and a "pages" table of pagename strings
+
-- * Rows are tables of a "use" string & a "pages" table of pagename strings
 +
-- * Blanks are left empty for defaulting elsewhere, but can terminate list
 
local forTable = {}
 
local forTable = {}
 
local i = from
 
local i = from
 
local terminated = false
 
local terminated = false
-- Repeat to generate and append each row
+
-- Loop to generate rows
 
repeat
 
repeat
 
-- New empty row
 
-- New empty row
 
local forRow = {}
 
local forRow = {}
-- If there's a blank use, assume the list's ended, leave empty for
+
-- On blank use, assume list's ended & break at end of this loop
-- defaulting elsewhere, and break at the end of this loop-through.
 
 
forRow.use = args[i]
 
forRow.use = args[i]
 
if not args[i] then terminated = true end
 
if not args[i] then terminated = true end
 
-- New empty list of pages
 
-- New empty list of pages
 
forRow.pages = {}
 
forRow.pages = {}
-- Try to insert first pages item; empty is ignored/defaulted elsewhere
+
-- Insert first pages item if present
 
table.insert(forRow.pages, args[i + 1])
 
table.insert(forRow.pages, args[i + 1])
-- If the option after next is "and", do an inner loop where we collect
+
-- If the param after next is "and", do inner loop to collect params
-- items following "and"'s until the "and"'s stop. If there's a blank
+
-- until the "and"'s stop. Blanks are ignored: "1|and||and|3" → {1, 3}
-- where we'd expect an item, ignore it: "1|and||and|3" → {1, 3}
+
while args[i + 2] == options.andKeyword do
while args[i + 2] == 'and' do
 
 
if args[i + 3] then  
 
if args[i + 3] then  
 
table.insert(forRow.pages, args[i + 3])
 
table.insert(forRow.pages, args[i + 3])
 
end
 
end
-- Increment to the next "and"
+
-- Increment to next "and"
 
i = i + 2
 
i = i + 2
 
end
 
end
-- Increment to the next use
+
-- Increment to next use
 
i = i + 2
 
i = i + 2
-- Add the row to the table
+
-- Append the row
 
table.insert(forTable, forRow)
 
table.insert(forTable, forRow)
 
until terminated or i > maxArg
 
until terminated or i > maxArg
Line 145: Line 154:
 
end
 
end
  
-- Takes a table as formatted by forSeeArgsToTable and stringifies it
+
-- Stringifies a table as formatted by forSeeArgsToTable
 
function p.forSeeTableToString (forSeeTable, options)
 
function p.forSeeTableToString (forSeeTable, options)
 
-- Type-checks and defaults
 
-- Type-checks and defaults
Line 154: Line 163:
 
if options[k] == nil then options[k] = v end
 
if options[k] == nil then options[k] = v end
 
end
 
end
-- Format each for-see item and make a table containing them
+
-- Stringify each for-see item into a list
 
local strList = {}
 
local strList = {}
 
for k, v in pairs(forSeeTable) do
 
for k, v in pairs(forSeeTable) do
Line 160: Line 169:
 
local pagesStr = p.andList(v.pages, true) or
 
local pagesStr = p.andList(v.pages, true) or
 
mHatnote._formatLink(mHatnote.disambiguate(options.title))
 
mHatnote._formatLink(mHatnote.disambiguate(options.title))
local forSeeStr = string.format(options.forseeForm, useStr, pagesStr)
+
local forSeeStr = string.format(options.forSeeForm, useStr, pagesStr)
for k, v in pairs(options.punctuationCollapseReplacements) do
+
forSeeStr = punctuationCollapse(forSeeStr)
forSeeStr = string.gsub(forSeeStr, k, v)
 
end
 
 
table.insert(strList, forSeeStr)
 
table.insert(strList, forSeeStr)
 
end
 
end
Line 171: Line 178:
  
 
-- Produces a "For X, see [[Y]]" string from arguments. Expects index gaps
 
-- Produces a "For X, see [[Y]]" string from arguments. Expects index gaps
-- but not blank or whitespace values; those should be filtered. Ignores
+
-- but not blank/whitespace values. Ignores named args and args < "from".
-- arguments less than "from", and named arguments.
 
 
function p._forSee (args, from, options)
 
function p._forSee (args, from, options)
 
local forSeeTable = p.forSeeArgsToTable(args, from, options)
 
local forSeeTable = p.forSeeArgsToTable(args, from, options)
Line 178: Line 184:
 
end
 
end
  
-- Calls _forSee but pulls from the frame.
+
-- As _forSee, but uses the frame.
 
function p.forSee (frame, from, options)
 
function p.forSee (frame, from, options)
 
mArguments = require('Module:Arguments')
 
mArguments = require('Module:Arguments')

Revision as of 01:41, 31 May 2017


-- Module:Hatnote list -- -- -- -- This module produces and formats lists for use in hatnotes. In particular, -- -- it implements the for-see list, i.e. lists of "For X, see Y" statements, --

-- as used in

, Error: missing redirect parameter (help)., and their variants. Also introduced --

-- are andList & orList helpers for formatting lists with those conjunctions. --


local mArguments --initialize lazily local mHatnote = require('Module:Hatnote') local libraryUtil = require('libraryUtil') local checkType = libraryUtil.checkType local p = {}


-- List stringification helper functions -- -- These functions are used for stringifying lists, usually page lists inside -- the "Y" portion of "For X, see Y" for-see items.


--default options table used across the list stringification functions local stringifyListDefaultOptions = { conjunction = "and", separator = ",", altSeparator = ";", space = " ", formatted = false }

-- Stringifies a list generically; probably shouldn't be used directly function stringifyList(list, options) -- Type-checks, defaults, and a shortcut checkType("stringifyList", 1, list, "table") if #list == 0 then return nil end checkType("stringifyList", 2, options, "table", true) options = options or {} for k, v in pairs(stringifyListDefaultOptions) do if options[k] == nil then options[k] = v end end local s = options.space -- Format the list if requested if options.formatted then list = mHatnote.formatPages(unpack(list)) end -- Set the separator; if any item contains it, use the alternate separator local separator = options.separator --searches display text only function searchDisp(t, f) return string.find(string.sub(t, (string.find(t, '|') or 0) + 1), f) end for k, v in pairs(list) do if searchDisp(v, separator) then separator = options.altSeparator break end end -- Set the conjunction, apply Oxford comma, and force a comma if #1 has "§" local conjunction = s .. options.conjunction .. s if #list == 2 and searchDisp(list[1], "§") or #list > 2 then conjunction = separator .. conjunction end -- Return the formatted string return mw.text.listToText(list, separator .. s, conjunction) end

--DRY function function conjList (conj, list, fmt) return stringifyList(list, {conjunction = conj, formatted = fmt}) end

-- Stringifies lists with "and" or "or" function p.andList (...) return conjList("and", ...) end function p.orList (...) return conjList("or", ...) end


-- For see -- -- Makes a "For X, see Y." list from raw parameters. Intended for the

--

and Error: missing redirect parameter (help). templates and their variants.


--default options table used across the forSee family of functions local forSeeDefaultOptions = { andKeyword = 'and', title = mw.title.getCurrentTitle().text, otherText = 'other uses', forSeeForm = 'For %s, see %s.' }

--Collapses duplicate punctuation function punctuationCollapse (text) local replacements = { ["%.%.$"] = ".", ["%?%.$"] = "?", ["%!%.$"] = "!", ["%.%]%]%.$"] = ".]]", ["%?%]%]%.$"] = "?]]", ["%!%]%]%.$"] = "!]]" } for k, v in pairs(replacements) do text = string.gsub(text, k, v) end return text end

-- Structures arguments into a table for stringification, & options function p.forSeeArgsToTable (args, from, options) -- Type-checks and defaults checkType("forSeeArgsToTable", 1, args, 'table') checkType("forSeeArgsToTable", 2, from, 'number', true) from = from or 1 checkType("forSeeArgsToTable", 3, options, 'table', true) options = options or {} for k, v in pairs(forSeeDefaultOptions) do if options[k] == nil then options[k] = v end end -- maxArg's gotten manually because getArgs() and table.maxn aren't friends local maxArg = 0 for k, v in pairs(args) do if type(k) == 'number' and k > maxArg then maxArg = k end end -- Structure the data out from the parameter list: -- * forTable is the wrapper table, with forRow rows -- * Rows are tables of a "use" string & a "pages" table of pagename strings -- * Blanks are left empty for defaulting elsewhere, but can terminate list local forTable = {} local i = from local terminated = false -- Loop to generate rows repeat -- New empty row local forRow = {} -- On blank use, assume list's ended & break at end of this loop forRow.use = args[i] if not args[i] then terminated = true end -- New empty list of pages forRow.pages = {} -- Insert first pages item if present table.insert(forRow.pages, args[i + 1]) -- If the param after next is "and", do inner loop to collect params -- until the "and"'s stop. Blanks are ignored: "1|and||and|3" → {1, 3} while args[i + 2] == options.andKeyword do if args[i + 3] then table.insert(forRow.pages, args[i + 3]) end -- Increment to next "and" i = i + 2 end -- Increment to next use i = i + 2 -- Append the row table.insert(forTable, forRow) until terminated or i > maxArg

return forTable end

-- Stringifies a table as formatted by forSeeArgsToTable function p.forSeeTableToString (forSeeTable, options) -- Type-checks and defaults checkType("forSeeTableToString", 1, forSeeTable, "table") checkType("forSeeTableToString", 2, options, "table", true) options = options or {} for k, v in pairs(forSeeDefaultOptions) do if options[k] == nil then options[k] = v end end -- Stringify each for-see item into a list local strList = {} for k, v in pairs(forSeeTable) do local useStr = v.use or options.otherText local pagesStr = p.andList(v.pages, true) or mHatnote._formatLink(mHatnote.disambiguate(options.title)) local forSeeStr = string.format(options.forSeeForm, useStr, pagesStr) forSeeStr = punctuationCollapse(forSeeStr) table.insert(strList, forSeeStr) end -- Return the concatenated list return table.concat(strList, ' ') end

-- Produces a "For X, see Y" string from arguments. Expects index gaps -- but not blank/whitespace values. Ignores named args and args < "from". function p._forSee (args, from, options) local forSeeTable = p.forSeeArgsToTable(args, from, options) return p.forSeeTableToString(forSeeTable, options) end

-- As _forSee, but uses the frame. function p.forSee (frame, from, options) mArguments = require('Module:Arguments') return p._forSee(mArguments.getArgs(frame), from, options) end

return p