From 69a916aa0c8afe9be0473028a3212a5d2aa481c9 Mon Sep 17 00:00:00 2001
From: Christian Hesse <mail@eworm.de>
Date: Thu, 17 Sep 2020 18:00:45 +0200
Subject: lexers: add Mikrotik RouterOS script lexer

Add a lexer for Mikrotik RouterOS scripts.

https://wiki.mikrotik.com/wiki/Manual:Scripting
---
 lua/lexers/routeros.lua  | 116 +++++++++++++++++++++++++++++++++++++++++++++++
 lua/plugins/filetype.lua |   6 +++
 2 files changed, 122 insertions(+)
 create mode 100644 lua/lexers/routeros.lua

(limited to 'lua')

diff --git a/lua/lexers/routeros.lua b/lua/lexers/routeros.lua
new file mode 100644
index 0000000..7544393
--- /dev/null
+++ b/lua/lexers/routeros.lua
@@ -0,0 +1,116 @@
+-- Copyright 2020 Christian Hesse
+-- Mikrotik RouterOS script LPeg lexer.
+
+local l = require('lexer')
+local token, word_match = l.token, l.word_match
+local P, R, S = lpeg.P, lpeg.R, lpeg.S
+
+local M = {_NAME = 'routeros'}
+
+-- Whitespace.
+local ws = token(l.WHITESPACE, l.space^1)
+
+-- Comments.
+local comment = token(l.COMMENT, '#' * l.nonnewline^0)
+
+-- Strings.
+local string = token(l.STRING, l.delimited_range('"'))
+
+-- Numbers.
+local number = token(l.NUMBER, l.float + l.integer)
+
+-- Keywords.
+local keyword = token(l.KEYWORD, word_match({
+  -- control
+  ':delay',
+  ':do', 'on-error', 'while',
+  ':error',
+  ':foreach', 'in', 'do',
+  ':for', 'from', 'to', 'step',
+  ':if', 'do', 'else',
+  ':return',
+  ':while', 'do',
+  -- menu specific commands
+  'add',
+  'disable',
+  'edit',
+  'enable',
+  'export',
+  'find',
+  'get',
+  'info',
+  'monitor',
+  'print', 'append', 'as-value', 'brief', 'count-only', 'detail', 'file',
+      'follow', 'follow-only', 'from', 'interval', 'terse', 'value-list',
+      'where', 'without-paging',
+  'remove',
+  'set',
+  -- output & string handling
+  ':beep',
+  ':blink',
+  ':environment',
+  ':execute',
+  ':find',
+  ':len',
+  ':log', 'alert', 'critical', 'debug', 'emergency', 'error', 'info',
+      'notice', 'warning',
+  ':parse',
+  ':pick',
+  ':put',
+  ':terminal',
+  ':time',
+  ':typeof',
+  -- variable declaration
+  ':global',
+  ':local',
+  ':set',
+  -- variable casting
+  ':toarray',
+  ':tobool',
+  ':toid',
+  ':toip',
+  ':toip6',
+  ':tonum',
+  ':tostr',
+  ':totime',
+  -- boolean values and logical operators
+  'false', 'no',
+  'true', 'yes',
+  'and',
+  'in',
+  'or',
+  -- networking
+  ':ping',
+  ':resolve'
+}, ':-'))
+
+-- Identifiers.
+local identifier = token(l.IDENTIFIER, l.word)
+
+-- Variables.
+local variable = token(l.VARIABLE,
+                       '$' * (S('!#?*@$') + l.digit^1 + l.word +
+                              l.delimited_range('{}', true, true, true)))
+
+-- Operators.
+local operator = token(l.OPERATOR, S('=!%<>+-/*&|~.,;()[]{}'))
+
+M._rules = {
+  {'whitespace', ws},
+  {'keyword', keyword},
+  {'identifier', identifier},
+  {'string', string},
+  {'comment', comment},
+  {'number', number},
+  {'variable', variable},
+  {'operator', operator},
+}
+
+M._foldsymbols = {
+  _patterns = {'[a-z]+', '[{}]', '#'},
+  [l.KEYWORD] = { },
+  [l.OPERATOR] = {['{'] = 1, ['}'] = -1},
+  [l.COMMENT] = {['#'] = l.fold_line_comments('#')}
+}
+
+return M
diff --git a/lua/plugins/filetype.lua b/lua/plugins/filetype.lua
index 048b35d..fa1717e 100644
--- a/lua/plugins/filetype.lua
+++ b/lua/plugins/filetype.lua
@@ -343,6 +343,12 @@ vis.ftdetect.filetypes = {
 	rhtml = {
 		ext = { "%.erb$", "%.rhtml$" },
 	},
+	routeros = {
+		ext = { "%.rsc" },
+		detect = function(file, data)
+			return data:match("^#.* by RouterOS")
+		end
+	},
 	rstats = {
 		ext = { "%.R$", "%.Rout$", "%.Rhistory$", "%.Rt$", "Rout.save", "Rout.fail" },
 	},
-- 
cgit v1.2.3