opencroc 1.4.3 → 1.5.1

package/dist/cli/index.js

@@ -1270,6 +1270,10 @@ var init_config_validator = __esm({
 });
 
 // src/pipeline/index.ts
+var pipeline_exports = {};
+__export(pipeline_exports, {
+  createPipeline: () => createPipeline
+});
 import * as fs4 from "fs";
 import * as path5 from "path";
 function createPipeline(config) {
@@ -1285,11 +1289,32 @@ function createPipeline(config) {
         validationErrors: [],
         duration: 0
       };
+      const backendRoot = path5.resolve(config.backendRoot);
+      const findDir = (name) => {
+        const candidates = [
+          path5.join(backendRoot, name),
+          // ./models
+          path5.join(backendRoot, "src", name),
+          // ./src/models
+          path5.join(backendRoot, "backend", "src", name),
+          // ./backend/src/models
+          path5.join(backendRoot, "backend", name),
+          // ./backend/models
+          path5.join(backendRoot, "server", "src", name),
+          // ./server/src/models
+          path5.join(backendRoot, "app", name)
+          // ./app/models
+        ];
+        for (const c of candidates) {
+          if (fs4.existsSync(c)) return c;
+        }
+        return null;
+      };
+      const modelsRoot = findDir("models");
+      const controllersRoot = findDir("controllers");
       if (activeSteps.includes("scan")) {
-        const backendRoot = path5.resolve(config.backendRoot);
-        const modelsDir = path5.join(backendRoot, "models");
-        if (fs4.existsSync(modelsDir)) {
-          const dirs = fs4.readdirSync(modelsDir, { withFileTypes: true }).filter((d) => d.isDirectory()).map((d) => d.name);
+        if (modelsRoot) {
+          const dirs = fs4.readdirSync(modelsRoot, { withFileTypes: true }).filter((d) => d.isDirectory()).map((d) => d.name);
           const moduleFilter = config.modules;
           for (const dir of dirs) {
             if (moduleFilter && !moduleFilter.includes(dir)) continue;
@@ -1298,20 +1323,20 @@ function createPipeline(config) {
           if (result.modules.length === 0) {
             result.modules.push("default");
           } else {
-            const rootFiles = fs4.readdirSync(modelsDir).filter((f) => f.endsWith(".ts") && !f.endsWith(".test.ts") && f !== "index.ts");
+            const rootFiles = fs4.readdirSync(modelsRoot).filter((f) => f.endsWith(".ts") && !f.endsWith(".test.ts") && f !== "index.ts");
             if (rootFiles.length > 0) {
               result.modules.unshift("default");
             }
           }
         }
       }
-      const resolveModelDir = (backendRoot, mod) => mod === "default" ? path5.join(backendRoot, "models") : path5.join(backendRoot, "models", mod);
-      const resolveControllerDir = (backendRoot, mod) => mod === "default" ? path5.join(backendRoot, "controllers") : path5.join(backendRoot, "controllers", mod);
+      const resolveModelDir = (_backendRoot, mod) => mod === "default" ? modelsRoot || path5.join(backendRoot, "models") : path5.join(modelsRoot || path5.join(backendRoot, "models"), mod);
+      const resolveControllerDir = (_backendRoot, mod) => mod === "default" ? controllersRoot || path5.join(backendRoot, "controllers") : path5.join(controllersRoot || path5.join(backendRoot, "controllers"), mod);
       if (activeSteps.includes("er-diagram")) {
         const erGen = createERDiagramGenerator();
-        const backendRoot = path5.resolve(config.backendRoot);
+        const backendRoot2 = path5.resolve(config.backendRoot);
         for (const mod of result.modules) {
-          const modelDir = resolveModelDir(backendRoot, mod);
+          const modelDir = resolveModelDir(backendRoot2, mod);
           const tables = fs4.existsSync(modelDir) ? parseModuleModels(modelDir) : [];
           const relations = [];
           const assocFile = path5.join(modelDir, "associations.ts");
@@ -1334,9 +1359,9 @@ function createPipeline(config) {
       }
       if (activeSteps.includes("api-chain")) {
         const chainAnalyzer = createApiChainAnalyzer();
-        const backendRoot = path5.resolve(config.backendRoot);
+        const backendRoot2 = path5.resolve(config.backendRoot);
         for (const mod of result.modules) {
-          const controllerDir = resolveControllerDir(backendRoot, mod);
+          const controllerDir = resolveControllerDir(backendRoot2, mod);
           const endpoints = fs4.existsSync(controllerDir) ? parseControllerDirectory(controllerDir) : [];
           const analysis = chainAnalyzer.analyze(endpoints);
           analysis.moduleName = mod;
@@ -1353,10 +1378,10 @@ function createPipeline(config) {
         }
       }
       if (activeSteps.includes("plan")) {
-        const backendRoot = path5.resolve(config.backendRoot);
+        const backendRoot2 = path5.resolve(config.backendRoot);
         const chainAnalyzer = createApiChainAnalyzer();
         for (const mod of result.modules) {
-          const controllerDir = resolveControllerDir(backendRoot, mod);
+          const controllerDir = resolveControllerDir(backendRoot2, mod);
           const endpoints = fs4.existsSync(controllerDir) ? parseControllerDirectory(controllerDir) : [];
           const analysis = chainAnalyzer.analyze(endpoints);
           const topoOrder = topologicalSort(analysis.dag);
@@ -3491,6 +3516,39 @@ function registerAgentRoutes(app, office) {
       agents: office.getAgents()
     };
   });
+  app.get("/api/files", async () => {
+    const files = office.getGeneratedFiles();
+    return files.map((f) => ({
+      filePath: f.filePath,
+      module: f.module,
+      chain: f.chain,
+      lines: f.content.split("\n").length,
+      size: f.content.length
+    }));
+  });
+  app.get("/api/files/:index", async (req, reply) => {
+    const files = office.getGeneratedFiles();
+    const idx = parseInt(req.params.index, 10);
+    if (isNaN(idx) || idx < 0 || idx >= files.length) {
+      reply.code(404).send({ error: "File not found" });
+      return;
+    }
+    return files[idx];
+  });
+  app.get("/api/pipeline/result", async () => {
+    const result = office.getLastPipelineResult();
+    if (!result) return { ok: false, message: "No pipeline has been run yet" };
+    return {
+      ok: true,
+      modules: result.modules,
+      erDiagramCount: result.erDiagrams.size,
+      chainCount: [...result.chainPlans.values()].reduce((s, p) => s + p.chains.length, 0),
+      totalSteps: [...result.chainPlans.values()].reduce((s, p) => s + p.totalSteps, 0),
+      filesGenerated: result.generatedFiles.length,
+      validationErrors: result.validationErrors.length,
+      duration: result.duration
+    };
+  });
 }
 var init_agents = __esm({
   "src/server/routes/agents.ts"() {
@@ -3520,6 +3578,8 @@ var init_croc_office = __esm({
       agents;
       cachedGraph = null;
       running = false;
+      lastPipelineResult = null;
+      lastGeneratedFiles = [];
       constructor(config, cwd) {
         this.config = config;
         this.cwd = cwd;
@@ -3589,41 +3649,92 @@ var init_croc_office = __esm({
           this.running = false;
         }
       }
-      /** Run the pipeline: scan → er-diagram → api-chain → plan → codegen */
+      /** Run the real pipeline: scan → er-diagram → api-chain → plan → codegen → report */
       async runPipeline() {
         if (this.running) return { ok: false, task: "pipeline", duration: 0, error: "Another task is running" };
         this.running = true;
         const start = Date.now();
         try {
+          const { resolve: resolvePath } = await import("path");
+          const { createPipeline: createPipeline2 } = await Promise.resolve().then(() => (init_pipeline(), pipeline_exports));
+          const backendRoot = resolvePath(this.cwd, this.config.backendRoot);
+          const pipelineConfig = { ...this.config, backendRoot };
+          const pipeline = createPipeline2(pipelineConfig);
           this.updateAgent("parser-croc", { status: "working", currentTask: "Scanning source code...", progress: 10 });
-          this.log("\u{1F40A} \u89E3\u6790\u9CC4 is scanning source code...");
+          this.log(`\u{1F40A} \u89E3\u6790\u9CC4 scanning from: ${backendRoot}`);
           this.invalidateCache();
           await this.buildKnowledgeGraph();
           this.updateNodeStatus("module", "testing");
-          this.updateAgent("parser-croc", { status: "done", currentTask: "Scan complete", progress: 100 });
+          this.updateAgent("parser-croc", { currentTask: "Parsing models & ER diagrams...", progress: 40 });
+          const scanResult = await pipeline.run(["scan", "er-diagram"]);
+          const moduleCount = scanResult.modules.length;
+          const erCount = scanResult.erDiagrams.size;
+          this.log(`\u{1F4CA} Found ${moduleCount} modules, ${erCount} ER diagrams`);
+          this.updateAgent("parser-croc", { status: "done", currentTask: `${moduleCount} modules parsed`, progress: 100 });
           this.updateAgent("analyzer-croc", { status: "working", currentTask: "Analyzing API chains...", progress: 0 });
           this.log("\u{1F40A} \u5206\u6790\u9CC4 is analyzing API dependencies...");
-          await this.delay(800);
+          const analyzeResult = await pipeline.run(["api-chain"]);
+          const warnings = analyzeResult.validationErrors.filter((e) => e.severity === "warning");
+          if (warnings.length > 0) {
+            this.log(`\u26A0\uFE0F ${warnings.length} API chain warnings`, "warn");
+          }
           this.updateAgent("analyzer-croc", { status: "done", currentTask: "Analysis complete", progress: 100 });
           this.updateAgent("planner-croc", { status: "thinking", currentTask: "Planning test chains...", progress: 0 });
           this.log("\u{1F40A} \u89C4\u5212\u9CC4 is planning test chains...");
-          await this.delay(600);
-          this.updateAgent("planner-croc", { status: "done", currentTask: "Plan ready", progress: 100 });
+          const planResult = await pipeline.run(["plan"]);
+          let totalChains = 0, totalSteps = 0;
+          for (const [, plan] of planResult.chainPlans) {
+            totalChains += plan.chains.length;
+            totalSteps += plan.totalSteps;
+          }
+          this.log(`\u{1F4CB} Planned ${totalChains} test chains with ${totalSteps} steps`);
+          this.updateAgent("planner-croc", { status: "done", currentTask: `${totalChains} chains planned`, progress: 100 });
           this.updateAgent("tester-croc", { status: "working", currentTask: "Generating test code...", progress: 0 });
-          this.log("\u{1F40A} \u6D4B\u8BD5\u9CC4 is generating test code...");
+          this.log("\u{1F40A} \u6D4B\u8BD5\u9CC4 is generating Playwright test code...");
           this.updateNodeStatus("controller", "testing");
-          await this.delay(500);
+          const fullResult = await pipeline.run(["scan", "er-diagram", "api-chain", "plan", "codegen"]);
+          this.lastPipelineResult = fullResult;
+          this.lastGeneratedFiles = fullResult.generatedFiles;
+          const { writeFileSync: writeFileSync10, mkdirSync: mkdirSync10 } = await import("fs");
+          const { dirname: dirname6 } = await import("path");
+          let filesWritten = 0;
+          for (const file of fullResult.generatedFiles) {
+            const fullPath = resolvePath(this.cwd, file.filePath);
+            mkdirSync10(dirname6(fullPath), { recursive: true });
+            writeFileSync10(fullPath, file.content, "utf-8");
+            filesWritten++;
+          }
           this.updateNodeStatus("controller", "passed");
-          this.updateAgent("tester-croc", { status: "done", currentTask: "Tests generated", progress: 100 });
-          this.updateAgent("reporter-croc", { status: "working", currentTask: "Building report...", progress: 0 });
+          this.log(`\u2705 Generated ${filesWritten} test files`);
+          this.updateAgent("tester-croc", { status: "done", currentTask: `${filesWritten} files generated`, progress: 100 });
+          this.broadcast("files:generated", fullResult.generatedFiles.map((f) => ({
+            filePath: f.filePath,
+            module: f.module,
+            chain: f.chain,
+            lines: f.content.split("\n").length
+          })));
+          this.updateAgent("reporter-croc", { status: "working", currentTask: "Compiling report...", progress: 0 });
           this.log("\u{1F40A} \u6C47\u62A5\u9CC4 is compiling results...");
-          await this.delay(400);
+          const validateResult = await pipeline.run(["validate"]);
+          const errors = validateResult.validationErrors.filter((e) => e.severity === "error");
+          if (errors.length > 0) {
+            this.log(`\u26A0\uFE0F ${errors.length} validation errors`, "warn");
+          }
           this.updateNodeStatus("module", "passed");
           this.updateAgent("reporter-croc", { status: "done", currentTask: "Report ready", progress: 100 });
           const duration = Date.now() - start;
-          this.log(`\u2705 Pipeline complete in ${duration}ms`);
-          this.broadcast("pipeline:complete", { duration, status: "success" });
-          return { ok: true, task: "pipeline", duration };
+          this.log(`\u2705 Pipeline complete in ${duration}ms \u2014 ${moduleCount} modules, ${totalChains} chains, ${filesWritten} files`);
+          this.broadcast("pipeline:complete", {
+            duration,
+            status: "success",
+            summary: { modules: moduleCount, chains: totalChains, steps: totalSteps, files: filesWritten }
+          });
+          return { ok: true, task: "pipeline", duration, details: {
+            modules: moduleCount,
+            chains: totalChains,
+            steps: totalSteps,
+            files: filesWritten
+          } };
         } catch (err) {
           this.updateAgent("tester-croc", { status: "error", currentTask: String(err) });
           this.log(`\u274C Pipeline failed: ${err}`, "error");
@@ -3642,6 +3753,14 @@ var init_croc_office = __esm({
         }
         this.broadcast("agent:update", this.agents);
       }
+      /** Get last pipeline result */
+      getLastPipelineResult() {
+        return this.lastPipelineResult;
+      }
+      /** Get generated test files from last pipeline run */
+      getGeneratedFiles() {
+        return this.lastGeneratedFiles;
+      }
       // ============ Graph Helpers ============
       updateNodeStatus(type, status) {
         if (!this.cachedGraph) return;
@@ -3652,9 +3771,6 @@ var init_croc_office = __esm({
         }
         this.broadcast("graph:update", this.cachedGraph);
       }
-      delay(ms) {
-        return new Promise((resolve9) => setTimeout(resolve9, ms));
-      }
       /** Build knowledge graph from project source code */
       async buildKnowledgeGraph() {
         if (this.cachedGraph) return this.cachedGraph;