@kage-core/kage-graph-mcp 1.1.1 → 1.1.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@kage-core/kage-graph-mcp",
-  "version": "1.1.1",
+  "version": "1.1.2",
   "description": "Local-first repo memory, code graph, and recall MCP server for coding agents",
   "main": "dist/index.js",
   "files": [

package/viewer/app.js

@@ -35,21 +35,37 @@
 
   var palette = {
     repo: "#41ff8f",
-    memory: "#b88cff",
+    memory: "#41ff8f",
     path: "#ff6b6b",
     tag: "#ffd166",
     package: "#6ad7ff",
     command: "#9be7c0",
     memory_type: "#41ff8f",
     file: "#6ad7ff",
-    symbol: "#b88cff",
-    route: "#ff8fab",
+    symbol: "#9be7c0",
+    route: "#6ad7ff",
     test: "#ffd166",
-    external: "#93a4a0",
+    external: "#62776b",
     script: "#6ad7ff",
     default: "#9be7c0"
   };
 
+  var graphPalette = {
+    background: "#020503",
+    grid: "rgba(65,255,143,0.040)",
+    gridStrong: "rgba(65,255,143,0.070)",
+    text: "#d7f9df",
+    muted: "#6ea77d",
+    memory: "#41ff8f",
+    code: "#6ad7ff",
+    amber: "#ffd166",
+    danger: "#ff6b6b",
+    dependency: "#62776b",
+    body: "rgba(4,12,8,0.88)",
+    bodyCode: "rgba(5,16,18,0.88)",
+    bodyMemory: "rgba(5,18,10,0.90)"
+  };
+
   var els = {
     graphFile: document.getElementById("graphFile"),
     graphSummary: document.getElementById("graphSummary"),
@@ -454,7 +470,7 @@
   function render() {
     if (!state.graph) return;
 
-    var query = normalize(els.searchInput.value);
+    var query = parseSearchQuery(els.searchInput.value);
     var mode = els.viewMode.value;
     var type = els.typeFilter.value;
     var relation = els.relationFilter.value;
@@ -464,7 +480,7 @@
     state.entities.forEach(function (entity) {
       if (mode !== "combined" && entity.graph_kind !== mode) return;
       var passesType = !type || entity.type === type;
-      var passesSearch = !query || searchableText(entity).indexOf(query) !== -1;
+      var passesSearch = matchesSearchQuery(entity, query);
       if (passesType && passesSearch) matchedEntityIds.add(entity.id);
     });
 
@@ -472,7 +488,7 @@
       if (mode !== "combined" && edge.graph_kind !== mode) return;
       var fromMatched = matchedEntityIds.has(edge.from);
       var toMatched = matchedEntityIds.has(edge.to);
-      var edgeMatchesSearch = !query || searchableText(edge).indexOf(query) !== -1;
+      var edgeMatchesSearch = matchesSearchQuery(edge, query);
       var passesRelation = !relation || edge.relation === relation;
       if (passesRelation && (edgeMatchesSearch || fromMatched || toMatched)) {
         matchedEdgeIds.add(edge.id);
@@ -483,7 +499,7 @@
       }
     });
 
-    if (!query && !type && !relation) {
+    if (!query.active && !type && !relation) {
       matchedEntityIds = new Set(state.entities.filter(function (entity) { return mode === "combined" || entity.graph_kind === mode; }).map(function (entity) { return entity.id; }));
       matchedEdgeIds = new Set(state.edges.filter(function (edge) { return mode === "combined" || edge.graph_kind === mode; }).map(function (edge) { return edge.id; }));
     }
@@ -520,7 +536,7 @@
       Array.from(entities).forEach(function (id) {
         var entity = state.entityById.get(id);
         if (!entity) return;
-        if (isDependencyEntity(entity) && !(options.query && searchableText(entity).indexOf(options.query) !== -1)) {
+        if (isDependencyEntity(entity) && !matchesSearchQuery(entity, options.query)) {
           entities.delete(id);
         }
       });
@@ -751,9 +767,15 @@
 
   function drawCanvasGrid(ctx, width, height) {
     ctx.save();
-    ctx.fillStyle = "#020503";
+    ctx.fillStyle = graphPalette.background;
     ctx.fillRect(0, 0, width, height);
-    ctx.strokeStyle = "rgba(65,255,143,0.045)";
+    var gradient = ctx.createRadialGradient(width * 0.52, height * 0.44, 40, width * 0.52, height * 0.44, Math.max(width, height) * 0.72);
+    gradient.addColorStop(0, "rgba(65,255,143,0.080)");
+    gradient.addColorStop(0.48, "rgba(65,255,143,0.018)");
+    gradient.addColorStop(1, "rgba(2,5,3,0)");
+    ctx.fillStyle = gradient;
+    ctx.fillRect(0, 0, width, height);
+    ctx.strokeStyle = graphPalette.grid;
     ctx.lineWidth = 1;
     var grid = 28;
     for (var x = 0; x < width; x += grid) {
@@ -768,22 +790,29 @@
       ctx.lineTo(width, y);
       ctx.stroke();
     }
+    ctx.strokeStyle = graphPalette.gridStrong;
+    ctx.beginPath();
+    ctx.moveTo(0, height / 2);
+    ctx.lineTo(width, height / 2);
+    ctx.moveTo(width / 2, 0);
+    ctx.lineTo(width / 2, height);
+    ctx.stroke();
     ctx.restore();
   }
 
   function drawCanvasEdges(ctx) {
     var nodeMap = state.sim.nodeById;
     var focusId = focusedCanvasNodeId();
-    var query = normalize(els.searchInput.value);
+    var query = parseSearchQuery(els.searchInput.value);
     var dense = state.sim.nodes.length > 55;
     state.sim.edges.forEach(function (edge) {
       var from = nodeMap.get(edge.from);
       var to = nodeMap.get(edge.to);
       if (!from || !to) return;
       var connected = focusId && (edge.from === focusId || edge.to === focusId);
-      var matches = !query || searchableText(edge).indexOf(query) !== -1 || searchableText(from.entity).indexOf(query) !== -1 || searchableText(to.entity).indexOf(query) !== -1;
-      var alpha = !matches ? 0.04 : focusId ? (connected ? 0.78 : 0.07) : (dense ? 0.18 : 0.34);
-      var color = hexToRgb(palette[from.entity.type] || palette[from.entity.graph_kind] || palette.default);
+      var matches = matchesSearchQuery(edge, query) || matchesSearchQuery(from.entity, query) || matchesSearchQuery(to.entity, query);
+      var alpha = !matches ? 0.035 : focusId ? (connected ? 0.62 : 0.055) : (dense ? 0.13 : 0.22);
+      var color = hexToRgb(edgeThemeColor(edge, from.entity, to.entity));
       var dx = to.x - from.x;
       var dy = to.y - from.y;
       var dist = Math.max(1, Math.sqrt(dx * dx + dy * dy));
@@ -794,7 +823,7 @@
       ctx.moveTo(from.x, from.y);
       ctx.quadraticCurveTo(cx, cy, to.x, to.y);
       ctx.strokeStyle = "rgba(" + color.r + "," + color.g + "," + color.b + "," + alpha + ")";
-      ctx.lineWidth = connected ? 2.8 : 1.2;
+      ctx.lineWidth = connected ? 2.2 : 1;
       ctx.stroke();
       if (connected || (!dense && state.sim.zoom > 1.25)) drawArrow(ctx, from, to, cx, cy, color, alpha);
       if (connected && state.sim.zoom > 0.62) drawEdgeLabel(ctx, edge, cx, cy);
@@ -803,7 +832,7 @@
 
   function drawCanvasNodes(ctx) {
     var focusId = focusedCanvasNodeId();
-    var query = normalize(els.searchInput.value);
+    var query = parseSearchQuery(els.searchInput.value);
     var dense = state.sim.nodes.length > 55;
     state.sim.nodes.forEach(function (node) {
       var entity = node.entity;
@@ -812,35 +841,42 @@
       var connected = focusId && (node.id === focusId || state.sim.edges.some(function (edge) {
         return (edge.from === focusId && edge.to === node.id) || (edge.to === focusId && edge.from === node.id);
       }));
-      var matches = !query || searchableText(entity).indexOf(query) !== -1;
+      var matches = matchesSearchQuery(entity, query);
       var alpha = !matches ? 0.12 : focusId && !connected ? 0.20 : 1;
-      var color = palette[entity.type] || palette[entity.graph_kind] || palette.default;
+      var color = nodeThemeColor(entity);
       ctx.save();
       ctx.globalAlpha = alpha;
-      if (selected || hovered || (!focusId && !dense)) {
+      if (selected || hovered) {
         ctx.shadowColor = color;
-        ctx.shadowBlur = selected ? 20 : hovered ? 16 : 5;
+        ctx.shadowBlur = selected ? 14 : 10;
       }
       drawNodeShape(ctx, node.x, node.y, node.r, entity);
-      var grad = ctx.createRadialGradient(node.x - node.r * 0.3, node.y - node.r * 0.3, 1, node.x, node.y, node.r * 1.35);
-      grad.addColorStop(0, brighten(color, 54));
-      grad.addColorStop(1, color);
-      ctx.fillStyle = grad;
+      ctx.fillStyle = nodeFillColor(entity);
       ctx.fill();
+      ctx.strokeStyle = color;
+      ctx.lineWidth = selected || hovered ? 2.2 : 1.2;
+      ctx.stroke();
+      if (entity.graph_kind === "memory") {
+        ctx.fillStyle = color;
+        ctx.globalAlpha = alpha * 0.85;
+        ctx.beginPath();
+        ctx.arc(node.x, node.y, Math.max(2.4, node.r * 0.18), 0, Math.PI * 2);
+        ctx.fill();
+      }
       ctx.restore();
 
       if (selected || hovered) {
         ctx.save();
         drawNodeShape(ctx, node.x, node.y, node.r + 4, entity);
         ctx.strokeStyle = color;
-        ctx.lineWidth = selected ? 3 : 2;
+        ctx.lineWidth = selected ? 2.6 : 1.8;
         ctx.shadowColor = color;
-        ctx.shadowBlur = 12;
+        ctx.shadowBlur = 8;
         ctx.stroke();
         ctx.restore();
       }
 
-      var shouldLabel = matches && (selected || hovered || (query && matches) || (!dense && state.sim.zoom > 0.75) || (dense && state.sim.zoom > 1.55 && node.r > 13));
+      var shouldLabel = matches && (selected || hovered || (query.active && matches) || (!dense && state.sim.zoom > 0.75) || (dense && state.sim.zoom > 1.55 && node.r > 13));
       if (shouldLabel) drawNodeLabel(ctx, node, selected || hovered);
     });
   }
@@ -851,8 +887,8 @@
     var tipY = to.y - to.r * Math.sin(angle);
     ctx.beginPath();
     ctx.moveTo(tipX, tipY);
-    ctx.lineTo(tipX - 8 * Math.cos(angle - 0.35), tipY - 8 * Math.sin(angle - 0.35));
-    ctx.lineTo(tipX - 8 * Math.cos(angle + 0.35), tipY - 8 * Math.sin(angle + 0.35));
+    ctx.lineTo(tipX - 6 * Math.cos(angle - 0.32), tipY - 6 * Math.sin(angle - 0.32));
+    ctx.lineTo(tipX - 6 * Math.cos(angle + 0.32), tipY - 6 * Math.sin(angle + 0.32));
     ctx.closePath();
     ctx.fillStyle = "rgba(" + color.r + "," + color.g + "," + color.b + "," + Math.min(0.85, alpha + 0.10) + ")";
     ctx.fill();
@@ -861,9 +897,9 @@
   function drawEdgeLabel(ctx, edge, x, y) {
     var inv = 1 / state.sim.zoom;
     ctx.save();
-    ctx.font = "700 " + (10 * inv).toFixed(1) + "px ui-monospace, Menlo, monospace";
+    ctx.font = "700 " + (9 * inv).toFixed(1) + "px ui-monospace, Menlo, monospace";
     ctx.textAlign = "center";
-    ctx.fillStyle = "rgba(215,249,223,0.82)";
+    ctx.fillStyle = "rgba(155,231,192,0.72)";
     ctx.fillText(shortName(edge.relation || "related", 22), x, y - 5 * inv);
     ctx.restore();
   }
@@ -872,17 +908,17 @@
     var inv = 1 / state.sim.zoom;
     var label = shortName(displayName(node.entity), strong ? 30 : 20);
     ctx.save();
-    ctx.font = (strong ? "800 " : "700 ") + (12 * inv).toFixed(1) + "px ui-monospace, Menlo, monospace";
+    ctx.font = (strong ? "800 " : "700 ") + (11 * inv).toFixed(1) + "px ui-monospace, Menlo, monospace";
     var width = ctx.measureText(label).width + 16 * inv;
     var height = 20 * inv;
     var x = node.x - width / 2;
     var y = node.y + node.r + 8 * inv;
-    ctx.fillStyle = "rgba(3,6,4,0.88)";
+    ctx.fillStyle = "rgba(2,5,3,0.92)";
     roundedRect(ctx, x, y, width, height, 4 * inv);
     ctx.fill();
-    ctx.strokeStyle = strong ? "rgba(65,255,143,0.55)" : "rgba(65,255,143,0.18)";
+    ctx.strokeStyle = strong ? "rgba(65,255,143,0.45)" : "rgba(65,255,143,0.14)";
     ctx.stroke();
-    ctx.fillStyle = strong ? "#d7f9df" : "#9be7c0";
+    ctx.fillStyle = strong ? graphPalette.text : graphPalette.muted;
     ctx.textAlign = "center";
     ctx.fillText(label, node.x, y + 14 * inv);
     ctx.restore();
@@ -919,6 +955,30 @@
     ctx.arc(x, y, r, 0, Math.PI * 2);
   }
 
+  function nodeThemeColor(entity) {
+    if (isDependencyEntity(entity) || entity.type === "external") return graphPalette.dependency;
+    if (entity.type === "test" || entity.type === "tag") return graphPalette.amber;
+    if (entity.type === "bug_fix" || entity.type === "path") return graphPalette.danger;
+    if (entity.graph_kind === "memory" || ["memory", "repo", "memory_type", "decision", "runbook", "workflow", "convention", "gotcha", "reference", "policy"].indexOf(entity.type) !== -1) return graphPalette.memory;
+    if (entity.graph_kind === "code" || ["file", "symbol", "route", "script", "command", "package"].indexOf(entity.type) !== -1) return graphPalette.code;
+    return graphPalette.muted;
+  }
+
+  function nodeFillColor(entity) {
+    if (isDependencyEntity(entity) || entity.type === "external") return "rgba(7,13,10,0.88)";
+    if (entity.graph_kind === "memory" || entity.type === "memory") return graphPalette.bodyMemory;
+    if (entity.graph_kind === "code") return graphPalette.bodyCode;
+    return graphPalette.body;
+  }
+
+  function edgeThemeColor(edge, fromEntity, toEntity) {
+    if (edge.relation && /test|covers/i.test(edge.relation)) return graphPalette.amber;
+    if (edge.relation && /invalid|risk|missing|bug/i.test(edge.relation)) return graphPalette.danger;
+    if (isDependencyEntity(fromEntity) || isDependencyEntity(toEntity)) return graphPalette.dependency;
+    if (fromEntity.graph_kind === "memory" || toEntity.graph_kind === "memory") return graphPalette.memory;
+    return graphPalette.code;
+  }
+
   function renderSvg() {
     els.edgeLayer.textContent = "";
     els.nodeLayer.textContent = "";
@@ -1461,7 +1521,7 @@
     }
     var entity = node.entity;
     var relationCount = state.sim.edges.filter(function (edge) { return edge.from === node.id || edge.to === node.id; }).length;
-    var color = palette[entity.type] || palette[entity.graph_kind] || palette.default;
+    var color = nodeThemeColor(entity);
     els.tooltip.innerHTML = [
       "<div class=\"tt-name\"></div>",
       "<div class=\"tt-type\"></div>",
@@ -1653,6 +1713,77 @@
     return normalize(JSON.stringify(value || {}));
   }
 
+  function parseSearchQuery(value) {
+    var raw = normalize(value);
+    var tokens = raw
+      .replace(/[^a-z0-9_./:-]+/g, " ")
+      .split(/\s+/)
+      .map(searchStem)
+      .filter(Boolean)
+      .filter(function (token) { return !SEARCH_STOP_WORDS.has(token); });
+    if ((tokens.indexOf("run") !== -1 || tokens.indexOf("runn") !== -1) && tokens.indexOf("test") !== -1) tokens.push("runtest");
+    var groups = tokens.map(function (token) {
+      return unique([token].concat(SEARCH_SYNONYMS[token] || []).map(searchStem).filter(Boolean));
+    });
+    return {
+      active: raw.trim().length > 0,
+      raw: raw,
+      tokens: unique(groups.reduce(function (all, group) { return all.concat(group); }, [])),
+      groups: groups
+    };
+  }
+
+  var SEARCH_STOP_WORDS = new Set([
+    "a", "an", "and", "are", "about", "can", "do", "does", "for", "from", "how", "i", "in", "is", "it", "me", "of", "on", "or", "please", "show", "that", "the", "there", "this", "to", "what", "when", "where", "which", "who", "why", "with"
+  ]);
+
+  var SEARCH_SYNONYMS = {
+    memory: ["packet", "runbook", "decision", "workflow", "gotcha", "reference"],
+    test: ["tests", "testing", "vitest", "jest", "pytest", "spec"],
+    run: ["running", "command", "script", "npm", "pnpm", "yarn"],
+    runn: ["run", "running", "command", "script", "npm", "pnpm", "yarn"],
+    runtest: ["run", "test", "runbook", "command"],
+    start: ["serve", "dev", "launch"],
+    build: ["compile", "tsc"],
+    bug: ["fix", "gotcha", "error"],
+    route: ["endpoint", "api"],
+    file: ["path"],
+    dependency: ["package", "external", "deps"]
+  };
+
+  function matchesSearchQuery(value, query) {
+    if (!query || !query.active) return true;
+    var text = searchableText(value);
+    if (query.raw && text.indexOf(query.raw) !== -1) return true;
+    if (!query.tokens.length) return true;
+    var textTokens = new Set(text
+      .replace(/[^a-z0-9_./:-]+/g, " ")
+      .split(/\s+/)
+      .map(searchStem)
+      .filter(Boolean));
+    var requiredGroups = (query.groups || []).filter(function (group) {
+      return group.some(function (token) { return SEARCH_SOFT_TOKENS.has(token) ? text.indexOf(token) !== -1 || textTokens.has(token) : true; });
+    });
+    if (!requiredGroups.length) requiredGroups = query.groups || [query.tokens];
+    return requiredGroups.every(function (group) {
+      return group.some(function (token) {
+        if (text.indexOf(token) !== -1) return true;
+        return textTokens.has(token);
+      });
+    });
+  }
+
+  var SEARCH_SOFT_TOKENS = new Set(["memory", "packet", "about"]);
+
+  function searchStem(value) {
+    var token = String(value || "").toLowerCase();
+    if (token.length > 5 && token.endsWith("ing")) token = token.slice(0, -3);
+    if (token.length > 4 && token.endsWith("ies")) token = token.slice(0, -3) + "y";
+    if (token.length > 4 && token.endsWith("es")) token = token.slice(0, -2);
+    if (token.length > 3 && token.endsWith("s")) token = token.slice(0, -1);
+    return token;
+  }
+
   function shortName(value, max) {
     var text = String(value || "");
     return text.length > max ? text.slice(0, Math.max(1, max - 1)) + "..." : text;

package/viewer/styles.css

@@ -331,7 +331,7 @@ input:focus, select:focus, button:focus, .file-picker:focus-within {
   height: 100%;
   overflow: hidden;
   background:
-    radial-gradient(circle at 52% 46%, rgba(65, 255, 143, 0.10), transparent 44%),
+    radial-gradient(circle at 52% 46%, rgba(65, 255, 143, 0.055), transparent 44%),
     #020503;
 }
 
@@ -367,10 +367,10 @@ input:focus, select:focus, button:focus, .file-picker:focus-within {
   display: none;
   max-width: min(320px, calc(100% - 28px));
   padding: 10px 12px;
-  border: 1px solid rgba(65, 255, 143, 0.32);
-  border-radius: 6px;
-  background: rgba(3, 6, 4, 0.90);
-  box-shadow: 0 14px 34px rgba(0, 0, 0, 0.38), inset 0 0 0 1px rgba(65, 255, 143, 0.08);
+  border: 1px solid rgba(65, 255, 143, 0.28);
+  border-radius: 4px;
+  background: rgba(2, 5, 3, 0.94);
+  box-shadow: 0 14px 34px rgba(0, 0, 0, 0.42), inset 0 0 0 1px rgba(65, 255, 143, 0.06);
   backdrop-filter: blur(12px);
   color: var(--text);
   pointer-events: none;
@@ -378,14 +378,14 @@ input:focus, select:focus, button:focus, .file-picker:focus-within {
 .graph-tooltip.visible { display: block; }
 .tt-name {
   color: var(--terminal);
-  font-weight: 850;
+  font-weight: 820;
   overflow-wrap: anywhere;
 }
 .tt-type {
   margin-top: 4px;
   font-size: 10px;
   font-weight: 800;
-  letter-spacing: 0.08em;
+  letter-spacing: 0.10em;
   text-transform: uppercase;
 }
 .tt-summary {