@everstateai/mcp 1.3.12 → 1.3.13

package/src/setup/auto-update.ts

@@ -95,6 +95,59 @@ export async function checkAndApplyUpdates(): Promise<UpdateCheckResult> {
       result.needsUpdate = true;
     }
 
+    // Install gotcha-check hook if not present
+    const gotchaCheckPath = `${hooksDir}/gotcha-check.js`;
+    if (!fs.existsSync(gotchaCheckPath)) {
+      try {
+        const { getGotchaCheckHook } = await import('./hooks/templates.js');
+        fs.mkdirSync(hooksDir, { recursive: true });
+        fs.writeFileSync(gotchaCheckPath, getGotchaCheckHook());
+        fs.chmodSync(gotchaCheckPath, '755');
+        result.updates.push('gotcha-check.js installed');
+        result.needsUpdate = true;
+
+        // Register PreToolUse hook for Edit and Write
+        const gotchaRegistered = await ensureGotchaHookRegistration(gotchaCheckPath);
+        if (gotchaRegistered) {
+          result.updates.push('PreToolUse gotcha hook registered');
+        }
+      } catch (error) {
+        result.errors.push(`Failed to install gotcha-check.js: ${String(error)}`);
+      }
+    }
+
+    // Install implicit-progress hook if not present
+    const implicitProgressPath = `${hooksDir}/implicit-progress.js`;
+    if (!fs.existsSync(implicitProgressPath)) {
+      try {
+        const { getImplicitProgressHook } = await import('./hooks/templates.js');
+        fs.mkdirSync(hooksDir, { recursive: true });
+        fs.writeFileSync(implicitProgressPath, getImplicitProgressHook());
+        fs.chmodSync(implicitProgressPath, '755');
+        result.updates.push('implicit-progress.js installed');
+        result.needsUpdate = true;
+
+        // Register PostToolUse hook for Edit, Write, and Bash
+        const progressRegistered = await ensureImplicitProgressRegistration(implicitProgressPath);
+        if (progressRegistered) {
+          result.updates.push('PostToolUse implicit-progress hook registered');
+        }
+      } catch (error) {
+        result.errors.push(`Failed to install implicit-progress.js: ${String(error)}`);
+      }
+    }
+
+    // Migrate Stop→SessionEnd and ensure PreCompact/permissions in known projects
+    try {
+      const migrateResult = await migrateProjectSettings(versionFile);
+      if (migrateResult.length > 0) {
+        result.updates.push(...migrateResult);
+        result.needsUpdate = true;
+      }
+    } catch (error) {
+      result.errors.push(`Project migration failed: ${String(error)}`);
+    }
+
     // Update MCP proxy version in version file
     const currentVersion = getMcpProxyVersion();
     if (versionFile.components.mcpProxy !== currentVersion) {
@@ -326,3 +379,213 @@ async function ensureHookRegistration(syncTodosPath: string): Promise<boolean> {
   }
 }
 
