add tokenizer for graphql language variables

microsoft · Jan 14, 2020 · 3655d9a · 3655d9a
1 parent 03db2f6
commit 3655d9a
Showing 1 changed file with 18 additions and 4 deletions.
diff --git a/src/graphql/graphql.ts b/src/graphql/graphql.ts
@@ -69,16 +69,30 @@ export const language = <ILanguage>{
 	// The main tokenizer for our languages
 	tokenizer: {
 		root: [
-			// identifiers and keywords
+
+			// fields and argument names
 			[
-				/[a-z_$][\w$]*/,
+				/[a-z_][\w$]*/,
 				{
 					cases: {
 						'@keywords': 'keyword',
-						'@default': 'identifier',
+						'@default': 'key.identifier',
 					},
 				},
 			],
+
+			// identify typed input variables
+			[
+				/[$][\w$]*/,
+				{
+					cases: {
+						'@keywords': 'keyword',
+						'@default': 'argument.identifier',
+					},
+				},
+			],
+
+			// to show class names nicely
 			[
 				/[A-Z][\w\$]*/,
 				{
@@ -87,7 +101,7 @@ export const language = <ILanguage>{
 						'@default': 'type.identifier',
 					},
 				},
-			], // to show class names nicely
+			],
 
 			// whitespace
 			{ include: '@whitespace' },