convex 1.34.0 → 1.34.1

package/src/cli/lib/{ai → aiFiles}/integration.test.ts

@@ -4,12 +4,12 @@ import os from "os";
 import path from "path";
 import {
   checkAiFilesStaleness,
-  disableAiFiles,
+  safelyAttemptToDisableAiFiles,
   enableAiFiles,
   removeAiFiles,
-  statusAiFiles,
-  updateAiFiles,
+  installAiFiles,
 } from "./index.js";
+import { statusAiFiles } from "./status.js";
 import { logMessage } from "../../../bundler/log.js";
 import { AGENTS_MD_START_MARKER } from "../../codegen_templates/agentsmd.js";
 import { CLAUDE_MD_START_MARKER } from "../../codegen_templates/claudemd.js";
@@ -22,10 +22,13 @@ vi.mock("../versionApi.js", () => ({
   downloadGuidelines: vi.fn(async () => "integration guidelines content"),
   fetchAgentSkillsSha: vi.fn(async () => "integration-sha"),
   getVersion: vi.fn(async () => ({
-    message: null,
-    guidelinesHash: "integration-guidelines-hash",
-    agentSkillsSha: "integration-agent-skills-sha",
-    disableSkillsCli: false,
+    kind: "ok",
+    data: {
+      message: null,
+      guidelinesHash: "integration-guidelines-hash",
+      agentSkillsSha: "integration-agent-skills-sha",
+      disableSkillsCli: false,
+    },
   })),
 }));
 
