Module:template parser

---

The following documentation is located at Module:template parser/documentation. ^[edit]

Useful links: subpage list • links • transclusions • testcases • sandbox

This module provides functions for parsing and finding template invocations found in wikitext.

parseTemplate(text, not_transcluded)

Parses text as a template invocation and returns a pair of values, the template name and the arguments (containing anonymous, numbered and named arguments). If the text could not be parsed as a template invocation, the function returns nil. The parser will correctly parse any wikitext given as template arguments (such as subtemplates, arguments, tables etc), but if the string does not form a valid template in markup, then it will return nil.

findTemplates(text, not_transcluded)

Finds all template invocations in the text. This is designed to be used as an iterator in for statements, and returns four values for each invocation:

1. The template name.
2. The template arguments.
3. The full template invocation as it appears in the original text.
4. The index the template appears at within the given text; as with Lua in general, the beginning of the text is index 1.

For convenience, template names will be normalized in two ways:

1. They are preprocessed, which means that any templates ({{ }}) and parameters ({{{ }}}) they contain will be resolved.
2. Any redirects will be converted to their canonical equivalents (e.g. {{l}} is treated as {{link}}).

Note that any templates with invalid names (after preprocessing) will be skipped over. For performance reasons, preprocessing is only applied to the keys in a template's table of arguments, so it should be applied (selectively) to the values by the calling module when needed.

Note that the parser will respect <noinclude>, <includeonly> and <onlyinclude> tags. By default, text is treated as though it has been transcluded, which means that text between <noinclude> tags will be ignored, and <onlyinclude> tags will be respected if present. If the parameter not_transcluded is set to true, then text will be treated as though it has not been transcluded, which means text between <includeonly> tags will be ignored instead.

Although the parser is very accurate, some discrepancies may still exist between it and the native parser in certain cases.

--[[
NOTE: This module works by using recursive backtracking to build a node tree, which can then be traversed as necessary.

Because it is called by a number of high-use modules, it has been optimised for speed using a profiler, since it is used to scrape data from large numbers of pages very quickly. To that end, it rolls some of its own methods in cases where this is faster than using a function from one of the standard libraries. Please DO NOT "simplify" the code by removing these, since you are almost guaranteed to slow things down, which could seriously impact performance on pages which call this module hundreds or thousands of times.

It has also been designed to emulate the native parser's behaviour as much as possible, which in some cases means replicating bugs or unintuitive behaviours in that code; these should not be "fixed", since it is important that the outputs are the same. Most of these originate from deficient regular expressions, which can't be used here, so the bugs have to be manually reintroduced as special cases (e.g. onlyinclude tags being case-sensitive and whitespace intolerant, unlike all other tags). If any of these are fixed, this module should also be updated accordingly.
]]
local export = {}

local data_module = "Module:template parser/data"
local load_module = "Module:load"
local magic_words_data_module = "Module:data/magic words"
local pages_module = "Module:pages"
local parser_extension_tags_data_module = "Module:data/parser extension tags"
local parser_module = "Module:parser"
local string_utilities_module = "Module:string utilities"
local table_module = "Module:table"

local require = require

local m_parser = require(parser_module)
local mw = mw
local mw_title = mw.title
local mw_uri = mw.uri
local string = string
local table = table

local anchor_encode = mw_uri.anchorEncode
local build_template -- defined as export.buildTemplate below
local class_else_type = m_parser.class_else_type
local concat = table.concat
local encode_uri = mw_uri.encode
local find = string.find
local format = string.format
local gsub = string.gsub
local html_create = mw.html.create
local insert = table.insert
local is_node = m_parser.is_node
local lower = string.lower
local make_title = mw_title.makeTitle -- unconditionally adds the specified namespace prefix
local match = string.match
local new_title = mw_title.new -- specified namespace prefix is only added if the input doesn't contain one
local next = next
local pairs = pairs
local parse -- defined as export.parse below
local parse_template_name -- defined below
local pcall = pcall
local rep = string.rep
local reverse = string.reverse
local select = select
local sub = string.sub
local title_equals = mw_title.equals
local tostring = m_parser.tostring
local type = type
local umatch = mw.ustring.match

--[==[
Loaders for functions in other modules, which overwrite themselves with the target function when called. This ensures modules are only loaded when needed, retains the speed/convenience of locally-declared pre-loaded functions, and has no overhead after the first call, since the target functions are called directly in any subsequent calls.]==]
	local function decode_entities(...)
		decode_entities = require(string_utilities_module).decode_entities
		return decode_entities(...)
	end
	
	local function encode_entities(...)
		encode_entities = require(string_utilities_module).encode_entities
		return encode_entities(...)
	end
	
	local function get_namespace_shortcut(...)
		get_namespace_shortcut = require(pages_module).get_namespace_shortcut
		return get_namespace_shortcut(...)
	end
	
	local function is_internal_title(...)
		is_internal_title = require(pages_module).is_internal_title
		return is_internal_title(...)
	end
	
	local function load_data(...)
		load_data = require(load_module).load_data
		return load_data(...)
	end
	
	local function pattern_escape(...)
		pattern_escape = require(string_utilities_module).pattern_escape
		return pattern_escape(...)
	end
	
	local function php_trim(...)
		php_trim = require(string_utilities_module).php_trim
		return php_trim(...)
	end
	
	local function replacement_escape(...)
		replacement_escape = require(string_utilities_module).replacement_escape
		return replacement_escape(...)
	end
	
	local function scribunto_param_key(...)
		scribunto_param_key = require(string_utilities_module).scribunto_param_key
		return scribunto_param_key(...)
	end
	
	local function shallowcopy(...)
		shallowcopy = require(table_module).shallowcopy
		return shallowcopy(...)
	end
	
	local function sorted_pairs(...)
		sorted_pairs = require(table_module).sortedPairs
		return sorted_pairs(...)
	end
	
	local function split(...)
		split = require(string_utilities_module).split
		return split(...)
	end
	
	local function table_len(...)
		table_len = require(table_module).length
		return table_len(...)
	end
	
	local function uupper(...)
		uupper = require(string_utilities_module).upper
		return uupper(...)
	end

--[==[
Loaders for objects, which load data (or some other object) into some variable, which can then be accessed as "foo or get_foo()", where the function get_foo sets the object to "foo" and then returns it. This ensures they are only loaded when needed, and avoids the need to check for the existence of the object each time, since once "foo" has been set, "get_foo" will not be called again.]==]
	local data
	local function get_data()
		data, get_data = load_data(data_module), nil
		return data
	end
	
	local frame
	local function get_frame()
		frame, get_frame = mw.getCurrentFrame(), nil
		return frame
	end
	
	local magic_words
	local function get_magic_words()
		magic_words, get_magic_words = load_data(magic_words_data_module), nil
		return magic_words
	end
	
	local parser_extension_tags
	local function get_parser_extension_tags()
		parser_extension_tags, get_parser_extension_tags = load_data(parser_extension_tags_data_module), nil
		return parser_extension_tags
	end

local Parser, Node = m_parser.new()

------------------------------------------------------------------------------------
--
-- Nodes
--
------------------------------------------------------------------------------------

Node.keys_to_remove = {"handler", "head", "pattern", "route", "step"}

local function expand(obj, frame_args)
	return is_node(obj) and obj:expand(frame_args) or obj
end
export.expand = expand

