mermaid chart sdk integrated

.vscode/settings.json

@@ -7,5 +7,6 @@
         "out": true // set this to false to include "out" folder in search results
     },
     // Turn off tsc task auto detection since we have the necessary tasks as npm scripts
-    "typescript.tsc.autoDetect": "off"
+    "typescript.tsc.autoDetect": "off",
+    "cSpell.words": ["mermaidchart"]
 }

package.json

@@ -514,8 +514,9 @@
     "typescript": "^4.9.5"
   },
   "dependencies": {
-    "axios": "^1.4.0",
     "uuid": "^9.0.0",
+    "@mermaidchart/sdk": "^0.2.0",
     "yaml": "^2.7.0"
-  }
+  },
+  "packageManager": "[email protected]+sha512.faf344af2d6ca65c4c5c8c2224ea77a81a5e8859cbc4e06b1511ddce2f0151512431dd19e6aff31f2c6a8f5f2aced9bd2273e1fed7dd4de1868984059d2c4247"
 }

src/extension.ts

@@ -1,5 +1,5 @@
 import * as vscode from "vscode";
-import { MermaidChartProvider, MCTreeItem, getAllTreeViewProjectsCache } from "./mermaidChartProvider";
+import { MermaidChartProvider, MCTreeItem, getAllTreeViewProjectsCache, getProjectIdForDocument } from "./mermaidChartProvider";
 import { MermaidChartVSCode } from "./mermaidChartVSCode";
 import {
   applyMermaidChartTokenHighlighting,
@@ -193,7 +193,7 @@ context.subscriptions.push(
 );
 
 context.subscriptions.push(
-  vscode.commands.registerCommand('mermaid.connectDiagram',async(uri:vscode.Uri, range:vscode.Range)=>{
+  vscode.commands.registerCommand('mermaid.connectDiagram', async (uri: vscode.Uri, range: vscode.Range) => {
     const document = await vscode.workspace.openTextDocument(uri);
     const content = document.getText();
     const fileExt = path.extname(document.fileName);
@@ -205,31 +205,57 @@ context.subscriptions.push(
       projects.map((p) => ({ label: p.title, description: p.title, projectId: p.uuid })),
       { placeHolder: "Select a project to save the diagram" }
     );
-    
+
     if (!selectedProject || !selectedProject?.projectId) {
-        vscode.window.showInformationMessage("Operation cancelled.");
-        return;
+      vscode.window.showInformationMessage("Operation cancelled.");
+      return;
     }
 
-    const response = await mcAPI.createDocumentWithDiagram(diagramCode, selectedProject.projectId)
+    try {
+      const newDocument = await mcAPI.createDocument(selectedProject.projectId);
+
+      if (!newDocument || !newDocument.documentID) {
+        vscode.window.showErrorMessage("Failed to create a new document.");
+        return;
+      }
+
+      await mcAPI.setDocument({
+        documentID: newDocument.documentID,
+        projectID: selectedProject.projectId,
+        code: diagramCode,
+      });
+
+      const processedCode = ensureIdField(diagramCode, newDocument.documentID);
+      mermaidChartProvider.refresh();
+      const editor = await createMermaidFile(context, processedCode, true);
 
-    const processedCode = ensureIdField(diagramCode, response.documentID);
-       const editor= await await createMermaidFile(context, processedCode, true);
-       if(editor){
-        syncAuxFile(editor.document.uri.toString(), uri,range);
-       }
+      if (editor) {
+        syncAuxFile(editor.document.uri.toString(), uri, range);
+      }
+    } catch (error) {
+        vscode.window.showErrorMessage(`Error creating or connecting document: ${error instanceof Error ? error.message : "Unknown error occurred."}`);
+    }
   })
-)
-  vscode.workspace.onWillSaveTextDocument(async (event) => {
-    if (event.document.languageId.startsWith("mermaid")) {
-      event.waitUntil(Promise.resolve([]));
-      const content = event.document.getText();
-      const diagramId = extractIdFromCode(content);
-      if (diagramId) {
-          await mcAPI.saveDocumentCode(content, diagramId);
+);
+
+vscode.workspace.onWillSaveTextDocument(async (event) => {
+  if (event.document.languageId.startsWith("mermaid")) {
+    event.waitUntil(Promise.resolve([]));
+    const content = event.document.getText();
+    const diagramId = extractIdFromCode(content);
+    if (diagramId) {
+      const projectId = getProjectIdForDocument(diagramId);
+
+      if (projectId) {
+        await mcAPI.setDocument({
+          documentID: diagramId,
+          projectID: projectId,
+          code: content,
+        });
       }
     }
-  });
+  }
+});
 
   context.subscriptions.push(
     vscode.commands.registerCommand('mermaidChart.syncDiagramWithMermaid', async () => {
@@ -245,13 +271,29 @@ context.subscriptions.push(
         try {
             const diagramId = extractIdFromCode(content);
             if (TempFileCache.hasTempUri(context, document.uri.toString()) && diagramId) {
-                await mcAPI.saveDocumentCode(content, diagramId);
+                const projectId = getProjectIdForDocument(diagramId);
+
+                if (projectId) {
+                  await mcAPI.setDocument({
+                    documentID: diagramId,
+                    projectID: projectId,
+                    code: content,
+                  });
+                }
                 vscode.window.showInformationMessage(`Diagram synced successfully with Mermaid chart. Diagram ID: ${diagramId}`);
             } else if (TempFileCache.hasTempUri(context, document.uri.toString())){
               vscode.window.showInformationMessage('This is temporary buffer, this can not be saved locally');
           } else if (!TempFileCache.hasTempUri(context, document.uri.toString()) && diagramId) {
               await vscode.commands.executeCommand('workbench.action.files.save');
-                await mcAPI.saveDocumentCode(content, diagramId);
+              const projectId = getProjectIdForDocument(diagramId);
+
+                if (projectId) {
+                  await mcAPI.setDocument({
+                    documentID: diagramId,
+                    projectID: projectId,
+                    code: content,
+                  });
+                }
                 vscode.window.showInformationMessage(`Diagram synced successfully with Mermaid chart. Diagram ID: ${diagramId}`);
             } else {
               await vscode.commands.executeCommand('workbench.action.files.save');