@aabadin/project-memory-context 0.2.4 → 0.2.5

package/cli/context.mjs

@@ -1,5 +1,5 @@
 #!/usr/bin/env node
-import { access } from 'node:fs/promises';
+import { existsSync, readFileSync, writeFileSync, mkdirSync } from 'node:fs';
 import { dirname, join, resolve } from 'node:path';
 import { fileURLToPath } from 'node:url';
 
@@ -22,7 +22,7 @@ export async function findProjectRoot(startDir = process.cwd()) {
 
   while (true) {
     const installPath = join(currentDir, '.planning', 'project-memory-context', 'install.json');
-    if (await fileExists(installPath)) {
+    if (existsSync(installPath)) {
       return currentDir;
     }
 
@@ -35,6 +35,23 @@ export async function findProjectRoot(startDir = process.cwd()) {
   }
 }
 
+async function markContext(projectRoot, nodeIds) {
+  if (!nodeIds || nodeIds.length === 0) return;
+  const trackerPath = join(projectRoot, '.planning', 'project-memory-context', 'context-tracker.json');
+  let tracker = { activeNodeIds: [] };
+  try {
+    if (existsSync(trackerPath)) {
+      tracker = JSON.parse(readFileSync(trackerPath, 'utf-8'));
+    }
+  } catch {}
+  const existing = new Set(tracker.activeNodeIds || []);
+  nodeIds.forEach((id) => existing.add(id));
+  tracker.activeNodeIds = [...existing];
+  const dir = dirname(trackerPath);
+  if (!existsSync(dir)) mkdirSync(dir, { recursive: true });
+  writeFileSync(trackerPath, JSON.stringify(tracker, null, 2));
+}
+
 export function parseArgs(args) {
   const DEPTH_VALUES = ['compact', 'extended', 'deep', 'disk'];
   const FOCUS_VALUES = ['dependencies', 'callers', 'containment', 'impact', 'all'];
@@ -190,12 +207,12 @@ export function buildRenderInput(engine, resolved, { depth, focus }) {
   };
 }
 
-export async function runTargetContext({ projectRoot, target, explicitMode, depth, focus }) {
-  const artifacts = await loadArtifacts(projectRoot);
+export async function runTargetContext({ projectRoot, target, explicitMode, depth, focus, artifacts }) {
+  const artfs = artifacts ?? await loadArtifacts(projectRoot);
   const engine = createQueryEngine({
-    graph: artifacts.graph,
-    symbolIndex: artifacts.symbolIndex,
-    worklist: artifacts.worklist,
+    graph: artfs.graph,
+    symbolIndex: artfs.symbolIndex,
+    worklist: artfs.worklist,
     enrichmentDir: join(projectRoot, '.planning', 'project-memory-context', 'enrichment'),
     projectSlug: 'project',
   });
@@ -205,7 +222,7 @@ export async function runTargetContext({ projectRoot, target, explicitMode, dept
   const input = buildRenderInput(engine, resolved, { depth, focus });
   const output = renderTargetContext(input);
 
-  return { output, resolved, input };
+  return { output, resolved, input, artifacts: artfs };
 }
 
 export async function runProjectContext(projectRoot = process.cwd(), refresh = false) {
@@ -301,7 +318,8 @@ export async function main(args = process.argv.slice(2)) {
     return 1;
   }
 
-  const { output } = await runTargetContext({
+  const artifacts = await loadArtifacts(projectRoot);
+  const { output, resolved } = await runTargetContext({
     projectRoot,
     target: parsed.target,
     explicitMode: parsed.explicitMode,
@@ -309,6 +327,17 @@ export async function main(args = process.argv.slice(2)) {
     focus: parsed.focus,
   });
 
+  const nodeIdsToMark = [];
+  if (resolved.symbolKey) {
+    const entry = artifacts.symbolIndex[resolved.symbolKey];
+    if (entry?.graphNodeId) nodeIdsToMark.push(entry.graphNodeId);
+  } else if (resolved.target && (resolved.mode === 'file' || resolved.mode === 'symbol-missing')) {
+    const fileNodeId = resolved.target.replace(/[/\\]/g, '_').replace(/[:.]/g, '_');
+    nodeIdsToMark.push(fileNodeId);
+  }
+
+  await markContext(projectRoot, nodeIdsToMark);
+
   console.log(output);
   return 0;
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aabadin/project-memory-context",
-  "version": "0.2.4",
+  "version": "0.2.5",
   "description": "Portable project memory context CLI — bootstraps semantic enrichment workflows for any AI coding agent.",
   "license": "GPL-3.0-or-later",
   "type": "module",