@everstateai/mcp 1.3.11 → 1.3.13

package/src/setup.ts

@@ -24,6 +24,8 @@ import {
 import {
   getSessionStartHook,
   getSessionEndHook,
+  getPreCompactHook,
+  getSubagentContextHook,
   getSyncTodosHook,
 } from "./setup/hooks/templates.js";
 
@@ -336,11 +338,13 @@ function cleanupOldHooks(projectDir: string) {
 
   const oldHookFiles = [
     "everstate-session-start.sh",
-    "everstate-session-stop.sh",
+    "everstate-session-end.sh",
+    "everstate-session-stop.sh",   // legacy name
     "everstate-sync-todos.js",
     "everstate-pre-compact.sh",
     "session-start.sh",
-    "session-stop.sh",
+    "session-end.sh",
+    "session-stop.sh",             // legacy name
     "sync-todos.js",
     "pre-compact.sh",
   ];
@@ -400,7 +404,9 @@ async function installHooks(projectDir: string, force: boolean = false) {
 
   // Use versioned templates for all hooks
   const sessionStartHook = getSessionStartHook(hookConfig);
-  const sessionStopHook = getSessionEndHook(hookConfig);
+  const sessionEndHook = getSessionEndHook(hookConfig);
+  const preCompactHook = getPreCompactHook(hookConfig);
+  const subagentContextHook = getSubagentContextHook(hookConfig);
 
   // Sync-todos hook: try download from server, fallback to embedded template
   let syncTodosHook: string;
@@ -412,10 +418,26 @@ async function installHooks(projectDir: string, force: boolean = false) {
 
   // Write project-local hook files
   const startHookPath = path.join(projectHooksDir, "everstate-session-start.sh");
-  const stopHookPath = path.join(projectHooksDir, "everstate-session-stop.sh");
+  const endHookPath = path.join(projectHooksDir, "everstate-session-end.sh");
+  const preCompactPath = path.join(projectHooksDir, "everstate-pre-compact.js");
+  const subagentContextPath = path.join(projectHooksDir, "everstate-subagent-context.js");
 
   fs.writeFileSync(startHookPath, sessionStartHook, { mode: 0o755 });
-  fs.writeFileSync(stopHookPath, sessionStopHook, { mode: 0o755 });
+  fs.writeFileSync(endHookPath, sessionEndHook, { mode: 0o755 });
+  fs.writeFileSync(preCompactPath, preCompactHook, { mode: 0o755 });
+  fs.writeFileSync(subagentContextPath, subagentContextHook, { mode: 0o755 });
+
+  // Clean up old bash-based pre-compact hook (replaced by JS version)
+  const oldPreCompactShPath = path.join(projectHooksDir, "everstate-pre-compact.sh");
+  if (fs.existsSync(oldPreCompactShPath)) {
+    fs.unlinkSync(oldPreCompactShPath);
+  }
+
+  // Clean up old session-stop.sh if it exists (renamed to session-end.sh)
+  const oldStopPath = path.join(projectHooksDir, "everstate-session-stop.sh");
+  if (fs.existsSync(oldStopPath)) {
+    fs.unlinkSync(oldStopPath);
+  }
 
   // Write sync-todos to global location
   const globalSyncTodosPath = path.join(globalHooksDir, "sync-todos.js");
@@ -483,7 +505,24 @@ async function installHooks(projectDir: string, force: boolean = false) {
   }
 
   addHookIfNotExists("SessionStart", "*", startHookPath);
-  addHookIfNotExists("Stop", "*", stopHookPath);
+  addHookIfNotExists("SessionEnd", "*", endHookPath);
+  addHookIfNotExists("PreCompact", "*", `node ${preCompactPath}`);
+  addHookIfNotExists("SubagentStart", "*", `node ${subagentContextPath}`);
+
+  // Migrate: remove stale Stop hooks from older installations
+  if (settings.hooks.Stop) {
+    const everstateStopEntries = settings.hooks.Stop.filter((entry: any) =>
+      entry.hooks?.some((h: any) => h.command?.includes("everstate"))
+    );
+    if (everstateStopEntries.length > 0) {
+      settings.hooks.Stop = settings.hooks.Stop.filter((entry: any) =>
+        !entry.hooks?.some((h: any) => h.command?.includes("everstate"))
+      );
+      if (settings.hooks.Stop.length === 0) {
+        settings.hooks.Stop = [];
+      }
+    }
+  }
 
   // Also register PostToolUse for TodoWrite at the project level as fallback
   // This ensures sync works even if global ~/.claude.json is missing the hook
@@ -498,8 +537,18 @@ async function installHooks(projectDir: string, force: boolean = false) {
     }
   }
 
+  // Add Everstate MCP tools to permissions allow list
+  if (!settings.permissions) settings.permissions = {};
+  if (!settings.permissions.allow) settings.permissions.allow = [];
+
+  const everstatePermission = "mcp__everstate__*";
+  if (!settings.permissions.allow.includes(everstatePermission)) {
+    settings.permissions.allow.push(everstatePermission);
+  }
+
   writeJsonFile(settingsPath, settings);
   logSuccess("Session hooks installed");
+  logSuccess("Everstate tools added to approved commands");
 }
 
 // ============================================================================
@@ -555,7 +604,7 @@ async function createProjectConfig(
 
 function updateGitignore(projectDir: string) {
   const gitignorePath = path.join(projectDir, ".gitignore");
-  const entries = ["# Everstate", ".claude/settings.local.json", ".claude/hooks/"];
+  const entries = ["# Everstate", ".claude/settings.local.json", ".claude/hooks/", ".claude/rules/"];
 
   let content = fs.existsSync(gitignorePath) ? fs.readFileSync(gitignorePath, "utf8") : "";
   const toAdd = entries.filter((e) => !content.includes(e.replace("# Everstate", "")));
@@ -566,6 +615,56 @@ function updateGitignore(projectDir: string) {
   }
 }
 
+// ============================================================================
+// Everstate Rules (.claude/rules/everstate.md)
+// ============================================================================
+// Claude Code auto-loads all .md files from .claude/rules/ — this keeps
+// Everstate instructions separate from the user's CLAUDE.md.
+
+const EVERSTATE_CLAUDE_MD_MARKER = "<!-- everstate -->";
+
+const EVERSTATE_RULES_CONTENT = `# Everstate - Project Memory
+
+This project uses [Everstate](https://www.everstate.ai) for persistent memory across sessions.
+
+- Sessions start and end automatically via hooks — no manual calls needed
+- Use \`recall("query")\` to search past session context
+- Use \`log({ type: "achievement", message: "..." })\` to record progress
+- Use \`everstate({ action: "help" })\` to discover all available actions
+- Call \`done()\` before ending for a detailed summary (auto-done runs otherwise)
+`;
+
+function ensureClaudeMd(projectDir: string) {
+  // Write instructions to .claude/rules/everstate.md (auto-loaded by Claude Code)
+  const rulesDir = path.join(projectDir, ".claude", "rules");
+  const rulesPath = path.join(rulesDir, "everstate.md");
+
+  if (!fs.existsSync(rulesDir)) {
+    fs.mkdirSync(rulesDir, { recursive: true });
+  }
+  fs.writeFileSync(rulesPath, EVERSTATE_RULES_CONTENT);
+  logSuccess("Created .claude/rules/everstate.md");
+
+  // Clean up: remove old <!-- everstate --> section from CLAUDE.md if present
+  const claudeMdPath = path.join(projectDir, "CLAUDE.md");
+  if (fs.existsSync(claudeMdPath)) {
+    const content = fs.readFileSync(claudeMdPath, "utf8");
+    if (content.includes(EVERSTATE_CLAUDE_MD_MARKER)) {
+      // Remove the old Everstate section (marker through end of section)
+      const markerIdx = content.indexOf(EVERSTATE_CLAUDE_MD_MARKER);
+      const cleaned = content.substring(0, markerIdx).trimEnd();
+      if (cleaned.length > 0) {
+        fs.writeFileSync(claudeMdPath, cleaned + "\n");
+        logSuccess("Removed old Everstate section from CLAUDE.md (moved to .claude/rules/)");
+      } else {
+        // CLAUDE.md was entirely the Everstate section — remove the file
+        fs.unlinkSync(claudeMdPath);
+        logSuccess("Removed CLAUDE.md (was only Everstate content, now in .claude/rules/)");
+      }
+    }
+  }
+}
+
 // ============================================================================
 // Setup summary
 // ============================================================================
@@ -679,6 +778,19 @@ async function runUpgrade(args: SetupArgs) {
     logInfo("Updating hooks...");
     await installHooks(projectDir, true);
 
+    // Also run auto-update to install any new hook types (gotcha-check, implicit-progress)
+    try {
+      const { checkAndApplyUpdates } = await import('./setup/auto-update.js');
+      const updateResult = await checkAndApplyUpdates();
+      if (updateResult.updates.length > 0) {
+        for (const update of updateResult.updates) {
+          logSuccess(update);
+        }
+      }
+    } catch {
+      // Auto-update is optional - don't fail upgrade if it errors
+    }
+
     logInfo("Updating version file...");
     const configData = readJsonFile(configPath);
     const projectId = configData?.projectId;
@@ -833,7 +945,10 @@ export async function runSetup(argv: string[]) {
     // ── Step 7: Update .gitignore ────────────────────────────────────
     updateGitignore(projectDir);
 
-    // ── Step 8: Write version file ───────────────────────────────────
+    // ── Step 8: Create .claude/rules/everstate.md ─────────────────────
+    ensureClaudeMd(projectDir);
+
+    // ── Step 9: Write version file ───────────────────────────────────
     const versionFile = createVersionFile("setup");
     versionFile.projects[projectConfig.projectId] = {
       projectId: projectConfig.projectId,