From d7a6ac2aa7dd650c1932a4c84475fc9ceebcae57 Mon Sep 17 00:00:00 2001
From: Yash-Singh1 <saiansh2525@gmail.com>
Date: Sun, 22 Aug 2021 10:57:20 -0700
Subject: [PATCH] stateDiagram-v2

---
 src/lib/components/editor/util.ts | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/lib/components/editor/util.ts b/src/lib/components/editor/util.ts
index ea2825dd..9409e89c 100644
--- a/src/lib/components/editor/util.ts
+++ b/src/lib/components/editor/util.ts
@@ -9,6 +9,7 @@ export const initEditor = (monacoEditor): void => {
 	const typeKeywords = [
 		'flowchart',
 		'graph',
+		'stateDiagram-v2',
 		'stateDiagram',
 		'sequenceDiagram',
 		'classDiagram',
@@ -79,7 +80,7 @@ export const initEditor = (monacoEditor): void => {
 				[/[-+>]+/, { cases: { '@eatingSequenceArrows': 'transition' } }],
 				[':::', 'transition'],
 				[
-					/[a-z_$][\w$]*/,
+					/[a-z_$][\w$]*((?<=stateDiagram)-v2)?/,
 					{
 						cases: {
 							'@typeKeywords': 'typeKeyword',