vis

a vi-like editor based on Plan 9's structural regular expressions
erlang.lua

(3588B)
      1 -- Copyright 2006-2025 Mitchell. See LICENSE.
      2 -- Erlang LPeg lexer.
      3 
      4 local lexer = require('lexer')
      5 local token, word_match = lexer.token, lexer.word_match
      6 local P, S = lpeg.P, lpeg.S
      7 
      8 local lex = lexer.new('erlang')
      9 
     10 -- Whitespace.
     11 lex:add_rule('whitespace', token(lexer.WHITESPACE, lexer.space^1))
     12 
     13 -- Keywords.
     14 lex:add_rule('keyword', token(lexer.KEYWORD, word_match{
     15 	'after', 'begin', 'case', 'catch', 'cond', 'end', 'fun', 'if', 'let', 'of', 'query', 'receive',
     16 	'try', 'when',
     17 	-- Operators.
     18 	'div', 'rem', 'or', 'xor', 'bor', 'bxor', 'bsl', 'bsr', 'and', 'band', 'not', 'bnot', 'badarg',
     19 	'nocookie', 'orelse', 'andalso', 'false', 'true'
     20 }))
     21 
     22 -- Functions.
     23 lex:add_rule('function', token(lexer.FUNCTION, word_match{
     24 	'abs', 'alive', 'apply', 'atom_to_list', 'binary_to_list', 'binary_to_term', 'concat_binary',
     25 	'date', 'disconnect_node', 'element', 'erase', 'exit', 'float', 'float_to_list', 'get',
     26 	'get_keys', 'group_leader', 'halt', 'hd', 'integer_to_list', 'is_alive', 'is_record', 'length',
     27 	'link', 'list_to_atom', 'list_to_binary', 'list_to_float', 'list_to_integer', 'list_to_pid',
     28 	'list_to_tuple', 'load_module', 'make_ref', 'monitor_node', 'node', 'nodes', 'now', 'open_port',
     29 	'pid_to_list', 'process_flag', 'process_info', 'process', 'put', 'register', 'registered',
     30 	'round', 'self', 'setelement', 'size', 'spawn', 'spawn_link', 'split_binary', 'statistics',
     31 	'term_to_binary', 'throw', 'time', 'tl', 'trunc', 'tuple_to_list', 'unlink', 'unregister',
     32 	'whereis',
     33 	-- Others.
     34 	'any', 'atom', 'binary', 'bitstring', 'byte', 'constant', 'function', 'integer', 'list', 'map',
     35 	'mfa', 'non_neg_integer', 'number', 'pid', 'ports', 'port_close', 'port_info', 'pos_integer',
     36 	'reference', 'record',
     37 	-- Erlang.
     38 	'check_process_code', 'delete_module', 'get_cookie', 'hash', 'math', 'module_loaded', 'preloaded',
     39 	'processes', 'purge_module', 'set_cookie', 'set_node',
     40 	-- Math.
     41 	'acos', 'asin', 'atan', 'atan2', 'cos', 'cosh', 'exp', 'log', 'log10', 'min', 'max', 'pi', 'pow',
     42 	'power', 'sin', 'sinh', 'sqrt', 'tan', 'tanh'
     43 }))
     44 
     45 -- Identifiers.
     46 lex:add_rule('identifier', token(lexer.IDENTIFIER, lexer.lower * ('_' + lexer.alnum)^0))
     47 
     48 -- Variables.
     49 lex:add_rule('variable', token(lexer.VARIABLE, P('_')^0 * lexer.upper * ('_' + lexer.alnum)^0))
     50 
     51 -- Directives.
     52 lex:add_rule('directive', token(lexer.PREPROCESSOR, '-' * word_match{
     53 	'author', 'behaviour', 'behavior', 'compile', 'copyright', 'define', 'doc', 'else', 'endif',
     54 	'export', 'file', 'ifdef', 'ifndef', 'import', 'include', 'include_lib', 'module', 'record',
     55 	'spec', 'type', 'undef'
     56 }))
     57 
     58 -- Strings.
     59 local sq_str = lexer.range("'", true)
     60 local dq_str = lexer.range('"')
     61 lex:add_rule('string', token(lexer.STRING, sq_str + dq_str + '$' * lexer.any * lexer.alnum^0))
     62 
     63 -- Comments.
     64 lex:add_rule('comment', token(lexer.COMMENT, lexer.to_eol('%')))
     65 
     66 -- Numbers.
     67 lex:add_rule('number', token(lexer.NUMBER, lexer.number))
     68 
     69 -- Operators.
     70 lex:add_rule('operator', token(lexer.OPERATOR, S('-<>.;=/|+*:,!()[]{}')))
     71 
     72 -- Preprocessor.
     73 lex:add_rule('preprocessor', token(lexer.TYPE, '?' * lexer.word))
     74 
     75 -- Records.
     76 lex:add_rule('type', token(lexer.TYPE, '#' * lexer.word))
     77 
     78 -- Fold points.
     79 lex:add_fold_point(lexer.KEYWORD, 'case', 'end')
     80 lex:add_fold_point(lexer.KEYWORD, 'fun', 'end')
     81 lex:add_fold_point(lexer.KEYWORD, 'if', 'end')
     82 lex:add_fold_point(lexer.KEYWORD, 'query', 'end')
     83 lex:add_fold_point(lexer.KEYWORD, 'receive', 'end')
     84 lex:add_fold_point(lexer.OPERATOR, '(', ')')
     85 lex:add_fold_point(lexer.OPERATOR, '[', ']')
     86 lex:add_fold_point(lexer.OPERATOR, '{', '}')
     87 
     88 lexer.property['scintillua.comment'] = '%'
     89 
     90 return lex