Add `xs` lexer and mappings.

2 files changed, 79 insertions, 1 deletions

diff --git a/lua/lexers/xs.lua b/lua/lexers/xs.lua
new file mode 100644
index 0000000..2243a6f
--- /dev/null
+++ b/lua/lexers/xs.lua
@@ -0,0 +1,75 @@
+-- Copyright 2017 Michael Forney. See LICENSE.
+-- Copyright 2017 David B. Lamkins. See LICENSE.
+-- xs 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 = 'xs'}
+
+-- Whitespace.
+local ws = token(l.WHITESPACE, l.space^1)
+
+-- Comments.
+local comment = token(l.COMMENT, '#' * l.nonnewline^0)
+
+-- Strings.
+local str = l.delimited_range("'", false, true)
+local herestr = '<<<' * str
+local heredoc = '<<' * P(function(input, index)
+  local s, e, _, delimiter =
+    input:find('[ \t]*(["\']?)([%w!"%%+,-./:?@_~]+)%1', index)
+  if s == index and delimiter then
+    delimiter = delimiter:gsub('[%%+-.?]', '%%%1')
+    local _, e = input:find('[\n\r]'..delimiter..'[\n\r]', e)
+    return e and e + 1 or #input + 1
+  end
+end)
+local string = token(l.STRING, str + herestr + heredoc)
+
+-- Numbers.
+local number = token(l.NUMBER, l.integer + l.float)
+
+-- Keywords.
+local keyword = token(l.KEYWORD, word_match({
+  'access', 'alias', 'catch', 'cd', 'dirs', 'echo', 'else', 'escape', 'eval',
+  'exec', 'exit', 'false', 'fn-', 'fn', 'for', 'forever', 'fork', 'history',
+  'if', 'jobs', 'let', 'limit', 'local', 'map', 'omap', 'popd', 'printf',
+  'pushd', 'read', 'result', 'set-', 'switch', 'throw', 'time', 'true',
+  'umask', 'until', 'unwind-protect', 'var', 'vars', 'wait', 'whats', 'while',
+  ':lt', ':le', ':gt', ':ge', ':eq', ':ne', '~', '~~', '...', '.',
+}, '!"%*+,-./:?@[]~'))
+
+-- Constants.
+local constant = token(l.CONSTANT, '$&' * l.word)
+
+-- Identifiers.
+local identifier = token(l.IDENTIFIER, l.word)
+
+-- Variables.
+local variable = token(l.VARIABLE,
+                       '$' * S('"#')^-1 * ('*' + l.digit^1 + l.word))
+
+-- Operators.
+local operator = token(l.OPERATOR, S('@`=!<>*&^|;?()[]{}') + '\\\n')
+
+M._rules = {
+  {'whitespace', ws},
+  {'keyword', keyword},
+  {'constant', constant},
+  {'identifier', identifier},
+  {'string', string},
+  {'comment', comment},
+  {'number', number},
+  {'variable', variable},
+  {'operator', operator},
+}
+
+M._foldsymbols = {
+  _patterns = {'[{}]', '#'},
+  [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 945bff9..54a1d4d 100644
--- a/lua/plugins/filetype.lua
+++ b/lua/plugins/filetype.lua
@@ -301,7 +301,7 @@ vis.ftdetect.filetypes = {
 		mime = { "text/x-python" },
 	},
 	rc = {
-		ext = { "%.rc$", "%.es$", "%.xs" },
+		ext = { "%.rc$", "%.es$" },
 	},
 	rebol = {
 		ext = { "%.r$", "%.reb$" },
@@ -398,6 +398,9 @@ vis.ftdetect.filetypes = {
 	wsf = {
 		ext = { "%.wsf$" },
 	},
+	xs = {
+		ext = { "%.xs$", "%.xsin", "%.xsrc" },
+	},
 	xml = {
 		ext = { "%.dtd$", "%.svg$", "%.xml$", "%.xsd$", "%.xsl$", "%.xslt$", "%.xul$" },
 	},