1 files changed, 58 insertions, 0 deletions
diff --git a/lexlua/mediawiki.lua b/lexlua/mediawiki.lua
new file mode 100644
index 000000000..f3d7efff7
--- /dev/null
+++ b/lexlua/mediawiki.lua
@@ -0,0 +1,58 @@
+-- Copyright 2006-2019 Mitchell mitchell.att.foicica.com. See License.txt.
+-- MediaWiki LPeg lexer.
+-- Contributed by Alexander Misel.
+
+local lexer = require('lexer')
+local token, word_match = lexer.token, lexer.word_match
+local P, R, S, B = lpeg.P, lpeg.R, lpeg.S, lpeg.B
+
+local lex = lexer.new('mediawiki')
+
+-- Comments.
+lex:add_rule('comment', token(lexer.COMMENT, '<!--' * (lexer.any - '-->')^0 *
+                                             P('-->')^-1))
+
+-- HTML-like tags
+local tag_start = token('tag_start', '<' * P('/')^-1 * lexer.alnum^1 *
+                                     lexer.space^0)
+local tag_attr = token('tag_attr', lexer.alpha^1 * lexer.space^0 *
+                                   ('=' * lexer.space^0 *
+                                    ('"' * ((lexer.any - S('>"\\')) +
+                                     ('\\' * lexer.any))^0 * '"' +
+                                     (lexer.any - lexer.space - '>')^0)^-1)^0 *
+                                   lexer.space^0)
+local tag_end = token('tag_end', P('/')^-1 * '>')
+lex:add_rule('tag', tag_start * tag_attr^0 * tag_end)
+lex:add_style('tag_start', lexer.STYLE_KEYWORD)
+lex:add_style('tag_attr', lexer.STYLE_TYPE)
+lex:add_style('tag_end', lexer.STYLE_KEYWORD)
+
+-- Link
+lex:add_rule('link', token(lexer.STRING, S('[]')))
+lex:add_rule('internal_link', B('[[') *
+                              token('link_article', (lexer.any - '|' - ']]')^1))
+lex:add_style('link_article', lexer.STYLE_STRING..',underlined')
+
+-- Templates and parser functions.
+lex:add_rule('template', token(lexer.OPERATOR, S('{}')))
+lex:add_rule('parser_func', B('{{') *
+                            token('parser_func', P('#') * lexer.alpha^1 +
+                                                 lexer.upper^1 * ':'))
+lex:add_rule('template_name', B('{{') *
+                              token('template_name', (lexer.any - S('{}|'))^1))
+lex:add_style('parser_func', lexer.STYLE_FUNCTION)
+lex:add_style('template_name', lexer.STYLE_OPERATOR..',underlined')
+
+-- Operators.
+lex:add_rule('operator', token(lexer.OPERATOR, S('-=|#~!')))
+
+-- Behavior switches
+local start_pat = P(function(_, pos) return pos == 1 end)
+lex:add_rule('behavior_switch', (B(lexer.space) + start_pat) *
+                                token('behavior_switch',
+                                      '__' * (P('TOC') + 'FORCETOC' + 'NOTOC' +
+                                              'NOEDITSECTION' + 'NOCC' +
+                                              'NOINDEX') * '__') * #lexer.space)
+lex:add_style('behavior_switch', lexer.STYLE_KEYWORD)
+
+return lex