mirror of
https://github.com/jkingsman/Remote-Terminal-for-MeshCore.git
synced 2026-03-28 17:43:05 +01:00
Refactor visualizer
This commit is contained in:
@@ -51,6 +51,12 @@ frontend/src/
|
||||
├── components/
|
||||
│ ├── AppShell.tsx # App-shell layout: status, sidebar, search/settings panes, cracker, modals
|
||||
│ ├── ConversationPane.tsx # Active conversation surface selection (map/raw/repeater/chat/empty)
|
||||
│ ├── visualizer/
|
||||
│ │ ├── useVisualizerData3D.ts # Packet→graph data pipeline, repeat aggregation, simulation state
|
||||
│ │ ├── useVisualizer3DScene.ts # Three.js scene lifecycle, buffers, hover/pin interaction
|
||||
│ │ ├── VisualizerControls.tsx # Visualizer legends and control panel overlay
|
||||
│ │ ├── VisualizerTooltip.tsx # Hover/pin node detail overlay
|
||||
│ │ └── shared.ts # Graph node/link types and shared rendering helpers
|
||||
│ └── ...
|
||||
├── utils/
|
||||
│ ├── urlHash.ts # Hash parsing and encoding
|
||||
@@ -216,6 +222,7 @@ High-level state is delegated to hooks:
|
||||
### Visualizer behavior
|
||||
|
||||
- `VisualizerView.tsx` hosts `PacketVisualizer3D.tsx` (desktop split-pane and mobile tabs).
|
||||
- `PacketVisualizer3D.tsx` is now a thin composition shell over visualizer-specific hooks/components in `components/visualizer/`.
|
||||
- `PacketVisualizer3D` uses persistent Three.js geometries for links/highlights/particles and updates typed-array buffers in-place per frame.
|
||||
- Packet repeat aggregation keys prefer decoder `messageHash` (path-insensitive), with hash fallback for malformed packets.
|
||||
- Raw-packet decoding in `RawPacketList.tsx` and `visualizerUtils.ts` relies on the multibyte-aware decoder fork; keep frontend packet parsing aligned with backend `path_utils.py`.
|
||||
|
||||
@@ -12,7 +12,7 @@ The visualizer displays:
|
||||
|
||||
## Architecture
|
||||
|
||||
### Data Layer (`useVisualizerData3D` hook)
|
||||
### Data Layer (`components/visualizer/useVisualizerData3D.ts`)
|
||||
|
||||
The custom hook manages all graph state and simulation logic:
|
||||
|
||||
@@ -39,6 +39,8 @@ Packets → Parse → Aggregate by key → Observation window → Publish → An
|
||||
|
||||
### Rendering Layer (Three.js)
|
||||
|
||||
Scene creation, render-loop updates, raycasting hover, and click-to-pin interaction live in `components/visualizer/useVisualizer3DScene.ts`.
|
||||
|
||||
- `THREE.WebGLRenderer` + `CSS2DRenderer` (text labels overlaid on 3D scene)
|
||||
- `OrbitControls` for camera interaction (orbit, pan, zoom)
|
||||
- `THREE.Mesh` with `SphereGeometry` per node + `CSS2DObject` labels
|
||||
@@ -46,15 +48,20 @@ Packets → Parse → Aggregate by key → Observation window → Publish → An
|
||||
- `THREE.Points` with vertex colors for particles (persistent geometry + circular sprite texture)
|
||||
- `THREE.Raycaster` for hover/click detection on node spheres
|
||||
|
||||
### Shared Utilities (`utils/visualizerUtils.ts`)
|
||||
### Shared Utilities
|
||||
|
||||
Types, constants, and pure functions shared across the codebase:
|
||||
- `components/visualizer/shared.ts`
|
||||
- Graph-specific types: `GraphNode`, `GraphLink`, `NodeMeshData`
|
||||
- Shared rendering helpers: node colors, relative-time formatting, typed-array growth helpers
|
||||
- `utils/visualizerUtils.ts`
|
||||
- Packet parsing, identity helpers, ambiguous repeater heuristics, constants shared across visualizer code
|
||||
|
||||
- Types: `NodeType`, `PacketLabel`, `Particle`, `ObservedPath`, `PendingPacket`, `ParsedPacket`, `TrafficObservation`, `RepeaterTrafficData`, `RepeaterSplitAnalysis`
|
||||
- Constants: `COLORS`, `PARTICLE_COLOR_MAP`, `PARTICLE_SPEED`, `DEFAULT_OBSERVATION_WINDOW_SEC`, traffic thresholds, `PACKET_LEGEND_ITEMS`
|
||||
- Functions: `hashString` (from `utils/contactAvatar.ts`), `parsePacket`, `getPacketLabel`, `generatePacketKey`, `getLinkId`, `getNodeType`, `dedupeConsecutive`, `analyzeRepeaterTraffic`, `recordTrafficObservation`
|
||||
### UI Overlays
|
||||
|
||||
`GraphNode` and `GraphLink` are defined locally in the component — they extend `SimulationNodeDatum3D` and `SimulationLinkDatum` from `d3-force-3d`.
|
||||
- `components/visualizer/VisualizerControls.tsx`
|
||||
- Legends, settings toggles, repulsion/speed controls, reset/stretch actions
|
||||
- `components/visualizer/VisualizerTooltip.tsx`
|
||||
- Hovered/pinned node metadata and neighbor list
|
||||
|
||||
### Type Declarations (`types/d3-force-3d.d.ts`)
|
||||
|
||||
|
||||
File diff suppressed because it is too large
Load Diff
317
frontend/src/components/visualizer/VisualizerControls.tsx
Normal file
317
frontend/src/components/visualizer/VisualizerControls.tsx
Normal file
@@ -0,0 +1,317 @@
|
||||
import { Checkbox } from '../ui/checkbox';
|
||||
import { PACKET_LEGEND_ITEMS } from '../../utils/visualizerUtils';
|
||||
import { NODE_LEGEND_ITEMS } from './shared';
|
||||
|
||||
interface VisualizerControlsProps {
|
||||
showControls: boolean;
|
||||
setShowControls: (value: boolean) => void;
|
||||
fullScreen?: boolean;
|
||||
onFullScreenChange?: (fullScreen: boolean) => void;
|
||||
showAmbiguousPaths: boolean;
|
||||
setShowAmbiguousPaths: (value: boolean) => void;
|
||||
showAmbiguousNodes: boolean;
|
||||
setShowAmbiguousNodes: (value: boolean) => void;
|
||||
useAdvertPathHints: boolean;
|
||||
setUseAdvertPathHints: (value: boolean) => void;
|
||||
splitAmbiguousByTraffic: boolean;
|
||||
setSplitAmbiguousByTraffic: (value: boolean) => void;
|
||||
observationWindowSec: number;
|
||||
setObservationWindowSec: (value: number) => void;
|
||||
pruneStaleNodes: boolean;
|
||||
setPruneStaleNodes: (value: boolean) => void;
|
||||
pruneStaleMinutes: number;
|
||||
setPruneStaleMinutes: (value: number) => void;
|
||||
letEmDrift: boolean;
|
||||
setLetEmDrift: (value: boolean) => void;
|
||||
autoOrbit: boolean;
|
||||
setAutoOrbit: (value: boolean) => void;
|
||||
chargeStrength: number;
|
||||
setChargeStrength: (value: number) => void;
|
||||
particleSpeedMultiplier: number;
|
||||
setParticleSpeedMultiplier: (value: number) => void;
|
||||
nodeCount: number;
|
||||
linkCount: number;
|
||||
onExpandContract: () => void;
|
||||
onClearAndReset: () => void;
|
||||
}
|
||||
|
||||
export function VisualizerControls({
|
||||
showControls,
|
||||
setShowControls,
|
||||
fullScreen,
|
||||
onFullScreenChange,
|
||||
showAmbiguousPaths,
|
||||
setShowAmbiguousPaths,
|
||||
showAmbiguousNodes,
|
||||
setShowAmbiguousNodes,
|
||||
useAdvertPathHints,
|
||||
setUseAdvertPathHints,
|
||||
splitAmbiguousByTraffic,
|
||||
setSplitAmbiguousByTraffic,
|
||||
observationWindowSec,
|
||||
setObservationWindowSec,
|
||||
pruneStaleNodes,
|
||||
setPruneStaleNodes,
|
||||
pruneStaleMinutes,
|
||||
setPruneStaleMinutes,
|
||||
letEmDrift,
|
||||
setLetEmDrift,
|
||||
autoOrbit,
|
||||
setAutoOrbit,
|
||||
chargeStrength,
|
||||
setChargeStrength,
|
||||
particleSpeedMultiplier,
|
||||
setParticleSpeedMultiplier,
|
||||
nodeCount,
|
||||
linkCount,
|
||||
onExpandContract,
|
||||
onClearAndReset,
|
||||
}: VisualizerControlsProps) {
|
||||
return (
|
||||
<>
|
||||
{showControls && (
|
||||
<div className="absolute bottom-4 left-4 bg-background/80 backdrop-blur-sm rounded-lg p-3 text-xs border border-border z-10">
|
||||
<div className="flex gap-6">
|
||||
<div className="flex flex-col gap-1.5">
|
||||
<div className="text-muted-foreground font-medium mb-1">Packets</div>
|
||||
{PACKET_LEGEND_ITEMS.map((item) => (
|
||||
<div key={item.label} className="flex items-center gap-2">
|
||||
<div
|
||||
className="w-5 h-5 rounded-full flex items-center justify-center text-[8px] font-bold text-white"
|
||||
style={{ backgroundColor: item.color }}
|
||||
>
|
||||
{item.label}
|
||||
</div>
|
||||
<span>{item.description}</span>
|
||||
</div>
|
||||
))}
|
||||
</div>
|
||||
<div className="flex flex-col gap-1.5">
|
||||
<div className="text-muted-foreground font-medium mb-1">Nodes</div>
|
||||
{NODE_LEGEND_ITEMS.map((item) => (
|
||||
<div key={item.label} className="flex items-center gap-2">
|
||||
<div
|
||||
className="rounded-full"
|
||||
style={{
|
||||
width: item.size,
|
||||
height: item.size,
|
||||
backgroundColor: item.color,
|
||||
}}
|
||||
/>
|
||||
<span>{item.label}</span>
|
||||
</div>
|
||||
))}
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
)}
|
||||
|
||||
<div
|
||||
className={`absolute top-4 left-4 bg-background/80 backdrop-blur-sm rounded-lg p-3 text-xs border border-border z-10 transition-opacity ${!showControls ? 'opacity-40 hover:opacity-100' : ''}`}
|
||||
>
|
||||
<div className="flex flex-col gap-2">
|
||||
<div className="flex flex-col gap-2">
|
||||
<label className="flex items-center gap-2 cursor-pointer">
|
||||
<Checkbox
|
||||
checked={showControls}
|
||||
onCheckedChange={(c) => setShowControls(c === true)}
|
||||
/>
|
||||
<span title="Toggle legends and controls visibility">Show controls</span>
|
||||
</label>
|
||||
{onFullScreenChange && (
|
||||
<label className="flex items-center gap-2 cursor-pointer">
|
||||
<Checkbox
|
||||
checked={!fullScreen}
|
||||
onCheckedChange={(c) => onFullScreenChange(c !== true)}
|
||||
/>
|
||||
<span title="Show or hide the packet feed sidebar">Show packet feed sidebar</span>
|
||||
</label>
|
||||
)}
|
||||
</div>
|
||||
{showControls && (
|
||||
<>
|
||||
<div className="border-t border-border pt-2 mt-1 flex flex-col gap-2">
|
||||
<label className="flex items-center gap-2 cursor-pointer">
|
||||
<Checkbox
|
||||
checked={showAmbiguousPaths}
|
||||
onCheckedChange={(c) => setShowAmbiguousPaths(c === true)}
|
||||
/>
|
||||
<span title="Show placeholder nodes for repeaters when the 1-byte prefix matches multiple contacts">
|
||||
Show ambiguous repeaters
|
||||
</span>
|
||||
</label>
|
||||
<label className="flex items-center gap-2 cursor-pointer">
|
||||
<Checkbox
|
||||
checked={showAmbiguousNodes}
|
||||
onCheckedChange={(c) => setShowAmbiguousNodes(c === true)}
|
||||
/>
|
||||
<span title="Show placeholder nodes for senders/recipients when only a 1-byte prefix is known">
|
||||
Show ambiguous sender/recipient
|
||||
</span>
|
||||
</label>
|
||||
<label className="flex items-center gap-2 cursor-pointer">
|
||||
<Checkbox
|
||||
checked={useAdvertPathHints}
|
||||
onCheckedChange={(c) => setUseAdvertPathHints(c === true)}
|
||||
disabled={!showAmbiguousPaths}
|
||||
/>
|
||||
<span
|
||||
title="Use stored repeater advert paths to assign likely identity labels for ambiguous repeater nodes"
|
||||
className={!showAmbiguousPaths ? 'text-muted-foreground' : ''}
|
||||
>
|
||||
Use repeater advert-path identity hints
|
||||
</span>
|
||||
</label>
|
||||
<label className="flex items-center gap-2 cursor-pointer">
|
||||
<Checkbox
|
||||
checked={splitAmbiguousByTraffic}
|
||||
onCheckedChange={(c) => setSplitAmbiguousByTraffic(c === true)}
|
||||
disabled={!showAmbiguousPaths}
|
||||
/>
|
||||
<span
|
||||
title="Split ambiguous repeaters into separate nodes based on traffic patterns (prev→next). Helps identify colliding prefixes representing different physical nodes, but requires enough traffic to disambiguate."
|
||||
className={!showAmbiguousPaths ? 'text-muted-foreground' : ''}
|
||||
>
|
||||
Heuristically group repeaters by traffic pattern
|
||||
</span>
|
||||
</label>
|
||||
<div className="flex items-center gap-2">
|
||||
<label
|
||||
htmlFor="observation-window-3d"
|
||||
className="text-muted-foreground"
|
||||
title="How long to wait for duplicate packets via different paths before animating"
|
||||
>
|
||||
Ack/echo listen window:
|
||||
</label>
|
||||
<input
|
||||
id="observation-window-3d"
|
||||
type="number"
|
||||
min="1"
|
||||
max="60"
|
||||
value={observationWindowSec}
|
||||
onChange={(e) =>
|
||||
setObservationWindowSec(
|
||||
Math.max(1, Math.min(60, parseInt(e.target.value, 10) || 1))
|
||||
)
|
||||
}
|
||||
className="w-12 px-1 py-0.5 bg-background border border-border rounded text-xs text-center"
|
||||
/>
|
||||
<span className="text-muted-foreground">sec</span>
|
||||
</div>
|
||||
<div className="border-t border-border pt-2 mt-1 flex flex-col gap-2">
|
||||
<label className="flex items-center gap-2 cursor-pointer">
|
||||
<Checkbox
|
||||
checked={pruneStaleNodes}
|
||||
onCheckedChange={(c) => setPruneStaleNodes(c === true)}
|
||||
/>
|
||||
<span title="Automatically remove nodes with no traffic within the configured window to keep the mesh manageable">
|
||||
Only show recently heard/in-a-path nodes
|
||||
</span>
|
||||
</label>
|
||||
{pruneStaleNodes && (
|
||||
<div className="flex items-center gap-2 pl-6">
|
||||
<label
|
||||
htmlFor="prune-window"
|
||||
className="text-muted-foreground whitespace-nowrap"
|
||||
>
|
||||
Window:
|
||||
</label>
|
||||
<input
|
||||
id="prune-window"
|
||||
type="number"
|
||||
min={1}
|
||||
max={60}
|
||||
value={pruneStaleMinutes}
|
||||
onChange={(e) => {
|
||||
const v = parseInt(e.target.value, 10);
|
||||
if (!isNaN(v) && v >= 1 && v <= 60) setPruneStaleMinutes(v);
|
||||
}}
|
||||
className="w-14 rounded border border-border bg-background px-2 py-0.5 text-sm"
|
||||
/>
|
||||
<span className="text-muted-foreground" aria-hidden="true">
|
||||
min
|
||||
</span>
|
||||
</div>
|
||||
)}
|
||||
<label className="flex items-center gap-2 cursor-pointer">
|
||||
<Checkbox
|
||||
checked={letEmDrift}
|
||||
onCheckedChange={(c) => setLetEmDrift(c === true)}
|
||||
/>
|
||||
<span title="When enabled, the graph continuously reorganizes itself into a better layout">
|
||||
Let 'em drift
|
||||
</span>
|
||||
</label>
|
||||
<label className="flex items-center gap-2 cursor-pointer">
|
||||
<Checkbox
|
||||
checked={autoOrbit}
|
||||
onCheckedChange={(c) => setAutoOrbit(c === true)}
|
||||
/>
|
||||
<span title="Automatically orbit the camera around the scene">
|
||||
Orbit the mesh
|
||||
</span>
|
||||
</label>
|
||||
<div className="flex flex-col gap-1 mt-1">
|
||||
<label
|
||||
htmlFor="viz-repulsion"
|
||||
className="text-muted-foreground"
|
||||
title="How strongly nodes repel each other. Higher values spread nodes out more."
|
||||
>
|
||||
Repulsion: {Math.abs(chargeStrength)}
|
||||
</label>
|
||||
<input
|
||||
id="viz-repulsion"
|
||||
type="range"
|
||||
min="50"
|
||||
max="2500"
|
||||
value={Math.abs(chargeStrength)}
|
||||
onChange={(e) => setChargeStrength(-parseInt(e.target.value, 10))}
|
||||
className="w-full h-2 bg-border rounded-lg appearance-none cursor-pointer accent-primary"
|
||||
/>
|
||||
</div>
|
||||
<div className="flex flex-col gap-1 mt-1">
|
||||
<label
|
||||
htmlFor="viz-packet-speed"
|
||||
className="text-muted-foreground"
|
||||
title="How fast particles travel along links. Higher values make packets move faster."
|
||||
>
|
||||
Packet speed: {particleSpeedMultiplier}x
|
||||
</label>
|
||||
<input
|
||||
id="viz-packet-speed"
|
||||
type="range"
|
||||
min="1"
|
||||
max="5"
|
||||
step="0.5"
|
||||
value={particleSpeedMultiplier}
|
||||
onChange={(e) => setParticleSpeedMultiplier(parseFloat(e.target.value))}
|
||||
className="w-full h-2 bg-border rounded-lg appearance-none cursor-pointer accent-primary"
|
||||
/>
|
||||
</div>
|
||||
</div>
|
||||
<button
|
||||
onClick={onExpandContract}
|
||||
className="mt-1 px-3 py-1.5 bg-primary/20 hover:bg-primary/30 text-primary rounded text-xs transition-colors"
|
||||
title="Expand nodes apart then contract back - can help untangle the graph"
|
||||
>
|
||||
Oooh Big Stretch!
|
||||
</button>
|
||||
<button
|
||||
onClick={onClearAndReset}
|
||||
className="mt-1 px-3 py-1.5 bg-yellow-500/20 hover:bg-yellow-500/30 text-yellow-500 rounded text-xs transition-colors"
|
||||
title="Clear all nodes and links from the visualization - packets are preserved"
|
||||
>
|
||||
Clear & Reset
|
||||
</button>
|
||||
</div>
|
||||
<div className="border-t border-border pt-2 mt-1">
|
||||
<div>Nodes: {nodeCount}</div>
|
||||
<div>Links: {linkCount}</div>
|
||||
</div>
|
||||
</>
|
||||
)}
|
||||
</div>
|
||||
</div>
|
||||
</>
|
||||
);
|
||||
}
|
||||
73
frontend/src/components/visualizer/VisualizerTooltip.tsx
Normal file
73
frontend/src/components/visualizer/VisualizerTooltip.tsx
Normal file
@@ -0,0 +1,73 @@
|
||||
import type { GraphNode } from './shared';
|
||||
import { formatRelativeTime } from './shared';
|
||||
|
||||
interface VisualizerTooltipProps {
|
||||
activeNodeId: string | null;
|
||||
nodes: Map<string, GraphNode>;
|
||||
neighborIds: string[];
|
||||
}
|
||||
|
||||
export function VisualizerTooltip({ activeNodeId, nodes, neighborIds }: VisualizerTooltipProps) {
|
||||
if (!activeNodeId) return null;
|
||||
|
||||
const node = nodes.get(activeNodeId);
|
||||
if (!node) return null;
|
||||
|
||||
const neighbors = neighborIds
|
||||
.map((nid) => {
|
||||
const neighbor = nodes.get(nid);
|
||||
if (!neighbor) return null;
|
||||
const displayName =
|
||||
neighbor.name || (neighbor.type === 'self' ? 'Me' : neighbor.id.slice(0, 8));
|
||||
return { id: nid, name: displayName, ambiguousNames: neighbor.ambiguousNames };
|
||||
})
|
||||
.filter((neighbor): neighbor is NonNullable<typeof neighbor> => neighbor !== null);
|
||||
|
||||
return (
|
||||
<div className="absolute top-4 right-4 bg-background/90 backdrop-blur-sm rounded-lg p-3 text-xs border border-border z-10 max-w-72 max-h-[calc(100%-2rem)] overflow-y-auto">
|
||||
<div className="flex flex-col gap-1">
|
||||
<div className="font-medium">
|
||||
{node.name || (node.type === 'self' ? 'Me' : node.id.slice(0, 8))}
|
||||
</div>
|
||||
<div className="text-muted-foreground">ID: {node.id}</div>
|
||||
<div className="text-muted-foreground">
|
||||
Type: {node.type}
|
||||
{node.isAmbiguous ? ' (ambiguous)' : ''}
|
||||
</div>
|
||||
{node.probableIdentity && (
|
||||
<div className="text-muted-foreground">Probably: {node.probableIdentity}</div>
|
||||
)}
|
||||
{node.ambiguousNames && node.ambiguousNames.length > 0 && (
|
||||
<div className="text-muted-foreground">
|
||||
{node.probableIdentity ? 'Other possible: ' : 'Possible: '}
|
||||
{node.ambiguousNames.join(', ')}
|
||||
</div>
|
||||
)}
|
||||
{node.type !== 'self' && (
|
||||
<div className="text-muted-foreground border-t border-border pt-1 mt-1">
|
||||
<div>Last active: {formatRelativeTime(node.lastActivity)}</div>
|
||||
{node.lastActivityReason && <div>Reason: {node.lastActivityReason}</div>}
|
||||
</div>
|
||||
)}
|
||||
{neighbors.length > 0 && (
|
||||
<div className="text-muted-foreground border-t border-border pt-1 mt-1">
|
||||
<div className="mb-0.5">Traffic exchanged with:</div>
|
||||
<ul className="pl-3 flex flex-col gap-0.5">
|
||||
{neighbors.map((neighbor) => (
|
||||
<li key={neighbor.id}>
|
||||
{neighbor.name}
|
||||
{neighbor.ambiguousNames && neighbor.ambiguousNames.length > 0 && (
|
||||
<span className="text-muted-foreground/60">
|
||||
{' '}
|
||||
({neighbor.ambiguousNames.join(', ')})
|
||||
</span>
|
||||
)}
|
||||
</li>
|
||||
))}
|
||||
</ul>
|
||||
</div>
|
||||
)}
|
||||
</div>
|
||||
</div>
|
||||
);
|
||||
}
|
||||
83
frontend/src/components/visualizer/shared.ts
Normal file
83
frontend/src/components/visualizer/shared.ts
Normal file
@@ -0,0 +1,83 @@
|
||||
import * as THREE from 'three';
|
||||
import type { SimulationLinkDatum } from 'd3-force';
|
||||
import type { SimulationNodeDatum3D } from 'd3-force-3d';
|
||||
import type { CSS2DObject } from 'three/examples/jsm/renderers/CSS2DRenderer.js';
|
||||
|
||||
import type { NodeType } from '../../utils/visualizerUtils';
|
||||
|
||||
export interface GraphNode extends SimulationNodeDatum3D {
|
||||
id: string;
|
||||
name: string | null;
|
||||
type: NodeType;
|
||||
isAmbiguous: boolean;
|
||||
lastActivity: number;
|
||||
lastActivityReason?: string;
|
||||
lastSeen?: number | null;
|
||||
probableIdentity?: string | null;
|
||||
ambiguousNames?: string[];
|
||||
}
|
||||
|
||||
export interface GraphLink extends SimulationLinkDatum<GraphNode> {
|
||||
source: string | GraphNode;
|
||||
target: string | GraphNode;
|
||||
lastActivity: number;
|
||||
}
|
||||
|
||||
export interface NodeMeshData {
|
||||
mesh: THREE.Mesh;
|
||||
label: CSS2DObject;
|
||||
labelDiv: HTMLDivElement;
|
||||
}
|
||||
|
||||
export const NODE_COLORS = {
|
||||
self: 0x22c55e,
|
||||
repeater: 0x3b82f6,
|
||||
client: 0xffffff,
|
||||
ambiguous: 0x9ca3af,
|
||||
} as const;
|
||||
|
||||
export const NODE_LEGEND_ITEMS = [
|
||||
{ color: '#22c55e', label: 'You', size: 14 },
|
||||
{ color: '#3b82f6', label: 'Repeater', size: 10 },
|
||||
{ color: '#ffffff', label: 'Node', size: 10 },
|
||||
{ color: '#9ca3af', label: 'Ambiguous', size: 10 },
|
||||
] as const;
|
||||
|
||||
export function getBaseNodeColor(node: Pick<GraphNode, 'type' | 'isAmbiguous'>): number {
|
||||
if (node.type === 'self') return NODE_COLORS.self;
|
||||
if (node.type === 'repeater') return NODE_COLORS.repeater;
|
||||
return node.isAmbiguous ? NODE_COLORS.ambiguous : NODE_COLORS.client;
|
||||
}
|
||||
|
||||
export function growFloat32Buffer(current: Float32Array, requiredLength: number): Float32Array {
|
||||
let nextLength = Math.max(12, current.length);
|
||||
while (nextLength < requiredLength) {
|
||||
nextLength *= 2;
|
||||
}
|
||||
return new Float32Array(nextLength);
|
||||
}
|
||||
|
||||
export function arraysEqual(a: string[], b: string[]): boolean {
|
||||
if (a.length !== b.length) return false;
|
||||
for (let i = 0; i < a.length; i++) {
|
||||
if (a[i] !== b[i]) return false;
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
export function formatRelativeTime(timestamp: number): string {
|
||||
const seconds = Math.floor((Date.now() - timestamp) / 1000);
|
||||
if (seconds < 5) return 'just now';
|
||||
if (seconds < 60) return `${seconds}s ago`;
|
||||
const minutes = Math.floor(seconds / 60);
|
||||
const secs = seconds % 60;
|
||||
return secs > 0 ? `${minutes}m ${secs}s ago` : `${minutes}m ago`;
|
||||
}
|
||||
|
||||
export function normalizePacketTimestampMs(timestamp: number | null | undefined): number {
|
||||
if (!Number.isFinite(timestamp) || !timestamp || timestamp <= 0) {
|
||||
return Date.now();
|
||||
}
|
||||
const ts = Number(timestamp);
|
||||
return ts > 1_000_000_000_000 ? ts : ts * 1000;
|
||||
}
|
||||
578
frontend/src/components/visualizer/useVisualizer3DScene.ts
Normal file
578
frontend/src/components/visualizer/useVisualizer3DScene.ts
Normal file
@@ -0,0 +1,578 @@
|
||||
import { useEffect, useRef, useState, type RefObject } from 'react';
|
||||
import * as THREE from 'three';
|
||||
import { OrbitControls } from 'three/examples/jsm/controls/OrbitControls.js';
|
||||
import { CSS2DObject, CSS2DRenderer } from 'three/examples/jsm/renderers/CSS2DRenderer.js';
|
||||
|
||||
import { COLORS, getLinkId } from '../../utils/visualizerUtils';
|
||||
import type { VisualizerData3D } from './useVisualizerData3D';
|
||||
import { arraysEqual, getBaseNodeColor, growFloat32Buffer, type NodeMeshData } from './shared';
|
||||
|
||||
interface UseVisualizer3DSceneArgs {
|
||||
containerRef: RefObject<HTMLDivElement | null>;
|
||||
data: VisualizerData3D;
|
||||
autoOrbit: boolean;
|
||||
}
|
||||
|
||||
interface UseVisualizer3DSceneResult {
|
||||
hoveredNodeId: string | null;
|
||||
hoveredNeighborIds: string[];
|
||||
pinnedNodeId: string | null;
|
||||
}
|
||||
|
||||
export function useVisualizer3DScene({
|
||||
containerRef,
|
||||
data,
|
||||
autoOrbit,
|
||||
}: UseVisualizer3DSceneArgs): UseVisualizer3DSceneResult {
|
||||
const rendererRef = useRef<THREE.WebGLRenderer | null>(null);
|
||||
const cssRendererRef = useRef<CSS2DRenderer | null>(null);
|
||||
const sceneRef = useRef<THREE.Scene | null>(null);
|
||||
const cameraRef = useRef<THREE.PerspectiveCamera | null>(null);
|
||||
const controlsRef = useRef<OrbitControls | null>(null);
|
||||
const nodeMeshesRef = useRef<Map<string, NodeMeshData>>(new Map());
|
||||
const raycastTargetsRef = useRef<THREE.Mesh[]>([]);
|
||||
const linkLineRef = useRef<THREE.LineSegments | null>(null);
|
||||
const highlightLineRef = useRef<THREE.LineSegments | null>(null);
|
||||
const particlePointsRef = useRef<THREE.Points | null>(null);
|
||||
const particleTextureRef = useRef<THREE.Texture | null>(null);
|
||||
const linkPositionBufferRef = useRef<Float32Array>(new Float32Array(0));
|
||||
const highlightPositionBufferRef = useRef<Float32Array>(new Float32Array(0));
|
||||
const particlePositionBufferRef = useRef<Float32Array>(new Float32Array(0));
|
||||
const particleColorBufferRef = useRef<Float32Array>(new Float32Array(0));
|
||||
const raycasterRef = useRef(new THREE.Raycaster());
|
||||
const mouseRef = useRef(new THREE.Vector2());
|
||||
const dataRef = useRef(data);
|
||||
|
||||
const [hoveredNodeId, setHoveredNodeId] = useState<string | null>(null);
|
||||
const hoveredNodeIdRef = useRef<string | null>(null);
|
||||
const [hoveredNeighborIds, setHoveredNeighborIds] = useState<string[]>([]);
|
||||
const hoveredNeighborIdsRef = useRef<string[]>([]);
|
||||
const pinnedNodeIdRef = useRef<string | null>(null);
|
||||
const [pinnedNodeId, setPinnedNodeId] = useState<string | null>(null);
|
||||
|
||||
useEffect(() => {
|
||||
dataRef.current = data;
|
||||
}, [data]);
|
||||
|
||||
useEffect(() => {
|
||||
const container = containerRef.current;
|
||||
if (!container) return;
|
||||
|
||||
const scene = new THREE.Scene();
|
||||
scene.background = new THREE.Color(COLORS.background);
|
||||
sceneRef.current = scene;
|
||||
|
||||
const camera = new THREE.PerspectiveCamera(60, 1, 1, 5000);
|
||||
camera.position.set(0, 0, 400);
|
||||
cameraRef.current = camera;
|
||||
|
||||
const renderer = new THREE.WebGLRenderer({ antialias: true });
|
||||
renderer.setPixelRatio(window.devicePixelRatio);
|
||||
container.appendChild(renderer.domElement);
|
||||
rendererRef.current = renderer;
|
||||
|
||||
const texSize = 64;
|
||||
const texCanvas = document.createElement('canvas');
|
||||
texCanvas.width = texSize;
|
||||
texCanvas.height = texSize;
|
||||
const texCtx = texCanvas.getContext('2d');
|
||||
if (!texCtx) {
|
||||
renderer.dispose();
|
||||
if (renderer.domElement.parentNode) {
|
||||
renderer.domElement.parentNode.removeChild(renderer.domElement);
|
||||
}
|
||||
return;
|
||||
}
|
||||
const gradient = texCtx.createRadialGradient(
|
||||
texSize / 2,
|
||||
texSize / 2,
|
||||
0,
|
||||
texSize / 2,
|
||||
texSize / 2,
|
||||
texSize / 2
|
||||
);
|
||||
gradient.addColorStop(0, 'rgba(255,255,255,1)');
|
||||
gradient.addColorStop(0.5, 'rgba(255,255,255,0.8)');
|
||||
gradient.addColorStop(1, 'rgba(255,255,255,0)');
|
||||
texCtx.fillStyle = gradient;
|
||||
texCtx.fillRect(0, 0, texSize, texSize);
|
||||
const particleTexture = new THREE.CanvasTexture(texCanvas);
|
||||
particleTextureRef.current = particleTexture;
|
||||
|
||||
const cssRenderer = new CSS2DRenderer();
|
||||
cssRenderer.domElement.style.position = 'absolute';
|
||||
cssRenderer.domElement.style.top = '0';
|
||||
cssRenderer.domElement.style.left = '0';
|
||||
cssRenderer.domElement.style.pointerEvents = 'none';
|
||||
cssRenderer.domElement.style.zIndex = '1';
|
||||
container.appendChild(cssRenderer.domElement);
|
||||
cssRendererRef.current = cssRenderer;
|
||||
|
||||
const controls = new OrbitControls(camera, renderer.domElement);
|
||||
controls.enableDamping = true;
|
||||
controls.dampingFactor = 0.1;
|
||||
controls.minDistance = 50;
|
||||
controls.maxDistance = 2000;
|
||||
controlsRef.current = controls;
|
||||
|
||||
const linkGeometry = new THREE.BufferGeometry();
|
||||
const linkMaterial = new THREE.LineBasicMaterial({
|
||||
color: COLORS.link,
|
||||
transparent: true,
|
||||
opacity: 0.6,
|
||||
});
|
||||
const linkSegments = new THREE.LineSegments(linkGeometry, linkMaterial);
|
||||
linkSegments.visible = false;
|
||||
scene.add(linkSegments);
|
||||
linkLineRef.current = linkSegments;
|
||||
|
||||
const highlightGeometry = new THREE.BufferGeometry();
|
||||
const highlightMaterial = new THREE.LineBasicMaterial({
|
||||
color: 0xffd700,
|
||||
transparent: true,
|
||||
opacity: 1,
|
||||
linewidth: 2,
|
||||
});
|
||||
const highlightSegments = new THREE.LineSegments(highlightGeometry, highlightMaterial);
|
||||
highlightSegments.visible = false;
|
||||
scene.add(highlightSegments);
|
||||
highlightLineRef.current = highlightSegments;
|
||||
|
||||
const particleGeometry = new THREE.BufferGeometry();
|
||||
const particleMaterial = new THREE.PointsMaterial({
|
||||
size: 20,
|
||||
map: particleTexture,
|
||||
vertexColors: true,
|
||||
sizeAttenuation: true,
|
||||
transparent: true,
|
||||
opacity: 0.9,
|
||||
depthWrite: false,
|
||||
});
|
||||
const particlePoints = new THREE.Points(particleGeometry, particleMaterial);
|
||||
particlePoints.visible = false;
|
||||
scene.add(particlePoints);
|
||||
particlePointsRef.current = particlePoints;
|
||||
|
||||
const rect = container.getBoundingClientRect();
|
||||
renderer.setSize(rect.width, rect.height);
|
||||
cssRenderer.setSize(rect.width, rect.height);
|
||||
camera.aspect = rect.width / rect.height;
|
||||
camera.updateProjectionMatrix();
|
||||
|
||||
const observer = new ResizeObserver((entries) => {
|
||||
for (const entry of entries) {
|
||||
const { width, height } = entry.contentRect;
|
||||
if (width === 0 || height === 0) continue;
|
||||
renderer.setSize(width, height);
|
||||
cssRenderer.setSize(width, height);
|
||||
camera.aspect = width / height;
|
||||
camera.updateProjectionMatrix();
|
||||
}
|
||||
});
|
||||
observer.observe(container);
|
||||
|
||||
const nodeMeshes = nodeMeshesRef.current;
|
||||
return () => {
|
||||
observer.disconnect();
|
||||
controls.dispose();
|
||||
renderer.dispose();
|
||||
if (renderer.domElement.parentNode) {
|
||||
renderer.domElement.parentNode.removeChild(renderer.domElement);
|
||||
}
|
||||
if (cssRenderer.domElement.parentNode) {
|
||||
cssRenderer.domElement.parentNode.removeChild(cssRenderer.domElement);
|
||||
}
|
||||
for (const nd of nodeMeshes.values()) {
|
||||
nd.mesh.remove(nd.label);
|
||||
nd.labelDiv.remove();
|
||||
scene.remove(nd.mesh);
|
||||
nd.mesh.geometry.dispose();
|
||||
(nd.mesh.material as THREE.Material).dispose();
|
||||
}
|
||||
nodeMeshes.clear();
|
||||
raycastTargetsRef.current = [];
|
||||
|
||||
if (linkLineRef.current) {
|
||||
scene.remove(linkLineRef.current);
|
||||
linkLineRef.current.geometry.dispose();
|
||||
(linkLineRef.current.material as THREE.Material).dispose();
|
||||
linkLineRef.current = null;
|
||||
}
|
||||
if (highlightLineRef.current) {
|
||||
scene.remove(highlightLineRef.current);
|
||||
highlightLineRef.current.geometry.dispose();
|
||||
(highlightLineRef.current.material as THREE.Material).dispose();
|
||||
highlightLineRef.current = null;
|
||||
}
|
||||
if (particlePointsRef.current) {
|
||||
scene.remove(particlePointsRef.current);
|
||||
particlePointsRef.current.geometry.dispose();
|
||||
(particlePointsRef.current.material as THREE.Material).dispose();
|
||||
particlePointsRef.current = null;
|
||||
}
|
||||
particleTexture.dispose();
|
||||
particleTextureRef.current = null;
|
||||
linkPositionBufferRef.current = new Float32Array(0);
|
||||
highlightPositionBufferRef.current = new Float32Array(0);
|
||||
particlePositionBufferRef.current = new Float32Array(0);
|
||||
particleColorBufferRef.current = new Float32Array(0);
|
||||
sceneRef.current = null;
|
||||
cameraRef.current = null;
|
||||
rendererRef.current = null;
|
||||
cssRendererRef.current = null;
|
||||
controlsRef.current = null;
|
||||
};
|
||||
}, [containerRef]);
|
||||
|
||||
useEffect(() => {
|
||||
const controls = controlsRef.current;
|
||||
if (!controls) return;
|
||||
controls.autoRotate = autoOrbit;
|
||||
controls.autoRotateSpeed = -0.5;
|
||||
}, [autoOrbit]);
|
||||
|
||||
useEffect(() => {
|
||||
const renderer = rendererRef.current;
|
||||
const camera = cameraRef.current;
|
||||
if (!renderer || !camera) return;
|
||||
|
||||
const onMouseMove = (event: MouseEvent) => {
|
||||
const rect = renderer.domElement.getBoundingClientRect();
|
||||
mouseRef.current.x = ((event.clientX - rect.left) / rect.width) * 2 - 1;
|
||||
mouseRef.current.y = -((event.clientY - rect.top) / rect.height) * 2 + 1;
|
||||
};
|
||||
|
||||
let mouseDownPos = { x: 0, y: 0 };
|
||||
|
||||
const onMouseDown = (event: MouseEvent) => {
|
||||
mouseDownPos = { x: event.clientX, y: event.clientY };
|
||||
};
|
||||
|
||||
const onMouseUp = (event: MouseEvent) => {
|
||||
const dx = event.clientX - mouseDownPos.x;
|
||||
const dy = event.clientY - mouseDownPos.y;
|
||||
if (dx * dx + dy * dy > 25) return;
|
||||
|
||||
const rect = renderer.domElement.getBoundingClientRect();
|
||||
const clickMouse = new THREE.Vector2(
|
||||
((event.clientX - rect.left) / rect.width) * 2 - 1,
|
||||
-((event.clientY - rect.top) / rect.height) * 2 + 1
|
||||
);
|
||||
|
||||
const raycaster = raycasterRef.current;
|
||||
raycaster.setFromCamera(clickMouse, camera);
|
||||
const intersects = raycaster.intersectObjects(raycastTargetsRef.current, false);
|
||||
const clickedObject = intersects[0]?.object as THREE.Mesh | undefined;
|
||||
const clickedId = (clickedObject?.userData?.nodeId as string | undefined) ?? null;
|
||||
|
||||
if (clickedId === pinnedNodeIdRef.current) {
|
||||
pinnedNodeIdRef.current = null;
|
||||
setPinnedNodeId(null);
|
||||
} else if (clickedId) {
|
||||
pinnedNodeIdRef.current = clickedId;
|
||||
setPinnedNodeId(clickedId);
|
||||
} else {
|
||||
pinnedNodeIdRef.current = null;
|
||||
setPinnedNodeId(null);
|
||||
}
|
||||
};
|
||||
|
||||
renderer.domElement.addEventListener('mousemove', onMouseMove);
|
||||
renderer.domElement.addEventListener('mousedown', onMouseDown);
|
||||
renderer.domElement.addEventListener('mouseup', onMouseUp);
|
||||
return () => {
|
||||
renderer.domElement.removeEventListener('mousemove', onMouseMove);
|
||||
renderer.domElement.removeEventListener('mousedown', onMouseDown);
|
||||
renderer.domElement.removeEventListener('mouseup', onMouseUp);
|
||||
};
|
||||
}, []);
|
||||
|
||||
useEffect(() => {
|
||||
const scene = sceneRef.current;
|
||||
const camera = cameraRef.current;
|
||||
const renderer = rendererRef.current;
|
||||
const cssRenderer = cssRendererRef.current;
|
||||
const controls = controlsRef.current;
|
||||
if (!scene || !camera || !renderer || !cssRenderer || !controls) return;
|
||||
|
||||
let running = true;
|
||||
|
||||
const animate = () => {
|
||||
if (!running) return;
|
||||
requestAnimationFrame(animate);
|
||||
|
||||
controls.update();
|
||||
|
||||
const { nodes, links, particles } = dataRef.current;
|
||||
const currentNodeIds = new Set<string>();
|
||||
|
||||
for (const node of nodes.values()) {
|
||||
currentNodeIds.add(node.id);
|
||||
|
||||
let nd = nodeMeshesRef.current.get(node.id);
|
||||
if (!nd) {
|
||||
const isSelf = node.type === 'self';
|
||||
const radius = isSelf ? 12 : 6;
|
||||
const geometry = new THREE.SphereGeometry(radius, 16, 12);
|
||||
const material = new THREE.MeshBasicMaterial({ color: getBaseNodeColor(node) });
|
||||
const mesh = new THREE.Mesh(geometry, material);
|
||||
mesh.userData.nodeId = node.id;
|
||||
scene.add(mesh);
|
||||
|
||||
const labelDiv = document.createElement('div');
|
||||
labelDiv.style.color = node.isAmbiguous ? COLORS.ambiguous : '#e5e7eb';
|
||||
labelDiv.style.fontSize = '11px';
|
||||
labelDiv.style.fontFamily = 'sans-serif';
|
||||
labelDiv.style.textAlign = 'center';
|
||||
labelDiv.style.whiteSpace = 'nowrap';
|
||||
labelDiv.style.textShadow = '0 0 4px #000, 0 0 2px #000';
|
||||
const label = new CSS2DObject(labelDiv);
|
||||
label.position.set(0, -(radius + 6), 0);
|
||||
mesh.add(label);
|
||||
|
||||
nd = { mesh, label, labelDiv };
|
||||
nodeMeshesRef.current.set(node.id, nd);
|
||||
raycastTargetsRef.current.push(mesh);
|
||||
}
|
||||
|
||||
nd.mesh.position.set(node.x ?? 0, node.y ?? 0, node.z ?? 0);
|
||||
const labelColor = node.isAmbiguous ? COLORS.ambiguous : '#e5e7eb';
|
||||
if (nd.labelDiv.style.color !== labelColor) {
|
||||
nd.labelDiv.style.color = labelColor;
|
||||
}
|
||||
const labelText = node.name || (node.type === 'self' ? 'Me' : node.id.slice(0, 8));
|
||||
if (nd.labelDiv.textContent !== labelText) {
|
||||
nd.labelDiv.textContent = labelText;
|
||||
}
|
||||
}
|
||||
|
||||
for (const [id, nd] of nodeMeshesRef.current) {
|
||||
if (!currentNodeIds.has(id)) {
|
||||
nd.mesh.remove(nd.label);
|
||||
nd.labelDiv.remove();
|
||||
scene.remove(nd.mesh);
|
||||
nd.mesh.geometry.dispose();
|
||||
(nd.mesh.material as THREE.Material).dispose();
|
||||
const meshIdx = raycastTargetsRef.current.indexOf(nd.mesh);
|
||||
if (meshIdx >= 0) raycastTargetsRef.current.splice(meshIdx, 1);
|
||||
nodeMeshesRef.current.delete(id);
|
||||
}
|
||||
}
|
||||
|
||||
raycasterRef.current.setFromCamera(mouseRef.current, camera);
|
||||
const intersects = raycasterRef.current.intersectObjects(raycastTargetsRef.current, false);
|
||||
const hitObject = intersects[0]?.object as THREE.Mesh | undefined;
|
||||
const hitId = (hitObject?.userData?.nodeId as string | undefined) ?? null;
|
||||
if (hitId !== hoveredNodeIdRef.current) {
|
||||
hoveredNodeIdRef.current = hitId;
|
||||
setHoveredNodeId(hitId);
|
||||
}
|
||||
const activeId = pinnedNodeIdRef.current ?? hoveredNodeIdRef.current;
|
||||
|
||||
const visibleLinks = [];
|
||||
for (const link of links.values()) {
|
||||
const { sourceId, targetId } = getLinkId(link);
|
||||
if (currentNodeIds.has(sourceId) && currentNodeIds.has(targetId)) {
|
||||
visibleLinks.push(link);
|
||||
}
|
||||
}
|
||||
|
||||
const connectedIds = activeId ? new Set<string>([activeId]) : null;
|
||||
|
||||
const linkLine = linkLineRef.current;
|
||||
if (linkLine) {
|
||||
const geometry = linkLine.geometry as THREE.BufferGeometry;
|
||||
const requiredLength = visibleLinks.length * 6;
|
||||
if (linkPositionBufferRef.current.length < requiredLength) {
|
||||
linkPositionBufferRef.current = growFloat32Buffer(
|
||||
linkPositionBufferRef.current,
|
||||
requiredLength
|
||||
);
|
||||
geometry.setAttribute(
|
||||
'position',
|
||||
new THREE.BufferAttribute(linkPositionBufferRef.current, 3).setUsage(
|
||||
THREE.DynamicDrawUsage
|
||||
)
|
||||
);
|
||||
}
|
||||
|
||||
const highlightLine = highlightLineRef.current;
|
||||
if (highlightLine && highlightPositionBufferRef.current.length < requiredLength) {
|
||||
highlightPositionBufferRef.current = growFloat32Buffer(
|
||||
highlightPositionBufferRef.current,
|
||||
requiredLength
|
||||
);
|
||||
(highlightLine.geometry as THREE.BufferGeometry).setAttribute(
|
||||
'position',
|
||||
new THREE.BufferAttribute(highlightPositionBufferRef.current, 3).setUsage(
|
||||
THREE.DynamicDrawUsage
|
||||
)
|
||||
);
|
||||
}
|
||||
|
||||
const positions = linkPositionBufferRef.current;
|
||||
const hlPositions = highlightPositionBufferRef.current;
|
||||
let idx = 0;
|
||||
let hlIdx = 0;
|
||||
|
||||
for (const link of visibleLinks) {
|
||||
const { sourceId, targetId } = getLinkId(link);
|
||||
const sNode = nodes.get(sourceId);
|
||||
const tNode = nodes.get(targetId);
|
||||
if (!sNode || !tNode) continue;
|
||||
|
||||
const sx = sNode.x ?? 0;
|
||||
const sy = sNode.y ?? 0;
|
||||
const sz = sNode.z ?? 0;
|
||||
const tx = tNode.x ?? 0;
|
||||
const ty = tNode.y ?? 0;
|
||||
const tz = tNode.z ?? 0;
|
||||
|
||||
positions[idx++] = sx;
|
||||
positions[idx++] = sy;
|
||||
positions[idx++] = sz;
|
||||
positions[idx++] = tx;
|
||||
positions[idx++] = ty;
|
||||
positions[idx++] = tz;
|
||||
|
||||
if (activeId && (sourceId === activeId || targetId === activeId)) {
|
||||
connectedIds?.add(sourceId === activeId ? targetId : sourceId);
|
||||
hlPositions[hlIdx++] = sx;
|
||||
hlPositions[hlIdx++] = sy;
|
||||
hlPositions[hlIdx++] = sz;
|
||||
hlPositions[hlIdx++] = tx;
|
||||
hlPositions[hlIdx++] = ty;
|
||||
hlPositions[hlIdx++] = tz;
|
||||
}
|
||||
}
|
||||
|
||||
const positionAttr = geometry.getAttribute('position') as THREE.BufferAttribute | undefined;
|
||||
if (positionAttr) {
|
||||
positionAttr.needsUpdate = true;
|
||||
}
|
||||
geometry.setDrawRange(0, idx / 3);
|
||||
linkLine.visible = idx > 0;
|
||||
|
||||
if (highlightLine) {
|
||||
const hlGeometry = highlightLine.geometry as THREE.BufferGeometry;
|
||||
const hlAttr = hlGeometry.getAttribute('position') as THREE.BufferAttribute | undefined;
|
||||
if (hlAttr) {
|
||||
hlAttr.needsUpdate = true;
|
||||
}
|
||||
hlGeometry.setDrawRange(0, hlIdx / 3);
|
||||
highlightLine.visible = hlIdx > 0;
|
||||
}
|
||||
}
|
||||
|
||||
let writeIdx = 0;
|
||||
for (let readIdx = 0; readIdx < particles.length; readIdx++) {
|
||||
const particle = particles[readIdx];
|
||||
particle.progress += particle.speed;
|
||||
if (particle.progress <= 1) {
|
||||
particles[writeIdx++] = particle;
|
||||
}
|
||||
}
|
||||
particles.length = writeIdx;
|
||||
|
||||
const particlePoints = particlePointsRef.current;
|
||||
if (particlePoints) {
|
||||
const geometry = particlePoints.geometry as THREE.BufferGeometry;
|
||||
const requiredLength = particles.length * 3;
|
||||
|
||||
if (particlePositionBufferRef.current.length < requiredLength) {
|
||||
particlePositionBufferRef.current = growFloat32Buffer(
|
||||
particlePositionBufferRef.current,
|
||||
requiredLength
|
||||
);
|
||||
geometry.setAttribute(
|
||||
'position',
|
||||
new THREE.BufferAttribute(particlePositionBufferRef.current, 3).setUsage(
|
||||
THREE.DynamicDrawUsage
|
||||
)
|
||||
);
|
||||
}
|
||||
if (particleColorBufferRef.current.length < requiredLength) {
|
||||
particleColorBufferRef.current = growFloat32Buffer(
|
||||
particleColorBufferRef.current,
|
||||
requiredLength
|
||||
);
|
||||
geometry.setAttribute(
|
||||
'color',
|
||||
new THREE.BufferAttribute(particleColorBufferRef.current, 3).setUsage(
|
||||
THREE.DynamicDrawUsage
|
||||
)
|
||||
);
|
||||
}
|
||||
|
||||
const pPositions = particlePositionBufferRef.current;
|
||||
const pColors = particleColorBufferRef.current;
|
||||
const color = new THREE.Color();
|
||||
let visibleCount = 0;
|
||||
|
||||
for (const p of particles) {
|
||||
if (p.progress < 0) continue;
|
||||
if (!currentNodeIds.has(p.fromNodeId) || !currentNodeIds.has(p.toNodeId)) continue;
|
||||
|
||||
const fromNode = nodes.get(p.fromNodeId);
|
||||
const toNode = nodes.get(p.toNodeId);
|
||||
if (!fromNode || !toNode) continue;
|
||||
|
||||
const t = p.progress;
|
||||
const x = (fromNode.x ?? 0) + ((toNode.x ?? 0) - (fromNode.x ?? 0)) * t;
|
||||
const y = (fromNode.y ?? 0) + ((toNode.y ?? 0) - (fromNode.y ?? 0)) * t;
|
||||
const z = (fromNode.z ?? 0) + ((toNode.z ?? 0) - (fromNode.z ?? 0)) * t;
|
||||
|
||||
pPositions[visibleCount * 3] = x;
|
||||
pPositions[visibleCount * 3 + 1] = y;
|
||||
pPositions[visibleCount * 3 + 2] = z;
|
||||
|
||||
color.set(p.color);
|
||||
pColors[visibleCount * 3] = color.r;
|
||||
pColors[visibleCount * 3 + 1] = color.g;
|
||||
pColors[visibleCount * 3 + 2] = color.b;
|
||||
visibleCount++;
|
||||
}
|
||||
|
||||
const posAttr = geometry.getAttribute('position') as THREE.BufferAttribute | undefined;
|
||||
const colorAttr = geometry.getAttribute('color') as THREE.BufferAttribute | undefined;
|
||||
if (posAttr) posAttr.needsUpdate = true;
|
||||
if (colorAttr) colorAttr.needsUpdate = true;
|
||||
geometry.setDrawRange(0, visibleCount);
|
||||
particlePoints.visible = visibleCount > 0;
|
||||
}
|
||||
|
||||
const nextNeighbors = connectedIds
|
||||
? Array.from(connectedIds)
|
||||
.filter((id) => id !== activeId)
|
||||
.sort()
|
||||
: [];
|
||||
if (!arraysEqual(hoveredNeighborIdsRef.current, nextNeighbors)) {
|
||||
hoveredNeighborIdsRef.current = nextNeighbors;
|
||||
setHoveredNeighborIds(nextNeighbors);
|
||||
}
|
||||
|
||||
for (const [id, nd] of nodeMeshesRef.current) {
|
||||
const node = nodes.get(id);
|
||||
if (!node) continue;
|
||||
const mat = nd.mesh.material as THREE.MeshBasicMaterial;
|
||||
if (id === activeId) {
|
||||
mat.color.set(0xffd700);
|
||||
} else if (connectedIds?.has(id)) {
|
||||
mat.color.set(0xfff0b3);
|
||||
} else {
|
||||
mat.color.set(getBaseNodeColor(node));
|
||||
}
|
||||
}
|
||||
|
||||
renderer.render(scene, camera);
|
||||
cssRenderer.render(scene, camera);
|
||||
};
|
||||
|
||||
animate();
|
||||
return () => {
|
||||
running = false;
|
||||
};
|
||||
}, []);
|
||||
|
||||
return { hoveredNodeId, hoveredNeighborIds, pinnedNodeId };
|
||||
}
|
||||
922
frontend/src/components/visualizer/useVisualizerData3D.ts
Normal file
922
frontend/src/components/visualizer/useVisualizerData3D.ts
Normal file
@@ -0,0 +1,922 @@
|
||||
import { useCallback, useEffect, useMemo, useRef, useState } from 'react';
|
||||
import {
|
||||
forceCenter,
|
||||
forceLink,
|
||||
forceManyBody,
|
||||
forceSimulation,
|
||||
forceX,
|
||||
forceY,
|
||||
forceZ,
|
||||
type ForceLink3D,
|
||||
type Simulation3D,
|
||||
} from 'd3-force-3d';
|
||||
import { PayloadType } from '@michaelhart/meshcore-decoder';
|
||||
|
||||
import {
|
||||
CONTACT_TYPE_REPEATER,
|
||||
type Contact,
|
||||
type ContactAdvertPathSummary,
|
||||
type RadioConfig,
|
||||
type RawPacket,
|
||||
} from '../../types';
|
||||
import { getRawPacketObservationKey } from '../../utils/rawPacketIdentity';
|
||||
import {
|
||||
type Particle,
|
||||
type PendingPacket,
|
||||
type RepeaterTrafficData,
|
||||
PARTICLE_COLOR_MAP,
|
||||
PARTICLE_SPEED,
|
||||
analyzeRepeaterTraffic,
|
||||
buildAmbiguousRepeaterLabel,
|
||||
buildAmbiguousRepeaterNodeId,
|
||||
dedupeConsecutive,
|
||||
generatePacketKey,
|
||||
getNodeType,
|
||||
getPacketLabel,
|
||||
parsePacket,
|
||||
recordTrafficObservation,
|
||||
} from '../../utils/visualizerUtils';
|
||||
import { type GraphLink, type GraphNode, normalizePacketTimestampMs } from './shared';
|
||||
|
||||
export interface UseVisualizerData3DOptions {
|
||||
packets: RawPacket[];
|
||||
contacts: Contact[];
|
||||
config: RadioConfig | null;
|
||||
repeaterAdvertPaths: ContactAdvertPathSummary[];
|
||||
showAmbiguousPaths: boolean;
|
||||
showAmbiguousNodes: boolean;
|
||||
useAdvertPathHints: boolean;
|
||||
splitAmbiguousByTraffic: boolean;
|
||||
chargeStrength: number;
|
||||
letEmDrift: boolean;
|
||||
particleSpeedMultiplier: number;
|
||||
observationWindowSec: number;
|
||||
pruneStaleNodes: boolean;
|
||||
pruneStaleMinutes: number;
|
||||
}
|
||||
|
||||
export interface VisualizerData3D {
|
||||
nodes: Map<string, GraphNode>;
|
||||
links: Map<string, GraphLink>;
|
||||
particles: Particle[];
|
||||
stats: { processed: number; animated: number; nodes: number; links: number };
|
||||
expandContract: () => void;
|
||||
clearAndReset: () => void;
|
||||
}
|
||||
|
||||
export function useVisualizerData3D({
|
||||
packets,
|
||||
contacts,
|
||||
config,
|
||||
repeaterAdvertPaths,
|
||||
showAmbiguousPaths,
|
||||
showAmbiguousNodes,
|
||||
useAdvertPathHints,
|
||||
splitAmbiguousByTraffic,
|
||||
chargeStrength,
|
||||
letEmDrift,
|
||||
particleSpeedMultiplier,
|
||||
observationWindowSec,
|
||||
pruneStaleNodes,
|
||||
pruneStaleMinutes,
|
||||
}: UseVisualizerData3DOptions): VisualizerData3D {
|
||||
const nodesRef = useRef<Map<string, GraphNode>>(new Map());
|
||||
const linksRef = useRef<Map<string, GraphLink>>(new Map());
|
||||
const particlesRef = useRef<Particle[]>([]);
|
||||
const simulationRef = useRef<Simulation3D<GraphNode, GraphLink> | null>(null);
|
||||
const processedRef = useRef<Set<string>>(new Set());
|
||||
const pendingRef = useRef<Map<string, PendingPacket>>(new Map());
|
||||
const timersRef = useRef<Map<string, ReturnType<typeof setTimeout>>>(new Map());
|
||||
const trafficPatternsRef = useRef<Map<string, RepeaterTrafficData>>(new Map());
|
||||
const speedMultiplierRef = useRef(particleSpeedMultiplier);
|
||||
const observationWindowRef = useRef(observationWindowSec * 1000);
|
||||
const stretchRafRef = useRef<number | null>(null);
|
||||
const [stats, setStats] = useState({ processed: 0, animated: 0, nodes: 0, links: 0 });
|
||||
|
||||
const contactIndex = useMemo(() => {
|
||||
const byPrefix12 = new Map<string, Contact>();
|
||||
const byName = new Map<string, Contact>();
|
||||
const byPrefix = new Map<string, Contact[]>();
|
||||
|
||||
for (const contact of contacts) {
|
||||
const prefix12 = contact.public_key.slice(0, 12).toLowerCase();
|
||||
byPrefix12.set(prefix12, contact);
|
||||
|
||||
if (contact.name && !byName.has(contact.name)) {
|
||||
byName.set(contact.name, contact);
|
||||
}
|
||||
|
||||
for (let len = 1; len <= 12; len++) {
|
||||
const prefix = prefix12.slice(0, len);
|
||||
const matches = byPrefix.get(prefix);
|
||||
if (matches) {
|
||||
matches.push(contact);
|
||||
} else {
|
||||
byPrefix.set(prefix, [contact]);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return { byPrefix12, byName, byPrefix };
|
||||
}, [contacts]);
|
||||
|
||||
const advertPathIndex = useMemo(() => {
|
||||
const byRepeater = new Map<string, ContactAdvertPathSummary['paths']>();
|
||||
for (const summary of repeaterAdvertPaths) {
|
||||
const key = summary.public_key.slice(0, 12).toLowerCase();
|
||||
byRepeater.set(key, summary.paths);
|
||||
}
|
||||
return { byRepeater };
|
||||
}, [repeaterAdvertPaths]);
|
||||
|
||||
useEffect(() => {
|
||||
speedMultiplierRef.current = particleSpeedMultiplier;
|
||||
}, [particleSpeedMultiplier]);
|
||||
|
||||
useEffect(() => {
|
||||
observationWindowRef.current = observationWindowSec * 1000;
|
||||
}, [observationWindowSec]);
|
||||
|
||||
useEffect(() => {
|
||||
const sim = forceSimulation<GraphNode, GraphLink>([])
|
||||
.numDimensions(3)
|
||||
.force(
|
||||
'link',
|
||||
forceLink<GraphNode, GraphLink>([])
|
||||
.id((d) => d.id)
|
||||
.distance(120)
|
||||
.strength(0.3)
|
||||
)
|
||||
.force(
|
||||
'charge',
|
||||
forceManyBody<GraphNode>()
|
||||
.strength((d) => (d.id === 'self' ? -1200 : -200))
|
||||
.distanceMax(800)
|
||||
)
|
||||
.force('center', forceCenter(0, 0, 0))
|
||||
.force(
|
||||
'selfX',
|
||||
forceX<GraphNode>(0).strength((d) => (d.id === 'self' ? 0.1 : 0))
|
||||
)
|
||||
.force(
|
||||
'selfY',
|
||||
forceY<GraphNode>(0).strength((d) => (d.id === 'self' ? 0.1 : 0))
|
||||
)
|
||||
.force(
|
||||
'selfZ',
|
||||
forceZ<GraphNode>(0).strength((d) => (d.id === 'self' ? 0.1 : 0))
|
||||
)
|
||||
.alphaDecay(0.02)
|
||||
.velocityDecay(0.5)
|
||||
.alphaTarget(0.03);
|
||||
|
||||
simulationRef.current = sim;
|
||||
return () => {
|
||||
sim.stop();
|
||||
};
|
||||
}, []);
|
||||
|
||||
useEffect(() => {
|
||||
const sim = simulationRef.current;
|
||||
if (!sim) return;
|
||||
|
||||
sim.force(
|
||||
'charge',
|
||||
forceManyBody<GraphNode>()
|
||||
.strength((d) => (d.id === 'self' ? chargeStrength * 6 : chargeStrength))
|
||||
.distanceMax(800)
|
||||
);
|
||||
sim.alpha(0.3).restart();
|
||||
}, [chargeStrength]);
|
||||
|
||||
useEffect(() => {
|
||||
const sim = simulationRef.current;
|
||||
if (!sim) return;
|
||||
sim.alphaTarget(letEmDrift ? 0.05 : 0);
|
||||
}, [letEmDrift]);
|
||||
|
||||
const syncSimulation = useCallback(() => {
|
||||
const sim = simulationRef.current;
|
||||
if (!sim) return;
|
||||
|
||||
const nodes = Array.from(nodesRef.current.values());
|
||||
const links = Array.from(linksRef.current.values());
|
||||
|
||||
sim.nodes(nodes);
|
||||
const linkForce = sim.force('link') as ForceLink3D<GraphNode, GraphLink> | undefined;
|
||||
linkForce?.links(links);
|
||||
|
||||
sim.alpha(0.15).restart();
|
||||
|
||||
setStats((prev) =>
|
||||
prev.nodes === nodes.length && prev.links === links.length
|
||||
? prev
|
||||
: { ...prev, nodes: nodes.length, links: links.length }
|
||||
);
|
||||
}, []);
|
||||
|
||||
useEffect(() => {
|
||||
if (!nodesRef.current.has('self')) {
|
||||
nodesRef.current.set('self', {
|
||||
id: 'self',
|
||||
name: config?.name || 'Me',
|
||||
type: 'self',
|
||||
isAmbiguous: false,
|
||||
lastActivity: Date.now(),
|
||||
x: 0,
|
||||
y: 0,
|
||||
z: 0,
|
||||
});
|
||||
syncSimulation();
|
||||
}
|
||||
}, [config, syncSimulation]);
|
||||
|
||||
useEffect(() => {
|
||||
processedRef.current.clear();
|
||||
const selfNode = nodesRef.current.get('self');
|
||||
nodesRef.current.clear();
|
||||
if (selfNode) nodesRef.current.set('self', selfNode);
|
||||
linksRef.current.clear();
|
||||
particlesRef.current = [];
|
||||
pendingRef.current.clear();
|
||||
timersRef.current.forEach((t) => clearTimeout(t));
|
||||
timersRef.current.clear();
|
||||
trafficPatternsRef.current.clear();
|
||||
setStats({ processed: 0, animated: 0, nodes: selfNode ? 1 : 0, links: 0 });
|
||||
syncSimulation();
|
||||
}, [
|
||||
showAmbiguousPaths,
|
||||
showAmbiguousNodes,
|
||||
useAdvertPathHints,
|
||||
splitAmbiguousByTraffic,
|
||||
syncSimulation,
|
||||
]);
|
||||
|
||||
const addNode = useCallback(
|
||||
(
|
||||
id: string,
|
||||
name: string | null,
|
||||
type: GraphNode['type'],
|
||||
isAmbiguous: boolean,
|
||||
probableIdentity?: string | null,
|
||||
ambiguousNames?: string[],
|
||||
lastSeen?: number | null,
|
||||
activityAtMs?: number
|
||||
) => {
|
||||
const activityAt = activityAtMs ?? Date.now();
|
||||
const existing = nodesRef.current.get(id);
|
||||
if (existing) {
|
||||
existing.lastActivity = Math.max(existing.lastActivity, activityAt);
|
||||
if (name) existing.name = name;
|
||||
if (probableIdentity !== undefined) existing.probableIdentity = probableIdentity;
|
||||
if (ambiguousNames) existing.ambiguousNames = ambiguousNames;
|
||||
if (lastSeen !== undefined) existing.lastSeen = lastSeen;
|
||||
} else {
|
||||
const theta = Math.random() * Math.PI * 2;
|
||||
const phi = Math.acos(2 * Math.random() - 1);
|
||||
const r = 80 + Math.random() * 100;
|
||||
nodesRef.current.set(id, {
|
||||
id,
|
||||
name,
|
||||
type,
|
||||
isAmbiguous,
|
||||
lastActivity: activityAt,
|
||||
probableIdentity,
|
||||
lastSeen,
|
||||
ambiguousNames,
|
||||
x: r * Math.sin(phi) * Math.cos(theta),
|
||||
y: r * Math.sin(phi) * Math.sin(theta),
|
||||
z: r * Math.cos(phi),
|
||||
});
|
||||
}
|
||||
},
|
||||
[]
|
||||
);
|
||||
|
||||
const addLink = useCallback((sourceId: string, targetId: string, activityAtMs?: number) => {
|
||||
const activityAt = activityAtMs ?? Date.now();
|
||||
const key = [sourceId, targetId].sort().join('->');
|
||||
const existing = linksRef.current.get(key);
|
||||
if (existing) {
|
||||
existing.lastActivity = Math.max(existing.lastActivity, activityAt);
|
||||
} else {
|
||||
linksRef.current.set(key, { source: sourceId, target: targetId, lastActivity: activityAt });
|
||||
}
|
||||
}, []);
|
||||
|
||||
const publishPacket = useCallback((packetKey: string) => {
|
||||
const pending = pendingRef.current.get(packetKey);
|
||||
if (!pending) return;
|
||||
|
||||
pendingRef.current.delete(packetKey);
|
||||
timersRef.current.delete(packetKey);
|
||||
|
||||
if (document.hidden) return;
|
||||
|
||||
for (const path of pending.paths) {
|
||||
const dedupedPath = dedupeConsecutive(path.nodes);
|
||||
if (dedupedPath.length < 2) continue;
|
||||
|
||||
for (let i = 0; i < dedupedPath.length - 1; i++) {
|
||||
particlesRef.current.push({
|
||||
linkKey: [dedupedPath[i], dedupedPath[i + 1]].sort().join('->'),
|
||||
progress: -i,
|
||||
speed: PARTICLE_SPEED * speedMultiplierRef.current,
|
||||
color: PARTICLE_COLOR_MAP[pending.label],
|
||||
label: pending.label,
|
||||
fromNodeId: dedupedPath[i],
|
||||
toNodeId: dedupedPath[i + 1],
|
||||
});
|
||||
}
|
||||
}
|
||||
}, []);
|
||||
|
||||
const pickLikelyRepeaterByAdvertPath = useCallback(
|
||||
(candidates: Contact[], nextPrefix: string | null) => {
|
||||
const nextHop = nextPrefix?.toLowerCase() ?? null;
|
||||
const scored = candidates
|
||||
.map((candidate) => {
|
||||
const prefix12 = candidate.public_key.slice(0, 12).toLowerCase();
|
||||
const paths = advertPathIndex.byRepeater.get(prefix12) ?? [];
|
||||
let matchScore = 0;
|
||||
let totalScore = 0;
|
||||
|
||||
for (const path of paths) {
|
||||
totalScore += path.heard_count;
|
||||
const pathNextHop = path.next_hop?.toLowerCase() ?? null;
|
||||
if (pathNextHop === nextHop) {
|
||||
matchScore += path.heard_count;
|
||||
}
|
||||
}
|
||||
|
||||
return { candidate, matchScore, totalScore };
|
||||
})
|
||||
.filter((entry) => entry.totalScore > 0)
|
||||
.sort(
|
||||
(a, b) =>
|
||||
b.matchScore - a.matchScore ||
|
||||
b.totalScore - a.totalScore ||
|
||||
a.candidate.public_key.localeCompare(b.candidate.public_key)
|
||||
);
|
||||
|
||||
if (scored.length === 0) return null;
|
||||
|
||||
const top = scored[0];
|
||||
const second = scored[1] ?? null;
|
||||
|
||||
if (top.matchScore < 2) return null;
|
||||
if (second && top.matchScore < second.matchScore * 2) return null;
|
||||
|
||||
return top.candidate;
|
||||
},
|
||||
[advertPathIndex]
|
||||
);
|
||||
|
||||
const resolveNode = useCallback(
|
||||
(
|
||||
source: { type: 'prefix' | 'pubkey' | 'name'; value: string },
|
||||
isRepeater: boolean,
|
||||
showAmbiguous: boolean,
|
||||
myPrefix: string | null,
|
||||
activityAtMs: number,
|
||||
trafficContext?: { packetSource: string | null; nextPrefix: string | null }
|
||||
): string | null => {
|
||||
if (source.type === 'pubkey') {
|
||||
if (source.value.length < 12) return null;
|
||||
const nodeId = source.value.slice(0, 12).toLowerCase();
|
||||
if (myPrefix && nodeId === myPrefix) return 'self';
|
||||
const contact = contactIndex.byPrefix12.get(nodeId);
|
||||
addNode(
|
||||
nodeId,
|
||||
contact?.name || null,
|
||||
getNodeType(contact),
|
||||
false,
|
||||
undefined,
|
||||
undefined,
|
||||
contact?.last_seen,
|
||||
activityAtMs
|
||||
);
|
||||
return nodeId;
|
||||
}
|
||||
|
||||
if (source.type === 'name') {
|
||||
const contact = contactIndex.byName.get(source.value) ?? null;
|
||||
if (contact) {
|
||||
const nodeId = contact.public_key.slice(0, 12).toLowerCase();
|
||||
if (myPrefix && nodeId === myPrefix) return 'self';
|
||||
addNode(
|
||||
nodeId,
|
||||
contact.name,
|
||||
getNodeType(contact),
|
||||
false,
|
||||
undefined,
|
||||
undefined,
|
||||
contact.last_seen,
|
||||
activityAtMs
|
||||
);
|
||||
return nodeId;
|
||||
}
|
||||
const nodeId = `name:${source.value}`;
|
||||
addNode(
|
||||
nodeId,
|
||||
source.value,
|
||||
'client',
|
||||
false,
|
||||
undefined,
|
||||
undefined,
|
||||
undefined,
|
||||
activityAtMs
|
||||
);
|
||||
return nodeId;
|
||||
}
|
||||
|
||||
const lookupValue = source.value.toLowerCase();
|
||||
const matches = contactIndex.byPrefix.get(lookupValue) ?? [];
|
||||
const contact = matches.length === 1 ? matches[0] : null;
|
||||
if (contact) {
|
||||
const nodeId = contact.public_key.slice(0, 12).toLowerCase();
|
||||
if (myPrefix && nodeId === myPrefix) return 'self';
|
||||
addNode(
|
||||
nodeId,
|
||||
contact.name,
|
||||
getNodeType(contact),
|
||||
false,
|
||||
undefined,
|
||||
undefined,
|
||||
contact.last_seen,
|
||||
activityAtMs
|
||||
);
|
||||
return nodeId;
|
||||
}
|
||||
|
||||
if (showAmbiguous) {
|
||||
const filtered = isRepeater
|
||||
? matches.filter((c) => c.type === CONTACT_TYPE_REPEATER)
|
||||
: matches.filter((c) => c.type !== CONTACT_TYPE_REPEATER);
|
||||
|
||||
if (filtered.length === 1) {
|
||||
const c = filtered[0];
|
||||
const nodeId = c.public_key.slice(0, 12).toLowerCase();
|
||||
addNode(
|
||||
nodeId,
|
||||
c.name,
|
||||
getNodeType(c),
|
||||
false,
|
||||
undefined,
|
||||
undefined,
|
||||
c.last_seen,
|
||||
activityAtMs
|
||||
);
|
||||
return nodeId;
|
||||
}
|
||||
|
||||
if (filtered.length > 1 || (filtered.length === 0 && isRepeater)) {
|
||||
const names = filtered.map((c) => c.name || c.public_key.slice(0, 8));
|
||||
const lastSeen = filtered.reduce(
|
||||
(max, c) => (c.last_seen && (!max || c.last_seen > max) ? c.last_seen : max),
|
||||
null as number | null
|
||||
);
|
||||
|
||||
let nodeId = buildAmbiguousRepeaterNodeId(lookupValue);
|
||||
let displayName = buildAmbiguousRepeaterLabel(lookupValue);
|
||||
let probableIdentity: string | null = null;
|
||||
let ambiguousNames = names.length > 0 ? names : undefined;
|
||||
|
||||
if (useAdvertPathHints && isRepeater && trafficContext) {
|
||||
const normalizedNext = trafficContext.nextPrefix?.toLowerCase() ?? null;
|
||||
const likely = pickLikelyRepeaterByAdvertPath(filtered, normalizedNext);
|
||||
if (likely) {
|
||||
const likelyName = likely.name || likely.public_key.slice(0, 12).toUpperCase();
|
||||
probableIdentity = likelyName;
|
||||
displayName = likelyName;
|
||||
ambiguousNames = filtered
|
||||
.filter((c) => c.public_key !== likely.public_key)
|
||||
.map((c) => c.name || c.public_key.slice(0, 8));
|
||||
}
|
||||
}
|
||||
|
||||
if (splitAmbiguousByTraffic && isRepeater && trafficContext) {
|
||||
const normalizedNext = trafficContext.nextPrefix?.toLowerCase() ?? null;
|
||||
|
||||
if (trafficContext.packetSource) {
|
||||
recordTrafficObservation(
|
||||
trafficPatternsRef.current,
|
||||
lookupValue,
|
||||
trafficContext.packetSource,
|
||||
normalizedNext
|
||||
);
|
||||
}
|
||||
|
||||
const trafficData = trafficPatternsRef.current.get(lookupValue);
|
||||
if (trafficData) {
|
||||
const analysis = analyzeRepeaterTraffic(trafficData);
|
||||
if (analysis.shouldSplit && normalizedNext) {
|
||||
nodeId = buildAmbiguousRepeaterNodeId(lookupValue, normalizedNext);
|
||||
if (!probableIdentity) {
|
||||
displayName = buildAmbiguousRepeaterLabel(lookupValue, normalizedNext);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
addNode(
|
||||
nodeId,
|
||||
displayName,
|
||||
isRepeater ? 'repeater' : 'client',
|
||||
true,
|
||||
probableIdentity,
|
||||
ambiguousNames,
|
||||
lastSeen,
|
||||
activityAtMs
|
||||
);
|
||||
return nodeId;
|
||||
}
|
||||
}
|
||||
|
||||
return null;
|
||||
},
|
||||
[
|
||||
contactIndex,
|
||||
addNode,
|
||||
useAdvertPathHints,
|
||||
pickLikelyRepeaterByAdvertPath,
|
||||
splitAmbiguousByTraffic,
|
||||
]
|
||||
);
|
||||
|
||||
const buildPath = useCallback(
|
||||
(
|
||||
parsed: ReturnType<typeof parsePacket>,
|
||||
packet: RawPacket,
|
||||
myPrefix: string | null,
|
||||
activityAtMs: number
|
||||
): string[] => {
|
||||
if (!parsed) return [];
|
||||
const path: string[] = [];
|
||||
let packetSource: string | null = null;
|
||||
|
||||
if (parsed.payloadType === PayloadType.Advert && parsed.advertPubkey) {
|
||||
const nodeId = resolveNode(
|
||||
{ type: 'pubkey', value: parsed.advertPubkey },
|
||||
false,
|
||||
false,
|
||||
myPrefix,
|
||||
activityAtMs
|
||||
);
|
||||
if (nodeId) {
|
||||
path.push(nodeId);
|
||||
packetSource = nodeId;
|
||||
}
|
||||
} else if (parsed.payloadType === PayloadType.AnonRequest && parsed.anonRequestPubkey) {
|
||||
const nodeId = resolveNode(
|
||||
{ type: 'pubkey', value: parsed.anonRequestPubkey },
|
||||
false,
|
||||
false,
|
||||
myPrefix,
|
||||
activityAtMs
|
||||
);
|
||||
if (nodeId) {
|
||||
path.push(nodeId);
|
||||
packetSource = nodeId;
|
||||
}
|
||||
} else if (parsed.payloadType === PayloadType.TextMessage && parsed.srcHash) {
|
||||
if (myPrefix && parsed.srcHash.toLowerCase() === myPrefix) {
|
||||
path.push('self');
|
||||
packetSource = 'self';
|
||||
} else {
|
||||
const nodeId = resolveNode(
|
||||
{ type: 'prefix', value: parsed.srcHash },
|
||||
false,
|
||||
showAmbiguousNodes,
|
||||
myPrefix,
|
||||
activityAtMs
|
||||
);
|
||||
if (nodeId) {
|
||||
path.push(nodeId);
|
||||
packetSource = nodeId;
|
||||
}
|
||||
}
|
||||
} else if (parsed.payloadType === PayloadType.GroupText) {
|
||||
const senderName = parsed.groupTextSender || packet.decrypted_info?.sender;
|
||||
if (senderName) {
|
||||
const resolved = resolveNode(
|
||||
{ type: 'name', value: senderName },
|
||||
false,
|
||||
false,
|
||||
myPrefix,
|
||||
activityAtMs
|
||||
);
|
||||
if (resolved) {
|
||||
path.push(resolved);
|
||||
packetSource = resolved;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
for (let i = 0; i < parsed.pathBytes.length; i++) {
|
||||
const hexPrefix = parsed.pathBytes[i];
|
||||
const nextPrefix = parsed.pathBytes[i + 1] || null;
|
||||
const nodeId = resolveNode(
|
||||
{ type: 'prefix', value: hexPrefix },
|
||||
true,
|
||||
showAmbiguousPaths,
|
||||
myPrefix,
|
||||
activityAtMs,
|
||||
{ packetSource, nextPrefix }
|
||||
);
|
||||
if (nodeId) path.push(nodeId);
|
||||
}
|
||||
|
||||
if (parsed.payloadType === PayloadType.TextMessage && parsed.dstHash) {
|
||||
if (myPrefix && parsed.dstHash.toLowerCase() === myPrefix) {
|
||||
path.push('self');
|
||||
} else {
|
||||
const nodeId = resolveNode(
|
||||
{ type: 'prefix', value: parsed.dstHash },
|
||||
false,
|
||||
showAmbiguousNodes,
|
||||
myPrefix,
|
||||
activityAtMs
|
||||
);
|
||||
if (nodeId) path.push(nodeId);
|
||||
else path.push('self');
|
||||
}
|
||||
} else if (path.length > 0) {
|
||||
path.push('self');
|
||||
}
|
||||
|
||||
if (path.length > 0 && path[path.length - 1] !== 'self') {
|
||||
path.push('self');
|
||||
}
|
||||
|
||||
return dedupeConsecutive(path);
|
||||
},
|
||||
[resolveNode, showAmbiguousPaths, showAmbiguousNodes]
|
||||
);
|
||||
|
||||
useEffect(() => {
|
||||
let newProcessed = 0;
|
||||
let newAnimated = 0;
|
||||
let needsUpdate = false;
|
||||
const myPrefix = config?.public_key?.slice(0, 12).toLowerCase() || null;
|
||||
|
||||
for (const packet of packets) {
|
||||
const observationKey = getRawPacketObservationKey(packet);
|
||||
if (processedRef.current.has(observationKey)) continue;
|
||||
processedRef.current.add(observationKey);
|
||||
newProcessed++;
|
||||
|
||||
if (processedRef.current.size > 1000) {
|
||||
processedRef.current = new Set(Array.from(processedRef.current).slice(-500));
|
||||
}
|
||||
|
||||
const parsed = parsePacket(packet.data);
|
||||
if (!parsed) continue;
|
||||
|
||||
const packetActivityAt = normalizePacketTimestampMs(packet.timestamp);
|
||||
const path = buildPath(parsed, packet, myPrefix, packetActivityAt);
|
||||
if (path.length < 2) continue;
|
||||
|
||||
const label = getPacketLabel(parsed.payloadType);
|
||||
for (let i = 0; i < path.length; i++) {
|
||||
const n = nodesRef.current.get(path[i]);
|
||||
if (n && n.id !== 'self') {
|
||||
n.lastActivityReason = i === 0 ? `${label} source` : `Relayed ${label}`;
|
||||
}
|
||||
}
|
||||
|
||||
for (let i = 0; i < path.length - 1; i++) {
|
||||
if (path[i] !== path[i + 1]) {
|
||||
addLink(path[i], path[i + 1], packetActivityAt);
|
||||
needsUpdate = true;
|
||||
}
|
||||
}
|
||||
|
||||
const packetKey = generatePacketKey(parsed, packet);
|
||||
const now = Date.now();
|
||||
const existing = pendingRef.current.get(packetKey);
|
||||
|
||||
if (existing && now < existing.expiresAt) {
|
||||
existing.paths.push({ nodes: path, snr: packet.snr ?? null, timestamp: now });
|
||||
} else {
|
||||
const existingTimer = timersRef.current.get(packetKey);
|
||||
if (existingTimer) {
|
||||
clearTimeout(existingTimer);
|
||||
}
|
||||
const windowMs = observationWindowRef.current;
|
||||
pendingRef.current.set(packetKey, {
|
||||
key: packetKey,
|
||||
label: getPacketLabel(parsed.payloadType),
|
||||
paths: [{ nodes: path, snr: packet.snr ?? null, timestamp: now }],
|
||||
firstSeen: now,
|
||||
expiresAt: now + windowMs,
|
||||
});
|
||||
timersRef.current.set(
|
||||
packetKey,
|
||||
setTimeout(() => publishPacket(packetKey), windowMs)
|
||||
);
|
||||
}
|
||||
|
||||
if (pendingRef.current.size > 100) {
|
||||
const entries = Array.from(pendingRef.current.entries())
|
||||
.sort((a, b) => a[1].firstSeen - b[1].firstSeen)
|
||||
.slice(0, 50);
|
||||
for (const [key] of entries) {
|
||||
const timer = timersRef.current.get(key);
|
||||
if (timer) {
|
||||
clearTimeout(timer);
|
||||
}
|
||||
timersRef.current.delete(key);
|
||||
pendingRef.current.delete(key);
|
||||
}
|
||||
}
|
||||
|
||||
newAnimated++;
|
||||
}
|
||||
|
||||
if (needsUpdate) syncSimulation();
|
||||
if (newProcessed > 0) {
|
||||
setStats((prev) => ({
|
||||
...prev,
|
||||
processed: prev.processed + newProcessed,
|
||||
animated: prev.animated + newAnimated,
|
||||
}));
|
||||
}
|
||||
}, [packets, config, buildPath, addLink, syncSimulation, publishPacket]);
|
||||
|
||||
const expandContract = useCallback(() => {
|
||||
const sim = simulationRef.current;
|
||||
if (!sim) return;
|
||||
|
||||
if (stretchRafRef.current !== null) {
|
||||
cancelAnimationFrame(stretchRafRef.current);
|
||||
stretchRafRef.current = null;
|
||||
}
|
||||
|
||||
const startChargeStrength = chargeStrength;
|
||||
const peakChargeStrength = -5000;
|
||||
const startLinkStrength = 0.3;
|
||||
const minLinkStrength = 0.02;
|
||||
const expandDuration = 1000;
|
||||
const holdDuration = 2000;
|
||||
const contractDuration = 1000;
|
||||
const startTime = performance.now();
|
||||
|
||||
const animate = (now: number) => {
|
||||
const elapsed = now - startTime;
|
||||
let currentChargeStrength: number;
|
||||
let currentLinkStrength: number;
|
||||
|
||||
if (elapsed < expandDuration) {
|
||||
const t = elapsed / expandDuration;
|
||||
currentChargeStrength =
|
||||
startChargeStrength + (peakChargeStrength - startChargeStrength) * t;
|
||||
currentLinkStrength = startLinkStrength + (minLinkStrength - startLinkStrength) * t;
|
||||
} else if (elapsed < expandDuration + holdDuration) {
|
||||
currentChargeStrength = peakChargeStrength;
|
||||
currentLinkStrength = minLinkStrength;
|
||||
} else if (elapsed < expandDuration + holdDuration + contractDuration) {
|
||||
const t = (elapsed - expandDuration - holdDuration) / contractDuration;
|
||||
currentChargeStrength = peakChargeStrength + (startChargeStrength - peakChargeStrength) * t;
|
||||
currentLinkStrength = minLinkStrength + (startLinkStrength - minLinkStrength) * t;
|
||||
} else {
|
||||
sim.force(
|
||||
'charge',
|
||||
forceManyBody<GraphNode>()
|
||||
.strength((d) => (d.id === 'self' ? startChargeStrength * 6 : startChargeStrength))
|
||||
.distanceMax(800)
|
||||
);
|
||||
sim.force(
|
||||
'link',
|
||||
forceLink<GraphNode, GraphLink>(Array.from(linksRef.current.values()))
|
||||
.id((d) => d.id)
|
||||
.distance(120)
|
||||
.strength(startLinkStrength)
|
||||
);
|
||||
sim.alpha(0.3).restart();
|
||||
stretchRafRef.current = null;
|
||||
return;
|
||||
}
|
||||
|
||||
sim.force(
|
||||
'charge',
|
||||
forceManyBody<GraphNode>()
|
||||
.strength((d) => (d.id === 'self' ? currentChargeStrength * 6 : currentChargeStrength))
|
||||
.distanceMax(800)
|
||||
);
|
||||
sim.force(
|
||||
'link',
|
||||
forceLink<GraphNode, GraphLink>(Array.from(linksRef.current.values()))
|
||||
.id((d) => d.id)
|
||||
.distance(120)
|
||||
.strength(currentLinkStrength)
|
||||
);
|
||||
sim.alpha(0.5).restart();
|
||||
|
||||
stretchRafRef.current = requestAnimationFrame(animate);
|
||||
};
|
||||
|
||||
stretchRafRef.current = requestAnimationFrame(animate);
|
||||
}, [chargeStrength]);
|
||||
|
||||
const clearAndReset = useCallback(() => {
|
||||
if (stretchRafRef.current !== null) {
|
||||
cancelAnimationFrame(stretchRafRef.current);
|
||||
stretchRafRef.current = null;
|
||||
}
|
||||
|
||||
for (const timer of timersRef.current.values()) {
|
||||
clearTimeout(timer);
|
||||
}
|
||||
timersRef.current.clear();
|
||||
pendingRef.current.clear();
|
||||
processedRef.current.clear();
|
||||
trafficPatternsRef.current.clear();
|
||||
particlesRef.current.length = 0;
|
||||
linksRef.current.clear();
|
||||
|
||||
const selfNode = nodesRef.current.get('self');
|
||||
nodesRef.current.clear();
|
||||
if (selfNode) {
|
||||
selfNode.x = 0;
|
||||
selfNode.y = 0;
|
||||
selfNode.z = 0;
|
||||
selfNode.vx = 0;
|
||||
selfNode.vy = 0;
|
||||
selfNode.vz = 0;
|
||||
selfNode.lastActivity = Date.now();
|
||||
nodesRef.current.set('self', selfNode);
|
||||
}
|
||||
|
||||
const sim = simulationRef.current;
|
||||
if (sim) {
|
||||
sim.nodes(Array.from(nodesRef.current.values()));
|
||||
const linkForce = sim.force('link') as ForceLink3D<GraphNode, GraphLink> | undefined;
|
||||
linkForce?.links([]);
|
||||
sim.alpha(0.3).restart();
|
||||
}
|
||||
|
||||
setStats({ processed: 0, animated: 0, nodes: 1, links: 0 });
|
||||
}, []);
|
||||
|
||||
useEffect(() => {
|
||||
const stretchRaf = stretchRafRef;
|
||||
const timers = timersRef.current;
|
||||
const pending = pendingRef.current;
|
||||
return () => {
|
||||
if (stretchRaf.current !== null) {
|
||||
cancelAnimationFrame(stretchRaf.current);
|
||||
}
|
||||
for (const timer of timers.values()) {
|
||||
clearTimeout(timer);
|
||||
}
|
||||
timers.clear();
|
||||
pending.clear();
|
||||
};
|
||||
}, []);
|
||||
|
||||
useEffect(() => {
|
||||
if (!pruneStaleNodes) return;
|
||||
|
||||
const staleMs = pruneStaleMinutes * 60 * 1000;
|
||||
const pruneIntervalMs = 1000;
|
||||
|
||||
const interval = setInterval(() => {
|
||||
const cutoff = Date.now() - staleMs;
|
||||
let pruned = false;
|
||||
|
||||
for (const [id, node] of nodesRef.current) {
|
||||
if (id === 'self') continue;
|
||||
if (node.lastActivity < cutoff) {
|
||||
nodesRef.current.delete(id);
|
||||
pruned = true;
|
||||
}
|
||||
}
|
||||
|
||||
if (pruned) {
|
||||
for (const [key, link] of linksRef.current) {
|
||||
const sourceId = typeof link.source === 'string' ? link.source : link.source.id;
|
||||
const targetId = typeof link.target === 'string' ? link.target : link.target.id;
|
||||
if (!nodesRef.current.has(sourceId) || !nodesRef.current.has(targetId)) {
|
||||
linksRef.current.delete(key);
|
||||
}
|
||||
}
|
||||
syncSimulation();
|
||||
}
|
||||
}, pruneIntervalMs);
|
||||
|
||||
return () => clearInterval(interval);
|
||||
}, [pruneStaleNodes, pruneStaleMinutes, syncSimulation]);
|
||||
|
||||
return useMemo(
|
||||
() => ({
|
||||
nodes: nodesRef.current,
|
||||
links: linksRef.current,
|
||||
particles: particlesRef.current,
|
||||
stats,
|
||||
expandContract,
|
||||
clearAndReset,
|
||||
}),
|
||||
[stats, expandContract, clearAndReset]
|
||||
);
|
||||
}
|
||||
78
frontend/src/test/visualizerTooltip.test.tsx
Normal file
78
frontend/src/test/visualizerTooltip.test.tsx
Normal file
@@ -0,0 +1,78 @@
|
||||
import { render, screen } from '@testing-library/react';
|
||||
import { describe, expect, it, vi } from 'vitest';
|
||||
|
||||
import { VisualizerTooltip } from '../components/visualizer/VisualizerTooltip';
|
||||
import type { GraphNode } from '../components/visualizer/shared';
|
||||
|
||||
function createNode(overrides: Partial<GraphNode> & Pick<GraphNode, 'id' | 'type'>): GraphNode {
|
||||
return {
|
||||
id: overrides.id,
|
||||
type: overrides.type,
|
||||
name: overrides.name ?? null,
|
||||
isAmbiguous: overrides.isAmbiguous ?? false,
|
||||
lastActivity: overrides.lastActivity ?? Date.now(),
|
||||
x: overrides.x ?? 0,
|
||||
y: overrides.y ?? 0,
|
||||
z: overrides.z ?? 0,
|
||||
probableIdentity: overrides.probableIdentity,
|
||||
ambiguousNames: overrides.ambiguousNames,
|
||||
lastActivityReason: overrides.lastActivityReason,
|
||||
};
|
||||
}
|
||||
|
||||
describe('VisualizerTooltip', () => {
|
||||
it('renders nothing without an active node', () => {
|
||||
const { container } = render(
|
||||
<VisualizerTooltip activeNodeId={null} nodes={new Map()} neighborIds={[]} />
|
||||
);
|
||||
|
||||
expect(container).toBeEmptyDOMElement();
|
||||
});
|
||||
|
||||
it('renders ambiguous node details and neighbors', () => {
|
||||
vi.useFakeTimers();
|
||||
vi.setSystemTime(new Date('2026-03-10T22:00:00Z'));
|
||||
|
||||
const node = createNode({
|
||||
id: '?32',
|
||||
type: 'repeater',
|
||||
name: 'Likely Relay',
|
||||
isAmbiguous: true,
|
||||
probableIdentity: 'Likely Relay',
|
||||
ambiguousNames: ['Relay A', 'Relay B'],
|
||||
lastActivity: new Date('2026-03-10T21:58:30Z').getTime(),
|
||||
lastActivityReason: 'Relayed GT',
|
||||
});
|
||||
const neighbor = createNode({
|
||||
id: 'abcd1234ef56',
|
||||
type: 'client',
|
||||
name: 'Neighbor Node',
|
||||
ambiguousNames: ['Alt Neighbor'],
|
||||
});
|
||||
|
||||
render(
|
||||
<VisualizerTooltip
|
||||
activeNodeId={node.id}
|
||||
nodes={
|
||||
new Map([
|
||||
[node.id, node],
|
||||
[neighbor.id, neighbor],
|
||||
])
|
||||
}
|
||||
neighborIds={[neighbor.id]}
|
||||
/>
|
||||
);
|
||||
|
||||
expect(screen.getByText('Likely Relay')).toBeInTheDocument();
|
||||
expect(screen.getByText('ID: ?32')).toBeInTheDocument();
|
||||
expect(screen.getByText('Type: repeater (ambiguous)')).toBeInTheDocument();
|
||||
expect(screen.getByText('Probably: Likely Relay')).toBeInTheDocument();
|
||||
expect(screen.getByText('Other possible: Relay A, Relay B')).toBeInTheDocument();
|
||||
expect(screen.getByText('Last active: 1m 30s ago')).toBeInTheDocument();
|
||||
expect(screen.getByText('Reason: Relayed GT')).toBeInTheDocument();
|
||||
expect(screen.getByText('Neighbor Node')).toBeInTheDocument();
|
||||
expect(screen.getByText('(Alt Neighbor)')).toBeInTheDocument();
|
||||
|
||||
vi.useRealTimers();
|
||||
});
|
||||
});
|
||||
Reference in New Issue
Block a user