@@ -80,7 +83,7 @@ describe("ai-files integration with default convex/ directory", () => {
   });
 
   test("install creates guidelines, state, AGENTS.md, and CLAUDE.md", async () => {
-    await updateAiFiles(tmpDir, convexDir);
+    await installAiFiles({ projectDir: tmpDir, convexDir });
 
     expect(fs.existsSync(guidelinesPath())).toBe(true);
     expect(fs.existsSync(statePath())).toBe(true);
@@ -96,7 +99,7 @@ describe("ai-files integration with default convex/ directory", () => {
       path.join(tmpDir, "AGENTS.md"),
       "# My Project\n\nImportant team guidelines here.\n",
     );
-    await updateAiFiles(tmpDir, convexDir);
+    await installAiFiles({ projectDir: tmpDir, convexDir });
     const content = fs.readFileSync(path.join(tmpDir, "AGENTS.md"), "utf8");
     expect(content).toContain("# My Project");
     expect(content).toContain("Important team guidelines here.");
@@ -111,17 +114,17 @@ describe("ai-files integration with default convex/ directory", () => {
       path.join(tmpDir, "CLAUDE.md"),
       "My custom CLAUDE.md content\n",
     );
-    await updateAiFiles(tmpDir, convexDir);
+    await installAiFiles({ projectDir: tmpDir, convexDir });
     const content = fs.readFileSync(path.join(tmpDir, "CLAUDE.md"), "utf8");
     expect(content).toContain("My custom CLAUDE.md content");
   });
 
   test("second update is idempotent", async () => {
-    await updateAiFiles(tmpDir, convexDir);
+    await installAiFiles({ projectDir: tmpDir, convexDir });
     const firstGuidelines = fs.readFileSync(guidelinesPath(), "utf8");
     const firstState = fs.readFileSync(statePath(), "utf8");
 
-    await updateAiFiles(tmpDir, convexDir);
+    await installAiFiles({ projectDir: tmpDir, convexDir });
 
     expect(fs.readFileSync(guidelinesPath(), "utf8")).toBe(firstGuidelines);
     expect(fs.readFileSync(statePath(), "utf8")).toBe(firstState);
@@ -133,36 +136,24 @@ describe("ai-files integration with default convex/ directory", () => {
       path.join(tmpDir, ".cursor", "rules", "convex_rules.mdc"),
       "legacy",
     );
-    await updateAiFiles(tmpDir, convexDir);
+    await installAiFiles({ projectDir: tmpDir, convexDir });
     expect(
       fs.existsSync(path.join(tmpDir, ".cursor", "rules", "convex_rules.mdc")),
     ).toBe(false);
   });
 
-  test("skips locally modified guidelines", async () => {
-    await updateAiFiles(tmpDir, convexDir);
-    fs.appendFileSync(guidelinesPath(), "\n## My custom note\n");
-    const state = readJson(statePath());
-    state.guidelinesHash = "deliberately-stale-hash";
-    fs.writeFileSync(statePath(), JSON.stringify(state, null, 2) + "\n");
-
-    await updateAiFiles(tmpDir, convexDir);
-
-    expect(fs.readFileSync(guidelinesPath(), "utf8")).toContain(
-      "My custom note",
-    );
-    expect(vi.mocked(logMessage)).toHaveBeenCalledWith(
-      expect.stringContaining("modified locally"),
-    );
-  });
-
   test("staleness check nags when stored hash is stale", async () => {
-    await updateAiFiles(tmpDir, convexDir);
+    await installAiFiles({ projectDir: tmpDir, convexDir });
     const state = readJson(statePath());
     state.guidelinesHash = "deliberately-stale-hash";
     fs.writeFileSync(statePath(), JSON.stringify(state, null, 2) + "\n");
 
-    await checkAiFilesStaleness("canonical-hash", null, tmpDir, convexDir);
+    await checkAiFilesStaleness({
+      canonicalGuidelinesHash: "canonical-hash",
+      canonicalAgentSkillsSha: null,
+      projectDir: tmpDir,
+      convexDir,
+    });
 
     expect(vi.mocked(logMessage)).toHaveBeenCalledWith(
       expect.stringContaining("out of date"),
@@ -170,14 +161,19 @@ describe("ai-files integration with default convex/ directory", () => {
   });
 
   test("staleness check is silent when disabled in convex.json", async () => {
-    await updateAiFiles(tmpDir, convexDir);
-    await disableAiFiles(tmpDir);
+    await installAiFiles({ projectDir: tmpDir, convexDir });
+    await safelyAttemptToDisableAiFiles(tmpDir);
     const state = readJson(statePath());
     state.guidelinesHash = "deliberately-stale-hash";
     fs.writeFileSync(statePath(), JSON.stringify(state, null, 2) + "\n");
     vi.mocked(logMessage).mockClear();
 
-    await checkAiFilesStaleness("canonical-hash", null, tmpDir, convexDir);
+    await checkAiFilesStaleness({
+      canonicalGuidelinesHash: "canonical-hash",
+      canonicalAgentSkillsSha: null,
+      projectDir: tmpDir,
+      convexDir,
+    });
 
     const calls = vi.mocked(logMessage).mock.calls.map((c) => c[0]);
     expect(
@@ -186,40 +182,36 @@ describe("ai-files integration with default convex/ directory", () => {
   });
 
   test("disable keeps files but sets convex.json preference", async () => {
-    await updateAiFiles(tmpDir, convexDir);
-    await disableAiFiles(tmpDir);
+    await installAiFiles({ projectDir: tmpDir, convexDir });
+    await safelyAttemptToDisableAiFiles(tmpDir);
 
-    expect(readJson(projectConfigPath()).aiFiles.disableStalenessMessage).toBe(
-      true,
-    );
+    expect(readJson(projectConfigPath()).aiFiles.enabled).toBe(false);
     expect(fs.existsSync(guidelinesPath())).toBe(true);
     expect(fs.existsSync(path.join(tmpDir, "AGENTS.md"))).toBe(true);
     expect(fs.existsSync(path.join(tmpDir, "CLAUDE.md"))).toBe(true);
   });
 
   test("disable before install writes only convex.json and no AI state file", async () => {
-    await disableAiFiles(tmpDir);
+    await safelyAttemptToDisableAiFiles(tmpDir);
 
-    expect(readJson(projectConfigPath()).aiFiles.disableStalenessMessage).toBe(
-      true,
-    );
+    expect(readJson(projectConfigPath()).aiFiles.enabled).toBe(false);
     expect(fs.existsSync(statePath())).toBe(false);
     expect(fs.existsSync(guidelinesPath())).toBe(false);
   });
 
   test("remove deletes ai directory and AGENTS.md managed section", async () => {
-    await updateAiFiles(tmpDir, convexDir);
-    await removeAiFiles(tmpDir, convexDir);
+    await installAiFiles({ projectDir: tmpDir, convexDir });
+    await removeAiFiles({ projectDir: tmpDir, convexDir });
 
     expect(fs.existsSync(aiDir())).toBe(false);
   });
 
   test("status reports not installed after remove", async () => {
-    await updateAiFiles(tmpDir, convexDir);
-    await removeAiFiles(tmpDir, convexDir);
+    await installAiFiles({ projectDir: tmpDir, convexDir });
+    await removeAiFiles({ projectDir: tmpDir, convexDir });
     vi.mocked(logMessage).mockClear();
 
-    await statusAiFiles(tmpDir, convexDir);
+    await statusAiFiles({ projectDir: tmpDir, convexDir });
 
     expect(vi.mocked(logMessage)).toHaveBeenCalledWith(
       expect.stringContaining("not installed"),
@@ -227,10 +219,10 @@ describe("ai-files integration with default convex/ directory", () => {
   });
 
   test("status reports installed and enabled after install", async () => {
-    await updateAiFiles(tmpDir, convexDir);
+    await installAiFiles({ projectDir: tmpDir, convexDir });
     vi.mocked(logMessage).mockClear();
 
-    await statusAiFiles(tmpDir, convexDir);
+    await statusAiFiles({ projectDir: tmpDir, convexDir });
 
     expect(vi.mocked(logMessage)).toHaveBeenCalledWith(
       expect.stringContaining("enabled"),
@@ -238,52 +230,46 @@ describe("ai-files integration with default convex/ directory", () => {
   });
 
   test("disable after CLAUDE.md user edits preserves the file", async () => {
-    await updateAiFiles(tmpDir, convexDir);
+    await installAiFiles({ projectDir: tmpDir, convexDir });
     fs.appendFileSync(path.join(tmpDir, "CLAUDE.md"), "My custom note\n");
-    await disableAiFiles(tmpDir);
+    await safelyAttemptToDisableAiFiles(tmpDir);
     expect(fs.readFileSync(path.join(tmpDir, "CLAUDE.md"), "utf8")).toContain(
       "My custom note",
     );
   });
 
   test("update recreates missing CLAUDE.md", async () => {
-    await updateAiFiles(tmpDir, convexDir);
+    await installAiFiles({ projectDir: tmpDir, convexDir });
     fs.rmSync(path.join(tmpDir, "CLAUDE.md"), { force: true });
 
-    await updateAiFiles(tmpDir, convexDir);
+    await installAiFiles({ projectDir: tmpDir, convexDir });
 
     expect(fs.readFileSync(path.join(tmpDir, "CLAUDE.md"), "utf8")).toContain(
       "convex/_generated/ai/guidelines.md",
     );
   });
 
-  test("enable clears disableStalenessMessage", async () => {
-    await updateAiFiles(tmpDir, convexDir);
-    await disableAiFiles(tmpDir);
-    expect(readJson(projectConfigPath()).aiFiles.disableStalenessMessage).toBe(
-      true,
-    );
+  test("enable sets enabled flag to true", async () => {
+    await installAiFiles({ projectDir: tmpDir, convexDir });
+    await safelyAttemptToDisableAiFiles(tmpDir);
+    expect(readJson(projectConfigPath()).aiFiles.enabled).toBe(false);
 
-    await enableAiFiles(tmpDir, convexDir);
+    await enableAiFiles({ projectDir: tmpDir, convexDir });
 
-    expect(readJson(projectConfigPath()).aiFiles.disableStalenessMessage).toBe(
-      false,
-    );
+    expect(readJson(projectConfigPath()).aiFiles.enabled).toBe(true);
   });
 
   test("full cycle: disable -> remove -> enable reinstalls everything", async () => {
-    await updateAiFiles(tmpDir, convexDir);
-    await disableAiFiles(tmpDir);
-    await removeAiFiles(tmpDir, convexDir);
+    await installAiFiles({ projectDir: tmpDir, convexDir });
+    await safelyAttemptToDisableAiFiles(tmpDir);
+    await removeAiFiles({ projectDir: tmpDir, convexDir });
     expect(fs.existsSync(aiDir())).toBe(false);
 
-    await enableAiFiles(tmpDir, convexDir);
+    await enableAiFiles({ projectDir: tmpDir, convexDir });
 
     expect(fs.existsSync(guidelinesPath())).toBe(true);
     expect(fs.existsSync(path.join(tmpDir, "AGENTS.md"))).toBe(true);
-    expect(readJson(projectConfigPath()).aiFiles.disableStalenessMessage).toBe(
-      false,
-    );
+    expect(readJson(projectConfigPath()).aiFiles.enabled).toBe(true);
   });
 
   test("remove strips managed section from AGENTS.md but preserves user content", async () => {
@@ -291,11 +277,11 @@ describe("ai-files integration with default convex/ directory", () => {
       path.join(tmpDir, "AGENTS.md"),
       "# My Project\n\nTeam guidelines here.\n",
     );
-    await updateAiFiles(tmpDir, convexDir);
+    await installAiFiles({ projectDir: tmpDir, convexDir });
     const before = fs.readFileSync(path.join(tmpDir, "AGENTS.md"), "utf8");
     expect(before).toContain(AGENTS_MD_START_MARKER);
 
-    await removeAiFiles(tmpDir, convexDir);
+    await removeAiFiles({ projectDir: tmpDir, convexDir });
 
     const after = fs.readFileSync(path.join(tmpDir, "AGENTS.md"), "utf8");
     expect(after).toContain("# My Project");
@@ -304,19 +290,19 @@ describe("ai-files integration with default convex/ directory", () => {
   });
 
   test("remove on AGENTS.md with only Convex content deletes the file", async () => {
-    await updateAiFiles(tmpDir, convexDir);
+    await installAiFiles({ projectDir: tmpDir, convexDir });
     expect(fs.existsSync(path.join(tmpDir, "AGENTS.md"))).toBe(true);
 
-    await removeAiFiles(tmpDir, convexDir);
+    await removeAiFiles({ projectDir: tmpDir, convexDir });
 
     expect(fs.existsSync(path.join(tmpDir, "AGENTS.md"))).toBe(false);
   });
 
   test("remove deletes CLAUDE.md when empty after stripping managed section", async () => {
-    await updateAiFiles(tmpDir, convexDir);
+    await installAiFiles({ projectDir: tmpDir, convexDir });
     expect(fs.existsSync(path.join(tmpDir, "CLAUDE.md"))).toBe(true);
 
-    await removeAiFiles(tmpDir, convexDir);
+    await removeAiFiles({ projectDir: tmpDir, convexDir });
 
     expect(fs.existsSync(path.join(tmpDir, "CLAUDE.md"))).toBe(false);
   });
@@ -326,11 +312,11 @@ describe("ai-files integration with default convex/ directory", () => {
       path.join(tmpDir, "CLAUDE.md"),
       "My project-specific Claude instructions\n",
     );
-    await updateAiFiles(tmpDir, convexDir);
+    await installAiFiles({ projectDir: tmpDir, convexDir });
     const before = fs.readFileSync(path.join(tmpDir, "CLAUDE.md"), "utf8");
     expect(before).toContain(CLAUDE_MD_START_MARKER);
 
-    await removeAiFiles(tmpDir, convexDir);
+    await removeAiFiles({ projectDir: tmpDir, convexDir });
 
     const after = fs.readFileSync(path.join(tmpDir, "CLAUDE.md"), "utf8");
     expect(after).toContain("My project-specific Claude instructions");
@@ -340,7 +326,12 @@ describe("ai-files integration with default convex/ directory", () => {
   test("checkAiFilesStaleness nags when no state file exists", async () => {
     vi.mocked(logMessage).mockClear();
 
-    await checkAiFilesStaleness(null, null, tmpDir, convexDir);
+    await checkAiFilesStaleness({
+      canonicalGuidelinesHash: null,
+      canonicalAgentSkillsSha: null,
+      projectDir: tmpDir,
+      convexDir,
+    });
 
     expect(vi.mocked(logMessage)).toHaveBeenCalledWith(
       expect.stringContaining("not installed"),
@@ -348,16 +339,16 @@ describe("ai-files integration with default convex/ directory", () => {
   });
 
   test("checkAiFilesStaleness is silent when hashes match", async () => {
-    await updateAiFiles(tmpDir, convexDir);
+    await installAiFiles({ projectDir: tmpDir, convexDir });
     const state = readJson(statePath());
     vi.mocked(logMessage).mockClear();
 
-    await checkAiFilesStaleness(
-      state.guidelinesHash,
-      state.agentSkillsSha,
-      tmpDir,
+    await checkAiFilesStaleness({
+      canonicalGuidelinesHash: state.guidelinesHash,
+      canonicalAgentSkillsSha: state.agentSkillsSha,
+      projectDir: tmpDir,
       convexDir,
-    );
+    });
 
     const calls = vi.mocked(logMessage).mock.calls.map((c) => c[0]);
     expect(
@@ -369,9 +360,9 @@ describe("ai-files integration with default convex/ directory", () => {
   });
 
   test("AGENTS.md managed section is replaced not duplicated on repeated updates", async () => {
-    await updateAiFiles(tmpDir, convexDir);
-    await updateAiFiles(tmpDir, convexDir);
-    await updateAiFiles(tmpDir, convexDir);
+    await installAiFiles({ projectDir: tmpDir, convexDir });
+    await installAiFiles({ projectDir: tmpDir, convexDir });
+    await installAiFiles({ projectDir: tmpDir, convexDir });
 
     const content = fs.readFileSync(path.join(tmpDir, "AGENTS.md"), "utf8");
     const markerCount = content.split(AGENTS_MD_START_MARKER).length - 1;
@@ -379,11 +370,11 @@ describe("ai-files integration with default convex/ directory", () => {
   });
 
   test("status reports disabled state after disable", async () => {
-    await updateAiFiles(tmpDir, convexDir);
-    await disableAiFiles(tmpDir);
+    await installAiFiles({ projectDir: tmpDir, convexDir });
+    await safelyAttemptToDisableAiFiles(tmpDir);
     vi.mocked(logMessage).mockClear();
 
-    await statusAiFiles(tmpDir, convexDir);
+    await statusAiFiles({ projectDir: tmpDir, convexDir });
 
     expect(vi.mocked(logMessage)).toHaveBeenCalledWith(
       expect.stringContaining("disabled"),
@@ -421,7 +412,7 @@ describe("ai-files integration with functions directory override", () => {
   });
 
   test("installs into overridden functions directory, not default convex/", async () => {
-    await updateAiFiles(tmpDir, convexDir);
+    await installAiFiles({ projectDir: tmpDir, convexDir });
 
     expect(fs.existsSync(guidelinesPath())).toBe(true);
     expect(fs.existsSync(statePath())).toBe(true);
@@ -440,7 +431,7 @@ describe("ai-files integration with functions directory override", () => {
       path.join(tmpDir, "AGENTS.md"),
       "# Existing\n\nUser content.\n",
     );
-    await updateAiFiles(tmpDir, convexDir);
+    await installAiFiles({ projectDir: tmpDir, convexDir });
     const content = fs.readFileSync(path.join(tmpDir, "AGENTS.md"), "utf8");
     expect(content).toContain("# Existing");
     expect(content).toContain("User content.");
@@ -456,17 +447,17 @@ describe("ai-files integration with functions directory override", () => {
       "My custom CLAUDE.md content\n",
       "utf8",
     );
-    await updateAiFiles(tmpDir, convexDir);
+    await installAiFiles({ projectDir: tmpDir, convexDir });
     const content = fs.readFileSync(path.join(tmpDir, "CLAUDE.md"), "utf8");
     expect(content).toContain("My custom CLAUDE.md content");
   });
 
   test("second update is idempotent", async () => {
-    await updateAiFiles(tmpDir, convexDir);
+    await installAiFiles({ projectDir: tmpDir, convexDir });
     const firstGuidelines = fs.readFileSync(guidelinesPath(), "utf8");
     const firstState = fs.readFileSync(statePath(), "utf8");
 
-    await updateAiFiles(tmpDir, convexDir);
+    await installAiFiles({ projectDir: tmpDir, convexDir });
 
     expect(fs.readFileSync(guidelinesPath(), "utf8")).toBe(firstGuidelines);
     expect(fs.readFileSync(statePath(), "utf8")).toBe(firstState);
@@ -479,36 +470,14 @@ describe("ai-files integration with functions directory override", () => {
       "legacy",
       "utf8",
     );
-    await updateAiFiles(tmpDir, convexDir);
+    await installAiFiles({ projectDir: tmpDir, convexDir });
     expect(
       fs.existsSync(path.join(tmpDir, ".cursor", "rules", "convex_rules.mdc")),
     ).toBe(false);
   });
 
-  test("skips locally modified guidelines when stored hash is stale", async () => {
-    await updateAiFiles(tmpDir, convexDir);
-    const localNote = "\n## My custom note\n";
-    fs.appendFileSync(guidelinesPath(), localNote, "utf8");
-    const state = readJson(statePath());
-    state.guidelinesHash = "deliberately-stale-hash";
-    fs.writeFileSync(
-      statePath(),
-      JSON.stringify(state, null, 2) + "\n",
-      "utf8",
-    );
-
-    await updateAiFiles(tmpDir, convexDir);
-
-    expect(fs.readFileSync(guidelinesPath(), "utf8")).toContain(
-      localNote.trim(),
-    );
-    expect(vi.mocked(logMessage)).toHaveBeenCalledWith(
-      expect.stringContaining("modified locally"),
-    );
-  });
-
   test("staleness check logs update nag for stale stored hash", async () => {
-    await updateAiFiles(tmpDir, convexDir);
+    await installAiFiles({ projectDir: tmpDir, convexDir });
     const state = readJson(statePath());
     state.guidelinesHash = "deliberately-stale-hash";
     fs.writeFileSync(
@@ -517,7 +486,12 @@ describe("ai-files integration with functions directory override", () => {
       "utf8",
     );
 
-    await checkAiFilesStaleness("canonical-hash", null, tmpDir, convexDir);
+    await checkAiFilesStaleness({
+      canonicalGuidelinesHash: "canonical-hash",
+      canonicalAgentSkillsSha: null,
+      projectDir: tmpDir,
+      convexDir,
+    });
 
     expect(vi.mocked(logMessage)).toHaveBeenCalledWith(
       expect.stringContaining("out of date"),
@@ -525,21 +499,19 @@ describe("ai-files integration with functions directory override", () => {
   });
 
   test("disable sets convex.json preference and keeps files", async () => {
-    await updateAiFiles(tmpDir, convexDir);
-    await disableAiFiles(tmpDir);
+    await installAiFiles({ projectDir: tmpDir, convexDir });
+    await safelyAttemptToDisableAiFiles(tmpDir);
 
-    expect(readJson(projectConfigPath()).aiFiles.disableStalenessMessage).toBe(
-      true,
-    );
+    expect(readJson(projectConfigPath()).aiFiles.enabled).toBe(false);
     expect(fs.existsSync(guidelinesPath())).toBe(true);
     expect(fs.existsSync(path.join(tmpDir, "AGENTS.md"))).toBe(true);
     expect(fs.existsSync(path.join(tmpDir, "CLAUDE.md"))).toBe(true);
   });
 
   test("remove deletes files and status reports not installed", async () => {
-    await updateAiFiles(tmpDir, convexDir);
-    await removeAiFiles(tmpDir, convexDir);
-    await statusAiFiles(tmpDir, convexDir);
+    await installAiFiles({ projectDir: tmpDir, convexDir });
+    await removeAiFiles({ projectDir: tmpDir, convexDir });
+    await statusAiFiles({ projectDir: tmpDir, convexDir });
 
     expect(fs.existsSync(aiDir())).toBe(false);
     expect(vi.mocked(logMessage)).toHaveBeenCalledWith(
@@ -548,23 +520,23 @@ describe("ai-files integration with functions directory override", () => {
   });
 
   test("disable after CLAUDE.md user edits preserves file", async () => {
-    await updateAiFiles(tmpDir, convexDir);
+    await installAiFiles({ projectDir: tmpDir, convexDir });
     fs.appendFileSync(
       path.join(tmpDir, "CLAUDE.md"),
       "My custom note\n",
       "utf8",
     );
-    await disableAiFiles(tmpDir);
+    await safelyAttemptToDisableAiFiles(tmpDir);
     expect(fs.readFileSync(path.join(tmpDir, "CLAUDE.md"), "utf8")).toContain(
       "My custom note",
     );
   });
 
   test("update recreates missing CLAUDE.md", async () => {
-    await updateAiFiles(tmpDir, convexDir);
+    await installAiFiles({ projectDir: tmpDir, convexDir });
     fs.rmSync(path.join(tmpDir, "CLAUDE.md"), { force: true });
 
-    await updateAiFiles(tmpDir, convexDir);
+    await installAiFiles({ projectDir: tmpDir, convexDir });
 
     const content = fs.readFileSync(path.join(tmpDir, "CLAUDE.md"), "utf8");
     expect(content).toMatch(
@@ -572,30 +544,28 @@ describe("ai-files integration with functions directory override", () => {
     );
   });
 
-  test("enable clears disableStalenessMessage and re-enables status", async () => {
-    await updateAiFiles(tmpDir, convexDir);
-    await disableAiFiles(tmpDir);
-    await enableAiFiles(tmpDir, convexDir);
-    await statusAiFiles(tmpDir, convexDir);
+  test("enable sets enabled flag to true and re-enables status", async () => {
+    await installAiFiles({ projectDir: tmpDir, convexDir });
+    await safelyAttemptToDisableAiFiles(tmpDir);
+    await enableAiFiles({ projectDir: tmpDir, convexDir });
+    await statusAiFiles({ projectDir: tmpDir, convexDir });
 
-    expect(readJson(projectConfigPath()).aiFiles.disableStalenessMessage).toBe(
-      false,
-    );
+    expect(readJson(projectConfigPath()).aiFiles.enabled).toBe(true);
     expect(vi.mocked(logMessage)).toHaveBeenCalledWith(
       expect.stringContaining("enabled"),
     );
   });
 
   test("disable + remove + enable works with overridden functions directory", async () => {
-    await updateAiFiles(tmpDir, convexDir);
-    await disableAiFiles(tmpDir);
-    await removeAiFiles(tmpDir, convexDir);
+    await installAiFiles({ projectDir: tmpDir, convexDir });
+    await safelyAttemptToDisableAiFiles(tmpDir);
+    await removeAiFiles({ projectDir: tmpDir, convexDir });
 
     expect(
       fs.existsSync(path.join(tmpDir, "src", "convex", "_generated", "ai")),
     ).toBe(false);
 
-    await enableAiFiles(tmpDir, convexDir);
+    await enableAiFiles({ projectDir: tmpDir, convexDir });
 
     expect(fs.existsSync(guidelinesPath())).toBe(true);
   });

package/src/cli/lib/{ai → aiFiles}/paths.ts

@@ -1,5 +1,10 @@
 import path from "path";
 
+export type AiFilesPaths = {
+  projectDir: string;
+  convexDir: string;
+};
+
 const AI_FILES_PARENT_DIR = "_generated";
 const AI_FILES_DIR = "ai";