open the file as canvas view right when open

Author: trangdoan982

apps/obsidian/src/index.ts

@@ -25,6 +25,7 @@ export default class DiscourseGraphPlugin extends Plugin {
   private styleElement: HTMLStyleElement | null = null;
   private tagNodeHandler: TagNodeHandler | null = null;
   private currentViewActions: { leaf: WorkspaceLeaf; action: any }[] = [];
+  private pendingCanvasSwitches = new Set<string>();
 
   async onload() {
     await this.loadSettings();
@@ -40,26 +41,38 @@ export default class DiscourseGraphPlugin extends Plugin {
           if (!leaf) return;
 
           const view = leaf.view;
-          if (!(view instanceof MarkdownView)) return;
-
-          const file = view.file;
-          if (!file) return;
-
-          const cache = this.app.metadataCache.getFileCache(file);
-          if (cache?.frontmatter?.[FRONTMATTER_KEY]) {
-            // Add new action and track it
-            const action = view.addAction(
-              "layout",
-              "View as canvas",
-              async () => {
-                await leaf.setViewState({
+          const file = view instanceof MarkdownView ? view.file : null;
+
+          if (file) {
+            const cache = this.app.metadataCache.getFileCache(file);
+            const isCanvasFile = !!cache?.frontmatter?.[FRONTMATTER_KEY];
+
+            // If this file is pending a switch, switch it now
+            if (this.pendingCanvasSwitches.has(file.path) && isCanvasFile) {
+              if (view.getViewType() !== VIEW_TYPE_TLDRAW_DG_PREVIEW) {
+                void leaf.setViewState({
                   type: VIEW_TYPE_TLDRAW_DG_PREVIEW,
                   state: view.getState(),
                 });
-              },
-            );
+              }
+              this.pendingCanvasSwitches.delete(file.path);
+              return;
+            }
+
+            if (view instanceof MarkdownView && isCanvasFile) {
+              const action = view.addAction(
+                "layout",
+                "View as canvas",
+                async () => {
+                  await leaf.setViewState({
+                    type: VIEW_TYPE_TLDRAW_DG_PREVIEW,
+                    state: view.getState(),
+                  });
+                },
+              );
 
-            this.currentViewActions.push({ leaf, action });
+              this.currentViewActions.push({ leaf, action });
+            }
           }
         },
       ),
@@ -71,14 +84,9 @@ export default class DiscourseGraphPlugin extends Plugin {
 
         const cache = this.app.metadataCache.getFileCache(file);
         if (cache?.frontmatter?.[FRONTMATTER_KEY]) {
-          const leaf =
-            this.app.workspace.getActiveViewOfType(MarkdownView)?.leaf;
-          if (leaf) {
-            void leaf.setViewState({
-              type: VIEW_TYPE_TLDRAW_DG_PREVIEW,
-              state: leaf.view.getState(),
-            });
-          }
+          // Mark this file as needing a switch
+          // The actual switch will happen in active-leaf-change when the view is ready
+          this.pendingCanvasSwitches.add(file.path);
         }
       }),
     );