1package t
 2
 3import (
 4	. "github.com/alecthomas/chroma" // nolint
 5	"github.com/alecthomas/chroma/lexers/internal"
 6)
 7
 8// Tasm lexer.
 9var Tasm = internal.Register(MustNewLexer(
10	&Config{
11		Name:            "TASM",
12		Aliases:         []string{"tasm"},
13		Filenames:       []string{"*.asm", "*.ASM", "*.tasm"},
14		MimeTypes:       []string{"text/x-tasm"},
15		CaseInsensitive: true,
16	},
17	Rules{
18		"root": {
19			{`^\s*%`, CommentPreproc, Push("preproc")},
20			Include("whitespace"),
21			{`[@a-z$._?][\w$.?#@~]*:`, NameLabel, nil},
22			{`BITS|USE16|USE32|SECTION|SEGMENT|ABSOLUTE|EXTERN|GLOBAL|ORG|ALIGN|STRUC|ENDSTRUC|ENDS|COMMON|CPU|GROUP|UPPERCASE|INCLUDE|EXPORT|LIBRARY|MODULE|PROC|ENDP|USES|ARG|DATASEG|UDATASEG|END|IDEAL|P386|MODEL|ASSUME|CODESEG|SIZE`, Keyword, Push("instruction-args")},
23			{`([@a-z$._?][\w$.?#@~]*)(\s+)(db|dd|dw|T[A-Z][a-z]+)`, ByGroups(NameConstant, KeywordDeclaration, KeywordDeclaration), Push("instruction-args")},
24			{`(?:res|d)[bwdqt]|times`, KeywordDeclaration, Push("instruction-args")},
25			{`[@a-z$._?][\w$.?#@~]*`, NameFunction, Push("instruction-args")},
26			{`[\r\n]+`, Text, nil},
27		},
28		"instruction-args": {
29			{"\"(\\\\\"|[^\"\\n])*\"|'(\\\\'|[^'\\n])*'|`(\\\\`|[^`\\n])*`", LiteralString, nil},
30			{`(?:0x[0-9a-f]+|$0[0-9a-f]*|[0-9]+[0-9a-f]*h)`, LiteralNumberHex, nil},
31			{`[0-7]+q`, LiteralNumberOct, nil},
32			{`[01]+b`, LiteralNumberBin, nil},
33			{`[0-9]+\.e?[0-9]+`, LiteralNumberFloat, nil},
34			{`[0-9]+`, LiteralNumberInteger, nil},
35			Include("punctuation"),
36			{`r[0-9][0-5]?[bwd]|[a-d][lh]|[er]?[a-d]x|[er]?[sb]p|[er]?[sd]i|[c-gs]s|st[0-7]|mm[0-7]|cr[0-4]|dr[0-367]|tr[3-7]`, NameBuiltin, nil},
37			{`[@a-z$._?][\w$.?#@~]*`, NameVariable, nil},
38			{`(\\\s*)(;.*)([\r\n])`, ByGroups(Text, CommentSingle, Text), nil},
39			{`[\r\n]+`, Text, Pop(1)},
40			Include("whitespace"),
41		},
42		"preproc": {
43			{`[^;\n]+`, CommentPreproc, nil},
44			{`;.*?\n`, CommentSingle, Pop(1)},
45			{`\n`, CommentPreproc, Pop(1)},
46		},
47		"whitespace": {
48			{`[\n\r]`, Text, nil},
49			{`\\[\n\r]`, Text, nil},
50			{`[ \t]+`, Text, nil},
51			{`;.*`, CommentSingle, nil},
52		},
53		"punctuation": {
54			{`[,():\[\]]+`, Punctuation, nil},
55			{`[&|^<>+*=/%~-]+`, Operator, nil},
56			{`[$]+`, KeywordConstant, nil},
57			{`seg|wrt|strict`, OperatorWord, nil},
58			{`byte|[dq]?word`, KeywordType, nil},
59		},
60	},
61))