@basic-genomics/hivtrace-viz 1.1.3 → 1.1.5

package/dist/embed/index.html

@@ -918,8 +918,9 @@
         var attributes = null;
 
         // 从传入的options中获取配置
-        var enableClusterTracking = options && options.enableClusterTracking === true;
         var expandClusters = options && Array.isArray(options.expand) ? options.expand : [];
+        // 获取自定义上下文菜单项配置
+        var customContextMenuItems = options && Array.isArray(options.customContextMenuItems) ? options.customContextMenuItems : [];
 
         // 获取基因距离阈值并动态更新簇标签
         var threshold = options && typeof options.threshold === 'number' ? options.threshold : 0.015;
@@ -953,6 +954,61 @@
           }
         );
 
+        // 将自定义上下文菜单项赋值给 user_graph 实例
+        if (customContextMenuItems.length > 0) {
+          user_graph.customContextMenuItems = customContextMenuItems;
+          // 设置点击回调，通过 postMessage 通知宿主应用
+          user_graph.onCustomMenuItemClick = function (itemId, clusterInfo) {
+            // 提取节点的关键信息（避免循环引用和不可序列化的对象）
+            var extractNodeInfo = function (node) {
+              if (!node) return null;
+              return {
+                id: node.id,
+                // 基本属性
+                degree: node.degree,
+                // 节点属性（如果存在）
+                attributes: node.patient_attributes || node.attributes || undefined,
+                // 位置信息
+                x: typeof node.x === 'number' ? node.x : undefined,
+                y: typeof node.y === 'number' ? node.y : undefined,
+                // 状态信息
+                is_hidden: Boolean(node.is_hidden),
+                match_filter: Boolean(node.match_filter),
+                // 子簇信息（如果存在）
+                subcluster_id: node.subcluster_id,
+                subcluster_label: node.subcluster_label,
+              };
+            };
+
+            var essentialClusterData = {
+              cluster_id: clusterInfo?.cluster_id ?? clusterInfo?.id ?? undefined,
+              node_count: clusterInfo?.node_count ?? (Array.isArray(clusterInfo?.nodes) ? clusterInfo.nodes.length : 0),
+              cluster_size: typeof clusterInfo?.cluster_size === 'number' ? clusterInfo.cluster_size : undefined,
+              position: {
+                x: typeof clusterInfo?.x === 'number' ? clusterInfo.x : undefined,
+                y: typeof clusterInfo?.y === 'number' ? clusterInfo.y : undefined
+              },
+              state: {
+                expanded: Boolean(clusterInfo?.expanded),
+                fixed: Boolean(clusterInfo?.fixed)
+              },
+              // 节点 ID 列表（向后兼容）
+              node_ids: Array.isArray(clusterInfo?.nodes)
+                ? clusterInfo.nodes.map(function (node) { return node?.id }).filter(Boolean)
+                : [],
+              // 完整节点信息（用于扩展）
+              nodes: Array.isArray(clusterInfo?.nodes)
+                ? clusterInfo.nodes.map(extractNodeInfo).filter(Boolean)
+                : []
+            };
+            window.parent.postMessage({
+              type: 'CUSTOM_MENU_CLICK',
+              itemId: itemId,
+              clusterInfo: essentialClusterData
+            }, '*');
+          };
+        }
+
         if (user_graph.is_empty()) {
           HandleAppError(
             "This network contains no clusters and cannot be displayed"
@@ -1125,69 +1181,6 @@
 
 
 
-          // 可配置的追踪簇回调功能 - 基于传入参数决定是否启用
-          if (enableClusterTracking) {
-            user_graph.onTrackCluster = function (clusterInfo, allData) {
-              try {
-                // 直接提取需要的数据，避免复杂的数据清理
-                const essentialClusterData = {
-                  cluster_id: clusterInfo?.cluster_id ?? clusterInfo?.id ?? undefined,
-                  node_count: clusterInfo?.node_count ?? (Array.isArray(clusterInfo?.nodes) ? clusterInfo.nodes.length : 0),
-                  cluster_size: typeof clusterInfo?.cluster_size === 'number' ? clusterInfo.cluster_size : undefined,
-                  position: {
-                    x: typeof clusterInfo?.x === 'number' ? clusterInfo.x : undefined,
-                    y: typeof clusterInfo?.y === 'number' ? clusterInfo.y : undefined
-                  },
-                  state: {
-                    expanded: Boolean(clusterInfo?.expanded),
-                    fixed: Boolean(clusterInfo?.fixed)
-                  },
-                  // 只提取节点ID列表，避免传递复杂对象
-                  node_ids: Array.isArray(clusterInfo?.nodes)
-                    ? clusterInfo.nodes.slice(0, 20).map(node => node?.id).filter(Boolean)
-                    : []
-                };
-
-                // 提取网络统计信息
-                const networkStats = {
-                  total_nodes: allData?.network_info?.node_count ?? user_graph?.nodes?.length ?? 0,
-                  total_edges: allData?.network_info?.edge_count ?? user_graph?.edges?.length ?? 0,
-                  total_clusters: allData?.network_info?.cluster_count ?? 0
-                };
-
-                // 发送轻量级、精确的数据
-                window.parent.postMessage({
-                  type: 'TRACK_CLUSTER',
-                  clusterInfo: essentialClusterData,
-                  allData: { network_info: networkStats }
-                }, '*');
-
-              } catch (error) {
-                console.warn('簇追踪回调处理失败:', error.message);
-
-                // 最小化降级数据
-                window.parent.postMessage({
-                  type: 'TRACK_CLUSTER',
-                  clusterInfo: {
-                    cluster_id: 'fallback',
-                    node_count: 0,
-                    cluster_size: undefined,
-                    position: { x: undefined, y: undefined },
-                    state: { expanded: false, fixed: false },
-                    node_ids: []
-                  },
-                  allData: {
-                    network_info: {
-                      total_nodes: 0,
-                      total_edges: 0,
-                      total_clusters: 0
-                    }
-                  }
-                }, '*');
-              }
-            };
-          }
-
           [
             "#main-tab",
             "#graph-tab",
@@ -1303,11 +1296,17 @@
       overlay.innerHTML = `
           <i class="fa fa-exclamation-triangle hivtrace-error-icon"></i>
           <div class="hivtrace-error-message">${message}</div>
-          <button class="hivtrace-retry-btn" onclick="location.reload()">重试</button>
+          <button class="hivtrace-retry-btn" onclick="requestRetry()">重试</button>
         `;
       window.parent.postMessage({ type: 'ERROR', message: message }, '*');
     }
 
+    // 请求宿主应用重试
+    function requestRetry() {
+      window.parent.postMessage({ type: 'HIVTRACE_RETRY' }, '*');
+    }
+
+
 
     function in_progress() {
       return $(".progress").length > 0;