@massu/core 0.1.1 → 0.4.0

package/dist/hooks/post-tool-use.js

@@ -3,12 +3,13 @@ import{createRequire as __cr}from"module";const require=__cr(import.meta.url);
 
 // src/memory-db.ts
 import Database from "better-sqlite3";
-import { dirname as dirname2 } from "path";
+import { dirname as dirname2, basename } from "path";
 import { existsSync as existsSync2, mkdirSync } from "fs";
 
 // src/config.ts
 import { resolve, dirname } from "path";
 import { existsSync, readFileSync } from "fs";
+import { homedir } from "os";
 import { parse as parseYaml } from "yaml";
 import { z } from "zod";
 var DomainConfigSchema = z.object({
@@ -140,6 +141,49 @@ var CloudConfigSchema = z.object({
     audit: z.boolean().default(true)
   }).default({ memory: true, analytics: true, audit: true })
 }).optional();
+var ConventionsConfigSchema = z.object({
+  claudeDirName: z.string().default(".claude").refine(
+    (s) => !s.includes("..") && !s.startsWith("/"),
+    { message: 'claudeDirName must not contain ".." or start with "/"' }
+  ),
+  sessionStatePath: z.string().default(".claude/session-state/CURRENT.md").refine(
+    (s) => !s.includes("..") && !s.startsWith("/"),
+    { message: 'sessionStatePath must not contain ".." or start with "/"' }
+  ),
+  sessionArchivePath: z.string().default(".claude/session-state/archive").refine(
+    (s) => !s.includes("..") && !s.startsWith("/"),
+    { message: 'sessionArchivePath must not contain ".." or start with "/"' }
+  ),
+  knowledgeCategories: z.array(z.string()).default([
+    "patterns",
+    "commands",
+    "incidents",
+    "reference",
+    "protocols",
+    "checklists",
+    "playbooks",
+    "critical",
+    "scripts",
+    "status",
+    "templates",
+    "loop-state",
+    "session-state",
+    "agents"
+  ]),
+  knowledgeSourceFiles: z.array(z.string()).default(["CLAUDE.md", "MEMORY.md", "corrections.md"]),
+  excludePatterns: z.array(z.string()).default(["/ARCHIVE/", "/SESSION-HISTORY/"])
+}).optional();
+var PythonDomainConfigSchema = z.object({
+  name: z.string(),
+  packages: z.array(z.string()),
+  allowed_imports_from: z.array(z.string()).default([])
+});
+var PythonConfigSchema = z.object({
+  root: z.string(),
+  alembic_dir: z.string().optional(),
+  domains: z.array(PythonDomainConfigSchema).default([]),
+  exclude_dirs: z.array(z.string()).default(["__pycache__", ".venv", "venv", ".mypy_cache", ".pytest_cache"])
+}).optional();
 var PathsConfigSchema = z.object({
   source: z.string().default("src"),
   aliases: z.record(z.string(), z.string()).default({ "@": "src" }),
@@ -174,7 +218,9 @@ var RawConfigSchema = z.object({
   security: SecurityConfigSchema,
   team: TeamConfigSchema,
   regression: RegressionConfigSchema,
-  cloud: CloudConfigSchema
+  cloud: CloudConfigSchema,
+  conventions: ConventionsConfigSchema,
+  python: PythonConfigSchema
 }).passthrough();
 var _config = null;
 var _projectRoot = null;
@@ -238,13 +284,24 @@ function getConfig() {
     security: parsed.security,
     team: parsed.team,
     regression: parsed.regression,
-    cloud: parsed.cloud
+    cloud: parsed.cloud,
+    conventions: parsed.conventions,
+    python: parsed.python
   };
+  if (!_config.cloud?.apiKey && process.env.MASSU_API_KEY) {
+    _config.cloud = {
+      enabled: true,
+      sync: { memory: true, analytics: true, audit: true },
+      ..._config.cloud,
+      apiKey: process.env.MASSU_API_KEY
+    };
+  }
   return _config;
 }
 function getResolvedPaths() {
   const config = getConfig();
   const root = getProjectRoot();
+  const claudeDirName = config.conventions?.claudeDirName ?? ".claude";
   return {
     codegraphDbPath: resolve(root, ".codegraph/codegraph.db"),
     dataDbPath: resolve(root, ".massu/data.db"),
@@ -260,22 +317,43 @@ function getResolvedPaths() {
     ),
     extensions: [".ts", ".tsx", ".js", ".jsx"],
     indexFiles: ["index.ts", "index.tsx", "index.js", "index.jsx"],
-    patternsDir: resolve(root, ".claude/patterns"),
-    claudeMdPath: resolve(root, ".claude/CLAUDE.md"),
+    patternsDir: resolve(root, claudeDirName, "patterns"),
+    claudeMdPath: resolve(root, claudeDirName, "CLAUDE.md"),
     docsMapPath: resolve(root, ".massu/docs-map.json"),
     helpSitePath: resolve(root, "../" + config.project.name + "-help"),
-    memoryDbPath: resolve(root, ".massu/memory.db")
+    memoryDbPath: resolve(root, ".massu/memory.db"),
+    knowledgeDbPath: resolve(root, ".massu/knowledge.db"),
+    plansDir: resolve(root, "docs/plans"),
+    docsDir: resolve(root, "docs"),
+    claudeDir: resolve(root, claudeDirName),
+    memoryDir: resolve(homedir(), claudeDirName, "projects", root.replace(/\//g, "-"), "memory"),
+    sessionStatePath: resolve(root, config.conventions?.sessionStatePath ?? `${claudeDirName}/session-state/CURRENT.md`),
+    sessionArchivePath: resolve(root, config.conventions?.sessionArchivePath ?? `${claudeDirName}/session-state/archive`),
+    mcpJsonPath: resolve(root, ".mcp.json"),
+    settingsLocalPath: resolve(root, claudeDirName, "settings.local.json")
   };
 }
 
-// src/memory-schema.ts
+// src/memory-db.ts
+function getMemoryDb() {
+  const dbPath = getResolvedPaths().memoryDbPath;
+  const dir = dirname2(dbPath);
+  if (!existsSync2(dir)) {
+    mkdirSync(dir, { recursive: true });
+  }
+  const db = new Database(dbPath);
+  db.pragma("journal_mode = WAL");
+  db.pragma("foreign_keys = ON");
+  initMemorySchema(db);
+  return db;
+}
 function initMemorySchema(db) {
   db.exec(`
     -- Sessions table (linked to Claude Code session IDs)
     CREATE TABLE IF NOT EXISTS sessions (
       id INTEGER PRIMARY KEY AUTOINCREMENT,
       session_id TEXT UNIQUE NOT NULL,
-      project TEXT NOT NULL DEFAULT 'unknown',
+      project TEXT NOT NULL DEFAULT 'my-project',
       git_branch TEXT,
       started_at TEXT NOT NULL,
       started_at_epoch INTEGER NOT NULL,
@@ -330,9 +408,7 @@ function initMemorySchema(db) {
         content_rowid='id'
       );
     `);
-  } catch (e) {
-    process.stderr.write(`FTS5 setup warning: ${e instanceof Error ? e.message : String(e)}
-`);
+  } catch (_e) {
   }
   db.exec(`
     CREATE TRIGGER IF NOT EXISTS observations_ai AFTER INSERT ON observations BEGIN
@@ -392,9 +468,7 @@ function initMemorySchema(db) {
         content_rowid='id'
       );
     `);
-  } catch (e) {
-    process.stderr.write(`FTS5 setup warning: ${e instanceof Error ? e.message : String(e)}
-`);
+  } catch (_e) {
   }
   db.exec(`
     CREATE TRIGGER IF NOT EXISTS prompts_ai AFTER INSERT ON user_prompts BEGIN
@@ -464,9 +538,7 @@ function initMemorySchema(db) {
         content_rowid=id
       );
     `);
-  } catch (e) {
-    process.stderr.write(`FTS5 setup warning: ${e instanceof Error ? e.message : String(e)}
-`);
+  } catch (_e) {
   }
   db.exec(`
     CREATE TRIGGER IF NOT EXISTS ct_fts_insert AFTER INSERT ON conversation_turns BEGIN
@@ -490,7 +562,7 @@ function initMemorySchema(db) {
     CREATE TABLE IF NOT EXISTS session_quality_scores (
       id INTEGER PRIMARY KEY AUTOINCREMENT,
       session_id TEXT NOT NULL UNIQUE,
-      project TEXT NOT NULL DEFAULT 'unknown',
+      project TEXT NOT NULL DEFAULT 'my-project',
       score INTEGER NOT NULL DEFAULT 100,
       security_score INTEGER NOT NULL DEFAULT 100,
       architecture_score INTEGER NOT NULL DEFAULT 100,
@@ -513,7 +585,7 @@ function initMemorySchema(db) {
     CREATE TABLE IF NOT EXISTS session_costs (
       id INTEGER PRIMARY KEY AUTOINCREMENT,
       session_id TEXT NOT NULL UNIQUE,
-      project TEXT NOT NULL DEFAULT 'unknown',
+      project TEXT NOT NULL DEFAULT 'my-project',
       input_tokens INTEGER NOT NULL DEFAULT 0,
       output_tokens INTEGER NOT NULL DEFAULT 0,
       cache_read_tokens INTEGER NOT NULL DEFAULT 0,
@@ -740,10 +812,16 @@ function initMemorySchema(db) {
     );
     CREATE INDEX IF NOT EXISTS idx_pending_sync_created ON pending_sync(created_at ASC);
   `);
+  db.exec(`
+    CREATE TABLE IF NOT EXISTS license_cache (
+      api_key_hash TEXT PRIMARY KEY,
+      tier TEXT NOT NULL,
+      valid_until TEXT NOT NULL,
+      last_validated TEXT NOT NULL,
+      features TEXT DEFAULT '[]'
+    );
+  `);
 }