function Node:expand(frame_args)
	local output = {}
	for i = 1, #self do
		output[i] = expand(self[i], frame_args)
	end
	return concat(output)
end

local Wikitext = Node:new_class("wikitext")

-- force_node ensures the output will always be a node.
function Wikitext:new(this, force_node)
	if type(this) ~= "table" then
		return force_node and Node.new(self, {this}) or this
	elseif #this == 1 then
		local this1 = this[1]
		return force_node and not is_node(this1) and Node.new(self, this) or this1
	end
	local success, str = pcall(concat, this)
	if success then
		return force_node and Node.new(self, {str}) or str
	end
	return Node.new(self, this)
end

-- First value is the parameter name.
-- Second value is the parameter's default value.
-- Any additional values are ignored: e.g. "{{{a|b|c}}}" is parameter "a" with default value "b" (*not* "b|c").
local Parameter = Node:new_class("parameter")

function Parameter:new(this)
	local this2 = this[2]
	if class_else_type(this2) == "argument" then
		insert(this2, 2, "=")
		this2 = Wikitext:new(this2)
	end
	if this[3] == nil then
		this[2] = this2
	else
		this = {this[1], this2}
	end
	return Node.new(self, this)
end

function Parameter:__tostring()
	local output = {}
	for i = 1, #self do
		output[i] = tostring(self[i])
	end
	return "{{{" .. concat(output, "|") .. "}}}"
end

function Parameter:get_name(frame_args)
	return scribunto_param_key(expand(self[1], frame_args))
end

function Parameter:get_default(frame_args)
	local default = self[2]
	if default ~= nil then
		return expand(default, frame_args)
	end
	return "{{{" .. expand(self[1], frame_args) .. "}}}"
end

function Parameter:expand(frame_args)
	if frame_args == nil then
		return self:get_default()
	end
	local name = expand(self[1], frame_args)
	local val = frame_args[scribunto_param_key(name)] -- Parameter in use.
	if val ~= nil then
		return val
	end
	val = self[2] -- Default.
	if val ~= nil then
		return expand(val, frame_args)
	end
	return "{{{" .. name .. "}}}"
end

local Argument = Node:new_class("argument")

function Argument:__tostring()
	return tostring(self[1]) .. "=" .. tostring(self[2])
end

function Argument:expand(frame_args)
	return expand(self[1], frame_args) .. "=" .. expand(self[2], frame_args)
end

local Template = Node:new_class("template")

function Template:__tostring()
	local output = {}
	for i = 1, #self do
		output[i] = tostring(self[i])
	end
	return "{{" .. concat(output, "|") .. "}}"
end