+/**
+ * Ensure PreToolUse gotcha hook is registered in ~/.claude.json for Edit and Write
+ */
+async function ensureGotchaHookRegistration(gotchaCheckPath: string): Promise<boolean> {
+  const configPath = getClaudeConfigPath();
+
+  if (!fs.existsSync(configPath)) {
+    return false;
+  }
+
+  try {
+    const config = JSON.parse(fs.readFileSync(configPath, 'utf8'));
+
+    if (!config.hooks) {
+      config.hooks = {};
+    }
+    if (!config.hooks.PreToolUse) {
+      config.hooks.PreToolUse = [];
+    }
+
+    const preToolUse = config.hooks.PreToolUse as Array<{
+      matcher: string;
+      hooks: Array<{ type?: string; command?: string; timeout?: number }>;
+    }>;
+
+    const expectedCommand = `node ${gotchaCheckPath}`;
+    let updated = false;
+
+    for (const toolName of ['Edit', 'Write']) {
+      let entry = preToolUse.find((e) => e.matcher === toolName);
+      if (!entry) {
+        entry = { matcher: toolName, hooks: [] };
+        preToolUse.push(entry);
+      }
+
+      const hasGotchaHook = entry.hooks.some((h) => h.command?.includes('gotcha-check'));
+      if (!hasGotchaHook) {
+        entry.hooks.push({
+          type: 'command',
+          command: expectedCommand,
+          timeout: 5,
+        });
+        updated = true;
+      }
+    }
+
+    if (updated) {
+      fs.writeFileSync(configPath, JSON.stringify(config, null, 2) + '\n');
+    }
+    return updated;
+  } catch {
+    return false;
+  }
+}
+
+/**
+ * Ensure PostToolUse implicit-progress hook is registered for Edit, Write, Bash
+ */
+async function ensureImplicitProgressRegistration(hookPath: string): Promise<boolean> {
+  const configPath = getClaudeConfigPath();
+
+  if (!fs.existsSync(configPath)) {
+    return false;
+  }
+
+  try {
+    const config = JSON.parse(fs.readFileSync(configPath, 'utf8'));
+
+    if (!config.hooks) {
+      config.hooks = {};
+    }
+    if (!config.hooks.PostToolUse) {
+      config.hooks.PostToolUse = [];
+    }
+
+    const postToolUse = config.hooks.PostToolUse as Array<{
+      matcher: string;
+      hooks: Array<{ type?: string; command?: string; timeout?: number }>;
+    }>;
+
+    const expectedCommand = `node ${hookPath}`;
+    let updated = false;
+
+    for (const toolName of ['Edit', 'Write', 'Bash']) {
+      let entry = postToolUse.find((e) => e.matcher === toolName);
+      if (!entry) {
+        entry = { matcher: toolName, hooks: [] };
+        postToolUse.push(entry);
+      }
+
+      const hasHook = entry.hooks.some((h) => h.command?.includes('implicit-progress'));
+      if (!hasHook) {
+        entry.hooks.push({
+          type: 'command',
+          command: expectedCommand,
+          timeout: 5,
+        });
+        updated = true;
+      }
+    }
+
+    if (updated) {
+      fs.writeFileSync(configPath, JSON.stringify(config, null, 2) + '\n');
+    }
+    return updated;
+  } catch {
+    return false;
+  }
+}
+
+/**
+ * Migrate project-level settings in known projects:
+ * 1. Move Stop hooks → SessionEnd (bug fix from older installs)
+ * 2. Ensure PreCompact hook is registered
+ * 3. Ensure mcp__everstate__* permission is present
+ */
+async function migrateProjectSettings(
+  versionFile: ReturnType<typeof readVersionFile>
+): Promise<string[]> {
+  const updates: string[] = [];
+
+  if (!versionFile?.projects) return updates;
+
+  for (const [projectId, projectInfo] of Object.entries(versionFile.projects)) {
+    const projectDir = (projectInfo as any).projectDir;
+    if (!projectDir || !fs.existsSync(projectDir)) continue;
+
+    const settingsPath = `${projectDir}/.claude/settings.local.json`;
+    if (!fs.existsSync(settingsPath)) continue;
+
+    try {
+      const content = fs.readFileSync(settingsPath, 'utf8');
+      const settings = JSON.parse(content);
+      if (!settings.hooks) continue;
+
+      let modified = false;
+
+      // 1. Migrate Stop → SessionEnd for everstate hooks
+      if (settings.hooks.Stop && Array.isArray(settings.hooks.Stop)) {
+        const everstateEntries = settings.hooks.Stop.filter((entry: any) =>
+          entry.hooks?.some((h: any) => h.command?.includes('everstate'))
+        );
+
+        if (everstateEntries.length > 0) {
+          settings.hooks.Stop = settings.hooks.Stop.filter((entry: any) =>
+            !entry.hooks?.some((h: any) => h.command?.includes('everstate'))
+          );
+
+          if (!settings.hooks.SessionEnd) settings.hooks.SessionEnd = [];
+          for (const entry of everstateEntries) {
+            for (const hook of (entry.hooks || [])) {
+              if (hook.command) {
+                hook.command = hook.command
+                  .replace('session-stop.sh', 'session-end.sh')
+                  .replace('everstate-session-stop', 'everstate-session-end');
+              }
+            }
+            const alreadyExists = settings.hooks.SessionEnd.some((e: any) =>
+              e.hooks?.some((h: any) => h.command?.includes('everstate'))
+            );
+            if (!alreadyExists) {
+              settings.hooks.SessionEnd.push(entry);
+            }
+          }
+
+          modified = true;
+          updates.push(`${projectId}: Migrated Stop→SessionEnd hooks`);
+        }
+      }
+
+      // 2. Ensure PreCompact hook is registered
+      if (!settings.hooks.PreCompact || !Array.isArray(settings.hooks.PreCompact)) {
+        settings.hooks.PreCompact = [];
+      }
+      const hasPreCompact = settings.hooks.PreCompact.some((entry: any) =>
+        entry.hooks?.some((h: any) => h.command?.includes('pre-compact'))
+      );
+      if (!hasPreCompact) {
+        const preCompactPath = `${projectDir}/.claude/hooks/everstate-pre-compact.sh`;
+        if (fs.existsSync(preCompactPath)) {
+          settings.hooks.PreCompact.push({
+            matcher: '*',
+            hooks: [{ type: 'command', command: preCompactPath }],
+          });
+          modified = true;
+          updates.push(`${projectId}: Registered PreCompact hook`);
+        }
+      }
+
+      // 3. Ensure mcp__everstate__* permission
+      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);
+        modified = true;
+        updates.push(`${projectId}: Added mcp__everstate__* permission`);
+      }
+
+      if (modified) {
+        fs.writeFileSync(settingsPath, JSON.stringify(settings, null, 2) + '\n');
+      }
+    } catch {
+      // Skip projects with invalid settings
+    }
+  }
+
+  return updates;
+}
+