-
-// src/memory-queries.ts
-import { basename } from "path";
 function assignImportance(type, vrResult) {
   switch (type) {
     case "decision":
@@ -841,27 +919,9 @@ function deduplicateFailedAttempt(db, sessionId, title, detail, opts) {
   });
 }
 
-// src/memory-db.ts
-var initializedPaths = /* @__PURE__ */ new Set();
-function getMemoryDb() {
-  const dbPath = getResolvedPaths().memoryDbPath;
-  const dir = dirname2(dbPath);
-  if (!existsSync2(dir)) {
-    mkdirSync(dir, { recursive: true });
-  }
-  const db = new Database(dbPath);
-  db.pragma("journal_mode = WAL");
-  db.pragma("foreign_keys = ON");
-  if (!initializedPaths.has(dbPath)) {
-    initMemorySchema(db);
-    initializedPaths.add(dbPath);
-  }
-  return db;
-}
-
 // src/transcript-parser.ts
-function estimateTokens(text2) {
-  return Math.ceil(text2.length / 4);
+function estimateTokens(text) {
+  return Math.ceil(text.length / 4);
 }
 
 // src/adr-generator.ts
@@ -869,13 +929,14 @@ var DEFAULT_DETECTION_PHRASES = ["chose", "decided", "switching to", "moving fro
 function getDetectionPhrases() {
   return getConfig().governance?.adr?.detection_phrases ?? DEFAULT_DETECTION_PHRASES;
 }
-function detectDecisionPatterns(text2) {
+function detectDecisionPatterns(text) {
   const phrases = getDetectionPhrases();
-  const lower = text2.toLowerCase();
+  const lower = text.toLowerCase();
   return phrases.some((phrase) => lower.includes(phrase));
 }
 
 // src/observation-extractor.ts
+import { homedir as homedir2 } from "os";
 var PRIVATE_PATTERNS = [
   /\/Users\/\w+/,
   // Absolute macOS paths
@@ -895,9 +956,9 @@ var PRIVATE_PATTERNS = [
   // Stripe keys
 ];
 function classifyVisibility(title, detail) {
-  const text2 = `${title} ${detail ?? ""}`;
+  const text = `${title} ${detail ?? ""}`;
   for (const pattern of PRIVATE_PATTERNS) {
-    if (pattern.test(text2)) return "private";
+    if (pattern.test(text)) return "private";
   }
   return "public";
 }
@@ -954,7 +1015,9 @@ function classifyToolCall(tc) {
     }
     case "Read": {
       const filePath = tc.input.file_path ?? "unknown";
-      if (filePath.includes("/plans/") || filePath.includes("CLAUDE.md") || filePath.includes("CURRENT.md")) {
+      const knowledgeSourceFiles = getConfig().conventions?.knowledgeSourceFiles ?? ["CLAUDE.md", "MEMORY.md", "corrections.md"];
+      const plansDir = getResolvedPaths().plansDir;
+      if (filePath.includes(plansDir) || knowledgeSourceFiles.some((f) => filePath.includes(f))) {
         const title = `Read: ${shortenPath(filePath)}`;
         return {
           type: "discovery",
@@ -1042,13 +1105,13 @@ function classifyToolCall(tc) {
       return null;
   }
 }
-function extractLinkedReferences(text2) {
+function extractLinkedReferences(text) {
   const result = {};
-  const crMatch = text2.match(/CR-(\d+)/);
+  const crMatch = text.match(/CR-(\d+)/);
   if (crMatch) result.crRule = `CR-${crMatch[1]}`;
-  const vrMatch = text2.match(/VR-([A-Z_]+)/);
+  const vrMatch = text.match(/VR-([A-Z_]+)/);
   if (vrMatch) result.vrType = `VR-${vrMatch[1]}`;
-  const planMatch = text2.match(/P(\d+)-(\d+)/);
+  const planMatch = text.match(/P(\d+)-(\d+)/);
   if (planMatch) result.planItem = `P${planMatch[1]}-${planMatch[2]}`;
   return result;
 }
@@ -1064,7 +1127,11 @@ function shortenPath(filePath) {
   if (filePath.startsWith(root + "/")) {
     return filePath.slice(root.length + 1);
   }
-  return filePath.replace(/^\/Users\/\w+\//, "~/");
+  const home = homedir2();
+  if (filePath.startsWith(home + "/")) {
+    return "~/" + filePath.slice(home.length + 1);
+  }
+  return filePath;
 }
 function classifyRealTimeToolCall(toolName, toolInput, toolResponse, seenReads2) {
   const tc = {
@@ -1175,53 +1242,7 @@ function trackModification(db, featureKey) {
 
 // src/import-resolver.ts
 import { readFileSync as readFileSync2, existsSync as existsSync3, statSync } from "fs";
-import { resolve as resolve2, dirname as dirname3, join } from "path";
-function resolveImportPath(specifier, fromFile) {
-  if (!specifier.startsWith(".") && !specifier.startsWith("@/")) {
-    return null;
-  }
-  let basePath;
-  if (specifier.startsWith("@/")) {
-    const paths = getResolvedPaths();
-    basePath = resolve2(paths.pathAlias["@"] ?? paths.srcDir, specifier.slice(2));
-  } else {
-    basePath = resolve2(dirname3(fromFile), specifier);
-  }
-  if (existsSync3(basePath) && !isDirectory(basePath)) {
-    return toRelative(basePath);
-  }
-  const resolvedPaths = getResolvedPaths();
-  for (const ext of resolvedPaths.extensions) {
-    const withExt = basePath + ext;
-    if (existsSync3(withExt)) {
-      return toRelative(withExt);
-    }
-  }
-  for (const indexFile of resolvedPaths.indexFiles) {
-    const indexPath = join(basePath, indexFile);
-    if (existsSync3(indexPath)) {
-      return toRelative(indexPath);
-    }
-  }
-  return null;
-}
-function isDirectory(path) {
-  try {
-    return statSync(path).isDirectory();
-  } catch {
-    return false;
-  }
-}
-function toRelative(absPath) {
-  const root = getProjectRoot();
-  if (absPath.startsWith(root)) {
-    return absPath.slice(root.length + 1);
-  }
-  return absPath;
-}
-
-// src/validation-engine.ts
-import { existsSync as existsSync4, readFileSync as readFileSync3 } from "fs";
+import { resolve as resolve4, dirname as dirname3, join } from "path";
 
 // src/security-utils.ts
 import { resolve as resolve3, normalize } from "path";
@@ -1269,7 +1290,53 @@ function enforceSeverityFloors(configWeights, defaults) {
   return result;
 }
 
+// src/import-resolver.ts
+function resolveImportPath(specifier, fromFile) {
+  if (!specifier.startsWith(".") && !specifier.startsWith("@/")) {
+    return null;
+  }
+  let basePath;
+  if (specifier.startsWith("@/")) {
+    const paths = getResolvedPaths();
+    basePath = resolve4(paths.pathAlias["@"] ?? paths.srcDir, specifier.slice(2));
+  } else {
+    basePath = resolve4(dirname3(fromFile), specifier);
+  }
+  if (existsSync3(basePath) && !isDirectory(basePath)) {
+    return toRelative(basePath);
+  }
+  const resolvedPaths = getResolvedPaths();
+  for (const ext of resolvedPaths.extensions) {
+    const withExt = basePath + ext;
+    if (existsSync3(withExt)) {
+      return toRelative(withExt);
+    }
+  }
+  for (const indexFile of resolvedPaths.indexFiles) {
+    const indexPath = join(basePath, indexFile);
+    if (existsSync3(indexPath)) {
+      return toRelative(indexPath);
+    }
+  }
+  return null;
+}
+function isDirectory(path) {
+  try {
+    return statSync(path).isDirectory();
+  } catch {
+    return false;
+  }
+}
+function toRelative(absPath) {
+  const root = getProjectRoot();
+  if (absPath.startsWith(root)) {
+    return absPath.slice(root.length + 1);
+  }
+  return absPath;
+}
+
 // src/validation-engine.ts
+import { existsSync as existsSync4, readFileSync as readFileSync3 } from "fs";
 function getValidationChecks() {
   return getConfig().governance?.validation?.checks ?? {
     rule_compliance: true,
@@ -1539,6 +1606,9 @@ function storeSecurityScore(db, sessionId, filePath, riskScore, findings) {
 }
 
 // src/hooks/post-tool-use.ts
+import { readFileSync as readFileSync5, existsSync as existsSync6 } from "fs";
+import { join as join2 } from "path";
+import { parse as parseYaml2 } from "yaml";
 var seenReads = /* @__PURE__ */ new Set();
 var currentSessionId = null;
 async function main() {
@@ -1616,6 +1686,41 @@ async function main() {
         }
       } catch (_securityErr) {
       }
+      try {
+        if (tool_name === "Edit" || tool_name === "Write") {
+          const filePath = tool_input.file_path ?? "";
+          if (filePath && filePath.endsWith("MEMORY.md") && filePath.includes("/memory/")) {
+            const issues = checkMemoryFileIntegrity(filePath);
+            if (issues.length > 0) {
+              addObservation(
+                db,
+                session_id,
+                "incident_near_miss",
+                "MEMORY.md integrity issue detected",
+                issues.join("; "),
+                { importance: 4 }
+              );
+            }
+          }
+        }
+      } catch (_memoryErr) {
+      }
+      try {
+        if (tool_name === "Edit" || tool_name === "Write") {
+          const filePath = tool_input.file_path ?? "";
+          if (filePath && isKnowledgeSourceFile(filePath)) {
+            addObservation(
+              db,
+              session_id,
+              "discovery",
+              "Knowledge source file modified - index may be stale",
+              `Edited ${filePath.split("/").pop() ?? filePath}. Run knowledge re-index to update.`,
+              { importance: 3 }
+            );
+          }
+        }
+      } catch (_knowledgeErr) {
+      }
     } finally {
       db.close();
     }
@@ -1630,29 +1735,86 @@ function updatePlanProgress(db, sessionId, progress) {
   if (existing) {
     try {
       const currentProgress = JSON.parse(existing.plan_progress);
-      for (const p2 of progress) {
-        currentProgress[p2.planItem] = p2.status;
+      for (const p of progress) {
+        currentProgress[p.planItem] = p.status;
       }
       db.prepare("UPDATE session_summaries SET plan_progress = ? WHERE id = ?").run(JSON.stringify(currentProgress), existing.id);
     } catch (_e) {
     }
   } else {
     const progressMap = {};
-    for (const p2 of progress) {
-      progressMap[p2.planItem] = p2.status;
+    for (const p of progress) {
+      progressMap[p.planItem] = p.status;
     }
     addSummary(db, sessionId, { planProgress: progressMap });
   }
 }
 function readStdin() {
-  return new Promise((resolve4) => {
+  return new Promise((resolve5) => {
     let data = "";
     process.stdin.setEncoding("utf-8");
     process.stdin.on("data", (chunk) => {
       data += chunk;
     });
-    process.stdin.on("end", () => resolve4(data));
-    setTimeout(() => resolve4(data), 3e3);
+    process.stdin.on("end", () => resolve5(data));
+    setTimeout(() => resolve5(data), 3e3);
   });
 }
+function readConventions(cwd) {
+  const defaults = {
+    knowledgeSourceFiles: ["CLAUDE.md", "MEMORY.md", "corrections.md"],
+    claudeDirName: ".claude"
+  };
+  try {
+    const projectRoot = cwd ?? process.cwd();
+    const configPath = join2(projectRoot, "massu.config.yaml");
+    if (!existsSync6(configPath)) return defaults;
+    const content = readFileSync5(configPath, "utf-8");
+    const parsed = parseYaml2(content);
+    if (!parsed || typeof parsed !== "object") return defaults;
+    const conventions = parsed.conventions;
+    if (!conventions || typeof conventions !== "object") return defaults;
+    return {
+      knowledgeSourceFiles: Array.isArray(conventions.knowledgeSourceFiles) ? conventions.knowledgeSourceFiles : defaults.knowledgeSourceFiles,
+      claudeDirName: typeof conventions.claudeDirName === "string" ? conventions.claudeDirName : defaults.claudeDirName
+    };
+  } catch {
+    return defaults;
+  }
+}
+function isKnowledgeSourceFile(filePath) {
+  const basename2 = filePath.split("/").pop() ?? "";
+  const conventions = readConventions();
+  const knowledgeSourcePatterns = [
+    ...conventions.knowledgeSourceFiles,
+    "file-index.md",
+    "knowledge-db.ts",
+    "knowledge-indexer.ts",
+    "knowledge-tools.ts"
+  ];
+  return knowledgeSourcePatterns.some((p) => basename2 === p) || filePath.includes("/memory/") || filePath.includes(conventions.claudeDirName + "/");
+}
+function checkMemoryFileIntegrity(filePath) {
+  const issues = [];
+  try {
+    if (!existsSync6(filePath)) {
+      issues.push("MEMORY.md file does not exist after write");
+      return issues;
+    }
+    const content = readFileSync5(filePath, "utf-8");
+    const lines = content.split("\n");
+    const MAX_LINES = 200;
+    if (lines.length > MAX_LINES) {
+      issues.push(`MEMORY.md exceeds ${MAX_LINES} lines (currently ${lines.length}). Consider archiving old entries.`);
+    }
+    const requiredSections = ["# Massu Memory", "## Key Learnings", "## Common Gotchas"];
+    for (const section of requiredSections) {
+      if (!content.includes(section)) {
+        issues.push(`Missing required section: "${section}"`);
+      }
+    }
+  } catch (_e) {
+  }
+  return issues;
+}
 main();