fix classes keyword

d2lsp/completion.go

@@ -145,6 +145,9 @@ func getKeywordContext(text string, m *d2ast.Map, line, column int) string {
 		if _, isBoard := d2ast.BoardKeywords[firstPart]; isBoard {
 			firstPart = ""
 		}
+		if firstPart == "classes" {
+			firstPart = ""
+		}
 
 		_, isHolder := d2ast.ReservedKeywordHolders[firstPart]
 		if !isHolder {

d2lsp/completion_test.go

@@ -30,6 +30,18 @@ func TestGetCompletionItems(t *testing.T) {
 			column: 8,
 			want:   getStyleCompletions(),
 		},
+		{
+			name: "classes shapes",
+			text: `classes: {
+  goal: {
+    shape:
+  }
+}
+`,
+			line:   2,
+			column: 10,
+			want:   getShapeCompletions(),
+		},
 		{
 			name: "nested style map suggestions",
 			text: `a: {
@@ -291,6 +303,15 @@ layers: {
 			column: 16,
 			want:   getShapeCompletions(),
 		},
+		{
+			name: "shape 2 suggestions",
+			text: `a: {
+  shape:
+}`,
+			line:   1,
+			column: 8,
+			want:   getShapeCompletions(),
+		},
 	}
 
 	for _, tt := range tests {