ai-spec-dev 0.17.0 → 0.25.0

package/core/mock-server-generator.ts

@@ -1,5 +1,6 @@
 import * as path from "path";
 import * as fs from "fs-extra";
+import { spawn } from "child_process";
 import { SpecDSL, ApiEndpoint, FieldMap } from "./dsl-types";
 
 // ─── Types ────────────────────────────────────────────────────────────────────
@@ -434,6 +435,215 @@ export const worker = setupWorker(...handlers);
 `;
 }
 
+// ─── Proxy Patching (applyMockProxy / restoreMockProxy) ──────────────────────
+
+const MOCK_LOCK_FILE = ".ai-spec-mock.lock.json";
+
+interface MockLock {
+  framework: string;
+  mockPort: number;
+  frontendDir: string;
+  mockServerPid?: number;
+  actions: Array<
+    | { type: "wrote-file"; filePath: string }
+    | { type: "patched-pkg-proxy"; originalProxy?: string | null }
+    | { type: "added-pkg-script"; key: string; originalValue?: string | null }
+  >;
+}
+
+export interface ProxyApplyResult {
+  framework: string;
+  applied: boolean;
+  devCommand: string | null;
+  note?: string;
+}
+
+function findViteConfigFile(projectDir: string): string | null {
+  for (const f of ["vite.config.ts", "vite.config.mts", "vite.config.js", "vite.config.mjs"]) {
+    if (fs.existsSync(path.join(projectDir, f))) return f;
+  }
+  return null;
+}
+
+function buildViteProxyEntries(endpoints: ApiEndpoint[], mockPort: number): string {
+  const prefixes = new Set<string>();
+  for (const ep of endpoints) {
+    const parts = ep.path.split("/").filter(Boolean);
+    if (parts.length > 0) prefixes.add(`/${parts[0]}`);
+  }
+  if (prefixes.size === 0) prefixes.add("/api");
+  const target = `http://localhost:${mockPort}`;
+  return Array.from(prefixes)
+    .map((p) => `        '${p}': { target: '${target}', changeOrigin: true },`)
+    .join("\n");
+}
+
+function generateViteMockConfigTs(baseConfigFile: string, mockPort: number, endpoints: ApiEndpoint[]): string {
+  const importPath = `./${baseConfigFile.replace(/\.(ts|mts|js|mjs)$/, "")}`;
+  const proxyEntries = buildViteProxyEntries(endpoints, mockPort);
+  return `// Auto-generated by ai-spec mock --serve
+// LOCAL DEVELOPMENT ONLY — do not commit this file
+// Remove with: ai-spec mock --restore
+import { defineConfig, mergeConfig } from 'vite';
+
+export default defineConfig(async (env) => {
+  const mod = await import('${importPath}');
+  const baseConfigOrFn = mod.default;
+  const baseConfig =
+    typeof baseConfigOrFn === 'function'
+      ? await baseConfigOrFn(env)
+      : baseConfigOrFn;
+
+  return mergeConfig(baseConfig ?? {}, {
+    server: {
+      proxy: {
+${proxyEntries}
+      },
+    },
+  });
+});
+`;
+}
+
+/**
+ * Patch the frontend project's proxy config to point to the mock server.
+ * Vite: writes vite.config.ai-spec-mock.ts + adds "dev:mock" npm script.
+ * CRA : patches package.json "proxy" field (original backed up in lock file).
+ * Saves .ai-spec-mock.lock.json so restoreMockProxy() can undo all changes.
+ */
+export async function applyMockProxy(
+  frontendDir: string,
+  mockPort: number,
+  endpoints: ApiEndpoint[] = []
+): Promise<ProxyApplyResult> {
+  const framework = detectFrontendFramework(frontendDir);
+  const actions: MockLock["actions"] = [];
+
+  if (framework === "vite") {
+    const viteConfigFile = findViteConfigFile(frontendDir) ?? "vite.config.ts";
+    const mockConfigContent = generateViteMockConfigTs(viteConfigFile, mockPort, endpoints);
+    const mockConfigPath = path.join(frontendDir, "vite.config.ai-spec-mock.ts");
+    await fs.writeFile(mockConfigPath, mockConfigContent, "utf-8");
+    actions.push({ type: "wrote-file", filePath: "vite.config.ai-spec-mock.ts" });
+
+    const pkgPath = path.join(frontendDir, "package.json");
+    if (await fs.pathExists(pkgPath)) {
+      const pkg = await fs.readJson(pkgPath);
+      pkg.scripts = pkg.scripts ?? {};
+      const originalValue: string | null = pkg.scripts["dev:mock"] ?? null;
+      pkg.scripts["dev:mock"] = "vite --config vite.config.ai-spec-mock.ts";
+      await fs.writeJson(pkgPath, pkg, { spaces: 2 });
+      actions.push({ type: "added-pkg-script", key: "dev:mock", originalValue });
+    }
+
+    const lock: MockLock = { framework, mockPort, frontendDir, actions };
+    await fs.writeJson(path.join(frontendDir, MOCK_LOCK_FILE), lock, { spaces: 2 });
+    return { framework, applied: true, devCommand: "npm run dev:mock" };
+  }
+
+  if (framework === "cra") {
+    const pkgPath = path.join(frontendDir, "package.json");
+    if (await fs.pathExists(pkgPath)) {
+      const pkg = await fs.readJson(pkgPath);
+      const originalProxy: string | null = pkg.proxy ?? null;
+      pkg.proxy = `http://localhost:${mockPort}`;
+      await fs.writeJson(pkgPath, pkg, { spaces: 2 });
+      actions.push({ type: "patched-pkg-proxy", originalProxy });
+      const lock: MockLock = { framework, mockPort, frontendDir, actions };
+      await fs.writeJson(path.join(frontendDir, MOCK_LOCK_FILE), lock, { spaces: 2 });
+      return { framework, applied: true, devCommand: "npm start" };
+    }
+    return { framework, applied: false, devCommand: null, note: "No package.json found." };
+  }
+
+  // next / webpack / unknown — save lock but no auto-patch
+  const lock: MockLock = { framework, mockPort, frontendDir, actions };
+  await fs.writeJson(path.join(frontendDir, MOCK_LOCK_FILE), lock, { spaces: 2 });
+  const manualNote =
+    framework === "next"
+      ? `Add rewrites in next.config.js to proxy API calls to http://localhost:${mockPort}`
+      : `Add proxy in webpack.config.js devServer to target http://localhost:${mockPort}`;
+  return { framework, applied: false, devCommand: null, note: manualNote };
+}
+
+/**
+ * Undo all proxy changes made by applyMockProxy().
+ * Also kills the mock server if its PID was stored in the lock file.
+ */
+export async function restoreMockProxy(
+  frontendDir: string
+): Promise<{ restored: boolean; note?: string }> {
+  const lockPath = path.join(frontendDir, MOCK_LOCK_FILE);
+  if (!(await fs.pathExists(lockPath))) {
+    return { restored: false, note: "No lock file found — nothing to restore." };
+  }
+
+  const lock: MockLock = await fs.readJson(lockPath);
+
+  for (const action of lock.actions) {
+    if (action.type === "wrote-file") {
+      const fp = path.join(frontendDir, action.filePath);
+      if (await fs.pathExists(fp)) await fs.remove(fp);
+    } else if (action.type === "added-pkg-script") {
+      const pkgPath = path.join(frontendDir, "package.json");
+      if (await fs.pathExists(pkgPath)) {
+        const pkg = await fs.readJson(pkgPath);
+        if (action.originalValue == null) {
+          delete pkg.scripts?.[action.key];
+        } else {
+          pkg.scripts = pkg.scripts ?? {};
+          pkg.scripts[action.key] = action.originalValue;
+        }
+        await fs.writeJson(pkgPath, pkg, { spaces: 2 });
+      }
+    } else if (action.type === "patched-pkg-proxy") {
+      const pkgPath = path.join(frontendDir, "package.json");
+      if (await fs.pathExists(pkgPath)) {
+        const pkg = await fs.readJson(pkgPath);
+        if (action.originalProxy == null) {
+          delete pkg.proxy;
+        } else {
+          pkg.proxy = action.originalProxy;
+        }
+        await fs.writeJson(pkgPath, pkg, { spaces: 2 });
+      }
+    }
+  }
+
+  if (lock.mockServerPid) {
+    try { process.kill(lock.mockServerPid, "SIGTERM"); } catch { /* already dead */ }
+  }
+
+  await fs.remove(lockPath);
+  return { restored: true };
+}
+
+/**
+ * Start mock/server.js as a detached background process.
+ * Returns the spawned PID so it can be stored for later cleanup.
+ */
+export function startMockServerBackground(serverJsPath: string, port: number): number {
+  const child = spawn("node", [serverJsPath], {
+    detached: true,
+    stdio: "ignore",
+    env: { ...process.env, MOCK_PORT: String(port) },
+  });
+  child.unref();
+  return child.pid!;
+}
+
+/**
+ * Save the mock server PID into an existing lock file.
+ */
+export async function saveMockServerPid(frontendDir: string, pid: number): Promise<void> {
+  const lockPath = path.join(frontendDir, MOCK_LOCK_FILE);
+  if (await fs.pathExists(lockPath)) {
+    const lock: MockLock = await fs.readJson(lockPath);
+    lock.mockServerPid = pid;
+    await fs.writeJson(lockPath, lock, { spaces: 2 });
+  }
+}
+
 // ─── Public API ───────────────────────────────────────────────────────────────
 
 /**

package/core/task-generator.ts

@@ -65,7 +65,7 @@ export function buildTaskPrompt(spec: string, context?: ProjectContext): string
   return parts.join("\n");
 }
 
-export type TaskLayer = "data" | "infra" | "service" | "api" | "test";
+export type TaskLayer = "data" | "infra" | "service" | "api" | "view" | "route" | "test";
 export type TaskPriority = "high" | "medium" | "low";
 export type TaskStatus = "pending" | "done" | "failed";
 
@@ -87,7 +87,9 @@ const LAYER_ORDER: Record<TaskLayer, number> = {
   infra: 1,
   service: 2,
   api: 3,
-  test: 4,
+  view: 4,
+  route: 5,
+  test: 6,
 };
 
 export class TaskGenerator {
@@ -135,6 +137,8 @@ export function printTasks(tasks: SpecTask[]): void {
     infra: chalk.gray,
     service: chalk.blue,
     api: chalk.cyan,
+    view: chalk.yellow,
+    route: chalk.white,
     test: chalk.green,
   };