Jianw
2025-05-14 f01dbecb66e96d528d1ec70feebe3e87611a936a
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
-- Copyright 2015-2018 Mitchell mitchell.att.foicica.com. See License.txt.
-- PowerShell LPeg lexer.
-- Contributed by Jeff Stone.
 
local lexer = require('lexer')
local token, word_match = lexer.token, lexer.word_match
local P, R, S = lpeg.P, lpeg.R, lpeg.S
 
local lex = lexer.new('powershell')
 
-- Whitespace.
lex:add_rule('whitespace', token(lexer.WHITESPACE, lexer.space^1))
 
-- Comments.
lex:add_rule('comment', token(lexer.COMMENT, '#' * lexer.nonnewline^0))
 
-- Keywords.
lex:add_rule('keyword', token(lexer.KEYWORD, word_match([[
  Begin Break Continue Do Else End Exit For ForEach ForEach-Object Get-Date
  Get-Random If Param Pause Powershell Process Read-Host Return Switch While
  Write-Host
]], true)))
 
-- Comparison Operators.
lex:add_rule('comparison', token(lexer.KEYWORD, '-' * word_match([[
  and as band bor contains eq ge gt is isnot le like lt match ne nomatch not
  notcontains notlike or replace
]], true)))
 
-- Parameters.
lex:add_rule('parameter', token(lexer.KEYWORD, '-' * word_match([[
  Confirm Debug ErrorAction ErrorVariable OutBuffer OutVariable Verbose WhatIf
]], true)))
 
-- Properties.
lex:add_rule('property', token(lexer.KEYWORD, '.' * word_match([[
  day dayofweek dayofyear hour millisecond minute month second timeofday year
]], true)))
 
-- Types.
lex:add_rule('type', token(lexer.KEYWORD, '[' * word_match([[
  array boolean byte char datetime decimal double hashtable int long single
  string xml
]], true) * ']'))
 
-- Variables.
lex:add_rule('variable', token(lexer.VARIABLE,
                               '$' * (lexer.digit^1 + lexer.word +
                                      lexer.delimited_range('{}', true, true))))
 
-- Strings.
lex:add_rule('string', token(lexer.STRING, lexer.delimited_range('"', true)))
 
-- Numbers.
lex:add_rule('number', token(lexer.NUMBER, lexer.float + lexer.integer))
 
-- Operators.
lex:add_rule('operator', token(lexer.OPERATOR, S('=!<>+-/*^&|~.,:;?()[]{}%`')))
 
-- Fold points.
lex:add_fold_point(lexer.OPERATOR, '{', '}')
 
return lex