-- Normalize the template name, check it's a valid template, then memoize results (using false for invalid titles).
-- Parser functions (e.g. {{#IF:a|b|c}}) need to have the first argument extracted from the title, as it comes after the colon. Because of this, the parser function and first argument are memoized as a table.
-- FIXME: Some parser functions have special argument handling (e.g. {{#SWITCH:}}).
do
	local page_title = mw_title.getCurrentTitle()
	local namespace_has_subpages = mw.site.namespaces[page_title.namespace].hasSubpages
	local raw_pagename = page_title.fullText
	
	local templates = {}
	local parser_variables = {}
	local parser_functions = {}
	
	local function retrieve_magic_word_data(chunk)
		local mgw_data = (magic_words or get_magic_words())[chunk]
		if mgw_data then
			return mgw_data
		end
		local normalized = uupper(chunk)
		mgw_data = magic_words[normalized]
		if mgw_data and not mgw_data.case_sensitive then
			return mgw_data
		end
	end
	
	local function get_prefixed_invocation_name(title, namespace, shortcut)
		return shortcut and get_namespace_shortcut(title) .. ":" .. title.text or
			namespace == 0 and ":" .. title.prefixedText or
			title.prefixedText
	end
	
	-- Returns the name required to transclude the title object `title` using
	-- template {{ }} syntax. If the `shortcut` flag is set, then any calls
	-- which require a namespace prefix will use the abbreviated form where one
	-- exists (e.g. "Template:PAGENAME" becomes "T:PAGENAME").
	local function get_template_invocation_name(title, shortcut)
		if not (title and is_internal_title(title)) then
			error("Template invocations require a valid page title, which cannot contain an interwiki prefix.")
		end
		local namespace = title.namespace
		-- If not in the template namespace, include the prefix (or ":" if
		-- mainspace).
		if namespace ~= 10 then
			return get_prefixed_invocation_name(title, namespace, shortcut)
		end
		-- If in the template namespace and it shares a name with a magic word,
		-- it needs the prefix "Template:".
		local text = title.text
		local colon = find(text, ":", nil, true)
		if not colon then
			local mgw_data = retrieve_magic_word_data(text)
			return mgw_data and mgw_data.parser_variable and get_prefixed_invocation_name(title, namespace, shortcut) or text
		end
		local mgw_data = retrieve_magic_word_data(sub(text, 1, colon - 1))
		if mgw_data and (mgw_data.parser_function or mgw_data.transclusion_modifier) then
			return get_prefixed_invocation_name(title, namespace, shortcut)
		end
		-- Also if "Template:" is necessary for disambiguation (e.g.
		-- "Template:Category:Foo" can't be called with "Category:Foo").
		local check = new_title(text, namespace)
		return check and title_equals(title, check) and text or get_prefixed_invocation_name(title, namespace, shortcut)
	end
	export.getTemplateInvocationName = get_template_invocation_name
	
	-- Returns whether a title is a redirect or not. Structured like this to
	-- allow the use of pcall, since it will throw an error if the expensive
	-- parser function limit has been reached.
	local function is_redirect(title)
		return title.isRedirect
	end
	
	function parse_template_name(name, has_args, fragment, force_transclusion)
		local chunks, colon, start, n, p = {}, find(name, ":", nil, true), 1, 0, 0
		while colon do
			-- Pattern applies PHP ltrim.
			local mgw_data = retrieve_magic_word_data(match(sub(name, start, colon - 1), "[^%z\t-\v\r ].*") or "")
			if not mgw_data then
				break
			end
			local priority = mgw_data.priority
			if not (priority and priority > p) then
				local pf = mgw_data.parser_function and mgw_data.name or nil
				if pf then
					n = n + 1
					chunks[n] = pf .. ":"
					return chunks, "parser function", sub(name, colon + 1)
				end
				break
			end
			n = n + 1
			chunks[n] = mgw_data.name .. ":"
			start, p = colon + 1, priority
			colon = find(name, ":", start, true)
		end
		if start > 1 then
			name = sub(name, start)
		end
		name = php_trim(name)
		-- Parser variables can only take SUBST:/SAFESUBST: as modifiers.
		if not has_args and p <= 1 then
			local mgw_data = retrieve_magic_word_data(name)
			local pv = mgw_data and mgw_data.parser_variable and mgw_data.name or nil
			if pv then
				n = n + 1
				chunks[n] = pv
				return chunks, "parser variable"
			end
		end
		-- Handle relative template names.
		if namespace_has_subpages then
			-- If the name starts with "/", it's treated as a subpage of the
			-- current page. Final slashes are trimmed, but this can't affect
			-- the intervening slash (e.g. {{///}} refers to "{{PAGENAME}}/").
			local initial = sub(name, 1, 1)
			if initial == "/" then
				name = raw_pagename .. (match(name, "^/.*[^/]") or "/")
			-- If it starts with "../", trim it and any that follow, and go up
			-- that many subpage levels. Then, treat any additional text as
			-- a subpage of that page; final slashes are trimmed.
			elseif initial == "." and sub(name, 2, 3) == "./" then
				local n = 4
				while sub(name, n, n + 2) == "../" do
					n = n + 3
				end
				-- Retain an initial "/".
				name = sub(name, n - 1)
				-- Trim the relevant number of subpages from the pagename.
				local pagename, i = reverse(raw_pagename), 0
				for _ = 1, (n - 1) / 3 do
					i = find(pagename, "/", i + 1, true)
					-- Fail if there aren't enough slashes.
					if not i then
						return nil
					end
				end
				-- Add the subpage text; since the intervening "/" is retained
				-- in `name`, it can be trimmed along with any other final
				-- slashes (e.g. {{..///}} refers to "{{BASEPAGENAME}}".)
				name = reverse(sub(pagename, i + 1)) .. (match(name, "^.*[^/]") or "")
			end
		end
		local title = new_title(name, 10)
		if not (title and is_internal_title(title)) then
			return nil
		end
		-- If `fragment` is set, save the original title's fragment, since it
		-- won't carry through to any redirect targets.
		if fragment then
			fragment = title.fragment
		end
		-- Resolve any redirects. Note that is_internal_title treats interwiki
		-- titles as invalid, which is correct in this case: if the redirect
		-- target is an interwiki link, the template won't fail, but the
		-- redirect does not get resolved (i.e. the redirect page itself gets
		-- transcluded, so the template name should not be normalized to the
		-- target). It also treats titles that only have fragments as invalid
		-- (e.g. "#foo"), but these can't be used as redirects anyway.
		-- title.redirectTarget increments the expensive parser function count,
		-- but avoids extraneous transclusions polluting template lists and the
		-- performance hit caused by indiscriminately grabbing redirectTarget.
		-- However, if the expensive parser function limit has already been hit,
		-- redirectTarget is used as a fallback. force_transclusion forces the
		-- use of the fallback.
		local redirect = true
		if not force_transclusion then
			local success, resolved = pcall(is_redirect, title)
			if success and not resolved then
				redirect = false
			end
		end
		if redirect then
			redirect = title.redirectTarget
			if redirect and is_internal_title(redirect) then
				title = redirect
			end
		end
		local chunk = get_template_invocation_name(title)
		-- Set the fragment (if applicable).
		if fragment then
			chunk = chunk .. "#" .. fragment
		end
		chunks[n + 1] = chunk
		return chunks, "template"
	end
	
	-- Note: force_transclusion avoids incrementing the expensive parser
	-- function count by forcing transclusion instead. This should only be used
	-- when there is a real risk that the expensive parser function limit of
	-- 500 will be hit.
	local function process_name(self, frame_args, force_transclusion)
		local name = expand(self[1], frame_args)
		local has_args, norm = #self > 1
		if not has_args then
			norm = parser_variables[name]
			if norm then
				return norm, "parser variable"
			end
		end
		norm = templates[name]
		if norm then
			local pf_arg1 = parser_functions[name]
			return norm, pf_arg1 and "parser function" or "template", pf_arg1
		elseif norm == false then
			return nil
		end
		local chunks, subclass, pf_arg1 = parse_template_name(name, has_args, nil, force_transclusion)
		-- Fail if invalid.
		if not chunks then
			templates[name] = false
			return nil
		end
		local chunk1 = chunks[1]
		-- Fail on SUBST:.
		if chunk1 == "SUBST:" then
			templates[name] = false
			return nil
		-- Any modifiers are ignored.
		elseif subclass == "parser function" then
			local pf = chunks[#chunks]
			templates[name] = pf
			parser_functions[name] = pf_arg1
			return pf, "parser function", pf_arg1
		end
		-- Ignore SAFESUBST:, and treat MSGNW: as a parser function with the pagename as its first argument (ignoring any RAW: that comes after).
		if chunks[chunk1 == "SAFESUBST:" and 2 or 1] == "MSGNW:" then
			pf_arg1 = chunks[#chunks]
			local pf = "MSGNW:"
			templates[name] = pf
			parser_functions[name] = pf_arg1
			return pf, "parser function", pf_arg1
		end
		-- Ignore any remaining modifiers, as they've done their job.
		local output = chunks[#chunks]
		if subclass == "parser variable" then
			parser_variables[name] = output
		else
			templates[name] = output
		end
		return output, subclass
	end
	
	function Template:get_name(frame_args, force_transclusion)
		-- Only return the first return value.
		return (process_name(self, frame_args, force_transclusion))
	end
	
	function Template:get_arguments(frame_args)
		local name, subclass, pf_arg1 = process_name(self, frame_args)
		if name == nil then
			return nil
		elseif subclass == "parser variable" then
			return {}
		end
		local template_args = {}
		if subclass == "parser function" then
			template_args[1] = pf_arg1
			for i = 2, #self do
				template_args[i] = expand(self[i], frame_args) -- Not trimmed.
			end
			return template_args
		end
		local implicit = 0
		for i = 2, #self do
			local arg = self[i]
			if class_else_type(arg) == "argument" then
				template_args[scribunto_param_key(expand(arg[1], frame_args))] = php_trim(expand(arg[2], frame_args))
			else
				implicit = implicit + 1
				template_args[implicit] = expand(arg, frame_args) -- Not trimmed.
			end
		end
		return template_args
	end
end

-- BIG TODO: manual template expansion.
function Template:expand()
	return (frame or get_frame()):preprocess(tostring(self))
end

local Tag = Node:new_class("tag")

do
	local php_htmlspecialchars_data
	
	local function get_php_htmlspecialchars_data()
		php_htmlspecialchars_data, get_php_htmlspecialchars_data = (data or get_data()).php_htmlspecialchars, nil
		return php_htmlspecialchars_data
	end
	
	local function php_htmlspecialchars(str, compat)
		return (gsub(str, compat and "[&\"<>]" or "[&\"'<>]", php_htmlspecialchars_data or get_php_htmlspecialchars_data()))
	end
	
	function Tag:__tostring()
		local open_tag, attributes, n = {"<", self.name}, self:get_attributes(), 2
		for attr, value in next, attributes do
			n = n + 1
			open_tag[n] = " " .. php_htmlspecialchars(attr) .. "=\"" .. php_htmlspecialchars(value, true) .. "\""
		end
		if self.self_closing then
			return concat(open_tag) .. "/>"
		end
		return concat(open_tag) .. ">" .. concat(self) .. "</" .. self.name .. ">"
	end
	
	local valid_attribute_name
	
	local function get_valid_attribute_name()
		valid_attribute_name, get_valid_attribute_name = (data or get_data()).valid_attribute_name, nil
		return valid_attribute_name
	end
	
	function Tag:get_attributes()
		local raw = self.attributes
		if not raw then
			self.attributes = {}
			return self.attributes
		elseif type(raw) == "table" then
			return raw
		end
		if sub(raw, -1) == "/" then
			raw = sub(raw, 1, -2)
		end
		local attributes, head = {}, 1
		-- Semi-manual implementation of the native regex.
		while true do
			local name, loc = match(raw, "([^\t\n\f\r />][^\t\n\f\r /=>]*)()", head)
			if not name then
				break
			end
			head = loc
			local value
			loc = match(raw, "^[\t\n\f\r ]*=[\t\n\f\r ]*()", head)
			if loc then
				head = loc
				-- Either "", '' or the value ends on a space/at the end. Missing
				-- end quotes are repaired by closing the value at the end.
				value, loc = match(raw, "^\"([^\"]*)\"?()", head)
				if not value then
					value, loc = match(raw, "^'([^']*)'?()", head)
					if not value then
						value, loc = match(raw, "^([^\t\n\f\r ]*)()", head)
					end
				end
				head = loc
			end
			-- valid_attribute_name is a pattern matching a valid attribute name.
			-- Defined in the data due to its length - see there for more info.
			if umatch(name, valid_attribute_name or get_valid_attribute_name()) then
				-- Sanitizer applies PHP strtolower (ASCII-only).
				attributes[lower(name)] = value and decode_entities(
					php_trim((gsub(value, "[\t\n\r ]+", " ")))
				) or ""
			end
		end
		self.attributes = attributes
		return attributes
	end
end

function Tag:expand()
	return (frame or get_frame()):preprocess(tostring(self))
end

local Heading = Node:new_class("heading")

function Heading:new(this)
	if #this > 1 then
		local success, str = pcall(concat, this)
		if success then
			return Node.new(self, {
				str,
				level = this.level,
				section = this.section,
				index = this.index
			})
		end
	end
	return Node.new(self, this)
end

function Heading:__tostring()
	local eq = rep("=", self.level)
	return eq .. Node.__tostring(self) .. eq
end

do
	local expand_node = Node.expand
	
	-- Expanded heading names can contain "\n" (e.g. inside nowiki tags), which
	-- causes any heading containing them to fail. However, in such cases, the
	-- native parser still treats it as a heading for the purpose of section
	-- numbers.
	local function validate_name(self, frame_args)
		local name = expand_node(self, frame_args)
		if find(name, "\n", nil, true) then
			return nil
		end
		return name
	end
	
	function Heading:get_name(frame_args)
		local name = validate_name(self, frame_args)
		return name ~= nil and php_trim(name) or nil
	end
	
	-- FIXME: account for anchor disambiguation.
	function Heading:get_anchor(frame_args)
		local name = validate_name(self, frame_args)
		return name ~= nil and decode_entities(anchor_encode(name)) or nil
	end
	
	function Heading:expand(frame_args)
		local eq = rep("=", self.level)
		return eq .. expand_node(self, frame_args) .. eq
	end
end

------------------------------------------------------------------------------------
--
-- Parser
--
------------------------------------------------------------------------------------

function Parser:read(i, j)
	local head, i = self.head, i or 0
	return sub(self.text, head + i, head + (j or i))
end

function Parser:advance(n)
	self.head = self.head + (n or self[-1].step or 1)
end

function Parser:jump(head)
	self.head = head
	self[-1].nxt = nil
end

function Parser:set_pattern(pattern)
	local layer = self[-1]
	layer.pattern = pattern
	layer.nxt = nil
end

function Parser:consume()
	local layer = self[-1]
	local this = layer.nxt
	if this then
		layer.nxt = nil
	else
		local text, head = self.text, self.head
		local loc1, loc2 = find(text, layer.pattern, head)
		if loc1 == head or not loc1 then
			this = sub(text, head, loc2)
		else
			this = sub(text, head, loc1 - 1)
			layer.nxt = sub(text, loc1, loc2)
		end
	end
	layer.step = #this
	return layer.handler(self, this)
end

-- Template or parameter.

-- Parsed by matching the opening braces innermost-to-outermost (ignoring lone closing braces). Parameters {{{ }}} take priority over templates {{ }} where possible, but a double closing brace will always result in a closure, even if there are 3+ opening braces.

-- For example, "{{{{foo}}}}" (4) is parsed as a parameter enclosed by single braces, and "{{{{{foo}}}}}" (5) is a parameter inside a template. However, "{{{{{foo }} }}}" is a template inside a parameter, due to "}}" forcing the closure of the inner node.
do
	-- Handlers.
	local handle_name
	local handle_argument
	
	local function do_template_or_parameter(self, inner_node)
		self:push_sublayer(handle_name)
		self:set_pattern("[\n<[{|}]")
		-- If a node has already been parsed, nest it at the start of the new
		-- outer node (e.g. when parsing"{{{{foo}}bar}}", the template "{{foo}}"
		-- is parsed first, since it's the innermost, and becomes the first
		-- node of the outer template.
		if inner_node then
			self:emit(inner_node)
		end
	end
	
	function handle_name(self, ...)
		handle_name = self:switch(handle_name, {
			["\n"] = Parser.heading_block,
			["<"] = Parser.tag,
			["["] = Parser.wikilink_block,
			["{"] = Parser.braces,
			
			["|"] = function(self)
				self:emit(Wikitext:new(self:pop_sublayer()))
				self:push_sublayer(handle_argument)
				self:set_pattern("[\n<=[{|}]")
			end,
			
			["}"] = function(self)
				if self:read(1) == "}" then
					self:emit(Wikitext:new(self:pop_sublayer()))
					return self:pop()
				end
				self:emit("}")
			end,
			
			[""] = Parser.fail_route,
			[false] = Parser.emit
		})
		return handle_name(self, ...)
	end
	
	function handle_argument(self, ...)
		local function emit_argument(self)
			local arg = Wikitext:new(self:pop_sublayer())
			local layer = self[-1]
			local key = layer.key
			if key then
				arg = Argument:new{key, arg}
				layer.key = nil
			end
			self:emit(arg)
		end
		
		handle_argument = self:switch(handle_argument, {
			["\n"] = function(self)
				return self:heading_block("\n", self[-1].key and "=" or "==")
			end,
			
			["<"] = Parser.tag,
			
			["="] = function(self)
				local key = Wikitext:new(self:pop_sublayer())
				self[-1].key = key
				self:push_sublayer(handle_argument)
				self:set_pattern("[\n<[{|}]")
			end,
			
			["["] = Parser.wikilink_block,
			["{"] = Parser.braces,
			
			["|"] = function(self)
				emit_argument(self)
				self:push_sublayer(handle_argument)
				self:set_pattern("[\n<=[{|}]")
			end,
			
			["}"] = function(self)
				if self:read(1) == "}" then
					emit_argument(self)
					return self:pop()
				end
				self:emit("}")
			end,
			
			[""] = Parser.fail_route,
			[false] = Parser.emit
		})
		return handle_argument(self, ...)
	end
	
	function Parser:template_or_parameter()
		local text, head, node_to_emit, failed = self.text, self.head
		-- Comments/tags interrupt the brace count.
		local braces = match(text, "^{+()", head) - head
		self:advance(braces)
		while true do
			local success, node = self:try(do_template_or_parameter, node_to_emit)
			-- Fail means no "}}" or "}}}" was found, so emit any remaining
			-- unmatched opening braces before any templates/parameters that
			-- were found.
			if not success then
				self:emit(rep("{", braces))
				failed = true
				break
			-- If there are 3+ opening and closing braces, it's a parameter.
			elseif braces >= 3 and self:read(2) == "}" then
				self:advance(3)
				braces = braces - 3
				node = Parameter:new(node)
			-- Otherwise, it's a template.
			else
				self:advance(2)
				braces = braces - 2
				node = Template:new(node)
			end
			local index = head + braces
			node.index = index
			node.raw = sub(text, index, self.head - 1)
			node_to_emit = node
			-- Terminate once not enough braces remain for further matches.
			if braces == 0 then
				break
			-- Emit any stray opening brace before any matched nodes.
			elseif braces == 1 then
				self:emit("{")
				break
			end
		end
		if node_to_emit then
			self:emit(node_to_emit)
		end
		return braces, failed
	end
end

-- Tag.
do
	local end_tags
	
	local function get_end_tags()
		end_tags, get_end_tags = (data or get_data()).end_tags, nil
		return end_tags
	end
	
	-- Handlers.
	local handle_start
	local handle_tag
	
	local function do_tag(self)
		local layer = self[-1]
		layer.handler, layer.index = handle_start, self.head
		self:set_pattern("[%s/>]")
		self:advance()
	end
	
	local function is_ignored_tag(self, this)
		if self.transcluded then
			return this == "includeonly"
		end
		return this == "noinclude" or this == "onlyinclude"
	end
	
	local function ignored_tag(self, text, head)
		local loc = find(text, ">", head, true)
		if not loc then
			return self:fail_route()
		end
		self:jump(loc)
		local tag = self:pop()
		tag.ignored = true
		return tag
	end
	
	function handle_start(self, this)
		if this == "/" then
			local text, head = self.text, self.head + 1
			local this = match(text, "^[^%s/>]+", head)
			if this and is_ignored_tag(self, lower(this)) then
				head = head + #this
				if not match(text, "^/[^>]", head) then
					return ignored_tag(self, text, head)
				end
			end
			return self:fail_route()
		elseif this == "" then
			return self:fail_route()
		end
		-- Tags are only case-insensitive with ASCII characters.
		local raw_name = this
		this = lower(this)
		local end_tag_pattern = (end_tags or get_end_tags())[this]
		if not end_tag_pattern then -- Validity check.
			return self:fail_route()
		end
		local layer = self[-1]
		local text, head = self.text, self.head + layer.step
		if match(text, "^/[^>]", head) then
			return self:fail_route()
		elseif is_ignored_tag(self, this) then
			return ignored_tag(self, text, head)
		-- If an onlyinclude tag is not ignored (and cannot be active since it
		-- would have triggered special handling earlier), it must be plaintext.
		elseif this == "onlyinclude" then
			return self:fail_route()
		elseif this == "noinclude" or this == "includeonly" then
			layer.ignored = true -- Ignored block.
			layer.raw_name = raw_name
		end
		layer.name, layer.handler, layer.end_tag_pattern = this, handle_tag, end_tag_pattern
		self:set_pattern(">")
	end
	
	function handle_tag(self, this)
		if this == "" then
			return self:fail_route()
		elseif this ~= ">" then
			self[-1].attributes = this
			return
		elseif self:read(-1) == "/" then
			self[-1].self_closing = true
			return self:pop()
		end
		local text, head, layer = self.text, self.head + 1, self[-1]
		local loc1, loc2 = find(text, layer.end_tag_pattern, head)
		if loc1 then
			if loc1 > head then
				self:emit(sub(text, head, loc1 - 1))
			end
			self:jump(loc2)
			return self:pop()
		-- noinclude and includeonly will tolerate having no closing tag, but
		-- only if given in lowercase. This is due to a preprocessor bug, as
		-- it uses a regex with the /i (case-insensitive) flag to check for
		-- end tags, but a simple array lookup with lowercase tag names when
		-- looking up which tags should tolerate no closing tag (exact match
		-- only, so case-sensitive).
		elseif layer.ignored then
			local raw_name = layer.raw_name
			if raw_name == "noinclude" or raw_name == "includeonly" then
				self:jump(#text)
				return self:pop()
			end
		end
		return self:fail_route()
	end
	
	function Parser:tag()
		-- HTML comment.
		if self:read(1, 3) == "!--" then
			local text = self.text
			self:jump(select(2, find(text, "-->", self.head + 4, true)) or #text)
		-- onlyinclude tags (which must be lowercase with no whitespace).
		elseif self.onlyinclude and self:read(1, 13) == "/onlyinclude>" then
			local text = self.text
			self:jump(select(2, find(text, "<onlyinclude>", self.head + 14, true)) or #text)
		else
			local success, tag = self:try(do_tag)
			if not success then
				self:emit("<")
			elseif not tag.ignored then
				tag.end_tag_pattern = nil
				self:emit(Tag:new(tag))
			end
		end
	end
end

-- Heading.
-- The preparser assigns each heading a number, which is used for things like section edit links. The preparser will only do this for heading blocks which aren't nested inside templates, parameters and parser tags. In some cases (e.g. when template blocks contain untrimmed newlines), a preparsed heading may not be treated as a heading in the final output. That does not affect the preparser, however, which will always count sections based on the preparser heading count, since it can't know what a template's final output will be.
do
	-- Handlers.
	local handle_start
	local handle_body
	local handle_possible_end
	
	local function do_heading(self)
		local layer, head = self[-1], self.head
		layer.handler, layer.index = handle_start, head
		self:set_pattern("[\t\n ]")
		-- Comments/tags interrupt the equals count.
		local eq = match(self.text, "^=+()", head) - head
		layer.level = eq
		self:advance(eq)
	end
	
	local function do_heading_possible_end(self)
		local layer = self[-1]
		layer.handler = handle_possible_end
		self:set_pattern("[\n<]")
	end
	
	function handle_start(self, ...)
		-- ===== is "=" as an L2; ======== is "==" as an L3 etc.
		local function newline(self)
			local layer = self[-1]
			local eq = layer.level
			if eq <= 2 then
				return self:fail_route()
			end
			-- Calculate which equals signs determine the heading level.
			local level_eq = eq - (2 - eq % 2)
			level_eq = level_eq > 12 and 12 or level_eq
			-- Emit the excess.
			self:emit(rep("=", eq - level_eq))
			layer.level = level_eq / 2
			return self:pop()
		end
		
		local function whitespace(self)
			local success, possible_end = self:try(do_heading_possible_end)
			if success then
				self:emit(Wikitext:new(possible_end))
				local layer = self[-1]
				layer.handler = handle_body
				self:set_pattern("[\n<=[{]")
				return self:consume()
			end
			return newline(self)
		end
		
		handle_start = self:switch(handle_start, {
			["\t"] = whitespace,
			["\n"] = newline,
			[" "] = whitespace,
			[""] = newline,
			
			[false] = function(self)
				-- Emit any excess = signs once we know it's a conventional heading. Up till now, we couldn't know if the heading is just a string of = signs (e.g. ========), so it wasn't guaranteed that the heading text starts after the 6th.
				local layer = self[-1]
				local eq = layer.level
				if eq > 6 then
					self:emit(1, rep("=", eq - 6))
					layer.level = 6
				end
				layer.handler = handle_body
				self:set_pattern("[\n<=[{]")
				return self:consume()
			end
		})
		return handle_start(self, ...)
	end
	
	function handle_body(self, ...)
		handle_body = self:switch(handle_body, {
			["\n"] = Parser.fail_route,
			["<"] = Parser.tag,
			
			["="] = function(self)
				-- Comments/tags interrupt the equals count.
				local eq = match(self.text, "^=+", self.head)
				local eq_len = #eq
				self:advance(eq_len)
				local success, possible_end = self:try(do_heading_possible_end)
				if success then
					self:emit(eq)
					self:emit(Wikitext:new(possible_end))
					return self:consume()
				end
				local layer = self[-1]
				local level = layer.level
				if eq_len > level then
					self:emit(rep("=", eq_len - level))
				elseif level > eq_len then
					layer.level = eq_len
					self:emit(1, rep("=", level - eq_len))
				end
				return self:pop()
			end,
			
			["["] = Parser.wikilink_block,
			
			["{"] = function(self, this)
				return self:braces("{", true)
			end,
			
			[""] = Parser.fail_route,
			[false] = Parser.emit
		})
		return handle_body(self, ...)
	end
	
	function handle_possible_end(self, ...)
		handle_possible_end = self:switch(handle_possible_end, {
			["\n"] = Parser.fail_route,
			
			["<"] = function(self)
				if self:read(1, 3) ~= "!--" then
					return self:pop()
				end
				local head = select(2, find(self.text, "-->", self.head + 4, true))
				if not head then
					return self:pop()
				end
				self:jump(head)
			end,
			
			[""] = Parser.fail_route,
			
			[false] = function(self, this)
				if not match(this, "^[\t ]+()$") then
					return self:pop()
				end
				self:emit(this)
			end
		})
		return handle_possible_end(self, ...)
	end
	
	function Parser:heading()
		local success, heading = self:try(do_heading)
		if success then
			local section = self.section + 1
			heading.section = section
			self.section = section
			self:emit(Heading:new(heading))
			return self:consume()
		else
			self:emit("=")
		end
	end
end

------------------------------------------------------------------------------------
--
-- Block handlers
--
------------------------------------------------------------------------------------

-- Block handlers.

-- These are blocks which can affect template/parameter parsing, since they're also parsed by Parsoid at the same time (even though they aren't processed until later).

-- All blocks (including templates/parameters) can nest inside each other, but an inner block must be closed before the outer block which contains it. This is why, for example, the wikitext "{{template| [[ }}" will result in an unprocessed template, since the inner "[[" is treated as the opening of a wikilink block, which prevents "}}" from being treated as the closure of the template block. On the other hand, "{{template| [[ ]] }}" will process correctly, since the wikilink block is closed before the template closure. It makes no difference whether the block will be treated as valid or not when it's processed later on, so "{{template| [[ }} ]] }}" would also work, even though "[[ }} ]]" is not a valid wikilink.

-- Note that nesting also affects pipes and equals signs, in addition to block closures.

-- These blocks can be nested to any degree, so "{{template| [[ [[ [[ ]] }}" will not work, since only one of the three wikilink blocks has been closed. On the other hand, "{{template| [[ [[ [[ ]] ]] ]] }}" will work.

-- All blocks are implicitly closed by the end of the text, since their validity is irrelevant at this stage.

-- Language conversion block.
-- Opens with "-{" and closes with "}-". However, templates/parameters take priority, so "-{{" is parsed as "-" followed by the opening of a template/parameter block (depending on what comes after).
-- Note: Language conversion blocks aren't actually enabled on the English Wiktionary, but Parsoid still parses them at this stage, so they can affect the closure of outer blocks: e.g. "[[ -{ ]]" is not a valid wikilink block, since the "]]" falls inside the new language conversion block.
do
	--Handler.
	local handle_language_conversion_block
	
	local function do_language_conversion_block(self)
		local layer = self[-1]
		layer.handler = handle_language_conversion_block
		self:set_pattern("[\n<[{}]")
	end
	
	function handle_language_conversion_block(self, ...)
		handle_language_conversion_block = self:switch(handle_language_conversion_block, {
			["\n"] = Parser.heading_block,
			["<"] = Parser.tag,
			["["] = Parser.wikilink_block,
			["{"] = Parser.braces,
			
			["}"] = function(self)
				if self:read(1) == "-" then
					self:emit("}-")
					self:advance()
					return self:pop()
				end
				self:emit("}")
			end,
			
			[""] = Parser.pop,
			[false] = Parser.emit
		})
		return handle_language_conversion_block(self, ...)
	end
	
	function Parser:braces(this, fail_on_unclosed_braces)
		local language_conversion_block = self:read(-1) == "-"
		if self:read(1) == "{" then
			local braces, failed = self:template_or_parameter()
			-- Headings will fail if they contain an unclosed brace block.
			if failed and fail_on_unclosed_braces then
				return self:fail_route()
			-- Language conversion blocks cannot begin "-{{", but can begin
			-- "-{{{" iff parsed as "-{" + "{{".
			elseif not (language_conversion_block and braces == 1) then
				return self:consume()
			end
		else
			self:emit(this)
			if not language_conversion_block then
				return
			end
			self:advance()
		end
		self:emit(Wikitext:new(self:get(do_language_conversion_block)))
	end
end

--[==[
Headings

Opens with "\n=" (or "=" at the start of the text), and closes with "\n" or the end of the text. Note that it doesn't matter whether the heading will fail to process due to a premature newline (e.g. if there are no closing signs), so at this stage the only thing that matters for closure is the newline or end of text.

Note: Heading blocks are only parsed like this if they occur inside a template, since they do not iterate the preparser's heading count (i.e. they aren't proper headings).

Note 2: if directly inside a template argument with no previous equals signs, a newline followed by a single equals sign is parsed as an argument equals sign, not the opening of a new L1 heading block. This does not apply to any other heading levels. As such, {{template|key\n=}}, {{template|key\n=value}} or even {{template|\n=}} will successfully close, but {{template|key\n==}}, {{template|key=value\n=more value}}, {{template\n=}} etc. will not, since in the latter cases the "}}" would fall inside the new heading block.
]==]
do
	--Handler.
	local handle_heading_block
	
	local function do_heading_block(self)
		local layer = self[-1]
		layer.handler = handle_heading_block
		self:set_pattern("[\n<[{]")
	end
	
	function handle_heading_block(self, ...)
		handle_heading_block = self:switch(handle_heading_block, {
			["\n"] = function(self)
				self:newline()
				return self:pop()
			end,
			
			["<"] = Parser.tag,
			["["] = Parser.wikilink_block,
			["{"] = Parser.braces,
			[""] = Parser.pop,
			[false] = Parser.emit
		})
		return handle_heading_block(self, ...)
	end
	
	function Parser:heading_block(this, nxt)
		self:newline()
		this = this .. (nxt or "=")
		local loc = #this - 1
		while self:read(0, loc) == this do
			self:advance()
			self:emit(Wikitext:new(self:get(do_heading_block)))
		end
	end
end

-- Wikilink block.
-- Opens with "[[" and closes with "]]".
do
	-- Handler.
	local handle_wikilink_block
	
	local function do_wikilink_block(self)
		local layer = self[-1]
		layer.handler = handle_wikilink_block
		self:set_pattern("[\n<[%]{]")
	end
	
	function handle_wikilink_block(self, ...)
		handle_wikilink_block = self:switch(handle_wikilink_block, {
			["\n"] = Parser.heading_block,
			["<"] = Parser.tag,
			["["] = Parser.wikilink_block,
			
			["]"] = function(self)
				if self:read(1) == "]" then
					self:emit("]]")
					self:advance()
					return self:pop()
				end
				self:emit("]")
			end,
			
			["{"] = Parser.braces,
			[""] = Parser.pop,
			[false] = Parser.emit
		})
		return handle_wikilink_block(self, ...)
	end
	
	function Parser:wikilink_block()
		if self:read(1) == "[" then
			self:emit("[[")
			self:advance(2)
			self:emit(Wikitext:new(self:get(do_wikilink_block)))
		else
			self:emit("[")
		end
	end
end

-- Lines which only contain comments, " " and "\t" are eaten, so long as
-- they're bookended by "\n" (i.e. not the first or last line).
function Parser:newline()
	local text, head = self.text, self.head
	while true do
		repeat
			local loc = match(text, "^[\t ]*<!%-%-()", head + 1)
			if not loc then
				break
			end
			loc = select(2, find(text, "-->", loc, true))
			head = loc or head
		until not loc
		-- Fail if no comments found.
		if head == self.head then
			break
		end
		head = match(text, "^[\t ]*()\n", head + 1)
		if not head then
			break
		end
		self:jump(head)
	end
	self:emit("\n")
end

do
	-- Handlers.
	local handle_start
	local main_handler
	
	-- If `transcluded` is true, then the text is checked for a pair of
	-- onlyinclude tags. If these are found (even if they're in the wrong
	-- order), then the start of the page is treated as though it is preceded
	-- by a closing onlyinclude tag.
	-- Note 1: unlike other parser extension tags, onlyinclude tags are case-
	-- sensitive and cannot contain whitespace.
	-- Note 2: onlyinclude tags *can* be implicitly closed by the end of the
	-- text, but the hard requirement above means this can only happen if
	-- either the tags are in the wrong order or there are multiple onlyinclude
	-- blocks.
	local function do_parse(self, transcluded)
		local layer = self[-1]
		layer.handler = handle_start
		self:set_pattern(".")
		self.section = 0
		if not transcluded then
			return
		end
		self.transcluded = true
		local text = self.text
		if find(text, "</onlyinclude>", nil, true) then
			local head = find(text, "<onlyinclude>", nil, true)
			if head then
				self.onlyinclude = true
				self:jump(head + 13)
			end
		end
	end
	
	-- If the first character is "=", try parsing it as a heading.
	function handle_start(self, this)
		local layer = self[-1]
		layer.handler = main_handler
		self:set_pattern("[\n<{]")
		if this == "=" then
			return self:heading()
		end
		return self:consume()
	end
	
	function main_handler(self, ...)
		main_handler = self:switch(main_handler, {
			["\n"] = function(self)
				self:newline()
				if self:read(1) == "=" then
					self:advance()
					return self:heading()
				end
			end,
			
			["<"] = Parser.tag,
			
			["{"] = function(self)
				if self:read(1) == "{" then
					self:template_or_parameter()
					return self:consume()
				end
				self:emit("{")
			end,
			
			[""] = Parser.pop,
			[false] = Parser.emit
		})
		return main_handler(self, ...)
	end
	
	function export.parse(text, transcluded)
		local text_type = type(text)
		return (select(2, Parser:parse{
			text = text_type == "string" and text or
				text_type == "number" and tostring(text) or
				error("bad argument #1 (string expected, got " .. text_type .. ")"),
			node = {Wikitext, true},
			route = {do_parse, transcluded}
		}))
	end
	parse = export.parse
end

do
	local function is_template(v)
		return class_else_type(v) == "template"
	end
	
	function export.find_templates(text, not_transcluded)
		return parse(text, not not_transcluded):iterate(is_template)
	end
end

do
	local link_parameter_1, link_parameter_2
	
	local function get_link_parameter_1()
		link_parameter_1, get_link_parameter_1 = (data or get_data()).template_link_param_1, nil
		return link_parameter_1
	end
	
	local function get_link_parameter_2()
		link_parameter_2, get_link_parameter_2 = (data or get_data()).template_link_param_2, nil
		return link_parameter_2
	end
	
	-- Generate a link. If the target title doesn't have a fragment, use "#top"
	-- (which is an implicit anchor at the top of every page), as this ensures
	-- self-links still display as links, since bold display is distracting and
	-- unintuitive for template links.
	local function link_page(title, display)
		local fragment = title.fragment
		if fragment == "" then
			fragment = "top"
		end
		return format(
			"[[:%s|%s]]",
			encode_uri(title.prefixedText .. "#" .. fragment, "WIKI"),
			display
		)
	end
	
	-- pf_arg1 or pf_arg2 may need to be linked if a given parser function
	-- treats them as a pagename. If a key exists in `namespace`, the value is
	-- the namespace for the page: if not 0, then the namespace prefix will
	-- always be added to the input (e.g. {{#invoke:}} can only target the
	-- Module: namespace, so inputting "Template:foo" gives
	-- "Module:Template:foo", and "Module:foo" gives "Module:Module:foo").
	-- However, this isn't possible with mainspace (namespace 0), so prefixes
	-- are respected. make_title handles all of this automatically.
	local function finalize_arg(pagename, namespace)
		if namespace == nil then
			return pagename
		end
		local title = make_title(namespace, pagename)
		return title and is_internal_title(title) and link_page(title, pagename) or pagename
	end
	
	local function render_title(name, args)
		-- parse_template_name returns a table of transclusion modifiers plus
		-- the normalized template/magic word name, which will be used as link
		-- targets. The third return value pf_arg1 is the first argument of a
		-- a parser function, which comes after the colon (e.g. "foo" in
		-- "{{#IF:foo|bar|baz}}"). This means args[1] (i.e. the first argument
		-- that comes after a pipe is actually argument 2, and so on. Note: the
		-- second parameter of parse_template_name checks if there are any
		-- arguments, since parser variables cannot take arguments (e.g.
		-- {{CURRENTYEAR}} is a parser variable, but {{CURRENTYEAR|foo}}
		-- transcludes "Template:CURRENTYEAR"). In such cases, the returned
		-- table explicitly includes the "Template:" prefix in the template
		-- name. The third parameter instructs it to retain any fragment in the
		-- template name in the returned table, if present.
		local chunks, subclass, pf_arg1 = parse_template_name(
			name,
			args and pairs(args)(args) ~= nil,
			true
		)
		if chunks == nil then
			return name, args
		end
		local chunks_len = #chunks
		-- Additionally, generate the corresponding table `rawchunks`, which
		-- is a list of colon-separated chunks in the raw input. This is used
		-- to retrieve the display forms for each chunk.
		local rawchunks = split(name, ":")
		for i = 1, chunks_len - 1 do
			chunks[i] = format(
				"[[%s|%s]]",
				encode_uri((magic_words or get_magic_words())[sub(chunks[i], 1, -2)].transclusion_modifier, "WIKI"),
				rawchunks[i]
			)
		end
		local chunk = chunks[chunks_len]
		-- If it's a template, return a link to it with link_page, concatenating
		-- the remaining chunks in `rawchunks` to form the display text.
		-- Use new_title with the default namespace 10 (Template:) to generate
		-- a target title, which is the same setting used for retrieving
		-- templates (including those in other namespaces, as prefixes override
		-- the default).
		if subclass == "template" then
			chunks[chunks_len] = link_page(
				new_title(chunk, 10),
				concat(rawchunks, "&#58;", chunks_len) -- :
			)
			return concat(chunks, "&#58;"), args -- :
		elseif subclass == "parser variable" then
			chunks[chunks_len] = format(
				"[[%s|%s]]",
				encode_uri((magic_words or get_magic_words())[chunk].parser_variable, "WIKI"),
				rawchunks[chunks_len]
			)
			return concat(chunks, "&#58;"), args -- :
		end
		-- Otherwise, it must be a parser function.
		local mgw_data = (magic_words or get_magic_words())[sub(chunk, 1, -2)]
		local link = mgw_data.parser_function or mgw_data.transclusion_modifier
		local pf_arg2 = args and args[1] or nil
		-- Some magic words have different links,  depending on whether argument
		-- 2 is specified (e.g. "baz" in {{foo:bar|baz}}).
		if type(link) == "table" then
			link = pf_arg2 and link[2] or link[1]
		end
		chunks[chunks_len] = format(
			"[[%s|%s]]",
			encode_uri(link, "WIKI"),
			rawchunks[chunks_len]
		)
		-- #TAG: has special handling, because documentation links for parser
		-- extension tags come from [[Module:data/parser extension tags]].
		if chunk == "#TAG:" then
			-- Tags are only case-insensitive with ASCII characters.
			local tag = (parser_extension_tags or get_parser_extension_tags())[lower(php_trim(pf_arg1))]
			if tag then
				pf_arg1 = format(
					"[[%s|%s]]",
					encode_uri(tag, "WIKI"),
					pf_arg1
				)
			end
		-- Otherwise, finalize pf_arg1 and add it to `chunks`.
		else
			pf_arg1 = finalize_arg(pf_arg1, (link_parameter_1 or get_link_parameter_1())[chunk])
		end
		chunks[chunks_len + 1] = pf_arg1
		-- Finalize pf_arg2 (if applicable), then return.
		if pf_arg2 then
			args = shallowcopy(args) -- Avoid destructively modifying args.
			args[1] = finalize_arg(pf_arg2, (link_parameter_2 or get_link_parameter_2())[chunk])
		end
		return concat(chunks, "&#58;"), args -- :
	end
	
	function export.buildTemplate(title, args)
		local output = {title}
		if not args then
			return output
		end
		-- Iterate over all numbered parameters in order, followed by any
		-- remaining parameters in codepoint order. Implicit parameters are
		-- used wherever possible, even if explicit numbers are interpolated
		-- between them (e.g. 0 would go before any implicit parameters, and
		-- 2.5 between 2 and 3).
		-- TODO: handle "=" and "|" in params/values.
		local implicit
		for k, v in sorted_pairs(args) do
			if type(k) == "number" and k >= 1 and k % 1 == 0 then
				if implicit == nil then
					implicit = table_len(args)
				end
				insert(output, k <= implicit and v or k .. "=" .. v)
			else
				insert(output, k .. "=" .. v)
			end
		end
		return output
	end
	build_template = export.buildTemplate
	
	function export.templateLink(title, args, no_link)
		if not no_link then
			title, args = render_title(title, args)
		end
		local output = build_template(title, args)
		for i = 1, #output do
			output[i] = encode_entities(output[i], "={}", true, true)
		end
		return tostring(html_create("code")
			:css("white-space", "pre-wrap")
			:wikitext("&#123;&#123;" .. concat(output, "&#124;") .. "&#125;&#125;") -- {{ | }}
		)
	end
end

do
	local function is_parameter(v)
		return class_else_type(v) == "parameter"
	end
	
	function export.find_parameters(text, not_transcluded)
		return parse(text, not not_transcluded):iterate(is_parameter)
	end
	
	function export.displayParameter(name, default)
		return tostring(html_create("code")
			:css("white-space", "pre-wrap")
			:wikitext("&#123;&#123;&#123;" .. concat({name, default}, "&#124;") .. "&#125;&#125;&#125;") -- {{{ | }}}
		)
	end
end

do
	local function check_level(level)
		if type(level) ~= "number" then
			error("Heading levels must be numbers.")
		elseif level < 1 or level > 6 or level % 1 ~= 0 then
			error("Heading levels must be integers between 1 and 6.")
		end
		return level
	end
	
	local function is_heading(v)
		return class_else_type(v) == "heading"
	end
	
	-- FIXME: should headings which contain "\n" be returned? This may depend
	-- on variable factors, like template expansion. They iterate the heading
	-- count number, but fail on rendering. However, in some cases a different
	-- heading might still be rendered due to intermediate equals signs; it
	-- may even be of a different heading level: e.g., this is parsed as an
	-- L2 heading with a newline (due to the wikilink block), but renders as the
	-- L1 heading "=foo[[". Section edit links are sometimes (but not always)
	-- present in such cases.
	
	-- ==[[=
	-- ]]==
	
	-- TODO: section numbers for edit links seem to also include headings
	-- nested inside templates and parameters (but apparently not those in
	-- parser extension tags - need to test this more). If we ever want to add
	-- section edit links manually, this will need to be accounted for.
	function export.find_headings(text, i, j)
		local parsed = parse(text)
		if i == nil and j == nil then
			return parsed:iterate(is_heading)
		end
		i = i and check_level(i) or 1
		j = j and check_level(j) or 6
		return parsed:iterate(function(v)
			if class_else_type(v) == "heading" then
				local level = v.level
				return level >= i and level <= j
			end
		end)
	end
end

do
	local function make_tag(tag)
		return tostring(html_create("code")
			:css("white-space", "pre-wrap")
			:wikitext("&lt;" .. tag .. "&gt;")
		)
	end

	-- Note: invalid tags are returned without links.
	function export.wikitagLink(tag)
		-- ">" can't appear in tags (including attributes) since the parser
		-- unconditionally treats ">" as the end of a tag.
		if find(tag, ">", nil, true) then
			return make_tag(tag)
		end
		-- Tags must start "<tagname..." or "</tagname...", with no whitespace
		-- after "<" or "</".
		local slash, tagname, remainder = match(tag, "^(/?)([^/%s]+)(.*)$")
		if not tagname then
			return make_tag(tag)
		end
		-- Tags are only case-insensitive with ASCII characters.
		local link = lower(tagname)
		if (
			-- onlyinclude tags must be lowercase and are whitespace intolerant.
			link == "onlyinclude" and (link ~= tagname or remainder ~= "") or
			-- Closing wikitags (except onlyinclude) can only have whitespace
			-- after the tag name.
			slash == "/" and not match(remainder, "^%s*()$") or
			-- Tagnames cannot be followed immediately by "/", unless it comes
			-- at the end (e.g. "<nowiki/>", but not "<nowiki/ >").
			remainder ~= "/" and sub(remainder, 1, 1) == "/"
		) then
			-- Output with no link.
			return make_tag(tag)
		end
		-- Partial transclusion tags aren't in the table of parser extension
		-- tags.
		if link == "noinclude" or link == "includeonly" or link == "onlyinclude" then
			link = "mw:Transclusion#Partial transclusion"
		else
			link = (parser_extension_tags or get_parser_extension_tags())[link]
		end
		if link then
			tag = gsub(tag, pattern_escape(tagname), "[[" .. replacement_escape(encode_uri(link, "WIKI")) .. "|%0]]", 1)
		end
		return make_tag(tag)
	end
end

-- For convenience.
export.class_else_type = class_else_type

return export