@massu/core 0.1.1 → 0.4.0

package/dist/hooks/session-end.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,15 +812,27 @@ 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 enqueueSyncPayload(db, payload) {
   db.prepare("INSERT INTO pending_sync (payload) VALUES (?)").run(payload);
 }
 function dequeuePendingSync(db, limit = 10) {
-  db.prepare("DELETE FROM pending_sync WHERE retry_count >= 10").run();
+  const stale = db.prepare(
+    "SELECT id FROM pending_sync WHERE retry_count >= 10"
+  ).all();
+  if (stale.length > 0) {
+    const ids = stale.map((s) => s.id);
+    db.prepare(`DELETE FROM pending_sync WHERE id IN (${ids.map(() => "?").join(",")})`).run(...ids);
+  }
   return db.prepare(
     "SELECT id, payload, retry_count FROM pending_sync ORDER BY created_at ASC LIMIT ?"
   ).all(limit);
@@ -840,27 +924,9 @@ function setLastProcessedLine(db, sessionId, lineNumber) {
   db.prepare("INSERT OR REPLACE INTO memory_meta (key, value) VALUES (?, ?)").run(`last_processed_line:${sessionId}`, String(lineNumber));
 }
 
-// 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/session-archiver.ts
 import { existsSync as existsSync3, readFileSync as readFileSync2, writeFileSync, mkdirSync as mkdirSync2, renameSync } from "fs";
-import { resolve as resolve2, dirname as dirname3 } from "path";
+import { resolve as resolve3, dirname as dirname3 } from "path";
 
 // src/session-state-generator.ts
 function generateCurrentMd(db, sessionId) {
@@ -1005,17 +1071,17 @@ function safeParseJson(json, fallback) {
 }
 
 // src/session-archiver.ts
-var PROJECT_ROOT = getProjectRoot();
 function archiveAndRegenerate(db, sessionId) {
-  const currentMdPath = resolve2(PROJECT_ROOT, ".claude/session-state/CURRENT.md");
-  const archiveDir = resolve2(PROJECT_ROOT, ".claude/session-state/archive");
+  const resolved = getResolvedPaths();
+  const currentMdPath = resolved.sessionStatePath;
+  const archiveDir = resolved.sessionArchivePath;
   let archived = false;
   let archivePath;
   if (existsSync3(currentMdPath)) {
     const existingContent = readFileSync2(currentMdPath, "utf-8");
     if (existingContent.trim().length > 10) {
       const { date, slug } = extractArchiveInfo(existingContent);
-      archivePath = resolve2(archiveDir, `${date}-${slug}.md`);
+      archivePath = resolve3(archiveDir, `${date}-${slug}.md`);
       if (!existsSync3(archiveDir)) {
         mkdirSync2(archiveDir, { recursive: true });
       }
@@ -1129,8 +1195,8 @@ async function parseTranscriptFrom(filePath, startLine) {
   }
   return { entries, totalLines: lineNumber };
 }
-function estimateTokens(text2) {
-  return Math.ceil(text2.length / 4);
+function estimateTokens(text) {
+  return Math.ceil(text.length / 4);
 }
 
 // src/cloud-sync.ts
@@ -1211,30 +1277,25 @@ async function syncToCloud(db, payload) {
   };
 }
 async function drainSyncQueue(db) {
-  try {
-    const config = getConfig();
-    if (!config.cloud?.enabled || !config.cloud?.apiKey) return;
-    const pending = dequeuePendingSync(db, 10);
-    for (const item of pending) {
-      try {
-        const payload = JSON.parse(item.payload);
-        const result = await syncToCloud(db, payload);
-        if (result.success) {
-          removePendingSync(db, item.id);
-        } else {
-          incrementRetryCount(db, item.id, result.error ?? "Unknown error");
-        }
-      } catch (err) {
-        incrementRetryCount(db, item.id, err instanceof Error ? err.message : String(err));
+  const config = getConfig();
+  if (!config.cloud?.enabled || !config.cloud?.apiKey) return;
+  const pending = dequeuePendingSync(db, 10);
+  for (const item of pending) {
+    try {
+      const payload = JSON.parse(item.payload);
+      const result = await syncToCloud(db, payload);
+      if (result.success) {
+        removePendingSync(db, item.id);
+      } else {
+        incrementRetryCount(db, item.id, result.error ?? "Unknown error");
       }
+    } catch (err) {
+      incrementRetryCount(db, item.id, err instanceof Error ? err.message : String(err));
     }
-  } catch (err) {
-    process.stderr.write(`massu: drainSyncQueue failed: ${err instanceof Error ? err.message : String(err)}
-`);
   }
 }
 function sleep(ms) {
-  return new Promise((resolve3) => setTimeout(resolve3, ms));
+  return new Promise((resolve4) => setTimeout(resolve4, ms));
 }
 
 // src/analytics.ts
@@ -1300,7 +1361,6 @@ function backfillQualityScores(db) {
     FROM sessions s
     LEFT JOIN session_quality_scores q ON s.session_id = q.session_id
     WHERE q.session_id IS NULL
-    LIMIT 1000
   `).all();
   let backfilled = 0;
   for (const session of sessions) {
@@ -1313,11 +1373,10 @@ function backfillQualityScores(db) {
 
 // src/cost-tracker.ts
 var DEFAULT_MODEL_PRICING = {
-  "claude-opus-4-6": { input_per_million: 5, output_per_million: 25, cache_read_per_million: 0.5, cache_write_per_million: 6.25 },
+  "claude-opus-4-6": { input_per_million: 15, output_per_million: 75, cache_read_per_million: 1.5, cache_write_per_million: 18.75 },
   "claude-sonnet-4-6": { input_per_million: 3, output_per_million: 15, cache_read_per_million: 0.3, cache_write_per_million: 3.75 },
   "claude-sonnet-4-5": { input_per_million: 3, output_per_million: 15, cache_read_per_million: 0.3, cache_write_per_million: 3.75 },
-  "claude-3-5-haiku-20241022": { input_per_million: 0.8, output_per_million: 4, cache_read_per_million: 0.08, cache_write_per_million: 1 },
-  "claude-haiku-4-5-20251001": { input_per_million: 1, output_per_million: 5, cache_read_per_million: 0.1, cache_write_per_million: 1.25 },
+  "claude-haiku-4-5-20251001": { input_per_million: 0.8, output_per_million: 4, cache_read_per_million: 0.08, cache_write_per_million: 1 },
   "default": { input_per_million: 3, output_per_million: 15, cache_read_per_million: 0.3, cache_write_per_million: 3.75 }
 };
 function getModelPricing() {
@@ -1389,8 +1448,8 @@ import { createHash } from "crypto";
 function escapeRegex(str) {
   return str.replace(/[.*+?^${}()|[\]\\]/g, "\\$&");
 }
-function redactSensitiveContent(text2) {
-  return text2.replace(/\b(sk-|ghp_|gho_|xoxb-|xoxp-|AKIA)[A-Za-z0-9_-]{10,}\b/g, "[REDACTED_KEY]").replace(/Bearer\s+[A-Za-z0-9._~+/=-]{20,}/gi, "Bearer [REDACTED_TOKEN]").replace(/:\/\/[^:]+:[^@\s]+@/g, "://[REDACTED_CREDENTIALS]@").replace(/(https?:\/\/[^\s]+[?&](?:token|key|secret|password|auth)=)[^\s&]*/gi, "$1[REDACTED]").replace(/[a-zA-Z0-9._%+-]+@[a-zA-Z0-9.-]+\.[a-zA-Z]{2,}/g, "[REDACTED_EMAIL]").replace(/(?:\/Users\/|\/home\/|C:\\Users\\)[^\s"'`]+/g, "[REDACTED_PATH]");
+function redactSensitiveContent(text) {
+  return text.replace(/\b(sk-|ghp_|gho_|xoxb-|xoxp-|AKIA)[A-Za-z0-9_-]{10,}\b/g, "[REDACTED_KEY]").replace(/Bearer\s+[A-Za-z0-9._~+/=-]{20,}/gi, "Bearer [REDACTED_TOKEN]").replace(/:\/\/[^:]+:[^@\s]+@/g, "://[REDACTED_CREDENTIALS]@").replace(/(https?:\/\/[^\s]+[?&](?:token|key|secret|password|auth)=)[^\s&]*/gi, "$1[REDACTED]").replace(/[a-zA-Z0-9._%+-]+@[a-zA-Z0-9.-]+\.[a-zA-Z]{2,}/g, "[REDACTED_EMAIL]").replace(/(?:\/Users\/|\/home\/|C:\\Users\\)[^\s"'`]+/g, "[REDACTED_PATH]");
 }
 
 // src/prompt-analyzer.ts
@@ -1458,7 +1517,7 @@ function analyzeSessionPrompts(db, sessionId) {
   let stored = 0;
   for (let i = 0; i < prompts.length; i++) {
     const prompt = prompts[i];
-    const followUps = prompts.slice(i + 1, i + 4).map((p2) => p2.prompt_text);
+    const followUps = prompts.slice(i + 1, i + 4).map((p) => p.prompt_text);
     const category = categorizePrompt(prompt.prompt_text);
     const hash = hashPrompt(prompt.prompt_text);
     const { outcome, correctionsNeeded, followUpCount } = detectOutcome(followUps, []);
@@ -1510,7 +1569,7 @@ async function main() {
       }
       try {
         const { score, breakdown } = calculateQualityScore(db, session_id);
-        if (score > 0) {
+        if (score !== 50) {
           storeQualityScore(db, session_id, score, breakdown);
         }
         backfillQualityScores(db);
@@ -1683,18 +1742,18 @@ function groupEntriesIntoTurns(entries) {
       if (currentTurn) {
         turns.push(currentTurn);
       }
-      const text2 = getTextFromBlocks(entry.message.content);
-      if (text2.trim()) {
+      const text = getTextFromBlocks(entry.message.content);
+      if (text.trim()) {
         currentTurn = {
-          userPrompt: text2.trim(),
+          userPrompt: text.trim(),
           assistantText: null,
           toolCalls: []
         };
       }
     } else if (entry.type === "assistant" && entry.message && currentTurn) {
-      const text2 = getTextFromBlocks(entry.message.content);
-      if (text2.trim()) {
-        currentTurn.assistantText = currentTurn.assistantText ? currentTurn.assistantText + "\n" + text2.trim() : text2.trim();
+      const text = getTextFromBlocks(entry.message.content);
+      if (text.trim()) {
+        currentTurn.assistantText = currentTurn.assistantText ? currentTurn.assistantText + "\n" + text.trim() : text.trim();
       }
       for (const block of entry.message.content) {
         if (block.type === "tool_use") {
@@ -1764,14 +1823,14 @@ function extractFilesFromToolCall(toolName, input) {
   return [];
 }
 function readStdin() {
-  return new Promise((resolve3) => {
+  return new Promise((resolve4) => {
     let data = "";
     process.stdin.setEncoding("utf-8");
     process.stdin.on("data", (chunk) => {
       data += chunk;
     });
-    process.stdin.on("end", () => resolve3(data));
-    setTimeout(() => resolve3(data), 5e3);
+    process.stdin.on("end", () => resolve4(data));
+    setTimeout(() => resolve4(data), 5e3);
   });
 }
 main();