48,403
edits
(Created page with "local m_table local concat = table.concat local getmetatable = getmetatable local insert = table.insert local next = next local rawget = rawget local rawset = rawset local remove = table.remove local select = select local setmetatable = setmetatable local type = type local unpack = unpack local classes = {} local metamethods = mw.loadData("Module:parser/data").metamethods ------------------------------------------------------------------------------------ -- -- Helper...") |
No edit summary |
||
| Line 1: | Line 1: | ||
local | local export = {} | ||
local metamethods_data_module = "Module:data/metamethods" | |||
local table_module = "Module:table" | |||
local concat = table.concat | local concat = table.concat | ||
| Line 8: | Line 11: | ||
local rawset = rawset | local rawset = rawset | ||
local remove = table.remove | local remove = table.remove | ||
local | local require = require | ||
local setmetatable = setmetatable | local setmetatable = setmetatable | ||
local type = type | local type = type | ||
| Line 14: | Line 17: | ||
local classes = {} | local classes = {} | ||
local | |||
--[==[ | |||
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 deep_copy(...) | |||
deep_copy = require(table_module).deepCopy | |||
return deep_copy(...) | |||
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 metamethods | |||
local function get_metamethods() | |||
-- Use require, since lookup times are much slower with mw.loadData. | |||
metamethods, get_metamethods = require(metamethods_data_module), nil | |||
return metamethods | |||
end | |||
------------------------------------------------------------------------------------ | ------------------------------------------------------------------------------------ | ||
| Line 22: | Line 40: | ||
------------------------------------------------------------------------------------ | ------------------------------------------------------------------------------------ | ||
local function get_nested( | local function get_nested(t, k, ...) | ||
if | if t == nil then | ||
return nil | return nil | ||
elseif ... then | elseif ... == nil then | ||
return | return t[k] | ||
end | end | ||
return | return get_nested(t[k], ...) | ||
end | end | ||
local function set_nested( | local function set_nested(t, k, v, ...) | ||
if | if ... ~= nil then | ||
local t_next = t[k] | |||
if t_next == nil then | |||
t_next = {} | |||
return set_nested( | t[k] = t_next | ||
end | |||
return set_nested(t_next, v, ...) | |||
end | end | ||
t[k] = v | |||
end | end | ||
| Line 44: | Line 64: | ||
if parent then | if parent then | ||
for method, value in next, parent do | for method, value in next, parent do | ||
if metamethods[method] then | if child[method] == nil and (metamethods or get_metamethods())[method] ~= nil then | ||
child[method] = value | child[method] = value | ||
end | end | ||
| Line 57: | Line 77: | ||
local function is_node(value) | local function is_node(value) | ||
return | if value == nil then | ||
return false | |||
end | |||
local mt = getmetatable(value) | |||
return not (mt == nil or classes[mt] == nil) | |||
end | end | ||
-- Recursively calling tostring() adds to the C stack (limit: 200), whereas calling __tostring metamethods directly does not. Occasionally relevant when dealing with very deep nesting. | -- Recursively calling tostring() adds to the C stack (limit: 200), whereas | ||
-- calling __tostring metamethods directly does not. Occasionally relevant when | |||
-- dealing with very deep nesting. | |||
local tostring | local tostring | ||
do | do | ||
| Line 73: | Line 99: | ||
end | end | ||
local function | local function class_else_type(value) | ||
return | if value == nil then | ||
return type(value) | |||
end | |||
local mt = getmetatable(value) | |||
if mt == nil then | |||
return type(value) | |||
end | |||
local class = classes[mt] | |||
return class == nil and type(value) or class | |||
end | end | ||
| Line 88: | Line 122: | ||
function Node:next(i) | function Node:next(i) | ||
i = i + 1 | i = i + 1 | ||
return self[i], i | return self[i], self, i | ||
end | end | ||
--[==[ | |||
Implements recursive iteration over a node tree. | |||
By default, when a node is encountered (which may contain other nodes), it is returned on the first iteration, and then any child nodes are returned on each subsequent iteration; the same process is then followed if any of those children contain nodes themselves. Once a particular node has been fully traversed, the iterator then continues with any sibling nodes. The iterator will use the `next` method of each node to traverse it, which may differ depending on the node class. | |||
Each iteration returns three values: `value`, `node` and `key`. Together, these can be used to manipulate the node tree at any given point without needing to know the full structure. Note that when the input node is returned on the first iteration, `node` and `key` will be nil. | |||
The optional argument `test` can be used to limit the return values. This should be a function that returns a boolean value, where a return value of true means that the child will be returned by the iterator. If a node is not returned by the iterator, it will still be traversed, as it may contain children that should be returned. | |||
The method `iterate_nodes` is provided as a special instance of iterate which uses `is_node` as the test.]==] | |||
function Node:iterate(test) | |||
local node, k, n, nodes, keys, returned_self = self, 0, 0 | |||
-- Special case if `test` is `is_node`. | |||
local is_node_is_test = test == is_node | |||
function | return function() | ||
if not returned_self then | |||
returned_self = true | |||
if test == nil or test(self) then | |||
return self | |||
return self | |||
end | end | ||
end | end | ||
-- Get `v`, which is the value at the last-returned key of the current node; if `v` is a node, it will be iterated over (i.e. recursive iteration). By default, `v` will be the last-returned value, but looking it up here means that any modifications made to the node during the loop will be taken into account. This makes it possible to swap one node out for something else (e.g. another node), or to remove it entirely, without being locked into recursively iterating over the old node; instead, the new node (if any) will be iterated over. This means node trees can be modified on-the-fly during the course of a single loop. | |||
if is_node(v) then | local v, node_check = node[k], true | ||
while true do | |||
-- If `v` is a node, memoize the current node and key, then iterate over it. | |||
if node_check and is_node(v) then | |||
-- `n` is the current memo level. | |||
n = n + 1 | |||
if nodes then | |||
nodes[n], keys[n] = node, k | |||
else | |||
nodes, keys = {node}, {k} | |||
end | |||
node, k = v, 0 | |||
end | |||
v, node, k = node:next(k) | |||
-- If `v` is nil, move down one level, then continue iterating the node on that level (if any), or otherwise terminate the loop. | |||
if v == nil then | |||
if n == 0 then | |||
return nil | |||
end | |||
node, k, n = nodes[n], keys[n], n - 1 | |||
elseif test == nil or test(v) then | |||
return v, node, k | |||
-- If `test` is `is_node`, there's no point checking it again on the next loop. | |||
elseif node_check and is_node_is_test then | |||
node_check = false | |||
end | |||
end | end | ||
end | end | ||
end | end | ||
function Node: | function Node:iterate_nodes() | ||
return | return self:iterate(is_node) | ||
end | end | ||
| Line 176: | Line 191: | ||
function Node:clone() | function Node:clone() | ||
return | return deep_copy(self, "keep", true) | ||
end | end | ||
function Node:new_class(class) | function Node:new_class(class) | ||
local t = | local t = {type = class} | ||
t.__index = t | t.__index = t | ||
t = inherit_metamethods(t, self) | |||
classes[t] = class | classes[t] = class | ||
return setmetatable(t, self) | return setmetatable(t, self) | ||
end | end | ||
Node.keys_to_remove = {"fail", "handler", "head", "override", "route"} | |||
function Node:new(t) | function Node:new(t) | ||
setmetatable(t, nil) | setmetatable(t, nil) | ||
local keys_to_remove = self.keys_to_remove | |||
for i = 1, #keys_to_remove do | |||
t[keys_to_remove[i]] = nil | |||
end | |||
return setmetatable(t, self) | return setmetatable(t, self) | ||
end | end | ||
| Line 198: | Line 217: | ||
function Proxy:__index(k) | function Proxy:__index(k) | ||
local v = Proxy[k] | |||
if v ~= nil then | |||
return v | |||
end | |||
return self.__chars[k] | |||
end | end | ||
| Line 224: | Line 247: | ||
i = i + 1 | i = i + 1 | ||
local char = self.__chars[i] | local char = self.__chars[i] | ||
if char then | if char ~= nil then | ||
return i, self[i], self, self.__parents[i], self.__keys[i] | return i, self[i], self, self.__parents[i], self.__keys[i] | ||
end | end | ||
| Line 250: | Line 273: | ||
function Parser:read(delta) | function Parser:read(delta) | ||
local v = self.text[self.head + (delta or 0)] | |||
return v == nil and "" or v | |||
end | end | ||
function Parser:advance(n) | function Parser:advance(n) | ||
self.head = self.head + (n or | self.head = self.head + (n == nil and 1 or n) | ||
end | end | ||
function Parser:layer(n) | function Parser:layer(n) | ||
if n then | if n ~= nil then | ||
return rawget(self, #self + n) | return rawget(self, #self + n) | ||
end | end | ||
| Line 266: | Line 290: | ||
function Parser:emit(a, b) | function Parser:emit(a, b) | ||
local layer = self[-1] | local layer = self[-1] | ||
if b then | if b ~= nil then | ||
insert(layer, signed_index(layer, a), b) | insert(layer, signed_index(layer, a), b) | ||
else | else | ||
| Line 275: | Line 299: | ||
function Parser:emit_tokens(a, b) | function Parser:emit_tokens(a, b) | ||
local layer = self[-1] | local layer = self[-1] | ||
if b then | if b ~= nil then | ||
a = signed_index(layer, a) | a = signed_index(layer, a) | ||
for i = 1, #b do | for i = 1, #b do | ||
| Line 291: | Line 315: | ||
function Parser:remove(n) | function Parser:remove(n) | ||
local layer = self[-1] | local layer = self[-1] | ||
if n then | if n ~= nil then | ||
return remove(layer, signed_index(layer, n)) | return remove(layer, signed_index(layer, n)) | ||
end | end | ||
| Line 307: | Line 331: | ||
-- Unlike default table.concat, this respects __tostring metamethods. | -- Unlike default table.concat, this respects __tostring metamethods. | ||
function Parser:concat(a, b, c) | function Parser:concat(a, b, c) | ||
if | if a == nil or a > 0 then | ||
return self:concat(0, a, b) | return self:concat(0, a, b) | ||
end | end | ||
local layer = self:layer(a) | local layer, ret, n = self:layer(a), {}, 0 | ||
for i = b and signed_index(layer, b) or 1, c and signed_index(layer, c) or #layer do | |||
for i = signed_index(layer, b) or 1, signed_index(layer, c) or #layer do | n = n + 1 | ||
ret[n] = tostring(layer[i]) | |||
end | end | ||
return concat(ret) | return concat(ret) | ||
| Line 319: | Line 343: | ||
function Parser:emitted(delta) | function Parser:emitted(delta) | ||
delta = delta | if delta == nil then | ||
delta = -1 | |||
end | |||
local i = 0 | local i = 0 | ||
while true do | while true do | ||
local layer = self:layer(i) | local layer = self:layer(i) | ||
if | if layer == nil then | ||
return nil | return nil | ||
end | end | ||
| Line 366: | Line 392: | ||
self[len] = nil | self[len] = nil | ||
len = len - 1 | len = len - 1 | ||
self[-1] = self | local new = self[len] | ||
if | self[-1] = new == nil and self or new | ||
if layer.sublayer == nil then | |||
break | break | ||
end | end | ||
| Line 378: | Line 405: | ||
local len, layer = #self, self[-1] | local len, layer = #self, self[-1] | ||
self[len] = nil | self[len] = nil | ||
self[-1] | local new = self[len - 1] | ||
self[-1] = new == nil and self or new | |||
setmetatable(layer, nil) | |||
layer.sublayer = nil | |||
return layer | |||
end | end | ||
function Parser:get(route, ...) | function Parser:get(route, ...) | ||
self:push(route) | self:push(route) | ||
local layer = | local layer = route(self, ...) | ||
if layer == nil then | if layer == nil then | ||
layer = self:traverse() | layer = self:traverse() | ||
end | end | ||
return layer | |||
return false, | end | ||
function Parser:try(route, ...) | |||
local failed_layer = get_nested(self.failed_routes, route, self.head) | |||
if failed_layer ~= nil then | |||
return false, failed_layer | |||
end | end | ||
return | local layer = self:get(route, ...) | ||
return not layer.fail, layer | |||
end | end | ||
function Parser:consume(this, ...) | function Parser:consume(this, ...) | ||
local layer = self[-1] | local layer = self[-1] | ||
return (layer.override or layer.handler)(self, this | if this == nil then | ||
this = self:read() | |||
end | |||
return (layer.override or layer.handler)(self, this, ...) | |||
end | end | ||
| Line 406: | Line 441: | ||
local layer = self:pop() | local layer = self:pop() | ||
layer.fail = true | layer.fail = true | ||
set_nested(self, "failed_routes", layer. | set_nested(self, "failed_routes", layer.route, layer.head, layer) | ||
self.head = layer.head | self.head = layer.head | ||
return layer | return layer | ||
| Line 414: | Line 449: | ||
while true do | while true do | ||
local layer = self:consume() | local layer = self:consume() | ||
if layer then | if layer ~= nil then | ||
return layer | return layer | ||
end | end | ||
| Line 442: | Line 477: | ||
-- Generate a new parser class object, which is used as the template for any parser objects. These should be customized with additional/modified methods as needed. | -- Generate a new parser class object, which is used as the template for any parser objects. These should be customized with additional/modified methods as needed. | ||
function Parser:new_class() | function Parser:new_class() | ||
local t = | local t = {} | ||
t.__index = t | t.__index = t | ||
return setmetatable(t, self) | return setmetatable(inherit_metamethods(t, self), self) | ||
end | end | ||
| Line 457: | Line 492: | ||
function Parser:parse(data) | function Parser:parse(data) | ||
local parser = self:new(data.text) | local parser = self:new(data.text) | ||
local success, tokens = parser: | local success, tokens = parser:try(unpack(data.route)) | ||
if #parser > 0 then | if #parser > 0 then | ||
-- This shouldn't happen. | -- This shouldn't happen. | ||
| Line 467: | Line 502: | ||
return false, nil, parser | return false, nil, parser | ||
end | end | ||
error("Parser exited with | error("Parser exited with failed route.") | ||
end | end | ||
export.class_else_type = class_else_type | |||
export.is_node = is_node | export.is_node = is_node | ||
export.tostring = tostring | export.tostring = tostring | ||
function export.new() | function export.new() | ||