@andespindola/brainlink 0.1.0-beta.99 → 1.0.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (46) hide show
  1. package/AGENTS.md +6 -6
  2. package/CHANGELOG.md +14 -0
  3. package/README.md +186 -38
  4. package/dist/application/add-note.js +13 -44
  5. package/dist/application/analyze-vault.js +1 -1
  6. package/dist/application/auto-migrate-configured-vault.js +37 -0
  7. package/dist/application/build-context.js +119 -20
  8. package/dist/application/canonical-context-links.js +209 -0
  9. package/dist/application/frontend/client-css.js +212 -42
  10. package/dist/application/frontend/client-html.js +42 -28
  11. package/dist/application/frontend/client-js.js +1294 -3222
  12. package/dist/application/frontend/client-render-worker-js.js +676 -0
  13. package/dist/application/get-graph-contexts.js +33 -0
  14. package/dist/application/get-graph-layout.js +62 -8
  15. package/dist/application/get-graph-stream-chunk.js +326 -0
  16. package/dist/application/get-graph-view.js +246 -0
  17. package/dist/application/graph-view-state.js +66 -0
  18. package/dist/application/import-legacy-sqlite.js +3 -33
  19. package/dist/application/index-vault.js +35 -22
  20. package/dist/application/migrate-context-links.js +79 -0
  21. package/dist/application/search-graph-node-ids.js +63 -3
  22. package/dist/application/server/routes.js +197 -12
  23. package/dist/cli/commands/read-commands.js +39 -3
  24. package/dist/cli/commands/vault-commands.js +182 -0
  25. package/dist/cli/commands/write-commands.js +147 -12
  26. package/dist/cli/main.js +2 -0
  27. package/dist/cli/runtime.js +10 -2
  28. package/dist/domain/context.js +1 -0
  29. package/dist/domain/graph-contexts.js +180 -0
  30. package/dist/domain/graph-layout.js +347 -21
  31. package/dist/domain/markdown.js +53 -9
  32. package/dist/infrastructure/config.js +105 -6
  33. package/dist/infrastructure/context-packs.js +122 -0
  34. package/dist/infrastructure/file-index.js +6 -3
  35. package/dist/infrastructure/index-state.js +2 -0
  36. package/dist/infrastructure/vault-migration-state.js +69 -0
  37. package/dist/infrastructure/volatile-memory.js +100 -0
  38. package/dist/mcp/http-server.js +97 -0
  39. package/dist/mcp/runtime.js +20 -0
  40. package/dist/mcp/server.js +36 -13
  41. package/dist/mcp/tools.js +203 -14
  42. package/docs/AGENT_USAGE.md +50 -5
  43. package/docs/ARCHITECTURE.md +11 -0
  44. package/docs/QUICKSTART.md +3 -1
  45. package/docs/RELEASE.md +4 -3
  46. package/package.json +3 -1
@@ -1,9 +1,34 @@
1
1
  import { createHash } from 'node:crypto';
2
- import { stat } from 'node:fs/promises';
2
+ import { mkdir, readFile, rename, stat, writeFile } from 'node:fs/promises';
3
+ import { dirname, join } from 'node:path';
4
+ import { addVisualContextEdges } from '../domain/graph-contexts.js';
3
5
  import { createCauliflowerGraphLayout } from '../domain/graph-layout.js';
4
6
  import { indexStoragePath } from '../infrastructure/file-index.js';
5
7
  import { getGraphSummary } from './get-graph-summary.js';
8
+ const graphLayoutVersion = 9;
6
9
  const graphLayoutCache = new Map();
