|
@@ -1,1951 +0,0 @@
|
|
|
-#!/usr/bin/env node
|
|
|
|
|
-
|
|
|
|
|
-const fs = require('fs');
|
|
|
|
|
-const path = require('path');
|
|
|
|
|
-const { build } = require('esbuild');
|
|
|
|
|
-
|
|
|
|
|
-// 读取命令行参数
|
|
|
|
|
-const args = process.argv.slice(2);
|
|
|
|
|
-if (args.length === 0) {
|
|
|
|
|
- console.error('Usage: node visualize_v2.js <path-to-query_graph.json> [output.html]');
|
|
|
|
|
- process.exit(1);
|
|
|
|
|
-}
|
|
|
|
|
-
|
|
|
|
|
-const inputFile = args[0];
|
|
|
|
|
-const outputFile = args[1] || 'query_graph_output.html';
|
|
|
|
|
-
|
|
|
|
|
-// 读取输入数据
|
|
|
|
|
-const data = JSON.parse(fs.readFileSync(inputFile, 'utf-8'));
|
|
|
|
|
-
|
|
|
|
|
-// 创建临时 React 组件文件
|
|
|
|
|
-const reactComponentPath = path.join(__dirname, 'temp_flow_component_v2.jsx');
|
|
|
|
|
-const reactComponent = `
|
|
|
|
|
-import React, { useState, useCallback, useMemo, useEffect } from 'react';
|
|
|
|
|
-import { createRoot } from 'react-dom/client';
|
|
|
|
|
-import {
|
|
|
|
|
- ReactFlow,
|
|
|
|
|
- Controls,
|
|
|
|
|
- Background,
|
|
|
|
|
- useNodesState,
|
|
|
|
|
- useEdgesState,
|
|
|
|
|
- Handle,
|
|
|
|
|
- Position,
|
|
|
|
|
- useReactFlow,
|
|
|
|
|
- ReactFlowProvider,
|
|
|
|
|
-} from '@xyflow/react';
|
|
|
|
|
-import '@xyflow/react/dist/style.css';
|
|
|
|
|
-
|
|
|
|
|
-const data = ${JSON.stringify(data, null, 2)};
|
|
|
|
|
-
|
|
|
|
|
-// 查询节点组件 - 卡片样式
|
|
|
|
|
-function QueryNode({ id, data, sourcePosition, targetPosition }) {
|
|
|
|
|
- // 所有节点默认展开
|
|
|
|
|
- const expanded = true;
|
|
|
|
|
-
|
|
|
|
|
- return (
|
|
|
|
|
- <div>
|
|
|
|
|
- <Handle
|
|
|
|
|
- type="target"
|
|
|
|
|
- position={targetPosition || Position.Left}
|
|
|
|
|
- style={{ background: '#667eea', width: 8, height: 8 }}
|
|
|
|
|
- />
|
|
|
|
|
- <div
|
|
|
|
|
- style={{
|
|
|
|
|
- padding: '12px',
|
|
|
|
|
- borderRadius: '8px',
|
|
|
|
|
- border: data.isHighlighted ? '3px solid #667eea' :
|
|
|
|
|
- data.isCollapsed ? '2px solid #667eea' :
|
|
|
|
|
- data.isSelected === false ? '2px dashed #d1d5db' :
|
|
|
|
|
- data.level === 0 ? '2px solid #8b5cf6' : '1px solid #e5e7eb',
|
|
|
|
|
- background: data.isHighlighted ? '#eef2ff' :
|
|
|
|
|
- data.isSelected === false ? '#f9fafb' : 'white',
|
|
|
|
|
- minWidth: '200px',
|
|
|
|
|
- maxWidth: '280px',
|
|
|
|
|
- boxShadow: data.isHighlighted ? '0 0 0 4px rgba(102, 126, 234, 0.25), 0 4px 16px rgba(102, 126, 234, 0.4)' :
|
|
|
|
|
- data.isCollapsed ? '0 4px 12px rgba(102, 126, 234, 0.15)' :
|
|
|
|
|
- data.level === 0 ? '0 4px 12px rgba(139, 92, 246, 0.15)' : '0 2px 6px rgba(0, 0, 0, 0.06)',
|
|
|
|
|
- transition: 'all 0.3s ease',
|
|
|
|
|
- cursor: 'pointer',
|
|
|
|
|
- position: 'relative',
|
|
|
|
|
- opacity: data.isSelected === false ? 0.6 : 1,
|
|
|
|
|
- }}
|
|
|
|
|
- >
|
|
|
|
|
- {/* 折叠当前节点按钮 - 左边 */}
|
|
|
|
|
- <div
|
|
|
|
|
- style={{
|
|
|
|
|
- position: 'absolute',
|
|
|
|
|
- top: '6px',
|
|
|
|
|
- left: '6px',
|
|
|
|
|
- width: '20px',
|
|
|
|
|
- height: '20px',
|
|
|
|
|
- borderRadius: '50%',
|
|
|
|
|
- background: '#f59e0b',
|
|
|
|
|
- color: 'white',
|
|
|
|
|
- display: 'flex',
|
|
|
|
|
- alignItems: 'center',
|
|
|
|
|
- justifyContent: 'center',
|
|
|
|
|
- fontSize: '11px',
|
|
|
|
|
- fontWeight: 'bold',
|
|
|
|
|
- cursor: 'pointer',
|
|
|
|
|
- transition: 'all 0.2s ease',
|
|
|
|
|
- zIndex: 10,
|
|
|
|
|
- }}
|
|
|
|
|
- onClick={(e) => {
|
|
|
|
|
- e.stopPropagation();
|
|
|
|
|
- if (data.onHideSelf) {
|
|
|
|
|
- data.onHideSelf();
|
|
|
|
|
- }
|
|
|
|
|
- }}
|
|
|
|
|
- onMouseEnter={(e) => {
|
|
|
|
|
- e.currentTarget.style.background = '#d97706';
|
|
|
|
|
- }}
|
|
|
|
|
- onMouseLeave={(e) => {
|
|
|
|
|
- e.currentTarget.style.background = '#f59e0b';
|
|
|
|
|
- }}
|
|
|
|
|
- title="隐藏当前节点"
|
|
|
|
|
- >
|
|
|
|
|
- ×
|
|
|
|
|
- </div>
|
|
|
|
|
-
|
|
|
|
|
- {/* 聚焦按钮 - 右上角 */}
|
|
|
|
|
- <div
|
|
|
|
|
- style={{
|
|
|
|
|
- position: 'absolute',
|
|
|
|
|
- top: '6px',
|
|
|
|
|
- right: '6px',
|
|
|
|
|
- width: '20px',
|
|
|
|
|
- height: '20px',
|
|
|
|
|
- borderRadius: '50%',
|
|
|
|
|
- background: data.isFocused ? '#10b981' : '#e5e7eb',
|
|
|
|
|
- color: data.isFocused ? 'white' : '#6b7280',
|
|
|
|
|
- display: 'flex',
|
|
|
|
|
- alignItems: 'center',
|
|
|
|
|
- justifyContent: 'center',
|
|
|
|
|
- fontSize: '11px',
|
|
|
|
|
- fontWeight: 'bold',
|
|
|
|
|
- cursor: 'pointer',
|
|
|
|
|
- transition: 'all 0.2s ease',
|
|
|
|
|
- zIndex: 10,
|
|
|
|
|
- }}
|
|
|
|
|
- onClick={(e) => {
|
|
|
|
|
- e.stopPropagation();
|
|
|
|
|
- if (data.onFocus) {
|
|
|
|
|
- data.onFocus();
|
|
|
|
|
- }
|
|
|
|
|
- }}
|
|
|
|
|
- onMouseEnter={(e) => {
|
|
|
|
|
- if (!data.isFocused) {
|
|
|
|
|
- e.currentTarget.style.background = '#d1d5db';
|
|
|
|
|
- }
|
|
|
|
|
- }}
|
|
|
|
|
- onMouseLeave={(e) => {
|
|
|
|
|
- if (!data.isFocused) {
|
|
|
|
|
- e.currentTarget.style.background = '#e5e7eb';
|
|
|
|
|
- }
|
|
|
|
|
- }}
|
|
|
|
|
- title={data.isFocused ? '取消聚焦' : '聚焦到此节点'}
|
|
|
|
|
- >
|
|
|
|
|
- 🎯
|
|
|
|
|
- </div>
|
|
|
|
|
-
|
|
|
|
|
- {/* 折叠/展开子节点按钮 - 右边第二个位置 */}
|
|
|
|
|
- {data.hasChildren && (
|
|
|
|
|
- <div
|
|
|
|
|
- style={{
|
|
|
|
|
- position: 'absolute',
|
|
|
|
|
- top: '6px',
|
|
|
|
|
- right: '30px',
|
|
|
|
|
- width: '20px',
|
|
|
|
|
- height: '20px',
|
|
|
|
|
- borderRadius: '50%',
|
|
|
|
|
- background: data.isCollapsed ? '#667eea' : '#e5e7eb',
|
|
|
|
|
- color: data.isCollapsed ? 'white' : '#6b7280',
|
|
|
|
|
- display: 'flex',
|
|
|
|
|
- alignItems: 'center',
|
|
|
|
|
- justifyContent: 'center',
|
|
|
|
|
- fontSize: '11px',
|
|
|
|
|
- fontWeight: 'bold',
|
|
|
|
|
- cursor: 'pointer',
|
|
|
|
|
- transition: 'all 0.2s ease',
|
|
|
|
|
- zIndex: 10,
|
|
|
|
|
- }}
|
|
|
|
|
- onClick={(e) => {
|
|
|
|
|
- e.stopPropagation();
|
|
|
|
|
- data.onToggleCollapse();
|
|
|
|
|
- }}
|
|
|
|
|
- title={data.isCollapsed ? '展开子节点' : '折叠子节点'}
|
|
|
|
|
- >
|
|
|
|
|
- {data.isCollapsed ? '+' : '−'}
|
|
|
|
|
- </div>
|
|
|
|
|
- )}
|
|
|
|
|
-
|
|
|
|
|
- {/* 卡片内容 */}
|
|
|
|
|
- <div>
|
|
|
|
|
- {/* 标题行 */}
|
|
|
|
|
- <div style={{ display: 'flex', justifyContent: 'space-between', alignItems: 'flex-start', marginBottom: '8px', paddingLeft: '24px', paddingRight: data.hasChildren ? '54px' : '28px' }}>
|
|
|
|
|
- <div style={{ flex: 1 }}>
|
|
|
|
|
- <div style={{ display: 'flex', alignItems: 'center', gap: '4px', marginBottom: '3px' }}>
|
|
|
|
|
- <div style={{
|
|
|
|
|
- fontSize: '13px',
|
|
|
|
|
- fontWeight: data.level === 0 ? '700' : '600',
|
|
|
|
|
- color: data.level === 0 ? '#6b21a8' : '#1f2937',
|
|
|
|
|
- lineHeight: '1.3',
|
|
|
|
|
- flex: 1,
|
|
|
|
|
- }}>
|
|
|
|
|
- {data.title}
|
|
|
|
|
- </div>
|
|
|
|
|
- {data.isSelected === false && (
|
|
|
|
|
- <div style={{
|
|
|
|
|
- fontSize: '9px',
|
|
|
|
|
- padding: '1px 4px',
|
|
|
|
|
- borderRadius: '3px',
|
|
|
|
|
- background: '#fee2e2',
|
|
|
|
|
- color: '#991b1b',
|
|
|
|
|
- fontWeight: '500',
|
|
|
|
|
- flexShrink: 0,
|
|
|
|
|
- }}>
|
|
|
|
|
- 未选中
|
|
|
|
|
- </div>
|
|
|
|
|
- )}
|
|
|
|
|
- </div>
|
|
|
|
|
- </div>
|
|
|
|
|
- </div>
|
|
|
|
|
-
|
|
|
|
|
- {/* 展开的详细信息 - 始终显示 */}
|
|
|
|
|
- <div style={{ fontSize: '11px', lineHeight: 1.4 }}>
|
|
|
|
|
- <div style={{ display: 'flex', gap: '4px', marginBottom: '6px', flexWrap: 'wrap' }}>
|
|
|
|
|
- <span style={{
|
|
|
|
|
- display: 'inline-block',
|
|
|
|
|
- padding: '1px 6px',
|
|
|
|
|
- borderRadius: '10px',
|
|
|
|
|
- background: '#eff6ff',
|
|
|
|
|
- color: '#3b82f6',
|
|
|
|
|
- fontSize: '10px',
|
|
|
|
|
- fontWeight: '500',
|
|
|
|
|
- }}>
|
|
|
|
|
- Lv.{data.level}
|
|
|
|
|
- </span>
|
|
|
|
|
- <span style={{
|
|
|
|
|
- display: 'inline-block',
|
|
|
|
|
- padding: '1px 6px',
|
|
|
|
|
- borderRadius: '10px',
|
|
|
|
|
- background: '#f0fdf4',
|
|
|
|
|
- color: '#16a34a',
|
|
|
|
|
- fontSize: '10px',
|
|
|
|
|
- fontWeight: '500',
|
|
|
|
|
- }}>
|
|
|
|
|
- {data.score}
|
|
|
|
|
- </span>
|
|
|
|
|
- {data.strategy && data.strategy !== 'root' && (
|
|
|
|
|
- <span style={{
|
|
|
|
|
- display: 'inline-block',
|
|
|
|
|
- padding: '1px 6px',
|
|
|
|
|
- borderRadius: '10px',
|
|
|
|
|
- background: '#fef3c7',
|
|
|
|
|
- color: '#92400e',
|
|
|
|
|
- fontSize: '10px',
|
|
|
|
|
- fontWeight: '500',
|
|
|
|
|
- }}>
|
|
|
|
|
- {data.strategy}
|
|
|
|
|
- </span>
|
|
|
|
|
- )}
|
|
|
|
|
- </div>
|
|
|
|
|
-
|
|
|
|
|
- {data.parent && (
|
|
|
|
|
- <div style={{ color: '#6b7280', fontSize: '10px', marginTop: '4px', paddingTop: '4px', borderTop: '1px solid #f3f4f6' }}>
|
|
|
|
|
- <strong>Parent:</strong> {data.parent}
|
|
|
|
|
- </div>
|
|
|
|
|
- )}
|
|
|
|
|
- {data.evaluationReason && (
|
|
|
|
|
- <div style={{
|
|
|
|
|
- marginTop: '6px',
|
|
|
|
|
- paddingTop: '6px',
|
|
|
|
|
- borderTop: '1px solid #f3f4f6',
|
|
|
|
|
- fontSize: '10px',
|
|
|
|
|
- color: '#6b7280',
|
|
|
|
|
- lineHeight: '1.5',
|
|
|
|
|
- }}>
|
|
|
|
|
- <strong style={{ color: '#4b5563' }}>评估:</strong>
|
|
|
|
|
- <div style={{ marginTop: '2px' }}>{data.evaluationReason}</div>
|
|
|
|
|
- </div>
|
|
|
|
|
- )}
|
|
|
|
|
- </div>
|
|
|
|
|
- </div>
|
|
|
|
|
- </div>
|
|
|
|
|
- <Handle
|
|
|
|
|
- type="source"
|
|
|
|
|
- position={sourcePosition || Position.Right}
|
|
|
|
|
- style={{ background: '#667eea', width: 8, height: 8 }}
|
|
|
|
|
- />
|
|
|
|
|
- </div>
|
|
|
|
|
- );
|
|
|
|
|
-}
|
|
|
|
|
-
|
|
|
|
|
-// 笔记节点组件 - 卡片样式,带轮播图
|
|
|
|
|
-function NoteNode({ id, data, sourcePosition, targetPosition }) {
|
|
|
|
|
- const [currentImageIndex, setCurrentImageIndex] = useState(0);
|
|
|
|
|
- const expanded = true;
|
|
|
|
|
- const hasImages = data.imageList && data.imageList.length > 0;
|
|
|
|
|
-
|
|
|
|
|
- const nextImage = (e) => {
|
|
|
|
|
- e.stopPropagation();
|
|
|
|
|
- if (hasImages) {
|
|
|
|
|
- setCurrentImageIndex((prev) => (prev + 1) % data.imageList.length);
|
|
|
|
|
- }
|
|
|
|
|
- };
|
|
|
|
|
-
|
|
|
|
|
- const prevImage = (e) => {
|
|
|
|
|
- e.stopPropagation();
|
|
|
|
|
- if (hasImages) {
|
|
|
|
|
- setCurrentImageIndex((prev) => (prev - 1 + data.imageList.length) % data.imageList.length);
|
|
|
|
|
- }
|
|
|
|
|
- };
|
|
|
|
|
-
|
|
|
|
|
- return (
|
|
|
|
|
- <div>
|
|
|
|
|
- <Handle
|
|
|
|
|
- type="target"
|
|
|
|
|
- position={targetPosition || Position.Left}
|
|
|
|
|
- style={{ background: '#ec4899', width: 8, height: 8 }}
|
|
|
|
|
- />
|
|
|
|
|
- <div
|
|
|
|
|
- style={{
|
|
|
|
|
- padding: '14px',
|
|
|
|
|
- borderRadius: '20px',
|
|
|
|
|
- border: data.isHighlighted ? '3px solid #ec4899' : '2px solid #fce7f3',
|
|
|
|
|
- background: data.isHighlighted ? '#eef2ff' : 'white',
|
|
|
|
|
- minWidth: '220px',
|
|
|
|
|
- maxWidth: '300px',
|
|
|
|
|
- boxShadow: data.isHighlighted ? '0 0 0 4px rgba(236, 72, 153, 0.25), 0 4px 16px rgba(236, 72, 153, 0.4)' : '0 4px 12px rgba(236, 72, 153, 0.15)',
|
|
|
|
|
- transition: 'all 0.3s ease',
|
|
|
|
|
- cursor: 'pointer',
|
|
|
|
|
- }}
|
|
|
|
|
- >
|
|
|
|
|
- {/* 笔记图标和标题 */}
|
|
|
|
|
- <div style={{ display: 'flex', alignItems: 'flex-start', marginBottom: '8px' }}>
|
|
|
|
|
- <span style={{ fontSize: '16px', marginRight: '8px' }}>📝</span>
|
|
|
|
|
- <div style={{ flex: 1 }}>
|
|
|
|
|
- <div style={{
|
|
|
|
|
- fontSize: '13px',
|
|
|
|
|
- fontWeight: '600',
|
|
|
|
|
- color: '#831843',
|
|
|
|
|
- lineHeight: '1.4',
|
|
|
|
|
- marginBottom: '4px',
|
|
|
|
|
- }}>
|
|
|
|
|
- {data.title}
|
|
|
|
|
- </div>
|
|
|
|
|
- </div>
|
|
|
|
|
- </div>
|
|
|
|
|
-
|
|
|
|
|
- {/* 轮播图 */}
|
|
|
|
|
- {hasImages && (
|
|
|
|
|
- <div style={{
|
|
|
|
|
- position: 'relative',
|
|
|
|
|
- marginBottom: '8px',
|
|
|
|
|
- borderRadius: '12px',
|
|
|
|
|
- overflow: 'hidden',
|
|
|
|
|
- }}>
|
|
|
|
|
- <img
|
|
|
|
|
- src={data.imageList[currentImageIndex].image_url}
|
|
|
|
|
- alt={\`Image \${currentImageIndex + 1}\`}
|
|
|
|
|
- style={{
|
|
|
|
|
- width: '100%',
|
|
|
|
|
- height: '160px',
|
|
|
|
|
- objectFit: 'cover',
|
|
|
|
|
- display: 'block',
|
|
|
|
|
- }}
|
|
|
|
|
- onError={(e) => {
|
|
|
|
|
- e.target.style.display = 'none';
|
|
|
|
|
- }}
|
|
|
|
|
- />
|
|
|
|
|
- {data.imageList.length > 1 && (
|
|
|
|
|
- <>
|
|
|
|
|
- {/* 左右切换按钮 */}
|
|
|
|
|
- <button
|
|
|
|
|
- onClick={prevImage}
|
|
|
|
|
- style={{
|
|
|
|
|
- position: 'absolute',
|
|
|
|
|
- left: '4px',
|
|
|
|
|
- top: '50%',
|
|
|
|
|
- transform: 'translateY(-50%)',
|
|
|
|
|
- background: 'rgba(0, 0, 0, 0.5)',
|
|
|
|
|
- color: 'white',
|
|
|
|
|
- border: 'none',
|
|
|
|
|
- borderRadius: '50%',
|
|
|
|
|
- width: '24px',
|
|
|
|
|
- height: '24px',
|
|
|
|
|
- cursor: 'pointer',
|
|
|
|
|
- display: 'flex',
|
|
|
|
|
- alignItems: 'center',
|
|
|
|
|
- justifyContent: 'center',
|
|
|
|
|
- fontSize: '14px',
|
|
|
|
|
- }}
|
|
|
|
|
- >
|
|
|
|
|
- ‹
|
|
|
|
|
- </button>
|
|
|
|
|
- <button
|
|
|
|
|
- onClick={nextImage}
|
|
|
|
|
- style={{
|
|
|
|
|
- position: 'absolute',
|
|
|
|
|
- right: '4px',
|
|
|
|
|
- top: '50%',
|
|
|
|
|
- transform: 'translateY(-50%)',
|
|
|
|
|
- background: 'rgba(0, 0, 0, 0.5)',
|
|
|
|
|
- color: 'white',
|
|
|
|
|
- border: 'none',
|
|
|
|
|
- borderRadius: '50%',
|
|
|
|
|
- width: '24px',
|
|
|
|
|
- height: '24px',
|
|
|
|
|
- cursor: 'pointer',
|
|
|
|
|
- display: 'flex',
|
|
|
|
|
- alignItems: 'center',
|
|
|
|
|
- justifyContent: 'center',
|
|
|
|
|
- fontSize: '14px',
|
|
|
|
|
- }}
|
|
|
|
|
- >
|
|
|
|
|
- ›
|
|
|
|
|
- </button>
|
|
|
|
|
- {/* 图片计数 */}
|
|
|
|
|
- <div style={{
|
|
|
|
|
- position: 'absolute',
|
|
|
|
|
- bottom: '4px',
|
|
|
|
|
- right: '4px',
|
|
|
|
|
- background: 'rgba(0, 0, 0, 0.6)',
|
|
|
|
|
- color: 'white',
|
|
|
|
|
- padding: '2px 6px',
|
|
|
|
|
- borderRadius: '10px',
|
|
|
|
|
- fontSize: '10px',
|
|
|
|
|
- }}>
|
|
|
|
|
- {currentImageIndex + 1}/{data.imageList.length}
|
|
|
|
|
- </div>
|
|
|
|
|
- </>
|
|
|
|
|
- )}
|
|
|
|
|
- </div>
|
|
|
|
|
- )}
|
|
|
|
|
-
|
|
|
|
|
- {/* 标签 */}
|
|
|
|
|
- <div style={{ display: 'flex', gap: '6px', marginBottom: '8px', flexWrap: 'wrap' }}>
|
|
|
|
|
- <span style={{
|
|
|
|
|
- display: 'inline-block',
|
|
|
|
|
- padding: '2px 8px',
|
|
|
|
|
- borderRadius: '12px',
|
|
|
|
|
- background: '#fff1f2',
|
|
|
|
|
- color: '#be123c',
|
|
|
|
|
- fontSize: '10px',
|
|
|
|
|
- fontWeight: '500',
|
|
|
|
|
- }}>
|
|
|
|
|
- {data.matchLevel}
|
|
|
|
|
- </span>
|
|
|
|
|
- <span style={{
|
|
|
|
|
- display: 'inline-block',
|
|
|
|
|
- padding: '2px 8px',
|
|
|
|
|
- borderRadius: '12px',
|
|
|
|
|
- background: '#fff7ed',
|
|
|
|
|
- color: '#c2410c',
|
|
|
|
|
- fontSize: '10px',
|
|
|
|
|
- fontWeight: '500',
|
|
|
|
|
- }}>
|
|
|
|
|
- Score: {data.score}
|
|
|
|
|
- </span>
|
|
|
|
|
- </div>
|
|
|
|
|
-
|
|
|
|
|
- {/* 描述 */}
|
|
|
|
|
- {expanded && data.description && (
|
|
|
|
|
- <div style={{
|
|
|
|
|
- fontSize: '11px',
|
|
|
|
|
- color: '#9f1239',
|
|
|
|
|
- lineHeight: '1.5',
|
|
|
|
|
- paddingTop: '8px',
|
|
|
|
|
- borderTop: '1px solid #fbcfe8',
|
|
|
|
|
- }}>
|
|
|
|
|
- {data.description}
|
|
|
|
|
- </div>
|
|
|
|
|
- )}
|
|
|
|
|
-
|
|
|
|
|
- {/* 评估理由 */}
|
|
|
|
|
- {expanded && data.evaluationReason && (
|
|
|
|
|
- <div style={{
|
|
|
|
|
- fontSize: '10px',
|
|
|
|
|
- color: '#831843',
|
|
|
|
|
- lineHeight: '1.5',
|
|
|
|
|
- paddingTop: '8px',
|
|
|
|
|
- marginTop: '8px',
|
|
|
|
|
- borderTop: '1px solid #fbcfe8',
|
|
|
|
|
- }}>
|
|
|
|
|
- <strong style={{ color: '#9f1239' }}>评估:</strong>
|
|
|
|
|
- <div style={{ marginTop: '2px' }}>{data.evaluationReason}</div>
|
|
|
|
|
- </div>
|
|
|
|
|
- )}
|
|
|
|
|
- </div>
|
|
|
|
|
- <Handle
|
|
|
|
|
- type="source"
|
|
|
|
|
- position={sourcePosition || Position.Right}
|
|
|
|
|
- style={{ background: '#ec4899', width: 8, height: 8 }}
|
|
|
|
|
- />
|
|
|
|
|
- </div>
|
|
|
|
|
- );
|
|
|
|
|
-}
|
|
|
|
|
-
|
|
|
|
|
-const nodeTypes = {
|
|
|
|
|
- query: QueryNode,
|
|
|
|
|
- note: NoteNode,
|
|
|
|
|
-};
|
|
|
|
|
-
|
|
|
|
|
-// 根据 score 获取颜色
|
|
|
|
|
-function getScoreColor(score) {
|
|
|
|
|
- if (score >= 0.7) return '#10b981'; // 绿色 - 高分
|
|
|
|
|
- if (score >= 0.4) return '#f59e0b'; // 橙色 - 中分
|
|
|
|
|
- return '#ef4444'; // 红色 - 低分
|
|
|
|
|
-}
|
|
|
|
|
-
|
|
|
|
|
-// 截断文本,保留头尾,中间显示省略号
|
|
|
|
|
-function truncateMiddle(text, maxLength = 20) {
|
|
|
|
|
- if (!text || text.length <= maxLength) return text;
|
|
|
|
|
- const headLength = Math.ceil(maxLength * 0.4);
|
|
|
|
|
- const tailLength = Math.floor(maxLength * 0.4);
|
|
|
|
|
- const head = text.substring(0, headLength);
|
|
|
|
|
- const tail = text.substring(text.length - tailLength);
|
|
|
|
|
- return \`\${head}...\${tail}\`;
|
|
|
|
|
-}
|
|
|
|
|
-
|
|
|
|
|
-// 根据策略获取颜色
|
|
|
|
|
-function getStrategyColor(strategy) {
|
|
|
|
|
- const strategyColors = {
|
|
|
|
|
- '初始分词': '#10b981',
|
|
|
|
|
- '调用sug': '#06b6d4',
|
|
|
|
|
- '同义改写': '#f59e0b',
|
|
|
|
|
- '加词': '#3b82f6',
|
|
|
|
|
- '抽象改写': '#8b5cf6',
|
|
|
|
|
- '基于部分匹配改进': '#ec4899',
|
|
|
|
|
- '结果分支-抽象改写': '#a855f7',
|
|
|
|
|
- '结果分支-同义改写': '#fb923c',
|
|
|
|
|
- };
|
|
|
|
|
- return strategyColors[strategy] || '#9ca3af';
|
|
|
|
|
-}
|
|
|
|
|
-
|
|
|
|
|
-// 树节点组件
|
|
|
|
|
-function TreeNode({ node, level, children, isCollapsed, onToggle, isSelected, onSelect }) {
|
|
|
|
|
- const hasChildren = children && children.length > 0;
|
|
|
|
|
- const score = node.data.score ? parseFloat(node.data.score) : 0;
|
|
|
|
|
- const strategy = node.data.strategy || '';
|
|
|
|
|
- const strategyColor = getStrategyColor(strategy);
|
|
|
|
|
-
|
|
|
|
|
- return (
|
|
|
|
|
- <div style={{ marginLeft: level * 12 + 'px' }}>
|
|
|
|
|
- <div
|
|
|
|
|
- style={{
|
|
|
|
|
- padding: '6px 8px',
|
|
|
|
|
- borderRadius: '4px',
|
|
|
|
|
- cursor: 'pointer',
|
|
|
|
|
- background: 'transparent',
|
|
|
|
|
- border: isSelected ? '1px solid #3b82f6' : '1px solid transparent',
|
|
|
|
|
- display: 'flex',
|
|
|
|
|
- alignItems: 'center',
|
|
|
|
|
- gap: '6px',
|
|
|
|
|
- transition: 'all 0.2s ease',
|
|
|
|
|
- position: 'relative',
|
|
|
|
|
- overflow: 'visible',
|
|
|
|
|
- }}
|
|
|
|
|
- onMouseEnter={(e) => {
|
|
|
|
|
- if (!isSelected) e.currentTarget.style.background = '#f9fafb';
|
|
|
|
|
- }}
|
|
|
|
|
- onMouseLeave={(e) => {
|
|
|
|
|
- if (!isSelected) e.currentTarget.style.background = 'transparent';
|
|
|
|
|
- }}
|
|
|
|
|
- >
|
|
|
|
|
- {/* 策略类型竖线 */}
|
|
|
|
|
- <div style={{
|
|
|
|
|
- width: '3px',
|
|
|
|
|
- height: '20px',
|
|
|
|
|
- background: strategyColor,
|
|
|
|
|
- borderRadius: '2px',
|
|
|
|
|
- flexShrink: 0,
|
|
|
|
|
- position: 'relative',
|
|
|
|
|
- zIndex: 1,
|
|
|
|
|
- }} />
|
|
|
|
|
-
|
|
|
|
|
- {hasChildren && (
|
|
|
|
|
- <span
|
|
|
|
|
- style={{
|
|
|
|
|
- fontSize: '10px',
|
|
|
|
|
- color: '#6b7280',
|
|
|
|
|
- cursor: 'pointer',
|
|
|
|
|
- width: '16px',
|
|
|
|
|
- textAlign: 'center',
|
|
|
|
|
- position: 'relative',
|
|
|
|
|
- zIndex: 1,
|
|
|
|
|
- }}
|
|
|
|
|
- onClick={(e) => {
|
|
|
|
|
- e.stopPropagation();
|
|
|
|
|
- onToggle();
|
|
|
|
|
- }}
|
|
|
|
|
- >
|
|
|
|
|
- {isCollapsed ? '▶' : '▼'}
|
|
|
|
|
- </span>
|
|
|
|
|
- )}
|
|
|
|
|
- {!hasChildren && <span style={{ width: '16px', position: 'relative', zIndex: 1 }}></span>}
|
|
|
|
|
-
|
|
|
|
|
- <div
|
|
|
|
|
- style={{
|
|
|
|
|
- flex: 1,
|
|
|
|
|
- fontSize: '12px',
|
|
|
|
|
- color: '#374151',
|
|
|
|
|
- position: 'relative',
|
|
|
|
|
- zIndex: 1,
|
|
|
|
|
- minWidth: 0,
|
|
|
|
|
- display: 'flex',
|
|
|
|
|
- flexDirection: 'column',
|
|
|
|
|
- gap: '4px',
|
|
|
|
|
- }}
|
|
|
|
|
- onClick={onSelect}
|
|
|
|
|
- >
|
|
|
|
|
- <div style={{
|
|
|
|
|
- display: 'flex',
|
|
|
|
|
- alignItems: 'center',
|
|
|
|
|
- gap: '8px',
|
|
|
|
|
- }}>
|
|
|
|
|
- {/* 节点类型图标 */}
|
|
|
|
|
- <span style={{
|
|
|
|
|
- fontSize: '12px',
|
|
|
|
|
- flexShrink: 0,
|
|
|
|
|
- }}>
|
|
|
|
|
- {node.type === 'note' ? '📝' : '🔍'}
|
|
|
|
|
- </span>
|
|
|
|
|
-
|
|
|
|
|
- <div style={{
|
|
|
|
|
- fontWeight: level === 0 ? '600' : '400',
|
|
|
|
|
- maxWidth: '180px',
|
|
|
|
|
- flex: 1,
|
|
|
|
|
- minWidth: 0,
|
|
|
|
|
- color: (node.type === 'note' ? node.data.matchLevel === 'unsatisfied' : node.data.isSelected === false) ? '#ef4444' : '#374151',
|
|
|
|
|
- }}
|
|
|
|
|
- title={node.data.title || node.id}
|
|
|
|
|
- >
|
|
|
|
|
- {truncateMiddle(node.data.title || node.id, 18)}
|
|
|
|
|
- </div>
|
|
|
|
|
-
|
|
|
|
|
- {/* 分数显示 */}
|
|
|
|
|
- <span style={{
|
|
|
|
|
- fontSize: '11px',
|
|
|
|
|
- color: '#6b7280',
|
|
|
|
|
- fontWeight: '500',
|
|
|
|
|
- flexShrink: 0,
|
|
|
|
|
- }}>
|
|
|
|
|
- {score.toFixed(2)}
|
|
|
|
|
- </span>
|
|
|
|
|
- </div>
|
|
|
|
|
-
|
|
|
|
|
- {/* 分数下划线 */}
|
|
|
|
|
- <div style={{
|
|
|
|
|
- width: (score * 100) + '%',
|
|
|
|
|
- height: '2px',
|
|
|
|
|
- background: getScoreColor(score),
|
|
|
|
|
- borderRadius: '1px',
|
|
|
|
|
- }} />
|
|
|
|
|
- </div>
|
|
|
|
|
- </div>
|
|
|
|
|
-
|
|
|
|
|
- {hasChildren && !isCollapsed && (
|
|
|
|
|
- <div>
|
|
|
|
|
- {children}
|
|
|
|
|
- </div>
|
|
|
|
|
- )}
|
|
|
|
|
- </div>
|
|
|
|
|
- );
|
|
|
|
|
-}
|
|
|
|
|
-
|
|
|
|
|
-// 使用 dagre 自动布局
|
|
|
|
|
-function getLayoutedElements(nodes, edges, direction = 'LR') {
|
|
|
|
|
- console.log('🎯 Starting layout with dagre...');
|
|
|
|
|
- console.log('Input:', nodes.length, 'nodes,', edges.length, 'edges');
|
|
|
|
|
-
|
|
|
|
|
- // 检查 dagre 是否加载
|
|
|
|
|
- if (typeof window === 'undefined' || typeof window.dagre === 'undefined') {
|
|
|
|
|
- console.warn('⚠️ Dagre not loaded, using fallback layout');
|
|
|
|
|
- // 降级到简单布局
|
|
|
|
|
- const levelGroups = {};
|
|
|
|
|
- nodes.forEach(node => {
|
|
|
|
|
- const level = node.data.level || 0;
|
|
|
|
|
- if (!levelGroups[level]) levelGroups[level] = [];
|
|
|
|
|
- levelGroups[level].push(node);
|
|
|
|
|
- });
|
|
|
|
|
-
|
|
|
|
|
- Object.entries(levelGroups).forEach(([level, nodeList]) => {
|
|
|
|
|
- const x = parseInt(level) * 350;
|
|
|
|
|
- nodeList.forEach((node, index) => {
|
|
|
|
|
- node.position = { x, y: index * 150 };
|
|
|
|
|
- node.targetPosition = 'left';
|
|
|
|
|
- node.sourcePosition = 'right';
|
|
|
|
|
- });
|
|
|
|
|
- });
|
|
|
|
|
-
|
|
|
|
|
- return { nodes, edges };
|
|
|
|
|
- }
|
|
|
|
|
-
|
|
|
|
|
- try {
|
|
|
|
|
- const dagreGraph = new window.dagre.graphlib.Graph();
|
|
|
|
|
- dagreGraph.setDefaultEdgeLabel(() => ({}));
|
|
|
|
|
-
|
|
|
|
|
- const isHorizontal = direction === 'LR';
|
|
|
|
|
- dagreGraph.setGraph({
|
|
|
|
|
- rankdir: direction,
|
|
|
|
|
- nodesep: 120, // 垂直间距 - 增加以避免节点重叠
|
|
|
|
|
- ranksep: 280, // 水平间距 - 增加以容纳更宽的节点
|
|
|
|
|
- });
|
|
|
|
|
-
|
|
|
|
|
- // 添加节点 - 根据节点类型设置不同的尺寸
|
|
|
|
|
- nodes.forEach((node) => {
|
|
|
|
|
- let nodeWidth = 280;
|
|
|
|
|
- let nodeHeight = 180;
|
|
|
|
|
-
|
|
|
|
|
- // note 节点有轮播图,需要更大的空间
|
|
|
|
|
- if (node.type === 'note') {
|
|
|
|
|
- nodeWidth = 320;
|
|
|
|
|
- nodeHeight = 350; // 增加高度以容纳轮播图
|
|
|
|
|
- }
|
|
|
|
|
-
|
|
|
|
|
- dagreGraph.setNode(node.id, { width: nodeWidth, height: nodeHeight });
|
|
|
|
|
- });
|
|
|
|
|
-
|
|
|
|
|
- // 添加边
|
|
|
|
|
- edges.forEach((edge) => {
|
|
|
|
|
- dagreGraph.setEdge(edge.source, edge.target);
|
|
|
|
|
- });
|
|
|
|
|
-
|
|
|
|
|
- // 计算布局
|
|
|
|
|
- window.dagre.layout(dagreGraph);
|
|
|
|
|
- console.log('✅ Dagre layout completed');
|
|
|
|
|
-
|
|
|
|
|
- // 更新节点位置和 handle 位置
|
|
|
|
|
- nodes.forEach((node) => {
|
|
|
|
|
- const nodeWithPosition = dagreGraph.node(node.id);
|
|
|
|
|
-
|
|
|
|
|
- if (!nodeWithPosition) {
|
|
|
|
|
- console.warn('Node position not found for:', node.id);
|
|
|
|
|
- return;
|
|
|
|
|
- }
|
|
|
|
|
-
|
|
|
|
|
- node.targetPosition = isHorizontal ? 'left' : 'top';
|
|
|
|
|
- node.sourcePosition = isHorizontal ? 'right' : 'bottom';
|
|
|
|
|
-
|
|
|
|
|
- // 将 dagre 的中心点位置转换为 React Flow 的左上角位置
|
|
|
|
|
- node.position = {
|
|
|
|
|
- x: nodeWithPosition.x - nodeWidth / 2,
|
|
|
|
|
- y: nodeWithPosition.y - nodeHeight / 2,
|
|
|
|
|
- };
|
|
|
|
|
- });
|
|
|
|
|
-
|
|
|
|
|
- console.log('✅ Layout completed, sample node:', nodes[0]);
|
|
|
|
|
- return { nodes, edges };
|
|
|
|
|
- } catch (error) {
|
|
|
|
|
- console.error('❌ Error in dagre layout:', error);
|
|
|
|
|
- console.error('Error details:', error.message, error.stack);
|
|
|
|
|
-
|
|
|
|
|
- // 降级处理
|
|
|
|
|
- console.log('Using fallback layout...');
|
|
|
|
|
- const levelGroups = {};
|
|
|
|
|
- nodes.forEach(node => {
|
|
|
|
|
- const level = node.data.level || 0;
|
|
|
|
|
- if (!levelGroups[level]) levelGroups[level] = [];
|
|
|
|
|
- levelGroups[level].push(node);
|
|
|
|
|
- });
|
|
|
|
|
-
|
|
|
|
|
- Object.entries(levelGroups).forEach(([level, nodeList]) => {
|
|
|
|
|
- const x = parseInt(level) * 350;
|
|
|
|
|
- nodeList.forEach((node, index) => {
|
|
|
|
|
- node.position = { x, y: index * 150 };
|
|
|
|
|
- node.targetPosition = 'left';
|
|
|
|
|
- node.sourcePosition = 'right';
|
|
|
|
|
- });
|
|
|
|
|
- });
|
|
|
|
|
-
|
|
|
|
|
- return { nodes, edges };
|
|
|
|
|
- }
|
|
|
|
|
-}
|
|
|
|
|
-
|
|
|
|
|
-function transformData(data) {
|
|
|
|
|
- const nodes = [];
|
|
|
|
|
- const edges = [];
|
|
|
|
|
-
|
|
|
|
|
- const originalIdToCanvasId = {}; // 原始ID -> 画布ID的映射
|
|
|
|
|
- const canvasIdToNodeData = {}; // 避免重复创建相同的节点
|
|
|
|
|
-
|
|
|
|
|
- // 创建节点
|
|
|
|
|
- Object.entries(data.nodes).forEach(([originalId, node]) => {
|
|
|
|
|
- if (node.type === 'query') {
|
|
|
|
|
- // 使用 query_level 作为唯一ID
|
|
|
|
|
- const canvasId = node.query + '_' + node.level;
|
|
|
|
|
- originalIdToCanvasId[originalId] = canvasId;
|
|
|
|
|
-
|
|
|
|
|
- // 如果这个 canvasId 还没有创建过节点,则创建
|
|
|
|
|
- if (!canvasIdToNodeData[canvasId]) {
|
|
|
|
|
- canvasIdToNodeData[canvasId] = true;
|
|
|
|
|
- nodes.push({
|
|
|
|
|
- id: canvasId, // 使用 query_level 格式
|
|
|
|
|
- originalId: originalId, // 保留原始ID用于调试
|
|
|
|
|
- type: 'query',
|
|
|
|
|
- data: {
|
|
|
|
|
- title: node.query,
|
|
|
|
|
- level: node.level,
|
|
|
|
|
- score: node.relevance_score.toFixed(2),
|
|
|
|
|
- strategy: node.strategy,
|
|
|
|
|
- parent: node.parent_query,
|
|
|
|
|
- isSelected: node.is_selected,
|
|
|
|
|
- evaluationReason: node.evaluation_reason || '',
|
|
|
|
|
- },
|
|
|
|
|
- position: { x: 0, y: 0 }, // 初始位置,会被 dagre 覆盖
|
|
|
|
|
- });
|
|
|
|
|
- }
|
|
|
|
|
- } else if (node.type === 'note') {
|
|
|
|
|
- // note节点直接使用原始ID
|
|
|
|
|
- originalIdToCanvasId[originalId] = originalId;
|
|
|
|
|
-
|
|
|
|
|
- if (!canvasIdToNodeData[originalId]) {
|
|
|
|
|
- canvasIdToNodeData[originalId] = true;
|
|
|
|
|
- nodes.push({
|
|
|
|
|
- id: originalId,
|
|
|
|
|
- originalId: originalId,
|
|
|
|
|
- type: 'note',
|
|
|
|
|
- data: {
|
|
|
|
|
- title: node.title,
|
|
|
|
|
- matchLevel: node.match_level,
|
|
|
|
|
- score: node.relevance_score.toFixed(2),
|
|
|
|
|
- description: node.desc,
|
|
|
|
|
- isSelected: node.is_selected !== undefined ? node.is_selected : true,
|
|
|
|
|
- imageList: node.image_list || [], // 添加图片列表
|
|
|
|
|
- noteUrl: node.note_url || '', // 添加帖子链接
|
|
|
|
|
- evaluationReason: node.evaluation_reason || '', // 添加评估理由
|
|
|
|
|
- },
|
|
|
|
|
- position: { x: 0, y: 0 },
|
|
|
|
|
- });
|
|
|
|
|
- }
|
|
|
|
|
- }
|
|
|
|
|
- });
|
|
|
|
|
-
|
|
|
|
|
- // 创建边 - 使用虚线样式,映射到画布ID
|
|
|
|
|
- data.edges.forEach((edge, index) => {
|
|
|
|
|
- const edgeColors = {
|
|
|
|
|
- '初始分词': '#10b981',
|
|
|
|
|
- '调用sug': '#06b6d4',
|
|
|
|
|
- '同义改写': '#f59e0b',
|
|
|
|
|
- '加词': '#3b82f6',
|
|
|
|
|
- '抽象改写': '#8b5cf6',
|
|
|
|
|
- '基于部分匹配改进': '#ec4899',
|
|
|
|
|
- '结果分支-抽象改写': '#a855f7',
|
|
|
|
|
- '结果分支-同义改写': '#fb923c',
|
|
|
|
|
- 'query_to_note': '#ec4899',
|
|
|
|
|
- };
|
|
|
|
|
-
|
|
|
|
|
- const color = edgeColors[edge.strategy] || edgeColors[edge.edge_type] || '#d1d5db';
|
|
|
|
|
- const isNoteEdge = edge.edge_type === 'query_to_note';
|
|
|
|
|
-
|
|
|
|
|
- edges.push({
|
|
|
|
|
- id: \`edge-\${index}\`,
|
|
|
|
|
- source: originalIdToCanvasId[edge.from], // 使用画布ID
|
|
|
|
|
- target: originalIdToCanvasId[edge.to], // 使用画布ID
|
|
|
|
|
- type: 'simplebezier', // 使用简单贝塞尔曲线
|
|
|
|
|
- animated: isNoteEdge,
|
|
|
|
|
- style: {
|
|
|
|
|
- stroke: color,
|
|
|
|
|
- strokeWidth: isNoteEdge ? 2.5 : 2,
|
|
|
|
|
- strokeDasharray: isNoteEdge ? '5,5' : '8,4',
|
|
|
|
|
- },
|
|
|
|
|
- markerEnd: {
|
|
|
|
|
- type: 'arrowclosed',
|
|
|
|
|
- color: color,
|
|
|
|
|
- width: 20,
|
|
|
|
|
- height: 20,
|
|
|
|
|
- },
|
|
|
|
|
- });
|
|
|
|
|
- });
|
|
|
|
|
-
|
|
|
|
|
- // 使用 dagre 自动计算布局 - 从左到右
|
|
|
|
|
- return getLayoutedElements(nodes, edges, 'LR');
|
|
|
|
|
-}
|
|
|
|
|
-
|
|
|
|
|
-function FlowContent() {
|
|
|
|
|
- const { nodes: initialNodes, edges: initialEdges } = useMemo(() => {
|
|
|
|
|
- console.log('🔍 Transforming data...');
|
|
|
|
|
- const result = transformData(data);
|
|
|
|
|
- console.log('✅ Transformed:', result.nodes.length, 'nodes,', result.edges.length, 'edges');
|
|
|
|
|
- return result;
|
|
|
|
|
- }, []);
|
|
|
|
|
-
|
|
|
|
|
- // 初始化:找出所有有子节点的节点,默认折叠(画布节点)
|
|
|
|
|
- const initialCollapsedNodes = useMemo(() => {
|
|
|
|
|
- const nodesWithChildren = new Set();
|
|
|
|
|
- initialEdges.forEach(edge => {
|
|
|
|
|
- nodesWithChildren.add(edge.source);
|
|
|
|
|
- });
|
|
|
|
|
- // 排除根节点(level 0),让根节点默认展开
|
|
|
|
|
- const rootNode = initialNodes.find(n => n.data.level === 0);
|
|
|
|
|
- if (rootNode) {
|
|
|
|
|
- nodesWithChildren.delete(rootNode.id);
|
|
|
|
|
- }
|
|
|
|
|
- return nodesWithChildren;
|
|
|
|
|
- }, [initialNodes, initialEdges]);
|
|
|
|
|
-
|
|
|
|
|
- // 树节点的折叠状态需要在树构建后初始化
|
|
|
|
|
- const [collapsedNodes, setCollapsedNodes] = useState(() => initialCollapsedNodes);
|
|
|
|
|
- const [collapsedTreeNodes, setCollapsedTreeNodes] = useState(new Set());
|
|
|
|
|
- const [selectedNodeId, setSelectedNodeId] = useState(null);
|
|
|
|
|
- const [hiddenNodes, setHiddenNodes] = useState(new Set()); // 用户手动隐藏的节点
|
|
|
|
|
- const [focusMode, setFocusMode] = useState(false); // 全局聚焦模式,默认关闭
|
|
|
|
|
- const [focusedNodeId, setFocusedNodeId] = useState(null); // 单独聚焦的节点ID
|
|
|
|
|
-
|
|
|
|
|
- // 获取 React Flow 实例以控制画布
|
|
|
|
|
- const { setCenter, fitView } = useReactFlow();
|
|
|
|
|
-
|
|
|
|
|
- // 获取某个节点的所有后代节点ID
|
|
|
|
|
- const getDescendants = useCallback((nodeId) => {
|
|
|
|
|
- const descendants = new Set();
|
|
|
|
|
- const queue = [nodeId];
|
|
|
|
|
-
|
|
|
|
|
- while (queue.length > 0) {
|
|
|
|
|
- const current = queue.shift();
|
|
|
|
|
- initialEdges.forEach(edge => {
|
|
|
|
|
- if (edge.source === current && !descendants.has(edge.target)) {
|
|
|
|
|
- descendants.add(edge.target);
|
|
|
|
|
- queue.push(edge.target);
|
|
|
|
|
- }
|
|
|
|
|
- });
|
|
|
|
|
- }
|
|
|
|
|
-
|
|
|
|
|
- return descendants;
|
|
|
|
|
- }, [initialEdges]);
|
|
|
|
|
-
|
|
|
|
|
- // 获取直接父节点
|
|
|
|
|
- const getDirectParents = useCallback((nodeId) => {
|
|
|
|
|
- const parents = [];
|
|
|
|
|
- initialEdges.forEach(edge => {
|
|
|
|
|
- if (edge.target === nodeId) {
|
|
|
|
|
- parents.push(edge.source);
|
|
|
|
|
- }
|
|
|
|
|
- });
|
|
|
|
|
- return parents;
|
|
|
|
|
- }, [initialEdges]);
|
|
|
|
|
-
|
|
|
|
|
- // 获取直接子节点
|
|
|
|
|
- const getDirectChildren = useCallback((nodeId) => {
|
|
|
|
|
- const children = [];
|
|
|
|
|
- initialEdges.forEach(edge => {
|
|
|
|
|
- if (edge.source === nodeId) {
|
|
|
|
|
- children.push(edge.target);
|
|
|
|
|
- }
|
|
|
|
|
- });
|
|
|
|
|
- return children;
|
|
|
|
|
- }, [initialEdges]);
|
|
|
|
|
-
|
|
|
|
|
- // 切换节点折叠状态
|
|
|
|
|
- const toggleNodeCollapse = useCallback((nodeId) => {
|
|
|
|
|
- setCollapsedNodes(prev => {
|
|
|
|
|
- const newSet = new Set(prev);
|
|
|
|
|
- const descendants = getDescendants(nodeId);
|
|
|
|
|
-
|
|
|
|
|
- if (newSet.has(nodeId)) {
|
|
|
|
|
- // 展开:移除此节点,但保持其他折叠的节点
|
|
|
|
|
- newSet.delete(nodeId);
|
|
|
|
|
- } else {
|
|
|
|
|
- // 折叠:添加此节点
|
|
|
|
|
- newSet.add(nodeId);
|
|
|
|
|
- }
|
|
|
|
|
-
|
|
|
|
|
- return newSet;
|
|
|
|
|
- });
|
|
|
|
|
- }, [getDescendants]);
|
|
|
|
|
-
|
|
|
|
|
- // 过滤可见的节点和边,并重新计算布局
|
|
|
|
|
- const { nodes, edges } = useMemo(() => {
|
|
|
|
|
- const nodesToHide = new Set();
|
|
|
|
|
-
|
|
|
|
|
- // 判断使用哪个节点ID进行聚焦:优先使用单独聚焦的节点,否则使用全局聚焦模式的选中节点
|
|
|
|
|
- const effectiveFocusNodeId = focusedNodeId || (focusMode ? selectedNodeId : null);
|
|
|
|
|
-
|
|
|
|
|
- // 聚焦模式:只显示聚焦节点、其父节点和直接子节点
|
|
|
|
|
- if (effectiveFocusNodeId) {
|
|
|
|
|
- const visibleInFocus = new Set([effectiveFocusNodeId]);
|
|
|
|
|
-
|
|
|
|
|
- // 添加所有父节点
|
|
|
|
|
- initialEdges.forEach(edge => {
|
|
|
|
|
- if (edge.target === effectiveFocusNodeId) {
|
|
|
|
|
- visibleInFocus.add(edge.source);
|
|
|
|
|
- }
|
|
|
|
|
- });
|
|
|
|
|
-
|
|
|
|
|
- // 添加所有直接子节点
|
|
|
|
|
- initialEdges.forEach(edge => {
|
|
|
|
|
- if (edge.source === effectiveFocusNodeId) {
|
|
|
|
|
- visibleInFocus.add(edge.target);
|
|
|
|
|
- }
|
|
|
|
|
- });
|
|
|
|
|
-
|
|
|
|
|
- // 隐藏不在聚焦范围内的节点
|
|
|
|
|
- initialNodes.forEach(node => {
|
|
|
|
|
- if (!visibleInFocus.has(node.id)) {
|
|
|
|
|
- nodesToHide.add(node.id);
|
|
|
|
|
- }
|
|
|
|
|
- });
|
|
|
|
|
- } else {
|
|
|
|
|
- // 非聚焦模式:使用原有的折叠逻辑
|
|
|
|
|
- // 收集所有被折叠节点的后代
|
|
|
|
|
- collapsedNodes.forEach(collapsedId => {
|
|
|
|
|
- const descendants = getDescendants(collapsedId);
|
|
|
|
|
- descendants.forEach(id => nodesToHide.add(id));
|
|
|
|
|
- });
|
|
|
|
|
- }
|
|
|
|
|
-
|
|
|
|
|
- // 添加用户手动隐藏的节点
|
|
|
|
|
- hiddenNodes.forEach(id => nodesToHide.add(id));
|
|
|
|
|
-
|
|
|
|
|
- const visibleNodes = initialNodes
|
|
|
|
|
- .filter(node => !nodesToHide.has(node.id))
|
|
|
|
|
- .map(node => ({
|
|
|
|
|
- ...node,
|
|
|
|
|
- data: {
|
|
|
|
|
- ...node.data,
|
|
|
|
|
- isCollapsed: collapsedNodes.has(node.id),
|
|
|
|
|
- hasChildren: initialEdges.some(e => e.source === node.id),
|
|
|
|
|
- onToggleCollapse: () => toggleNodeCollapse(node.id),
|
|
|
|
|
- onHideSelf: () => {
|
|
|
|
|
- setHiddenNodes(prev => {
|
|
|
|
|
- const newSet = new Set(prev);
|
|
|
|
|
- newSet.add(node.id);
|
|
|
|
|
- return newSet;
|
|
|
|
|
- });
|
|
|
|
|
- },
|
|
|
|
|
- onFocus: () => {
|
|
|
|
|
- // 切换聚焦状态
|
|
|
|
|
- if (focusedNodeId === node.id) {
|
|
|
|
|
- setFocusedNodeId(null); // 如果已经聚焦,则取消聚焦
|
|
|
|
|
- } else {
|
|
|
|
|
- // 先取消之前的聚焦,然后聚焦到当前节点
|
|
|
|
|
- setFocusedNodeId(node.id);
|
|
|
|
|
-
|
|
|
|
|
- // 延迟聚焦视图到该节点
|
|
|
|
|
- setTimeout(() => {
|
|
|
|
|
- fitView({
|
|
|
|
|
- nodes: [{ id: node.id }],
|
|
|
|
|
- duration: 800,
|
|
|
|
|
- padding: 0.3,
|
|
|
|
|
- });
|
|
|
|
|
- }, 100);
|
|
|
|
|
- }
|
|
|
|
|
- },
|
|
|
|
|
- isFocused: focusedNodeId === node.id,
|
|
|
|
|
- isHighlighted: selectedNodeId === node.id,
|
|
|
|
|
- }
|
|
|
|
|
- }));
|
|
|
|
|
-
|
|
|
|
|
- const visibleEdges = initialEdges.filter(
|
|
|
|
|
- edge => !nodesToHide.has(edge.source) && !nodesToHide.has(edge.target)
|
|
|
|
|
- );
|
|
|
|
|
-
|
|
|
|
|
- // 重新计算布局 - 只对可见节点
|
|
|
|
|
- if (typeof window !== 'undefined' && typeof window.dagre !== 'undefined') {
|
|
|
|
|
- try {
|
|
|
|
|
- const dagreGraph = new window.dagre.graphlib.Graph();
|
|
|
|
|
- dagreGraph.setDefaultEdgeLabel(() => ({}));
|
|
|
|
|
-
|
|
|
|
|
- dagreGraph.setGraph({
|
|
|
|
|
- rankdir: 'LR',
|
|
|
|
|
- nodesep: 120, // 垂直间距 - 增加以避免节点重叠
|
|
|
|
|
- ranksep: 280, // 水平间距 - 增加以容纳更宽的节点
|
|
|
|
|
- });
|
|
|
|
|
-
|
|
|
|
|
- visibleNodes.forEach((node) => {
|
|
|
|
|
- let nodeWidth = 280;
|
|
|
|
|
- let nodeHeight = 180;
|
|
|
|
|
-
|
|
|
|
|
- // note 节点有轮播图,需要更大的空间
|
|
|
|
|
- if (node.type === 'note') {
|
|
|
|
|
- nodeWidth = 320;
|
|
|
|
|
- nodeHeight = 350;
|
|
|
|
|
- }
|
|
|
|
|
-
|
|
|
|
|
- dagreGraph.setNode(node.id, { width: nodeWidth, height: nodeHeight });
|
|
|
|
|
- });
|
|
|
|
|
-
|
|
|
|
|
- visibleEdges.forEach((edge) => {
|
|
|
|
|
- dagreGraph.setEdge(edge.source, edge.target);
|
|
|
|
|
- });
|
|
|
|
|
-
|
|
|
|
|
- window.dagre.layout(dagreGraph);
|
|
|
|
|
-
|
|
|
|
|
- visibleNodes.forEach((node) => {
|
|
|
|
|
- const nodeWithPosition = dagreGraph.node(node.id);
|
|
|
|
|
- if (nodeWithPosition) {
|
|
|
|
|
- // 根据节点类型获取对应的尺寸
|
|
|
|
|
- let nodeWidth = 280;
|
|
|
|
|
- let nodeHeight = 180;
|
|
|
|
|
- if (node.type === 'note') {
|
|
|
|
|
- nodeWidth = 320;
|
|
|
|
|
- nodeHeight = 350;
|
|
|
|
|
- }
|
|
|
|
|
-
|
|
|
|
|
- node.position = {
|
|
|
|
|
- x: nodeWithPosition.x - nodeWidth / 2,
|
|
|
|
|
- y: nodeWithPosition.y - nodeHeight / 2,
|
|
|
|
|
- };
|
|
|
|
|
- node.targetPosition = 'left';
|
|
|
|
|
- node.sourcePosition = 'right';
|
|
|
|
|
- }
|
|
|
|
|
- });
|
|
|
|
|
-
|
|
|
|
|
- console.log('✅ Dynamic layout recalculated for', visibleNodes.length, 'visible nodes');
|
|
|
|
|
- } catch (error) {
|
|
|
|
|
- console.error('❌ Error in dynamic layout:', error);
|
|
|
|
|
- }
|
|
|
|
|
- }
|
|
|
|
|
-
|
|
|
|
|
- return { nodes: visibleNodes, edges: visibleEdges };
|
|
|
|
|
- }, [initialNodes, initialEdges, collapsedNodes, hiddenNodes, focusMode, focusedNodeId, getDescendants, toggleNodeCollapse, selectedNodeId]);
|
|
|
|
|
-
|
|
|
|
|
- // 构建树形结构 - 允许一个节点有多个父节点
|
|
|
|
|
- const buildTree = useCallback(() => {
|
|
|
|
|
- const nodeMap = new Map();
|
|
|
|
|
- initialNodes.forEach(node => {
|
|
|
|
|
- nodeMap.set(node.id, node);
|
|
|
|
|
- });
|
|
|
|
|
-
|
|
|
|
|
- // 为每个节点创建树节点的副本(允许多次出现)
|
|
|
|
|
- const createTreeNode = (nodeId, pathKey) => {
|
|
|
|
|
- const node = nodeMap.get(nodeId);
|
|
|
|
|
- if (!node) return null;
|
|
|
|
|
-
|
|
|
|
|
- return {
|
|
|
|
|
- ...node,
|
|
|
|
|
- treeKey: pathKey, // 唯一的树路径key,用于React key
|
|
|
|
|
- children: []
|
|
|
|
|
- };
|
|
|
|
|
- };
|
|
|
|
|
-
|
|
|
|
|
- // 构建父子关系映射:记录每个节点的所有父节点,去重边
|
|
|
|
|
- const parentToChildren = new Map();
|
|
|
|
|
- const childToParents = new Map();
|
|
|
|
|
-
|
|
|
|
|
- initialEdges.forEach(edge => {
|
|
|
|
|
- // 记录父->子关系(去重:同一个父节点到同一个子节点只记录一次)
|
|
|
|
|
- if (!parentToChildren.has(edge.source)) {
|
|
|
|
|
- parentToChildren.set(edge.source, []);
|
|
|
|
|
- }
|
|
|
|
|
- const children = parentToChildren.get(edge.source);
|
|
|
|
|
- if (!children.includes(edge.target)) {
|
|
|
|
|
- children.push(edge.target);
|
|
|
|
|
- }
|
|
|
|
|
-
|
|
|
|
|
- // 记录子->父关系(用于判断是否有多个父节点,也去重)
|
|
|
|
|
- if (!childToParents.has(edge.target)) {
|
|
|
|
|
- childToParents.set(edge.target, []);
|
|
|
|
|
- }
|
|
|
|
|
- const parents = childToParents.get(edge.target);
|
|
|
|
|
- if (!parents.includes(edge.source)) {
|
|
|
|
|
- parents.push(edge.source);
|
|
|
|
|
- }
|
|
|
|
|
- });
|
|
|
|
|
-
|
|
|
|
|
- // 递归构建树
|
|
|
|
|
- const buildSubtree = (nodeId, pathKey, visitedInPath) => {
|
|
|
|
|
- // 避免循环引用:如果当前路径中已经访问过这个节点,跳过
|
|
|
|
|
- if (visitedInPath.has(nodeId)) {
|
|
|
|
|
- return null;
|
|
|
|
|
- }
|
|
|
|
|
-
|
|
|
|
|
- const treeNode = createTreeNode(nodeId, pathKey);
|
|
|
|
|
- if (!treeNode) return null;
|
|
|
|
|
-
|
|
|
|
|
- const newVisitedInPath = new Set(visitedInPath);
|
|
|
|
|
- newVisitedInPath.add(nodeId);
|
|
|
|
|
-
|
|
|
|
|
- const children = parentToChildren.get(nodeId) || [];
|
|
|
|
|
- treeNode.children = children
|
|
|
|
|
- .map((childId, index) => buildSubtree(childId, pathKey + '-' + childId + '-' + index, newVisitedInPath))
|
|
|
|
|
- .filter(child => child !== null);
|
|
|
|
|
-
|
|
|
|
|
- return treeNode;
|
|
|
|
|
- };
|
|
|
|
|
-
|
|
|
|
|
- // 找出所有根节点(没有入边的节点)
|
|
|
|
|
- const hasParent = new Set();
|
|
|
|
|
- initialEdges.forEach(edge => {
|
|
|
|
|
- hasParent.add(edge.target);
|
|
|
|
|
- });
|
|
|
|
|
-
|
|
|
|
|
- const roots = [];
|
|
|
|
|
- initialNodes.forEach((node, index) => {
|
|
|
|
|
- if (!hasParent.has(node.id)) {
|
|
|
|
|
- const treeNode = buildSubtree(node.id, 'root-' + node.id + '-' + index, new Set());
|
|
|
|
|
- if (treeNode) roots.push(treeNode);
|
|
|
|
|
- }
|
|
|
|
|
- });
|
|
|
|
|
-
|
|
|
|
|
- return roots;
|
|
|
|
|
- }, [initialNodes, initialEdges]);
|
|
|
|
|
-
|
|
|
|
|
- const treeRoots = useMemo(() => buildTree(), [buildTree]);
|
|
|
|
|
-
|
|
|
|
|
- // 初始化树节点折叠状态
|
|
|
|
|
- useEffect(() => {
|
|
|
|
|
- const getAllTreeKeys = (nodes) => {
|
|
|
|
|
- const keys = new Set();
|
|
|
|
|
- const traverse = (node) => {
|
|
|
|
|
- if (node.children && node.children.length > 0) {
|
|
|
|
|
- // 排除根节点
|
|
|
|
|
- if (node.data.level !== 0) {
|
|
|
|
|
- keys.add(node.treeKey);
|
|
|
|
|
- }
|
|
|
|
|
- node.children.forEach(traverse);
|
|
|
|
|
- }
|
|
|
|
|
- };
|
|
|
|
|
- nodes.forEach(traverse);
|
|
|
|
|
- return keys;
|
|
|
|
|
- };
|
|
|
|
|
-
|
|
|
|
|
- setCollapsedTreeNodes(getAllTreeKeys(treeRoots));
|
|
|
|
|
- }, [treeRoots]);
|
|
|
|
|
-
|
|
|
|
|
- const renderTree = useCallback((treeNodes, level = 0) => {
|
|
|
|
|
- return treeNodes.map(node => {
|
|
|
|
|
- // 使用 treeKey 来区分树中的不同实例
|
|
|
|
|
- const isCollapsed = collapsedTreeNodes.has(node.treeKey);
|
|
|
|
|
- const isSelected = selectedNodeId === node.id;
|
|
|
|
|
-
|
|
|
|
|
- return (
|
|
|
|
|
- <TreeNode
|
|
|
|
|
- key={node.treeKey}
|
|
|
|
|
- node={node}
|
|
|
|
|
- level={level}
|
|
|
|
|
- isCollapsed={isCollapsed}
|
|
|
|
|
- isSelected={isSelected}
|
|
|
|
|
- onToggle={() => {
|
|
|
|
|
- setCollapsedTreeNodes(prev => {
|
|
|
|
|
- const newSet = new Set(prev);
|
|
|
|
|
- if (newSet.has(node.treeKey)) {
|
|
|
|
|
- newSet.delete(node.treeKey);
|
|
|
|
|
- } else {
|
|
|
|
|
- newSet.add(node.treeKey);
|
|
|
|
|
- }
|
|
|
|
|
- return newSet;
|
|
|
|
|
- });
|
|
|
|
|
- }}
|
|
|
|
|
- onSelect={() => {
|
|
|
|
|
- const nodeId = node.id;
|
|
|
|
|
-
|
|
|
|
|
- // 展开所有祖先节点
|
|
|
|
|
- const ancestorIds = [nodeId];
|
|
|
|
|
- const findAncestors = (id) => {
|
|
|
|
|
- initialEdges.forEach(edge => {
|
|
|
|
|
- if (edge.target === id && !ancestorIds.includes(edge.source)) {
|
|
|
|
|
- ancestorIds.push(edge.source);
|
|
|
|
|
- findAncestors(edge.source);
|
|
|
|
|
- }
|
|
|
|
|
- });
|
|
|
|
|
- };
|
|
|
|
|
- findAncestors(nodeId);
|
|
|
|
|
-
|
|
|
|
|
- // 如果节点或其祖先被隐藏,先恢复它们
|
|
|
|
|
- setHiddenNodes(prev => {
|
|
|
|
|
- const newSet = new Set(prev);
|
|
|
|
|
- ancestorIds.forEach(id => newSet.delete(id));
|
|
|
|
|
- return newSet;
|
|
|
|
|
- });
|
|
|
|
|
-
|
|
|
|
|
- setSelectedNodeId(nodeId);
|
|
|
|
|
-
|
|
|
|
|
- // 获取选中节点的直接子节点
|
|
|
|
|
- const childrenIds = [];
|
|
|
|
|
- initialEdges.forEach(edge => {
|
|
|
|
|
- if (edge.source === nodeId) {
|
|
|
|
|
- childrenIds.push(edge.target);
|
|
|
|
|
- }
|
|
|
|
|
- });
|
|
|
|
|
-
|
|
|
|
|
- setCollapsedNodes(prev => {
|
|
|
|
|
- const newSet = new Set(prev);
|
|
|
|
|
- // 展开所有祖先节点
|
|
|
|
|
- ancestorIds.forEach(id => newSet.delete(id));
|
|
|
|
|
- // 展开选中节点本身
|
|
|
|
|
- newSet.delete(nodeId);
|
|
|
|
|
- // 展开选中节点的直接子节点
|
|
|
|
|
- childrenIds.forEach(id => newSet.delete(id));
|
|
|
|
|
- return newSet;
|
|
|
|
|
- });
|
|
|
|
|
-
|
|
|
|
|
- // 延迟聚焦,等待节点展开和布局重新计算
|
|
|
|
|
- setTimeout(() => {
|
|
|
|
|
- fitView({
|
|
|
|
|
- nodes: [{ id: nodeId }],
|
|
|
|
|
- duration: 800,
|
|
|
|
|
- padding: 0.3,
|
|
|
|
|
- });
|
|
|
|
|
- }, 300);
|
|
|
|
|
- }}
|
|
|
|
|
- >
|
|
|
|
|
- {node.children && node.children.length > 0 && renderTree(node.children, level + 1)}
|
|
|
|
|
- </TreeNode>
|
|
|
|
|
- );
|
|
|
|
|
- });
|
|
|
|
|
- }, [collapsedTreeNodes, selectedNodeId, nodes, setCenter, initialEdges, setCollapsedNodes, fitView]);
|
|
|
|
|
-
|
|
|
|
|
- console.log('📊 Rendering with', nodes.length, 'visible nodes and', edges.length, 'visible edges');
|
|
|
|
|
-
|
|
|
|
|
- if (nodes.length === 0) {
|
|
|
|
|
- return (
|
|
|
|
|
- <div style={{ padding: 50, color: 'red', fontSize: 20 }}>
|
|
|
|
|
- ERROR: No nodes to display!
|
|
|
|
|
- </div>
|
|
|
|
|
- );
|
|
|
|
|
- }
|
|
|
|
|
-
|
|
|
|
|
- return (
|
|
|
|
|
- <div style={{ width: '100vw', height: '100vh', background: '#f9fafb', display: 'flex', flexDirection: 'column' }}>
|
|
|
|
|
- {/* 顶部面包屑导航栏 */}
|
|
|
|
|
- <div style={{
|
|
|
|
|
- minHeight: '48px',
|
|
|
|
|
- maxHeight: '120px',
|
|
|
|
|
- background: 'white',
|
|
|
|
|
- borderBottom: '1px solid #e5e7eb',
|
|
|
|
|
- display: 'flex',
|
|
|
|
|
- alignItems: 'flex-start',
|
|
|
|
|
- padding: '12px 24px',
|
|
|
|
|
- zIndex: 1000,
|
|
|
|
|
- boxShadow: '0 1px 3px rgba(0, 0, 0, 0.05)',
|
|
|
|
|
- flexShrink: 0,
|
|
|
|
|
- overflowY: 'auto',
|
|
|
|
|
- }}>
|
|
|
|
|
- <div style={{ width: '100%' }}>
|
|
|
|
|
- {selectedNodeId ? (
|
|
|
|
|
- <div style={{ fontSize: '12px', color: '#6b7280' }}>
|
|
|
|
|
- {/* 面包屑导航 - 显示所有路径 */}
|
|
|
|
|
- {(() => {
|
|
|
|
|
- const selectedNode = nodes.find(n => n.id === selectedNodeId);
|
|
|
|
|
- if (!selectedNode) return null;
|
|
|
|
|
-
|
|
|
|
|
- // 找到所有从根节点到当前节点的路径
|
|
|
|
|
- const findAllPaths = (targetId) => {
|
|
|
|
|
- const paths = [];
|
|
|
|
|
-
|
|
|
|
|
- const buildPath = (nodeId, currentPath) => {
|
|
|
|
|
- const node = initialNodes.find(n => n.id === nodeId);
|
|
|
|
|
- if (!node) return;
|
|
|
|
|
-
|
|
|
|
|
- const newPath = [node, ...currentPath];
|
|
|
|
|
-
|
|
|
|
|
- // 找到所有父节点
|
|
|
|
|
- const parents = initialEdges.filter(e => e.target === nodeId).map(e => e.source);
|
|
|
|
|
-
|
|
|
|
|
- if (parents.length === 0) {
|
|
|
|
|
- // 到达根节点
|
|
|
|
|
- paths.push(newPath);
|
|
|
|
|
- } else {
|
|
|
|
|
- // 递归处理所有父节点
|
|
|
|
|
- parents.forEach(parentId => {
|
|
|
|
|
- buildPath(parentId, newPath);
|
|
|
|
|
- });
|
|
|
|
|
- }
|
|
|
|
|
- };
|
|
|
|
|
-
|
|
|
|
|
- buildPath(targetId, []);
|
|
|
|
|
- return paths;
|
|
|
|
|
- };
|
|
|
|
|
-
|
|
|
|
|
- const allPaths = findAllPaths(selectedNodeId);
|
|
|
|
|
-
|
|
|
|
|
- // 去重:将路径转换为字符串进行比较
|
|
|
|
|
- const uniquePaths = [];
|
|
|
|
|
- const pathStrings = new Set();
|
|
|
|
|
- allPaths.forEach(path => {
|
|
|
|
|
- const pathString = path.map(n => n.id).join('->');
|
|
|
|
|
- if (!pathStrings.has(pathString)) {
|
|
|
|
|
- pathStrings.add(pathString);
|
|
|
|
|
- uniquePaths.push(path);
|
|
|
|
|
- }
|
|
|
|
|
- });
|
|
|
|
|
-
|
|
|
|
|
- return (
|
|
|
|
|
- <div style={{ display: 'flex', flexDirection: 'column', gap: '6px' }}>
|
|
|
|
|
- {uniquePaths.map((path, pathIndex) => (
|
|
|
|
|
- <div key={pathIndex} style={{ display: 'flex', alignItems: 'center', gap: '6px', flexWrap: 'wrap' }}>
|
|
|
|
|
- {pathIndex > 0 && <span style={{ color: '#d1d5db', marginRight: '4px' }}>或</span>}
|
|
|
|
|
- {path.map((node, index) => {
|
|
|
|
|
- // 获取节点的 score、strategy 和 isSelected
|
|
|
|
|
- const nodeScore = node.data.score ? parseFloat(node.data.score) : 0;
|
|
|
|
|
- const nodeStrategy = node.data.strategy || '';
|
|
|
|
|
- const strategyColor = getStrategyColor(nodeStrategy);
|
|
|
|
|
- const nodeIsSelected = node.type === 'note' ? node.data.matchLevel !== 'unsatisfied' : node.data.isSelected !== false;
|
|
|
|
|
-
|
|
|
|
|
- return (
|
|
|
|
|
- <React.Fragment key={node.id + '-' + index}>
|
|
|
|
|
- <span
|
|
|
|
|
- onClick={() => {
|
|
|
|
|
- const nodeId = node.id;
|
|
|
|
|
-
|
|
|
|
|
- // 找到所有祖先节点
|
|
|
|
|
- const ancestorIds = [nodeId];
|
|
|
|
|
- const findAncestors = (id) => {
|
|
|
|
|
- initialEdges.forEach(edge => {
|
|
|
|
|
- if (edge.target === id && !ancestorIds.includes(edge.source)) {
|
|
|
|
|
- ancestorIds.push(edge.source);
|
|
|
|
|
- findAncestors(edge.source);
|
|
|
|
|
- }
|
|
|
|
|
- });
|
|
|
|
|
- };
|
|
|
|
|
- findAncestors(nodeId);
|
|
|
|
|
-
|
|
|
|
|
- // 如果节点或其祖先被隐藏,先恢复它们
|
|
|
|
|
- setHiddenNodes(prev => {
|
|
|
|
|
- const newSet = new Set(prev);
|
|
|
|
|
- ancestorIds.forEach(id => newSet.delete(id));
|
|
|
|
|
- return newSet;
|
|
|
|
|
- });
|
|
|
|
|
-
|
|
|
|
|
- // 展开目录树中到达该节点的路径
|
|
|
|
|
- // 需要找到所有包含该节点的树路径的 treeKey,并展开它们的父节点
|
|
|
|
|
- setCollapsedTreeNodes(prev => {
|
|
|
|
|
- const newSet = new Set(prev);
|
|
|
|
|
- // 清空所有折叠状态,让目录树完全展开到选中节点
|
|
|
|
|
- // 这样可以确保选中节点在目录中可见
|
|
|
|
|
- return new Set();
|
|
|
|
|
- });
|
|
|
|
|
-
|
|
|
|
|
- setSelectedNodeId(nodeId);
|
|
|
|
|
- setTimeout(() => {
|
|
|
|
|
- fitView({
|
|
|
|
|
- nodes: [{ id: nodeId }],
|
|
|
|
|
- duration: 800,
|
|
|
|
|
- padding: 0.3,
|
|
|
|
|
- });
|
|
|
|
|
- }, 100);
|
|
|
|
|
- }}
|
|
|
|
|
- style={{
|
|
|
|
|
- padding: '6px 8px',
|
|
|
|
|
- borderRadius: '4px',
|
|
|
|
|
- background: 'white',
|
|
|
|
|
- border: index === path.length - 1 ? '2px solid #3b82f6' : '1px solid #d1d5db',
|
|
|
|
|
- color: '#374151',
|
|
|
|
|
- fontWeight: index === path.length - 1 ? '600' : '400',
|
|
|
|
|
- width: '180px',
|
|
|
|
|
- cursor: 'pointer',
|
|
|
|
|
- transition: 'all 0.2s ease',
|
|
|
|
|
- position: 'relative',
|
|
|
|
|
- display: 'inline-flex',
|
|
|
|
|
- flexDirection: 'column',
|
|
|
|
|
- gap: '4px',
|
|
|
|
|
- }}
|
|
|
|
|
- onMouseEnter={(e) => {
|
|
|
|
|
- e.currentTarget.style.opacity = '0.8';
|
|
|
|
|
- }}
|
|
|
|
|
- onMouseLeave={(e) => {
|
|
|
|
|
- e.currentTarget.style.opacity = '1';
|
|
|
|
|
- }}
|
|
|
|
|
- title={\`\${node.data.title || node.id} (Score: \${nodeScore.toFixed(2)}, Strategy: \${nodeStrategy}, Selected: \${nodeIsSelected})\`}
|
|
|
|
|
- >
|
|
|
|
|
- {/* 上半部分:竖线 + 图标 + 文字 + 分数 */}
|
|
|
|
|
- <div style={{
|
|
|
|
|
- display: 'flex',
|
|
|
|
|
- alignItems: 'center',
|
|
|
|
|
- gap: '6px',
|
|
|
|
|
- }}>
|
|
|
|
|
- {/* 策略类型竖线 */}
|
|
|
|
|
- <div style={{
|
|
|
|
|
- width: '3px',
|
|
|
|
|
- height: '16px',
|
|
|
|
|
- background: strategyColor,
|
|
|
|
|
- borderRadius: '2px',
|
|
|
|
|
- flexShrink: 0,
|
|
|
|
|
- }} />
|
|
|
|
|
-
|
|
|
|
|
- {/* 节点类型图标 */}
|
|
|
|
|
- <span style={{
|
|
|
|
|
- fontSize: '11px',
|
|
|
|
|
- flexShrink: 0,
|
|
|
|
|
- }}>
|
|
|
|
|
- {node.type === 'note' ? '📝' : '🔍'}
|
|
|
|
|
- </span>
|
|
|
|
|
-
|
|
|
|
|
- {/* 节点文字 */}
|
|
|
|
|
- <span style={{
|
|
|
|
|
- flex: 1,
|
|
|
|
|
- fontSize: '12px',
|
|
|
|
|
- color: nodeIsSelected ? '#374151' : '#ef4444',
|
|
|
|
|
- }}>
|
|
|
|
|
- {truncateMiddle(node.data.title || node.id, 18)}
|
|
|
|
|
- </span>
|
|
|
|
|
-
|
|
|
|
|
- {/* 分数显示 */}
|
|
|
|
|
- <span style={{
|
|
|
|
|
- fontSize: '10px',
|
|
|
|
|
- color: '#6b7280',
|
|
|
|
|
- fontWeight: '500',
|
|
|
|
|
- flexShrink: 0,
|
|
|
|
|
- }}>
|
|
|
|
|
- {nodeScore.toFixed(2)}
|
|
|
|
|
- </span>
|
|
|
|
|
- </div>
|
|
|
|
|
-
|
|
|
|
|
- {/* 分数下划线 */}
|
|
|
|
|
- <div style={{
|
|
|
|
|
- width: (nodeScore * 100) + '%',
|
|
|
|
|
- height: '2px',
|
|
|
|
|
- background: getScoreColor(nodeScore),
|
|
|
|
|
- borderRadius: '1px',
|
|
|
|
|
- marginLeft: '9px',
|
|
|
|
|
- }} />
|
|
|
|
|
- </span>
|
|
|
|
|
- {index < path.length - 1 && <span style={{ color: '#9ca3af' }}>›</span>}
|
|
|
|
|
- </React.Fragment>
|
|
|
|
|
- )})}
|
|
|
|
|
- </div>
|
|
|
|
|
- ))}
|
|
|
|
|
- </div>
|
|
|
|
|
- );
|
|
|
|
|
- })()}
|
|
|
|
|
- </div>
|
|
|
|
|
- ) : (
|
|
|
|
|
- <div style={{ fontSize: '13px', color: '#9ca3af', textAlign: 'center' }}>
|
|
|
|
|
- 选择一个节点查看路径
|
|
|
|
|
- </div>
|
|
|
|
|
- )}
|
|
|
|
|
- </div>
|
|
|
|
|
- </div>
|
|
|
|
|
-
|
|
|
|
|
- {/* 主内容区:目录 + 画布 */}
|
|
|
|
|
- <div style={{ display: 'flex', flex: 1, overflow: 'hidden' }}>
|
|
|
|
|
- {/* 左侧目录树 */}
|
|
|
|
|
- <div style={{
|
|
|
|
|
- width: '320px',
|
|
|
|
|
- background: 'white',
|
|
|
|
|
- borderRight: '1px solid #e5e7eb',
|
|
|
|
|
- display: 'flex',
|
|
|
|
|
- flexDirection: 'column',
|
|
|
|
|
- flexShrink: 0,
|
|
|
|
|
- }}>
|
|
|
|
|
- <div style={{
|
|
|
|
|
- padding: '12px 16px',
|
|
|
|
|
- borderBottom: '1px solid #e5e7eb',
|
|
|
|
|
- display: 'flex',
|
|
|
|
|
- justifyContent: 'space-between',
|
|
|
|
|
- alignItems: 'center',
|
|
|
|
|
- }}>
|
|
|
|
|
- <span style={{
|
|
|
|
|
- fontWeight: '600',
|
|
|
|
|
- fontSize: '14px',
|
|
|
|
|
- color: '#111827',
|
|
|
|
|
- }}>
|
|
|
|
|
- 节点目录
|
|
|
|
|
- </span>
|
|
|
|
|
- <div style={{ display: 'flex', gap: '6px' }}>
|
|
|
|
|
- <button
|
|
|
|
|
- onClick={() => {
|
|
|
|
|
- setCollapsedTreeNodes(new Set());
|
|
|
|
|
- }}
|
|
|
|
|
- style={{
|
|
|
|
|
- fontSize: '11px',
|
|
|
|
|
- padding: '4px 8px',
|
|
|
|
|
- borderRadius: '4px',
|
|
|
|
|
- border: '1px solid #d1d5db',
|
|
|
|
|
- background: 'white',
|
|
|
|
|
- color: '#6b7280',
|
|
|
|
|
- cursor: 'pointer',
|
|
|
|
|
- fontWeight: '500',
|
|
|
|
|
- }}
|
|
|
|
|
- title="展开全部节点"
|
|
|
|
|
- >
|
|
|
|
|
- 全部展开
|
|
|
|
|
- </button>
|
|
|
|
|
- <button
|
|
|
|
|
- onClick={() => {
|
|
|
|
|
- const getAllTreeKeys = (nodes) => {
|
|
|
|
|
- const keys = new Set();
|
|
|
|
|
- const traverse = (node) => {
|
|
|
|
|
- if (node.children && node.children.length > 0) {
|
|
|
|
|
- keys.add(node.treeKey);
|
|
|
|
|
- node.children.forEach(traverse);
|
|
|
|
|
- }
|
|
|
|
|
- };
|
|
|
|
|
- nodes.forEach(traverse);
|
|
|
|
|
- return keys;
|
|
|
|
|
- };
|
|
|
|
|
- setCollapsedTreeNodes(getAllTreeKeys(treeRoots));
|
|
|
|
|
- }}
|
|
|
|
|
- style={{
|
|
|
|
|
- fontSize: '11px',
|
|
|
|
|
- padding: '4px 8px',
|
|
|
|
|
- borderRadius: '4px',
|
|
|
|
|
- border: '1px solid #d1d5db',
|
|
|
|
|
- background: 'white',
|
|
|
|
|
- color: '#6b7280',
|
|
|
|
|
- cursor: 'pointer',
|
|
|
|
|
- fontWeight: '500',
|
|
|
|
|
- }}
|
|
|
|
|
- title="折叠全部节点"
|
|
|
|
|
- >
|
|
|
|
|
- 全部折叠
|
|
|
|
|
- </button>
|
|
|
|
|
- </div>
|
|
|
|
|
- </div>
|
|
|
|
|
- <div style={{
|
|
|
|
|
- flex: 1,
|
|
|
|
|
- overflowX: 'auto',
|
|
|
|
|
- overflowY: 'auto',
|
|
|
|
|
- padding: '8px',
|
|
|
|
|
- }}>
|
|
|
|
|
- <div style={{ minWidth: 'fit-content' }}>
|
|
|
|
|
- {renderTree(treeRoots)}
|
|
|
|
|
- </div>
|
|
|
|
|
- </div>
|
|
|
|
|
- </div>
|
|
|
|
|
-
|
|
|
|
|
- {/* 画布区域 */}
|
|
|
|
|
- <div style={{ flex: 1, position: 'relative' }}>
|
|
|
|
|
-
|
|
|
|
|
- {/* 右侧图例 */}
|
|
|
|
|
- <div style={{
|
|
|
|
|
- position: 'absolute',
|
|
|
|
|
- top: '20px',
|
|
|
|
|
- right: '20px',
|
|
|
|
|
- background: 'white',
|
|
|
|
|
- padding: '16px',
|
|
|
|
|
- borderRadius: '12px',
|
|
|
|
|
- boxShadow: '0 4px 12px rgba(0, 0, 0, 0.08)',
|
|
|
|
|
- zIndex: 1000,
|
|
|
|
|
- maxWidth: '260px',
|
|
|
|
|
- border: '1px solid #e5e7eb',
|
|
|
|
|
- }}>
|
|
|
|
|
- <div style={{ display: 'flex', justifyContent: 'space-between', alignItems: 'center', marginBottom: '12px' }}>
|
|
|
|
|
- <h3 style={{ fontSize: '14px', fontWeight: '600', color: '#111827', margin: 0 }}>图例</h3>
|
|
|
|
|
- <button
|
|
|
|
|
- onClick={() => setFocusMode(!focusMode)}
|
|
|
|
|
- style={{
|
|
|
|
|
- fontSize: '11px',
|
|
|
|
|
- padding: '4px 8px',
|
|
|
|
|
- borderRadius: '4px',
|
|
|
|
|
- border: '1px solid',
|
|
|
|
|
- borderColor: focusMode ? '#3b82f6' : '#d1d5db',
|
|
|
|
|
- background: focusMode ? '#3b82f6' : 'white',
|
|
|
|
|
- color: focusMode ? 'white' : '#6b7280',
|
|
|
|
|
- cursor: 'pointer',
|
|
|
|
|
- fontWeight: '500',
|
|
|
|
|
- }}
|
|
|
|
|
- title={focusMode ? '关闭聚焦模式' : '开启聚焦模式'}
|
|
|
|
|
- >
|
|
|
|
|
- {focusMode ? '🎯 聚焦' : '📊 全图'}
|
|
|
|
|
- </button>
|
|
|
|
|
- </div>
|
|
|
|
|
-
|
|
|
|
|
- <div style={{ fontSize: '12px' }}>
|
|
|
|
|
- {/* 画布节点展开/折叠控制 */}
|
|
|
|
|
- <div style={{ marginBottom: '12px', paddingBottom: '12px', borderBottom: '1px solid #f3f4f6' }}>
|
|
|
|
|
- <div style={{ fontSize: '12px', fontWeight: '500', marginBottom: '8px', color: '#374151' }}>节点控制</div>
|
|
|
|
|
- <div style={{ display: 'flex', gap: '6px' }}>
|
|
|
|
|
- <button
|
|
|
|
|
- onClick={() => {
|
|
|
|
|
- setCollapsedNodes(new Set());
|
|
|
|
|
- }}
|
|
|
|
|
- style={{
|
|
|
|
|
- fontSize: '11px',
|
|
|
|
|
- padding: '4px 8px',
|
|
|
|
|
- borderRadius: '4px',
|
|
|
|
|
- border: '1px solid #d1d5db',
|
|
|
|
|
- background: 'white',
|
|
|
|
|
- color: '#6b7280',
|
|
|
|
|
- cursor: 'pointer',
|
|
|
|
|
- fontWeight: '500',
|
|
|
|
|
- flex: 1,
|
|
|
|
|
- }}
|
|
|
|
|
- title="展开画布中所有节点的子节点"
|
|
|
|
|
- >
|
|
|
|
|
- 全部展开
|
|
|
|
|
- </button>
|
|
|
|
|
- <button
|
|
|
|
|
- onClick={() => {
|
|
|
|
|
- const allNodeIds = new Set(initialNodes.map(n => n.id));
|
|
|
|
|
- setCollapsedNodes(allNodeIds);
|
|
|
|
|
- }}
|
|
|
|
|
- style={{
|
|
|
|
|
- fontSize: '11px',
|
|
|
|
|
- padding: '4px 8px',
|
|
|
|
|
- borderRadius: '4px',
|
|
|
|
|
- border: '1px solid #d1d5db',
|
|
|
|
|
- background: 'white',
|
|
|
|
|
- color: '#6b7280',
|
|
|
|
|
- cursor: 'pointer',
|
|
|
|
|
- fontWeight: '500',
|
|
|
|
|
- flex: 1,
|
|
|
|
|
- }}
|
|
|
|
|
- title="折叠画布中所有节点的子节点"
|
|
|
|
|
- >
|
|
|
|
|
- 全部折叠
|
|
|
|
|
- </button>
|
|
|
|
|
- </div>
|
|
|
|
|
- </div>
|
|
|
|
|
-
|
|
|
|
|
- <div style={{ paddingTop: '12px', borderTop: '1px solid #f3f4f6' }}>
|
|
|
|
|
- <div style={{ fontSize: '12px', fontWeight: '500', marginBottom: '8px', color: '#374151' }}>策略类型</div>
|
|
|
|
|
- <div style={{ display: 'flex', alignItems: 'center', margin: '6px 0' }}>
|
|
|
|
|
- <div style={{ width: '20px', height: '2px', marginRight: '8px', background: '#10b981', opacity: 0.7 }}></div>
|
|
|
|
|
- <span style={{ color: '#6b7280', fontSize: '11px' }}>初始分词</span>
|
|
|
|
|
- </div>
|
|
|
|
|
- <div style={{ display: 'flex', alignItems: 'center', margin: '6px 0' }}>
|
|
|
|
|
- <div style={{ width: '20px', height: '2px', marginRight: '8px', background: '#06b6d4', opacity: 0.7 }}></div>
|
|
|
|
|
- <span style={{ color: '#6b7280', fontSize: '11px' }}>调用sug</span>
|
|
|
|
|
- </div>
|
|
|
|
|
- <div style={{ display: 'flex', alignItems: 'center', margin: '6px 0' }}>
|
|
|
|
|
- <div style={{ width: '20px', height: '2px', marginRight: '8px', background: '#f59e0b', opacity: 0.7 }}></div>
|
|
|
|
|
- <span style={{ color: '#6b7280', fontSize: '11px' }}>同义改写</span>
|
|
|
|
|
- </div>
|
|
|
|
|
- <div style={{ display: 'flex', alignItems: 'center', margin: '6px 0' }}>
|
|
|
|
|
- <div style={{ width: '20px', height: '2px', marginRight: '8px', background: '#3b82f6', opacity: 0.7 }}></div>
|
|
|
|
|
- <span style={{ color: '#6b7280', fontSize: '11px' }}>加词</span>
|
|
|
|
|
- </div>
|
|
|
|
|
- <div style={{ display: 'flex', alignItems: 'center', margin: '6px 0' }}>
|
|
|
|
|
- <div style={{ width: '20px', height: '2px', marginRight: '8px', background: '#8b5cf6', opacity: 0.7 }}></div>
|
|
|
|
|
- <span style={{ color: '#6b7280', fontSize: '11px' }}>抽象改写</span>
|
|
|
|
|
- </div>
|
|
|
|
|
- <div style={{ display: 'flex', alignItems: 'center', margin: '6px 0' }}>
|
|
|
|
|
- <div style={{ width: '20px', height: '2px', marginRight: '8px', background: '#ec4899', opacity: 0.7 }}></div>
|
|
|
|
|
- <span style={{ color: '#6b7280', fontSize: '11px' }}>基于部分匹配改进</span>
|
|
|
|
|
- </div>
|
|
|
|
|
- <div style={{ display: 'flex', alignItems: 'center', margin: '6px 0' }}>
|
|
|
|
|
- <div style={{ width: '20px', height: '2px', marginRight: '8px', background: '#a855f7', opacity: 0.7 }}></div>
|
|
|
|
|
- <span style={{ color: '#6b7280', fontSize: '11px' }}>结果分支-抽象改写</span>
|
|
|
|
|
- </div>
|
|
|
|
|
- <div style={{ display: 'flex', alignItems: 'center', margin: '6px 0' }}>
|
|
|
|
|
- <div style={{ width: '20px', height: '2px', marginRight: '8px', background: '#fb923c', opacity: 0.7 }}></div>
|
|
|
|
|
- <span style={{ color: '#6b7280', fontSize: '11px' }}>结果分支-同义改写</span>
|
|
|
|
|
- </div>
|
|
|
|
|
- </div>
|
|
|
|
|
-
|
|
|
|
|
- <div style={{
|
|
|
|
|
- marginTop: '12px',
|
|
|
|
|
- paddingTop: '12px',
|
|
|
|
|
- borderTop: '1px solid #f3f4f6',
|
|
|
|
|
- fontSize: '11px',
|
|
|
|
|
- color: '#9ca3af',
|
|
|
|
|
- lineHeight: '1.5',
|
|
|
|
|
- }}>
|
|
|
|
|
- 💡 点击节点左上角 × 隐藏节点
|
|
|
|
|
- </div>
|
|
|
|
|
-
|
|
|
|
|
- {/* 隐藏节点列表 - 在图例内部 */}
|
|
|
|
|
- {hiddenNodes.size > 0 && (
|
|
|
|
|
- <div style={{
|
|
|
|
|
- marginTop: '12px',
|
|
|
|
|
- paddingTop: '12px',
|
|
|
|
|
- borderTop: '1px solid #f3f4f6',
|
|
|
|
|
- }}>
|
|
|
|
|
- <div style={{ display: 'flex', justifyContent: 'space-between', alignItems: 'center', marginBottom: '8px' }}>
|
|
|
|
|
- <h4 style={{ fontSize: '12px', fontWeight: '600', color: '#111827' }}>已隐藏节点</h4>
|
|
|
|
|
- <button
|
|
|
|
|
- onClick={() => setHiddenNodes(new Set())}
|
|
|
|
|
- style={{
|
|
|
|
|
- fontSize: '10px',
|
|
|
|
|
- color: '#3b82f6',
|
|
|
|
|
- background: 'none',
|
|
|
|
|
- border: 'none',
|
|
|
|
|
- cursor: 'pointer',
|
|
|
|
|
- textDecoration: 'underline',
|
|
|
|
|
- }}
|
|
|
|
|
- >
|
|
|
|
|
- 全部恢复
|
|
|
|
|
- </button>
|
|
|
|
|
- </div>
|
|
|
|
|
- <div style={{ fontSize: '12px', maxHeight: '200px', overflow: 'auto' }}>
|
|
|
|
|
- {Array.from(hiddenNodes).map(nodeId => {
|
|
|
|
|
- const node = initialNodes.find(n => n.id === nodeId);
|
|
|
|
|
- if (!node) return null;
|
|
|
|
|
- return (
|
|
|
|
|
- <div
|
|
|
|
|
- key={nodeId}
|
|
|
|
|
- style={{
|
|
|
|
|
- display: 'flex',
|
|
|
|
|
- justifyContent: 'space-between',
|
|
|
|
|
- alignItems: 'center',
|
|
|
|
|
- padding: '6px 8px',
|
|
|
|
|
- margin: '4px 0',
|
|
|
|
|
- background: '#f9fafb',
|
|
|
|
|
- borderRadius: '6px',
|
|
|
|
|
- fontSize: '11px',
|
|
|
|
|
- }}
|
|
|
|
|
- >
|
|
|
|
|
- <span
|
|
|
|
|
- style={{
|
|
|
|
|
- flex: 1,
|
|
|
|
|
- overflow: 'hidden',
|
|
|
|
|
- textOverflow: 'ellipsis',
|
|
|
|
|
- whiteSpace: 'nowrap',
|
|
|
|
|
- color: '#374151',
|
|
|
|
|
- }}
|
|
|
|
|
- title={node.data.title || nodeId}
|
|
|
|
|
- >
|
|
|
|
|
- {node.data.title || nodeId}
|
|
|
|
|
- </span>
|
|
|
|
|
- <button
|
|
|
|
|
- onClick={() => {
|
|
|
|
|
- setHiddenNodes(prev => {
|
|
|
|
|
- const newSet = new Set(prev);
|
|
|
|
|
- newSet.delete(nodeId);
|
|
|
|
|
- return newSet;
|
|
|
|
|
- });
|
|
|
|
|
- }}
|
|
|
|
|
- style={{
|
|
|
|
|
- marginLeft: '8px',
|
|
|
|
|
- fontSize: '10px',
|
|
|
|
|
- color: '#10b981',
|
|
|
|
|
- background: 'none',
|
|
|
|
|
- border: 'none',
|
|
|
|
|
- cursor: 'pointer',
|
|
|
|
|
- flexShrink: 0,
|
|
|
|
|
- }}
|
|
|
|
|
- >
|
|
|
|
|
- 恢复
|
|
|
|
|
- </button>
|
|
|
|
|
- </div>
|
|
|
|
|
- );
|
|
|
|
|
- })}
|
|
|
|
|
- </div>
|
|
|
|
|
- </div>
|
|
|
|
|
- )}
|
|
|
|
|
- </div>
|
|
|
|
|
- </div>
|
|
|
|
|
-
|
|
|
|
|
- {/* React Flow 画布 */}
|
|
|
|
|
- <ReactFlow
|
|
|
|
|
- nodes={nodes}
|
|
|
|
|
- edges={edges}
|
|
|
|
|
- nodeTypes={nodeTypes}
|
|
|
|
|
- fitView
|
|
|
|
|
- fitViewOptions={{ padding: 0.2, duration: 500 }}
|
|
|
|
|
- minZoom={0.1}
|
|
|
|
|
- maxZoom={1.5}
|
|
|
|
|
- nodesDraggable={true}
|
|
|
|
|
- nodesConnectable={false}
|
|
|
|
|
- elementsSelectable={true}
|
|
|
|
|
- defaultEdgeOptions={{
|
|
|
|
|
- type: 'smoothstep',
|
|
|
|
|
- }}
|
|
|
|
|
- proOptions={{ hideAttribution: true }}
|
|
|
|
|
- onNodeClick={(event, clickedNode) => {
|
|
|
|
|
- setSelectedNodeId(clickedNode.id);
|
|
|
|
|
- }}
|
|
|
|
|
- >
|
|
|
|
|
- <Controls style={{ bottom: '20px', left: 'auto', right: '20px' }} />
|
|
|
|
|
- <Background variant="dots" gap={20} size={1} color="#e5e7eb" />
|
|
|
|
|
- </ReactFlow>
|
|
|
|
|
- </div>
|
|
|
|
|
- </div>
|
|
|
|
|
- </div>
|
|
|
|
|
- );
|
|
|
|
|
-}
|
|
|
|
|
-
|
|
|
|
|
-function App() {
|
|
|
|
|
- return (
|
|
|
|
|
- <ReactFlowProvider>
|
|
|
|
|
- <FlowContent />
|
|
|
|
|
- </ReactFlowProvider>
|
|
|
|
|
- );
|
|
|
|
|
-}
|
|
|
|
|
-
|
|
|
|
|
-const root = createRoot(document.getElementById('root'));
|
|
|
|
|
-root.render(<App />);
|
|
|
|
|
-`;
|
|
|
|
|
-
|
|
|
|
|
-fs.writeFileSync(reactComponentPath, reactComponent);
|
|
|
|
|
-
|
|
|
|
|
-// 使用 esbuild 打包
|
|
|
|
|
-console.log('🎨 Building modern visualization...');
|
|
|
|
|
-
|
|
|
|
|
-build({
|
|
|
|
|
- entryPoints: [reactComponentPath],
|
|
|
|
|
- bundle: true,
|
|
|
|
|
- outfile: path.join(__dirname, 'bundle_v2.js'),
|
|
|
|
|
- format: 'iife',
|
|
|
|
|
- loader: {
|
|
|
|
|
- '.css': 'css',
|
|
|
|
|
- },
|
|
|
|
|
- minify: false,
|
|
|
|
|
- sourcemap: 'inline',
|
|
|
|
|
-}).then(() => {
|
|
|
|
|
- // 读取打包后的 JS
|
|
|
|
|
- const bundleJs = fs.readFileSync(path.join(__dirname, 'bundle_v2.js'), 'utf-8');
|
|
|
|
|
-
|
|
|
|
|
- // 读取 CSS
|
|
|
|
|
- const cssPath = path.join(__dirname, 'node_modules/@xyflow/react/dist/style.css');
|
|
|
|
|
- const css = fs.readFileSync(cssPath, 'utf-8');
|
|
|
|
|
-
|
|
|
|
|
- // 生成最终 HTML
|
|
|
|
|
- const html = `<!DOCTYPE html>
|
|
|
|
|
-<html lang="zh-CN">
|
|
|
|
|
-<head>
|
|
|
|
|
- <meta charset="UTF-8">
|
|
|
|
|
- <meta name="viewport" content="width=device-width, initial-scale=1.0">
|
|
|
|
|
- <title>查询图可视化</title>
|
|
|
|
|
- <link rel="preconnect" href="https://fonts.googleapis.com">
|
|
|
|
|
- <link rel="preconnect" href="https://fonts.gstatic.com" crossorigin>
|
|
|
|
|
- <link href="https://fonts.googleapis.com/css2?family=Inter:wght@400;500;600&display=swap" rel="stylesheet">
|
|
|
|
|
- <script src="https://unpkg.com/dagre@0.8.5/dist/dagre.min.js"></script>
|
|
|
|
|
- <script>
|
|
|
|
|
- // 过滤特定的 React 警告
|
|
|
|
|
- const originalError = console.error;
|
|
|
|
|
- console.error = (...args) => {
|
|
|
|
|
- if (typeof args[0] === 'string' && args[0].includes('Each child in a list should have a unique "key" prop')) {
|
|
|
|
|
- return;
|
|
|
|
|
- }
|
|
|
|
|
- originalError.apply(console, args);
|
|
|
|
|
- };
|
|
|
|
|
- </script>
|
|
|
|
|
- <style>
|
|
|
|
|
- * {
|
|
|
|
|
- margin: 0;
|
|
|
|
|
- padding: 0;
|
|
|
|
|
- box-sizing: border-box;
|
|
|
|
|
- }
|
|
|
|
|
- body {
|
|
|
|
|
- font-family: 'Inter', -apple-system, BlinkMacSystemFont, 'Segoe UI', sans-serif;
|
|
|
|
|
- overflow: hidden;
|
|
|
|
|
- -webkit-font-smoothing: antialiased;
|
|
|
|
|
- -moz-osx-font-smoothing: grayscale;
|
|
|
|
|
- }
|
|
|
|
|
- #root {
|
|
|
|
|
- width: 100vw;
|
|
|
|
|
- height: 100vh;
|
|
|
|
|
- }
|
|
|
|
|
- ${css}
|
|
|
|
|
-
|
|
|
|
|
- /* 自定义样式覆盖 */
|
|
|
|
|
- .react-flow__edge-path {
|
|
|
|
|
- stroke-linecap: round;
|
|
|
|
|
- }
|
|
|
|
|
- .react-flow__controls {
|
|
|
|
|
- box-shadow: 0 4px 12px rgba(0, 0, 0, 0.1);
|
|
|
|
|
- border: 1px solid #e5e7eb;
|
|
|
|
|
- border-radius: 8px;
|
|
|
|
|
- }
|
|
|
|
|
- .react-flow__controls-button {
|
|
|
|
|
- border: none;
|
|
|
|
|
- border-bottom: 1px solid #e5e7eb;
|
|
|
|
|
- }
|
|
|
|
|
- .react-flow__controls-button:hover {
|
|
|
|
|
- background: #f9fafb;
|
|
|
|
|
- }
|
|
|
|
|
- </style>
|
|
|
|
|
-</head>
|
|
|
|
|
-<body>
|
|
|
|
|
- <div id="root"></div>
|
|
|
|
|
- <script>${bundleJs}</script>
|
|
|
|
|
-</body>
|
|
|
|
|
-</html>`;
|
|
|
|
|
-
|
|
|
|
|
- // 写入输出文件
|
|
|
|
|
- fs.writeFileSync(outputFile, html);
|
|
|
|
|
-
|
|
|
|
|
- // 清理临时文件
|
|
|
|
|
- fs.unlinkSync(reactComponentPath);
|
|
|
|
|
- fs.unlinkSync(path.join(__dirname, 'bundle_v2.js'));
|
|
|
|
|
-
|
|
|
|
|
- console.log('✅ Visualization generated: ' + outputFile);
|
|
|
|
|
- console.log('📊 Nodes: ' + Object.keys(data.nodes).length);
|
|
|
|
|
- console.log('🔗 Edges: ' + data.edges.length);
|
|
|
|
|
-}).catch(error => {
|
|
|
|
|
- console.error('❌ Build error:', error);
|
|
|
|
|
- process.exit(1);
|
|
|
|
|
-});
|
|
|