From 05034b74e3cba6bec075be89e55d8d80ef02bbad Mon Sep 17 00:00:00 2001
From: nomadoor <nomadoor@gmail.com>
Date: Sun, 19 Oct 2025 00:17:51 +0900
Subject: [PATCH] refactor: simplify main host management in ensureMainHosts
 function

---
 js/dual_panel_subgraph.js | 24 ++++--------------------
 1 file changed, 4 insertions(+), 20 deletions(-)

diff --git a/js/dual_panel_subgraph.js b/js/dual_panel_subgraph.js
index 75234fd..2d05a70 100644
--- a/js/dual_panel_subgraph.js
+++ b/js/dual_panel_subgraph.js
@@ -116,30 +116,14 @@ app.registerExtension({
         }
 
         function ensureMainHosts() {
-            const currentCanvas = getMainCanvasElement();
-            const isHostValid = (el) => el && el.isConnected && el.classList.contains(HOST_CLASS);
             if (state.mainHosts.length) {
-                if (state.mainHosts.every(isHostValid)) {
-                    const canvasParent = currentCanvas?.parentElement;
-                    if (canvasParent && state.mainHosts.includes(canvasParent)) {
-                        return;
-                    }
-                }
-                resetMainHosts();
-            }
-            if (!currentCanvas) {
                 return;
             }
-            const hosts = [];
-            let current = currentCanvas.parentElement;
-            for (let depth = 0; current && current !== document.body && depth < 3; depth += 1) {
-                if (!current.classList.contains(HOST_CLASS)) {
-                    current.classList.add(HOST_CLASS);
-                    hosts.push(current);
-                }
-                current = current.parentElement;
+            const mainHost = document.getElementById("graph-canvas");
+            if (mainHost) {
+                mainHost.classList.add(HOST_CLASS);
+                state.mainHosts.push(mainHost);
             }
-            state.mainHosts = hosts;
         }
 
         function resetMainHosts() {