@wix/ditto-codegen-public 1.0.313 → 1.0.314

package/dist/out.js

@@ -12095,7 +12095,8 @@ var require_app_builder = __commonJS({
         { filename: "codegen-instructions.md", content: codegen_rules_1.CODEGEN_RULES }
       ],
       askRules: [{ filename: "ask-instructions.md", content: ask_rules_1.ASK_RULES }],
-      decisionCheckers: [pre_run_decision_1.checkSiteUIDecision]
+      decisionCheckers: [pre_run_decision_1.checkSiteUIDecision],
+      hooks: []
     };
   }
 });
@@ -12113,20 +12114,93 @@ var require_studio_2 = __commonJS({
         { filename: "codegen-instructions.md", content: codegen_rules_1.CODEGEN_RULES }
       ],
       askRules: [{ filename: "ask-instructions.md", content: ask_rules_1.ASK_RULES }],
-      decisionCheckers: []
+      decisionCheckers: [],
+      hooks: []
     };
   }
 });
 
+// dist/environments/hooks/run-hooks.js
+var require_run_hooks = __commonJS({
+  "dist/environments/hooks/run-hooks.js"(exports2) {
+    "use strict";
+    Object.defineProperty(exports2, "__esModule", { value: true });
+    exports2.getActiveHooks = getActiveHooks;
+    exports2.runOnToolUse = runOnToolUse;
+    exports2.runOnFlowEnd = runOnFlowEnd;
+    var logger_12 = require_logger();
+    var job_context_storage_12 = require_job_context_storage();
+    function getActiveHooks() {
+      return job_context_storage_12.jobContextStorage.getStore()?.hooks ?? [];
+    }
+    async function runOnToolUse(hooks, event) {
+      for (const hook of hooks) {
+        if (!hook.onToolUse)
+          continue;
+        try {
+          await hook.onToolUse(event);
+        } catch (error) {
+          logger_12.logger.error("[Hook] onToolUse failed", {
+            hook: hook.name,
+            error: error instanceof Error ? error.message : String(error)
+          });
+        }
+      }
+    }
+    async function runOnFlowEnd(hooks, ctx) {
+      for (const hook of hooks) {
+        if (!hook.onFlowEnd)
+          continue;
+        try {
+          await hook.onFlowEnd(ctx);
+        } catch (hookError) {
+          if (ctx.failed) {
+            logger_12.logger.error("[Hook] onFlowEnd failed after flow failure", {
+              hook: hook.name,
+              error: hookError instanceof Error ? hookError.message : String(hookError)
+            });
+            continue;
+          }
+          throw hookError;
+        }
+      }
+    }
+  }
+});
+
+// dist/environments/hooks/index.js
+var require_hooks = __commonJS({
+  "dist/environments/hooks/index.js"(exports2) {
+    "use strict";
+    Object.defineProperty(exports2, "__esModule", { value: true });
+    exports2.runOnFlowEnd = exports2.runOnToolUse = exports2.getActiveHooks = void 0;
+    var run_hooks_1 = require_run_hooks();
+    Object.defineProperty(exports2, "getActiveHooks", { enumerable: true, get: function() {
+      return run_hooks_1.getActiveHooks;
+    } });
+    Object.defineProperty(exports2, "runOnToolUse", { enumerable: true, get: function() {
+      return run_hooks_1.runOnToolUse;
+    } });
+    Object.defineProperty(exports2, "runOnFlowEnd", { enumerable: true, get: function() {
+      return run_hooks_1.runOnFlowEnd;
+    } });
+  }
+});
+
 // dist/environments/index.js
 var require_environments = __commonJS({
   "dist/environments/index.js"(exports2) {
     "use strict";
     Object.defineProperty(exports2, "__esModule", { value: true });
+    exports2.getActiveHooks = void 0;
     exports2.getCodegenEnvironmentConfig = getCodegenEnvironmentConfig;
     var types_1 = require_types_impl();
     var app_builder_1 = require_app_builder();
     var studio_2_1 = require_studio_2();
+    var hooks_1 = require_hooks();
+    Object.defineProperty(exports2, "getActiveHooks", { enumerable: true, get: function() {
+      return hooks_1.getActiveHooks;
+    } });
     function getCodegenEnvironmentConfig(environment) {
       switch (environment) {
         case types_1.Environment.STUDIO_2:
@@ -12774,6 +12848,7 @@ var require_task_tracker = __commonJS({
     var constants_1 = require_constants5();
     var extension_handler_1 = require_extension_handler();
     var category_handler_1 = require_category_handler();
+    var hooks_1 = require_hooks();
     var OpenCodeTaskTracker = class {
       constructor(workingDirectory) {
         this.trackedTasks = /* @__PURE__ */ new Map();
@@ -12830,6 +12905,7 @@ var require_task_tracker = __commonJS({
         const ctx = this.handlerContext;
         await this.categoryHandler.handle(tool, ctx, state);
         this.taskCounter = ctx.taskCounter;
+        await (0, hooks_1.runOnToolUse)((0, hooks_1.getActiveHooks)(), { tool, state });
       }
       async handleWriteEvent(tool, state) {
         if (state.status !== "completed")
@@ -17895,6 +17971,7 @@ var require_opencode_init = __commonJS({
     var pre_run_decision_1 = require_pre_run_decision();
     var experiments_1 = require_experiments();
     var environments_12 = require_environments();
+    var hooks_1 = require_hooks();
     var runOpencodeInitFlow = async (blueprint, history, agentData) => {
       const store = job_context_storage_12.jobContextStorage.getStore();
       if (!store?.jobId || !store?.taskId) {
@@ -17919,6 +17996,7 @@ var require_opencode_init = __commonJS({
         Task_type: kind
       });
       const outputPath = (0, codegen_flow_helpers_12.getOutputPath)();
+      let caughtError;
       try {
         (0, agent_io_1.writeAgentInput)(outputPath, agentData);
         let finalBlueprint = blueprint;
@@ -17967,6 +18045,7 @@ var require_opencode_init = __commonJS({
         });
         logger_12.logger.info("[OpenCode Init] Completed task", { taskId });
       } catch (error) {
+        caughtError = error;
         const codegenError = (0, ditto_codegen_types_2.toCodegenError)(error);
         const agentOutput = (0, agent_io_1.readAgentOutput)(outputPath);
         await (0, codegen_flow_helpers_12.updateJobPayload)(localJobContext, {
@@ -17998,6 +18077,11 @@ var require_opencode_init = __commonJS({
           taskId,
           error: error instanceof Error ? error.message : String(error)
         });
+      } finally {
+        await (0, hooks_1.runOnFlowEnd)((0, hooks_1.getActiveHooks)(), {
+          failed: caughtError !== void 0,
+          error: caughtError
+        });
       }
     };
     exports2.runOpencodeInitFlow = runOpencodeInitFlow;
@@ -18073,6 +18157,7 @@ var require_opencode_iterate = __commonJS({
     var logger_12 = require_logger();
     var config_1 = require_config();
     var biEvents_1 = require_biEvents();
+    var hooks_1 = require_hooks();
     var runOpencodeIterateFlow = async (chatHistory, agentData) => {
       const store = job_context_storage_12.jobContextStorage.getStore();
       if (!store?.jobId || !store?.taskId) {
@@ -18097,6 +18182,7 @@ var require_opencode_iterate = __commonJS({
         Task_type: kind
       });
       const outputPath = (0, codegen_flow_helpers_12.getOutputPath)();
+      let caughtError;
       try {
         (0, agent_io_1.writeAgentInput)(outputPath, agentData);
         const iterationOrchestrator = new OpenCodeIterationOrchestrator_1.OpenCodeIterationOrchestrator();
@@ -18132,6 +18218,7 @@ var require_opencode_iterate = __commonJS({
         });
         logger_12.logger.info("[OpenCode Iterate] Completed task", { taskId });
       } catch (error) {
+        caughtError = error;
         const codegenError = (0, ditto_codegen_types_2.toCodegenError)(error);
         const agentOutput = (0, agent_io_1.readAgentOutput)(outputPath);
         await (0, codegen_flow_helpers_12.updateJobPayload)(localJobContext, {
@@ -18162,6 +18249,11 @@ var require_opencode_iterate = __commonJS({
           taskId,
           error: error instanceof Error ? error.message : String(error)
         });
+      } finally {
+        await (0, hooks_1.runOnFlowEnd)((0, hooks_1.getActiveHooks)(), {
+          failed: caughtError !== void 0,
+          error: caughtError
+        });
       }
     };
     exports2.runOpencodeIterateFlow = runOpencodeIterateFlow;
@@ -18648,6 +18740,27 @@ async function main(ctx) {
 var getFirstTask = (job) => {
   return job.tasks?.[0];
 };
+function buildHooksForTask(jobId, taskId, kind, payload) {
+  if (kind !== CodeGenService_1.TaskKind.INIT_CODEGEN && kind !== CodeGenService_1.TaskKind.ITERATE_CODEGEN) {
+    return void 0;
+  }
+  const userInput = extractUserInput(kind, payload);
+  return (0, environments_1.getCodegenEnvironmentConfig)().hooks.map((factory) => factory({ projectId: context_1.ctx.projectId, jobId, taskId, kind, userInput }));
+}
+function extractUserInput(kind, payload) {
+  if (kind === CodeGenService_1.TaskKind.INIT_CODEGEN) {
+    return payload.blueprint?.summary?.trim() || void 0;
+  }
+  if (kind === CodeGenService_1.TaskKind.ITERATE_CODEGEN) {
+    const history = payload.history ?? [];
+    for (let i = history.length - 1; i >= 0; i--) {
+      if (history[i].role === "USER") {
+        return history[i].text.trim() || void 0;
+      }
+    }
+  }
+  return void 0;
+}
 async function processJob(job) {
   if (!job.id) {
     throw new ditto_codegen_types_1.MissingJobIdError();
@@ -18658,14 +18771,15 @@ async function processJob(job) {
     logger_1.logger.warn("[Job] No initial task found, skipping", { jobId });
     return;
   }
-  await job_context_storage_1.jobContextStorage.run({ jobId, taskId: task.id, kind: task.kind }, async () => {
+  const payload = task.payload ?? {};
+  const hooks = buildHooksForTask(jobId, task.id, task.kind, payload);
+  await job_context_storage_1.jobContextStorage.run({ jobId, taskId: task.id, kind: task.kind, hooks }, async () => {
     logger_1.logger.info("[Job] Start processing", { taskKind: task.kind });
     context_1.ctx.setJobStatus(jobId, CodeGenService_1.Status.RUNNING);
     logger_1.logger.info("[Job] Marked job RUNNING", { initTaskId: task.id });
     jobCancellationMonitor.start(jobId);
     jobTimeoutMonitor.start(jobId);
     try {
-      const payload = task.payload ?? {};
       const history = payload.history ?? [];
       const blueprint = payload.blueprint;
       const agentData = payload.agentData;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@wix/ditto-codegen-public",
-  "version": "1.0.313",
+  "version": "1.0.314",
   "description": "AI-powered Wix CLI app generator - standalone executable",
   "scripts": {
     "build": "node build.mjs",
@@ -27,5 +27,5 @@
     "@wix/ditto-codegen": "1.0.0",
     "esbuild": "^0.27.2"
   },
-  "falconPackageHash": "e6d1a5f7a174a6739d369cde99966269aa4eca433ac218e6dd63e79f"
+  "falconPackageHash": "d9b12ec21f21078ba1ac9393765773db4870fefd00a8a85fc542cd50"
 }