update cdl/def diagnostic

client/src/extension.ts

@@ -1,11 +1,21 @@
 import * as vscode from "vscode"
+import {
+    LanguageClient,
+    LanguageClientOptions,
+    ServerOptions,
+    Executable
+} from "vscode-languageclient/node"
 import {
     formatCdlFile,
     completionHandlerCdl,
     hoverCdlHandler,
-    formatDefFile
+    formatDefFile,
+    isFirstLineMachine,
+    diagnosticHandler
 } from "./common/handlers"
 
+let client: LanguageClient
+
 export function activate(context: vscode.ExtensionContext) {
     const formatCdlProvider = vscode.languages.registerDocumentFormattingEditProvider(
         { scheme: "file", language: "cdl" },
@@ -70,6 +80,31 @@ export function activate(context: vscode.ExtensionContext) {
     )
 
     context.subscriptions.push(formatDefProvider)
-}
 
-export function deactivate() {}
+    // Diagnostics collection
+    const diagnosticCollectionCdl = vscode.languages.createDiagnosticCollection("cdl")
+    const diagnosticCollectionDef = vscode.languages.createDiagnosticCollection("def")
+    context.subscriptions.push(diagnosticCollectionCdl, diagnosticCollectionDef)
+
+    // Check if the first line of the CDL file contains "MACHINE"
+    vscode.workspace.onDidOpenTextDocument((document) => {
+        if (document.languageId === "cdl" || document.languageId === "def") {
+            if (document.languageId === "cdl") {
+                diagnosticCollectionCdl.set(document.uri, diagnosticHandler(document))
+            } else if (document.languageId === "def") {
+                diagnosticCollectionDef.set(document.uri, diagnosticHandler(document))
+            }
+        }
+    })
+
+    vscode.workspace.onDidChangeTextDocument((event) => {
+        const document = event.document
+        if (document.languageId === "cdl" || document.languageId === "def") {
+            if (document.languageId === "cdl") {
+                diagnosticCollectionCdl.set(document.uri, diagnosticHandler(document))
+            } else if (document.languageId === "def") {
+                diagnosticCollectionDef.set(document.uri, diagnosticHandler(document))
+            }
+        }
+    })
+}