10
+ const safeCacheSegment = (value, fallback) => value?.replace(/[^a-zA-Z0-9_-]/g, '_') || fallback;
11
+ const graphLayoutStoragePath = (vaultPath, options) => {
12
+ const agent = safeCacheSegment(options.agentId, 'all');
13
+ const context = safeCacheSegment(options.context, 'all-contexts');
14
+ return join(vaultPath, '.brainlink', `graph-layout-${agent}-${context}.json`);
15
+ };
16
+ const readPersistedLayout = async (vaultPath, databaseSignature, options) => {
17
+ try {
18
+ const parsed = JSON.parse(await readFile(graphLayoutStoragePath(vaultPath, options), 'utf8'));
19
+ return parsed.databaseSignature === databaseSignature && parsed.layoutVersion === graphLayoutVersion ? parsed : null;
20
+ }
21
+ catch {
22
+ return null;
23
+ }
24
+ };
25
+ const writePersistedLayout = async (vaultPath, options, cached) => {
26
+ const target = graphLayoutStoragePath(vaultPath, options);
27
+ const temp = `${target}.tmp`;
28
+ await mkdir(dirname(target), { recursive: true, mode: 0o700 });
29
+ await writeFile(temp, `${JSON.stringify(cached)}\n`, { encoding: 'utf8', mode: 0o600 });
30
+ await rename(temp, target);
31
+ };
7
32
  const readDatabaseSignature = async (vaultPath) => {
8
33
  try {
9
34
  const info = await stat(indexStoragePath(vaultPath));
@@ -22,20 +47,49 @@ const createGraphSignature = (graph) => {
22
47
  .update(`${graph.nodes.length}:${nodesSignature}|${graph.edges.length}:${edgesSignature}`)
23
48
  .digest('hex');
24
49
  };
25
- export const getGraphLayout = async (vaultPath, agentId) => {
50
+ const createLayout = (graph) => {
51
+ const rawLayout = createCauliflowerGraphLayout(graph);
52
+ return {
53
+ ...rawLayout,
54
+ nodes: rawLayout.nodes.map((node) => ({ ...node, content: '' }))
55
+ };
56
+ };
57
+ const filterGraphByContext = (graph, context) => {
58
+ const baseLayout = createCauliflowerGraphLayout(graph);
59
+ const selectedNodeIds = new Set(baseLayout.nodes
60
+ .filter((node) => node.segment === context)
61
+ .map((node) => node.id));
62
+ return {
63
+ nodes: graph.nodes.filter((node) => selectedNodeIds.has(node.id)),
64
+ edges: graph.edges.filter((edge) => selectedNodeIds.has(edge.source) && Boolean(edge.target && selectedNodeIds.has(edge.target)))
65
+ };
66
+ };
67
+ export const getGraphLayout = async (vaultPath, optionsOrAgentId) => {
68
+ const options = typeof optionsOrAgentId === 'string' ? { agentId: optionsOrAgentId } : optionsOrAgentId ?? {};
26
69
  const databaseSignature = await readDatabaseSignature(vaultPath);
27
- const cacheKey = `${vaultPath}:${agentId ?? ''}`;
70
+ const cacheKey = `${vaultPath}:${options.agentId ?? ''}:${options.context ?? ''}`;
28
71
  const cached = graphLayoutCache.get(cacheKey);
29
- if (cached?.databaseSignature === databaseSignature) {
72
+ if (cached?.databaseSignature === databaseSignature && cached.layoutVersion === graphLayoutVersion) {
30
73
  return {
31
74
  signature: cached.signature,
32
75
  layout: cached.layout
33
76
  };
34
77
  }
35
- const graph = await getGraphSummary(vaultPath, agentId);
36
- const signature = createGraphSignature(graph);
37
- const layout = createCauliflowerGraphLayout(graph);
38
- graphLayoutCache.set(cacheKey, { databaseSignature, signature, layout });
78
+ const persisted = await readPersistedLayout(vaultPath, databaseSignature, options);
79
+ if (persisted) {
80
+ graphLayoutCache.set(cacheKey, persisted);
81
+ return {
82
+ signature: persisted.signature,
83
+ layout: persisted.layout
84
+ };
85
+ }
86
+ const graph = addVisualContextEdges(await getGraphSummary(vaultPath, options.agentId));
87
+ const scopedGraph = options.context ? filterGraphByContext(graph, options.context) : graph;
88
+ const signature = createGraphSignature(scopedGraph);
89
+ const layout = createLayout(scopedGraph);
90
+ const nextCache = { layoutVersion: graphLayoutVersion, databaseSignature, signature, layout };
91
+ graphLayoutCache.set(cacheKey, nextCache);
92
+ await writePersistedLayout(vaultPath, options, nextCache);
39
93
  return {
40
94
  signature,
41
95
  layout
@@ -0,0 +1,326 @@
1
+ import { getGraphLayout } from './get-graph-layout.js';
2
+ const layoutCacheBySignature = new Map();
3
+ const maxLayoutCacheEntries = 6;
4
+ const farScaleThreshold = 0.22;
5
+ const midScaleThreshold = 0.78;
6
+ const viewportPaddingFactor = 0.18;
7
+ const maxNearEdgePerNode = 24;
8
+ const maxMidEdgePerNode = 12;
9
+ const maxFarEdgePerCluster = 8;
10
+ const inViewport = (item, input, padding) => {
11
+ const radius = item.radius ?? 24;
12
+ const paddedX = input.x - padding;
13
+ const paddedY = input.y - padding;
14
+ const paddedWidth = input.width + padding * 2;
15
+ const paddedHeight = input.height + padding * 2;
16
+ return (item.x + radius >= paddedX &&
17
+ item.x - radius <= paddedX + paddedWidth &&
18
+ item.y + radius >= paddedY &&
19
+ item.y - radius <= paddedY + paddedHeight);
20
+ };
21
+ const nodeDistanceToCenter = (item, input) => {
22
+ const cx = input.x + input.width / 2;
23
+ const cy = input.y + input.height / 2;
24
+ return Math.hypot(item.x - cx, item.y - cy);
25
+ };
26
+ const rankNodeRelevance = (node, input, degrees) => {
27
+ const degree = degrees.get(node.id) ?? 0;
28
+ const centerDistance = nodeDistanceToCenter(node, input);
29
+ const viewportRadius = Math.max(input.width, input.height) / 2;
30
+ const centerScore = 1 - Math.min(1, centerDistance / Math.max(viewportRadius, 1));
31
+ const degreeScore = Math.log1p(Math.max(0, degree));
32
+ const tagScore = Math.min(node.tags.length, 6) * 0.18;
33
+ return degreeScore * 1.1 + centerScore * 1.3 + tagScore;
34
+ };
35
+ const rankGroupRelevance = (group, input) => {
36
+ const centerDistance = nodeDistanceToCenter(group, input);
37
+ const viewportRadius = Math.max(input.width, input.height) / 2;
38
+ const centerScore = 1 - Math.min(1, centerDistance / Math.max(viewportRadius, 1));
39
+ const massScore = Math.log1p(group.nodeIds.length + group.childGroupIds.length * 2);
40
+ const edgeScore = Math.log1p(group.externalEdges.length + group.internalEdges.length);
41
+ return centerScore * 1.3 + massScore + edgeScore * 0.5;
42
+ };
43
+ const edgePriorityScore = (priority) => {
44
+ if (priority === 'critical')
45
+ return 4;
46
+ if (priority === 'high')
47
+ return 3;
48
+ if (priority === 'normal')
49
+ return 2;
50
+ return 1;
51
+ };
52
+ const edgeRank = (edge) => edge.weight * 1.35 + edgePriorityScore(edge.priority);
53
+ const createLayoutCache = (signature, nodes, edges, groups) => {
54
+ const nodeById = new Map(nodes.map((node) => [node.id, node]));
55
+ const groupById = new Map(groups.map((group) => [group.id, group]));
56
+ const degrees = new Map();
57
+ const adjacencyByNodeId = new Map();
58
+ const rankedAdjacencyByNodeId = new Map();
59
+ for (let index = 0; index < edges.length; index += 1) {
60
+ const edge = edges[index];
61
+ degrees.set(edge.source, (degrees.get(edge.source) ?? 0) + edge.weight);
62
+ const sourceAdjacency = adjacencyByNodeId.get(edge.source) ?? [];
63
+ sourceAdjacency.push(index);
64
+ adjacencyByNodeId.set(edge.source, sourceAdjacency);
65
+ if (!edge.target) {
66
+ continue;
67
+ }
68
+ degrees.set(edge.target, (degrees.get(edge.target) ?? 0) + edge.weight);
69
+ const targetAdjacency = adjacencyByNodeId.get(edge.target) ?? [];
70
+ targetAdjacency.push(index);
71
+ adjacencyByNodeId.set(edge.target, targetAdjacency);
72
+ }
73
+ for (const [nodeId, adjacency] of adjacencyByNodeId.entries()) {
74
+ rankedAdjacencyByNodeId.set(nodeId, [...adjacency].sort((leftIndex, rightIndex) => edgeRank(edges[rightIndex]) - edgeRank(edges[leftIndex])));
75
+ }
76
+ return {
77
+ signature,
78
+ degrees,
79
+ nodeById,
80
+ groupById,
81
+ adjacencyByNodeId,
82
+ rankedAdjacencyByNodeId
83
+ };
84
+ };
85
+ const getOrCreateLayoutCache = (signature, nodes, edges, groups) => {
86
+ const cached = layoutCacheBySignature.get(signature);
87
+ if (cached) {
88
+ return cached;
89
+ }
90
+ const next = createLayoutCache(signature, nodes, edges, groups);
91
+ layoutCacheBySignature.set(signature, next);
92
+ while (layoutCacheBySignature.size > maxLayoutCacheEntries) {
93
+ const oldest = layoutCacheBySignature.keys().next().value;
94
+ if (!oldest)
95
+ break;
96
+ layoutCacheBySignature.delete(oldest);
97
+ }
98
+ return next;
99
+ };
100
+ const selectTopByRelevance = (items, relevanceById, budget) => [...items]
101
+ .sort((left, right) => {
102
+ const relevanceDelta = (relevanceById.get(right.id) ?? 0) - (relevanceById.get(left.id) ?? 0);
103
+ if (relevanceDelta !== 0)
104
+ return relevanceDelta;
105
+ return left.id.localeCompare(right.id);
106
+ })
107
+ .slice(0, Math.max(1, budget));
108
+ const selectNearNodes = (nodes, cache, input) => {
109
+ const padding = Math.max(input.width, input.height) * viewportPaddingFactor;
110
+ const candidateNodes = nodes.length <= input.nodeBudget
111
+ ? nodes
112
+ : nodes.filter((node) => inViewport(node, input, padding));
113
+ const relevance = new Map();
114
+ for (let index = 0; index < candidateNodes.length; index += 1) {
115
+ const node = candidateNodes[index];
116
+ relevance.set(node.id, rankNodeRelevance(node, input, cache.degrees));
117
+ }
118
+ const selected = selectTopByRelevance(candidateNodes, relevance, input.nodeBudget);
119
+ return selected.map((node) => [
120
+ node.id,
121
+ node.title,
122
+ node.x,
123
+ node.y,
124
+ node.group,
125
+ node.segment,
126
+ 'node',
127
+ relevance.get(node.id) ?? 0
128
+ ]);
129
+ };
130
+ const selectMidNodes = (nodes, cache, input) => {
131
+ const padding = Math.max(input.width, input.height) * (viewportPaddingFactor + 0.08);
132
+ const candidateNodes = nodes.length <= input.nodeBudget
133
+ ? nodes
134
+ : nodes.filter((node) => inViewport(node, input, padding));
135
+ const relevance = new Map();
136
+ for (let index = 0; index < candidateNodes.length; index += 1) {
137
+ const node = candidateNodes[index];
138
+ relevance.set(node.id, rankNodeRelevance(node, input, cache.degrees) * 1.1);
139
+ }
140
+ const selected = selectTopByRelevance(candidateNodes, relevance, input.nodeBudget);
141
+ return selected.map((node) => [
142
+ node.id,
143
+ node.title,
144
+ node.x,
145
+ node.y,
146
+ node.group,
147
+ node.segment,
148
+ 'node',
149
+ relevance.get(node.id) ?? 0
150
+ ]);
151
+ };
152
+ const selectFarClusters = (nodes, input, nodeBudget) => {
153
+ const roots = createSegmentClusters(nodes);
154
+ const relevance = new Map();
155
+ for (let index = 0; index < roots.length; index += 1) {
156
+ const group = roots[index];
157
+ relevance.set(group.id, rankGroupRelevance(group, input));
158
+ }
159
+ const selected = selectTopByRelevance(roots, relevance, nodeBudget);
160
+ return selected.map((group) => [
161
+ `cluster:${group.id}`,
162
+ group.title,
163
+ group.x,
164
+ group.y,
165
+ group.group,
166
+ group.segment,
167
+ 'cluster',
168
+ relevance.get(group.id) ?? 0
169
+ ]);
170
+ };
171
+ const createSegmentClusters = (nodes) => {
172
+ const nodesBySegment = new Map();
173
+ nodes.forEach((node) => {
174
+ const bucket = nodesBySegment.get(node.segment) ?? [];
175
+ bucket.push(node);
176
+ nodesBySegment.set(node.segment, bucket);
177
+ });
178
+ return Array.from(nodesBySegment.entries()).map(([segment, segmentNodes]) => {
179
+ const center = segmentNodes.reduce((state, node) => ({
180
+ x: state.x + node.x,
181
+ y: state.y + node.y
182
+ }), { x: 0, y: 0 });
183
+ const x = center.x / Math.max(segmentNodes.length, 1);
184
+ const y = center.y / Math.max(segmentNodes.length, 1);
185
+ const radius = segmentNodes.reduce((largest, node) => Math.max(largest, Math.hypot(node.x - x, node.y - y)), 80);
186
+ return {
187
+ id: `segment:${segment}`,
188
+ level: 0,
189
+ parentId: null,
190
+ title: segment,
191
+ segment,
192
+ group: segmentNodes[0]?.group ?? segment,
193
+ x,
194
+ y,
195
+ radius: Math.max(radius + 96, 160),
196
+ nodeIds: segmentNodes.map((node) => node.id),
197
+ childGroupIds: [],
198
+ internalEdges: [],
199
+ externalEdges: []
200
+ };
201
+ });
202
+ };
203
+ const collectEdgesForNodes = (allEdges, cache, nodeRows, edgeBudget, maxEdgesPerNode) => {
204
+ const isClusterMode = nodeRows.length > 0 && nodeRows[0]?.[6] === 'cluster';
205
+ if (isClusterMode) {
206
+ const clusterSegments = new Set(nodeRows.map((row) => row[5]));
207
+ const clusterEdges = new Map();
208
+ for (let index = 0; index < allEdges.length; index += 1) {
209
+ const edge = allEdges[index];
210
+ if (!edge.target)
211
+ continue;
212
+ const sourceNode = cache.nodeById.get(edge.source);
213
+ const targetNode = cache.nodeById.get(edge.target);
214
+ if (!sourceNode || !targetNode)
215
+ continue;
216
+ const sourceCluster = sourceNode.segment;
217
+ const targetCluster = targetNode.segment;
218
+ if (!clusterSegments.has(sourceCluster) || !clusterSegments.has(targetCluster) || sourceCluster === targetCluster)
219
+ continue;
220
+ const key = sourceCluster < targetCluster ? `${sourceCluster}|${targetCluster}` : `${targetCluster}|${sourceCluster}`;
221
+ const current = clusterEdges.get(key);
222
+ if (!current || edgeRank(edge) > edgeRank(current)) {
223
+ clusterEdges.set(key, edge);
224
+ }
225
+ }
226
+ return [...clusterEdges.values()]
227
+ .sort((left, right) => edgeRank(right) - edgeRank(left))
228
+ .slice(0, Math.max(1, edgeBudget))
229
+ .map((edge) => [
230
+ `cluster:segment:${cache.nodeById.get(edge.source)?.segment ?? ''}`,
231
+ `cluster:segment:${cache.nodeById.get(edge.target ?? '')?.segment ?? ''}`,
232
+ edge.weight,
233
+ edge.priority
234
+ ])
235
+ .filter((edge) => edge[0] !== edge[1] && edge[0] !== 'cluster:segment:' && edge[1] !== 'cluster:segment:');
236
+ }
237
+ const nodeIds = new Set(nodeRows.map((row) => row[0]));
238
+ const collected = new Map();
239
+ for (const nodeId of nodeIds) {
240
+ const adjacency = cache.rankedAdjacencyByNodeId.get(nodeId);
241
+ if (!adjacency || adjacency.length === 0) {
242
+ continue;
243
+ }
244
+ let perNodeCount = 0;
245
+ for (let index = 0; index < adjacency.length; index += 1) {
246
+ if (perNodeCount >= maxEdgesPerNode) {
247
+ break;
248
+ }
249
+ const edge = allEdges[adjacency[index]];
250
+ if (!edge.target || !nodeIds.has(edge.source) || !nodeIds.has(edge.target)) {
251
+ continue;
252
+ }
253
+ const key = edge.source < edge.target ? `${edge.source}|${edge.target}` : `${edge.target}|${edge.source}`;
254
+ const current = collected.get(key);
255
+ if (!current || edgeRank(edge) > edgeRank(current)) {
256
+ collected.set(key, edge);
257
+ }
258
+ perNodeCount += 1;
259
+ }
260
+ }
261
+ return [...collected.values()]
262
+ .sort((left, right) => edgeRank(right) - edgeRank(left))
263
+ .slice(0, Math.max(1, edgeBudget))
264
+ .map((edge) => [edge.source, edge.target ?? '', edge.weight, edge.priority])
265
+ .filter((edge) => edge[1].length > 0);
266
+ };
267
+ const normalizeBudget = (value, fallback, min, max) => {
268
+ if (!Number.isFinite(value)) {
269
+ return fallback;
270
+ }
271
+ const rounded = Math.round(value);
272
+ if (rounded < min)
273
+ return min;
274
+ if (rounded > max)
275
+ return max;
276
+ return rounded;
277
+ };
278
+ export const getGraphStreamChunk = async (vaultPath, input) => {
279
+ const nodeBudget = normalizeBudget(input.nodeBudget, 1800, 80, 12_000);
280
+ const edgeBudget = normalizeBudget(input.edgeBudget, 5000, 120, 60_000);
281
+ const { signature, layout } = await getGraphLayout(vaultPath, {
282
+ agentId: input.agentId,
283
+ context: input.context
284
+ });
285
+ const groups = layout.groups ?? [];
286
+ const cache = getOrCreateLayoutCache(signature, layout.nodes, layout.edges, groups);
287
+ if (layout.nodes.length === 0) {
288
+ return {
289
+ signature,
290
+ mode: 'near',
291
+ nodes: [],
292
+ edges: [],
293
+ totals: {
294
+ nodes: 0,
295
+ edges: 0
296
+ }
297
+ };
298
+ }
299
+ const mode = input.scale < farScaleThreshold
300
+ ? 'far'
301
+ : input.scale < midScaleThreshold
302
+ ? 'mid'
303
+ : 'near';
304
+ const nodes = mode === 'far'
305
+ ? selectFarClusters(layout.nodes, input, nodeBudget)
306
+ : mode === 'mid'
307
+ ? selectMidNodes(layout.nodes, cache, {
308
+ ...input,
309
+ nodeBudget
310
+ })
311
+ : selectNearNodes(layout.nodes, cache, {
312
+ ...input,
313
+ nodeBudget
314
+ });
315
+ const edges = collectEdgesForNodes(layout.edges, cache, nodes, edgeBudget, mode === 'near' ? maxNearEdgePerNode : mode === 'mid' ? maxMidEdgePerNode : maxFarEdgePerCluster);
316
+ return {
317
+ signature,
318
+ mode,
319
+ nodes,
320
+ edges,
321
+ totals: {
322
+ nodes: layout.nodes.length,
323
+ edges: layout.edges.length
324
+ }
325
+ };
326
+ };
@@ -0,0 +1,246 @@
1
+ import { getGraphLayout } from './get-graph-layout.js';
2
+ const macroScale = 0.24;
3
+ const microCoverage = 0.72;
4
+ const nodeLimit = 1000;
5
+ const edgeLimit = 1400;
6
+ const groupEdgeLimit = 900;
7
+ const inViewport = (item, input) => {
8
+ const radius = item.radius ?? 48;
9
+ return (item.x + radius >= input.x &&
10
+ item.x - radius <= input.x + input.width &&
11
+ item.y + radius >= input.y &&
12
+ item.y - radius <= input.y + input.height);
13
+ };
14
+ const groupCoverage = (group, input) => {
15
+ const viewportRadius = Math.max(input.width, input.height) / 2;
16
+ const centerX = input.x + input.width / 2;
17
+ const centerY = input.y + input.height / 2;
18
+ const centerDistance = Math.hypot(group.x - centerX, group.y - centerY);
19
+ const fitCoverage = Math.min(1, childGraphRenderRadius(group) / Math.max(viewportRadius, 1));
20
+ const centerCoverage = 1 - Math.min(1, centerDistance / Math.max(viewportRadius, 1));
21
+ return fitCoverage * 0.72 + centerCoverage * 0.28;
22
+ };
23
+ const childGraphRenderRadius = (group) => {
24
+ const childCount = Math.max(group.nodeIds.length, group.childGroupIds.length, 1);
25
+ return Math.max(420, Math.min(1800, Math.sqrt(childCount) * 24));
26
+ };
27
+ const groupRenderRadius = (group) => {
28
+ const childCount = Math.max(group.nodeIds.length, group.childGroupIds.length, 1);
29
+ return 10 + Math.min(Math.log2(childCount + 1) * 4.2, 22);
30
+ };
31
+ const arrangeGraphLevelGroups = (groups) => {
32
+ if (groups.length <= 1) {
33
+ return groups.map((group) => ({ ...group, radius: groupRenderRadius(group) }));
34
+ }
35
+ const centerGroup = groups
36
+ .map((group) => ({
37
+ group,
38
+ score: Math.max(group.nodeIds.length, group.childGroupIds.length, 1) + group.externalEdges.length
39
+ }))
40
+ .sort((left, right) => right.score - left.score || left.group.title.localeCompare(right.group.title))[0]?.group;
41
+ const outerGroups = groups
42
+ .filter((group) => group.id !== centerGroup?.id)
43
+ .sort((left, right) => left.segment.localeCompare(right.segment) || left.title.localeCompare(right.title));
44
+ const baseRadius = Math.max(520, Math.min(2200, Math.sqrt(groups.length) * 135));
45
+ const goldenAngle = Math.PI * (3 - Math.sqrt(5));
46
+ const arranged = centerGroup
47
+ ? [{ ...centerGroup, x: 0, y: 0, radius: groupRenderRadius(centerGroup) }]
48
+ : [];
49
+ outerGroups.forEach((group, index) => {
50
+ const ringRadius = baseRadius * Math.sqrt((index + 1) / Math.max(outerGroups.length, 1));
51
+ const angle = index * goldenAngle;
52
+ arranged.push({
53
+ ...group,
54
+ x: Math.cos(angle) * ringRadius,
55
+ y: Math.sin(angle) * ringRadius,
56
+ radius: groupRenderRadius(group)
57
+ });
58
+ });
59
+ return arranged;
60
+ };
61
+ const distanceToViewportCenter = (item, input) => {
62
+ const centerX = input.x + input.width / 2;
63
+ const centerY = input.y + input.height / 2;
64
+ return Math.hypot(item.x - centerX, item.y - centerY);
65
+ };
66
+ const selectViewportItemsWithFill = (items, input, limit = nodeLimit) => {
67
+ const visible = items.filter((item) => inViewport(item, input));
68
+ if (visible.length >= limit) {
69
+ return visible.slice(0, limit);
70
+ }
71
+ const selectedIds = new Set(visible.map((item) => item.id));
72
+ const fill = items
73
+ .filter((item) => !selectedIds.has(item.id))
74
+ .sort((left, right) => distanceToViewportCenter(left, input) - distanceToViewportCenter(right, input) || left.id.localeCompare(right.id))
75
+ .slice(0, Math.max(0, limit - visible.length));
76
+ return visible.concat(fill);
77
+ };
78
+ const groupNode = (group) => [
79
+ `group:${group.id}`,
80
+ group.title,
81
+ group.x,
82
+ group.y,
83
+ group.group,
84
+ group.segment,
85
+ 'group'
86
+ ];
87
+ const realNode = (node) => [
88
+ node.id,
89
+ node.title,
90
+ node.x,
91
+ node.y,
92
+ node.group,
93
+ node.segment,
94
+ 'node'
95
+ ];
96
+ const descendants = (group, groupById) => group.nodeIds.length > 0
97
+ ? group.nodeIds
98
+ : group.childGroupIds.flatMap((childId) => {
99
+ const child = groupById.get(childId);
100
+ return child ? descendants(child, groupById) : [];
101
+ });
102
+ const aggregateGroupEdges = (groups, edges, groupById) => {
103
+ const groupNodeByNodeId = new Map();
104
+ groups.forEach((group) => {
105
+ descendants(group, groupById).forEach((nodeId) => groupNodeByNodeId.set(nodeId, `group:${group.id}`));
106
+ });
107
+ const selected = new Map();
108
+ edges.forEach((edge) => {
109
+ if (!edge.target)
110
+ return;
111
+ const source = groupNodeByNodeId.get(edge.source);
112
+ const target = groupNodeByNodeId.get(edge.target);
113
+ if (!source || !target || source === target)
114
+ return;
115
+ const key = source < target ? `${source}|${target}` : `${target}|${source}`;
116
+ const current = selected.get(key);
117
+ if (current && current[2] >= edge.weight)
118
+ return;
119
+ selected.set(key, [source, target, edge.weight, edge.priority]);
120
+ });
121
+ const degreeCounts = new Map();
122
+ return Array.from(selected.values())
123
+ .sort((left, right) => right[2] - left[2] || left[0].localeCompare(right[0]) || left[1].localeCompare(right[1]))
124
+ .filter((edge) => {
125
+ const sourceCount = degreeCounts.get(edge[0]) ?? 0;
126
+ const targetCount = degreeCounts.get(edge[1]) ?? 0;
127
+ if (sourceCount >= 3 || targetCount >= 3) {
128
+ return false;
129
+ }
130
+ degreeCounts.set(edge[0], sourceCount + 1);
131
+ degreeCounts.set(edge[1], targetCount + 1);
132
+ return true;
133
+ })
134
+ .slice(0, groupEdgeLimit);
135
+ };
136
+ const realEdges = (edges, nodeIds) => edges
137
+ .filter((edge) => Boolean(edge.target && nodeIds.has(edge.source) && nodeIds.has(edge.target)))
138
+ .slice(0, edgeLimit)
139
+ .map((edge) => [edge.source, edge.target, edge.weight, edge.priority]);
140
+ const degreeMap = (edges) => edges.reduce((degrees, edge) => {
141
+ degrees.set(edge.source, (degrees.get(edge.source) ?? 0) + edge.weight);
142
+ if (edge.target) {
143
+ degrees.set(edge.target, (degrees.get(edge.target) ?? 0) + edge.weight);
144
+ }
145
+ return degrees;
146
+ }, new Map());
147
+ const arrangeChildGraphNodes = (nodes, group, degrees) => {
148
+ if (nodes.length <= 1) {
149
+ return nodes.map((node) => ({ ...node, x: group.x, y: group.y }));
150
+ }
151
+ const centerNode = nodes
152
+ .map((node) => ({
153
+ node,
154
+ score: (degrees.get(node.id) ?? 0) + node.tags.length
155
+ }))
156
+ .sort((left, right) => right.score - left.score || left.node.title.localeCompare(right.node.title))[0]?.node;
157
+ const outerNodes = nodes
158
+ .filter((node) => node.id !== centerNode?.id)
159
+ .sort((left, right) => {
160
+ const degreeDelta = (degrees.get(right.id) ?? 0) - (degrees.get(left.id) ?? 0);
161
+ if (degreeDelta !== 0)
162
+ return degreeDelta;
163
+ return left.title.localeCompare(right.title);
164
+ });
165
+ const targetRadius = childGraphRenderRadius(group);
166
+ const goldenAngle = Math.PI * (3 - Math.sqrt(5));
167
+ const arranged = centerNode
168
+ ? [{ ...centerNode, x: group.x, y: group.y }]
169
+ : [];
170
+ outerNodes.forEach((node, index) => {
171
+ const ringRadius = targetRadius * Math.sqrt((index + 1) / Math.max(outerNodes.length, 1));
172
+ const angle = index * goldenAngle;
173
+ arranged.push({
174
+ ...node,
175
+ x: group.x + Math.cos(angle) * ringRadius,
176
+ y: group.y + Math.sin(angle) * ringRadius
177
+ });
178
+ });
179
+ return arranged;
180
+ };
181
+ const limitEdges = (edges) => edges.slice(0, edgeLimit);
182
+ export const getGraphView = async (vaultPath, input) => {
183
+ const { signature, layout } = await getGraphLayout(vaultPath, {
184
+ agentId: input.agentId,
185
+ context: input.context
186
+ });
187
+ const groups = layout.groups ?? [];
188
+ const degrees = degreeMap(layout.edges);
189
+ const groupById = new Map(groups.map((group) => [group.id, group]));
190
+ if (groups.length === 0) {
191
+ const nodes = layout.nodes.filter((node) => inViewport(node, input)).slice(0, nodeLimit);
192
+ const nodeIds = new Set(nodes.map((node) => node.id));
193
+ const viewNodes = nodes.map(realNode);
194
+ return {
195
+ signature,
196
+ mode: 'flat',
197
+ nodes: viewNodes,
198
+ edges: limitEdges(realEdges(layout.edges, nodeIds)),
199
+ totals: {
200
+ nodes: layout.nodes.length,
201
+ edges: layout.edges.length
202
+ }
203
+ };
204
+ }
205
+ const rootGroups = arrangeGraphLevelGroups(groups.filter((group) => group.parentId === null));
206
+ const leafGroups = arrangeGraphLevelGroups(groups.filter((group) => group.nodeIds.length > 0));
207
+ const visibleGroups = selectViewportItemsWithFill(rootGroups, input);
208
+ const focused = leafGroups
209
+ .filter((group) => group.nodeIds.length > 0 && inViewport(group, input))
210
+ .map((group) => ({ group, coverage: groupCoverage(group, input) }))
211
+ .sort((left, right) => right.coverage - left.coverage)[0];
212
+ if (focused && input.scale >= macroScale && focused.coverage >= microCoverage) {
213
+ const nodeIds = new Set(focused.group.nodeIds);
214
+ const arrangedNodes = arrangeChildGraphNodes(layout.nodes.filter((node) => nodeIds.has(node.id)), focused.group, degrees);
215
+ const nodesInViewport = arrangedNodes
216
+ .filter((node) => inViewport(node, input))
217
+ .slice(0, nodeLimit);
218
+ const nodes = nodesInViewport.length > 0
219
+ ? nodesInViewport
220
+ : arrangedNodes.slice(0, nodeLimit);
221
+ const visibleNodeIds = new Set(nodes.map((node) => node.id));
222
+ const viewNodes = nodes.map(realNode);
223
+ return {
224
+ signature,
225
+ mode: 'micro',
226
+ nodes: viewNodes,
227
+ edges: limitEdges(realEdges(layout.edges, visibleNodeIds)),
228
+ totals: {
229
+ nodes: layout.nodes.length,
230
+ edges: layout.edges.length
231
+ }
232
+ };
233
+ }
234
+ const groupsToRender = visibleGroups.slice(0, nodeLimit);
235
+ const viewNodes = groupsToRender.map(groupNode);
236
+ return {
237
+ signature,
238
+ mode: 'macro',
239
+ nodes: viewNodes,
240
+ edges: limitEdges(aggregateGroupEdges(groupsToRender, layout.edges, groupById)),
241
+ totals: {
242
+ nodes: layout.nodes.length,
243
+ edges: layout.edges.length
244
+ }
245
+ };
246
+ };