1. 1
2. 2
3. 3
4. 4
5. 5
6. 6
7. 7
8. 8
9. 9
10. 10
11. 11
12. 12
13. 13
14. 14
15. 15
16. 16
17. 17
18. 18
19. 19
20. 20
21. 21
22. 22
23. 23
24. 24
25. 25
26. 26
27. 27
28. 28
29. 29
30. 30
31. 31
32. 32
33. 33
34. 34
35. 35
36. 36
37. 37
38. 38
39. 39
40. 40
41. 41
42. 42
43. 43
44. 44
45. 45
46. 46
47. 47
48. 48
49. 49
50. 50
51. 51
52. 52
53. 53
54. 54
55. 55
56. 56
57. 57
58. 58
59. 59
60. 60
61. 61
62. 62
63. 63
64. 64
65. 65
66. 66
67. 67
68. 68
69. 69
70. 70
71. 71
72. 72
73. 73
74. 74
75. 75
76. 76
77. 77
78. 78
79. 79

package l

import (
	. "github.com/alecthomas/chroma" // nolint
	"github.com/alecthomas/chroma/lexers/internal"
)

// Lua lexer.
var Lua = internal.Register(MustNewLazyLexer(
	&Config{
		Name:      "Lua",
		Aliases:   []string{"lua"},
		Filenames: []string{"*.lua", "*.wlua"},
		MimeTypes: []string{"text/x-lua", "application/x-lua"},
	},
	luaRules,
))

func luaRules() Rules {
	return Rules{
		"root": {
			{`#!.*`, CommentPreproc, nil},
			Default(Push("base")),
		},
		"ws": {
			{`(?:--\[(=*)\[[\w\W]*?\](\1)\])`, CommentMultiline, nil},
			{`(?:--.*$)`, CommentSingle, nil},
			{`(?:\s+)`, Text, nil},
		},
		"base": {
			Include("ws"),
			{`(?i)0x[\da-f]*(\.[\da-f]*)?(p[+-]?\d+)?`, LiteralNumberHex, nil},
			{`(?i)(\d*\.\d+|\d+\.\d*)(e[+-]?\d+)?`, LiteralNumberFloat, nil},
			{`(?i)\d+e[+-]?\d+`, LiteralNumberFloat, nil},
			{`\d+`, LiteralNumberInteger, nil},
			{`(?s)\[(=*)\[.*?\]\1\]`, LiteralString, nil},
			{`::`, Punctuation, Push("label")},
			{`\.{3}`, Punctuation, nil},
			{`[=<>|~&+\-*/%#^]+|\.\.`, Operator, nil},
			{`[\[\]{}().,:;]`, Punctuation, nil},
			{`(and|or|not)\b`, OperatorWord, nil},
			{`(break|do|else|elseif|end|for|if|in|repeat|return|then|until|while)\b`, KeywordReserved, nil},
			{`goto\b`, KeywordReserved, Push("goto")},
			{`(local)\b`, KeywordDeclaration, nil},
			{`(true|false|nil)\b`, KeywordConstant, nil},
			{`(function)\b`, KeywordReserved, Push("funcname")},
			{`[A-Za-z_]\w*(\.[A-Za-z_]\w*)?`, Name, nil},
			{`'`, LiteralStringSingle, Combined("stringescape", "sqs")},
			{`"`, LiteralStringDouble, Combined("stringescape", "dqs")},
		},
		"funcname": {
			Include("ws"),
			{`[.:]`, Punctuation, nil},
			{`(?:[^\W\d]\w*)(?=(?:(?:--\[(=*)\[[\w\W]*?\](\2)\])|(?:--.*$)|(?:\s+))*[.:])`, NameClass, nil},
			{`(?:[^\W\d]\w*)`, NameFunction, Pop(1)},
			{`\(`, Punctuation, Pop(1)},
		},
		"goto": {
			Include("ws"),
			{`(?:[^\W\d]\w*)`, NameLabel, Pop(1)},
		},
		"label": {
			Include("ws"),
			{`::`, Punctuation, Pop(1)},
			{`(?:[^\W\d]\w*)`, NameLabel, nil},
		},
		"stringescape": {
			{`\\([abfnrtv\\"\']|[\r\n]{1,2}|z\s*|x[0-9a-fA-F]{2}|\d{1,3}|u\{[0-9a-fA-F]+\})`, LiteralStringEscape, nil},
		},
		"sqs": {
			{`'`, LiteralStringSingle, Pop(1)},
			{`[^\\']+`, LiteralStringSingle, nil},
		},
		"dqs": {
			{`"`, LiteralStringDouble, Pop(1)},
			{`[^\\"]+`, LiteralStringDouble, nil},
		},
	}
}