@wix/ditto-codegen-public 1.0.267 → 1.0.269

package/dist/out.js

@@ -47,6 +47,7 @@ var require_types_impl = __commonJS({
       TaskKind2["ITERATE_CODEGEN"] = "ITERATE_CODEGEN";
       TaskKind2["RUN_AGENT"] = "RUN_AGENT";
       TaskKind2["FIX_CODE"] = "FIX_CODE";
+      TaskKind2["ASK_CODEGEN"] = "ASK_CODEGEN";
     })(TaskKind || (exports2.TaskKind = TaskKind = {}));
     var Status;
     (function(Status2) {
@@ -48473,24 +48474,6 @@ var require_DittoEventEmitter = __commonJS({
   }
 });
 
-// dist/opencode-integration/types.js
-var require_types5 = __commonJS({
-  "dist/opencode-integration/types.js"(exports2) {
-    "use strict";
-    Object.defineProperty(exports2, "__esModule", { value: true });
-    exports2.ASK_MODE = void 0;
-    exports2.tryParseJson = tryParseJson;
-    exports2.ASK_MODE = "ask";
-    function tryParseJson(text) {
-      try {
-        return JSON.parse(text);
-      } catch {
-        return null;
-      }
-    }
-  }
-});
-
 // dist/opencode-integration/prompts/prompt-utils.js
 var require_prompt_utils = __commonJS({
   "dist/opencode-integration/prompts/prompt-utils.js"(exports2) {
@@ -48711,6 +48694,94 @@ var require_prompts = __commonJS({
   }
 });
 
+// dist/opencode-integration/task-tracker/constants.js
+var require_constants6 = __commonJS({
+  "dist/opencode-integration/task-tracker/constants.js"(exports2) {
+    "use strict";
+    Object.defineProperty(exports2, "__esModule", { value: true });
+    exports2.BUILDER_TO_EXTENSION_LABEL = exports2.WRITE_TOOLS = exports2.CODE_SEARCH_TOOLS = exports2.WIX_MCP_SERVER_NAME = exports2.CONTAINER_ROOT_PREFIX = void 0;
+    exports2.resolveFileOperation = resolveFileOperation;
+    exports2.isMcpTool = isMcpTool;
+    exports2.normalizeFilePath = normalizeFilePath;
+    exports2.isExtensionsFile = isExtensionsFile;
+    exports2.detectExtensionType = detectExtensionType;
+    exports2.toTitleCase = toTitleCase;
+    exports2.extractExtensionName = extractExtensionName;
+    exports2.getWrittenContent = getWrittenContent;
+    var ditto_codegen_types_12 = require_dist4();
+    exports2.CONTAINER_ROOT_PREFIX = /^\/user-code\//;
+    exports2.WIX_MCP_SERVER_NAME = "wix-mcp";
+    var EXTENSION_FILE_PATTERN = /\/extensions\.ts$|\.extension\.ts$/;
+    function resolveFileOperation(tool2, metadata) {
+      if (tool2 === "write") {
+        return metadata?.exists ? ditto_codegen_types_12.ExtensionGenerationOperation.EDIT : ditto_codegen_types_12.ExtensionGenerationOperation.INSERT;
+      }
+      if (tool2 === "edit") {
+        return ditto_codegen_types_12.ExtensionGenerationOperation.EDIT;
+      }
+      return null;
+    }
+    exports2.CODE_SEARCH_TOOLS = /* @__PURE__ */ new Set(["read", "grep", "glob", "list"]);
+    exports2.WRITE_TOOLS = /* @__PURE__ */ new Set(["write", "edit"]);
+    exports2.BUILDER_TO_EXTENSION_LABEL = {
+      dashboardPage: "dashboard page",
+      dashboardModal: "dashboard modal",
+      dashboardPlugin: "dashboard plugin",
+      dashboardMenuPlugin: "dashboard menu plugin",
+      embeddedScript: "embedded script",
+      event: "backend event",
+      genericExtension: "extension",
+      customElement: "custom element",
+      siteComponent: "site component",
+      sitePlugin: "site plugin",
+      ecomShippingRates: "service plugin",
+      ecomAdditionalFees: "service plugin",
+      ecomValidations: "service plugin",
+      ecomDiscountTriggers: "service plugin",
+      ecomGiftCards: "service plugin",
+      ecomPaymentSettings: "service plugin",
+      bookingsStaffSortingProvider: "service plugin"
+    };
+    function isMcpTool(tool2) {
+      return tool2.startsWith(exports2.WIX_MCP_SERVER_NAME);
+    }
+    function normalizeFilePath(filePath) {
+      return filePath.replace(exports2.CONTAINER_ROOT_PREFIX, "");
+    }
+    function isExtensionsFile(filePath) {
+      return EXTENSION_FILE_PATTERN.test(filePath);
+    }
+    function detectExtensionType(content) {
+      if (content.includes("genericExtension(") && content.includes("compType: 'DATA_COMPONENT'")) {
+        return "collection";
+      }
+      for (const [builder, label] of Object.entries(exports2.BUILDER_TO_EXTENSION_LABEL)) {
+        if (content.includes(`extensions.${builder}(`)) {
+          return label;
+        }
+      }
+      return null;
+    }
+    function toTitleCase(str) {
+      return str.split("-").map((word) => word.charAt(0).toUpperCase() + word.slice(1)).join(" ");
+    }
+    function extractExtensionName(filePath) {
+      const parts = normalizeFilePath(filePath).split("/");
+      const folderName = parts[parts.length - 2] || "";
+      return toTitleCase(folderName);
+    }
+    function getWrittenContent(tool2, state) {
+      if (!state.input)
+        return null;
+      if (tool2 === "write" && state.input.content)
+        return state.input.content;
+      if (tool2 === "edit" && state.input.new_string)
+        return state.input.new_string;
+      return null;
+    }
+  }
+});
+
 // dist/opencode-integration/config.js
 var require_config = __commonJS({
   "dist/opencode-integration/config.js"(exports2) {
@@ -48722,6 +48793,7 @@ var require_config = __commonJS({
     exports2.getOpenCodeAskEnv = getOpenCodeAskEnv;
     var os_1 = require("os");
     var codeGenerationService_12 = require_codeGenerationService();
+    var constants_1 = require_constants6();
     exports2.DEFAULT_TIMEOUT_MS = 20 * 60 * 1e3;
     exports2.DEFAULT_MODEL = "anthropic/claude-sonnet-4-6";
     function getAnthropicBaseUrl(projectId) {
@@ -48741,12 +48813,12 @@ var require_config = __commonJS({
       };
     }
     var WIX_MCP_CONFIG = {
-      "wix-mcp": {
+      [constants_1.WIX_MCP_SERVER_NAME]: {
         type: "local",
         command: [
           "npx",
           "-y",
-          "@wix/mcp@1.0.22",
+          "@wix/mcp",
           "--experimental",
           "GET_TO_KNOW_WIX,WIX_API_THROUGH_FS"
         ],
@@ -48830,6 +48902,22 @@ var require_config = __commonJS({
   }
 });
 
+// dist/opencode-integration/types.js
+var require_types5 = __commonJS({
+  "dist/opencode-integration/types.js"(exports2) {
+    "use strict";
+    Object.defineProperty(exports2, "__esModule", { value: true });
+    exports2.tryParseJson = tryParseJson;
+    function tryParseJson(text) {
+      try {
+        return JSON.parse(text);
+      } catch {
+        return null;
+      }
+    }
+  }
+});
+
 // dist/opencode-integration/parser.js
 var require_parser = __commonJS({
   "dist/opencode-integration/parser.js"(exports2) {
@@ -48844,10 +48932,7 @@ var require_parser = __commonJS({
     exports2.formatUsageStats = formatUsageStats;
     var ditto_codegen_types_12 = require_dist4();
     var types_1 = require_types5();
-    var OPENCODE_TOOL_TO_OPERATION = {
-      write: ditto_codegen_types_12.ExtensionGenerationOperation.INSERT,
-      edit: ditto_codegen_types_12.ExtensionGenerationOperation.EDIT
-    };
+    var constants_1 = require_constants6();
     function parseEvents(output, type) {
       return output.split("\n").filter((line) => line.trim()).map((line) => (0, types_1.tryParseJson)(line)).filter((event) => event !== null && event.type === type);
     }
@@ -48873,14 +48958,11 @@ var require_parser = __commonJS({
       }
       return [];
     }
-    function normalizeFilePath(filePath) {
-      return filePath.replace(/^\/user-code\//, "");
-    }
     function parseFilesChanged(output) {
       const filesChanged = [];
       const seenPaths = /* @__PURE__ */ new Set();
       const addFileChange = (filePath, operation) => {
-        const normalizedPath = normalizeFilePath(filePath);
+        const normalizedPath = (0, constants_1.normalizeFilePath)(filePath);
         if (normalizedPath && !seenPaths.has(normalizedPath)) {
           seenPaths.add(normalizedPath);
           filesChanged.push({ path: normalizedPath, operation });
@@ -48895,13 +48977,13 @@ var require_parser = __commonJS({
           continue;
         }
         const tool2 = event.part.tool;
-        const operation = OPENCODE_TOOL_TO_OPERATION[tool2];
+        const metadata = event.part.state.metadata;
+        const operation = (0, constants_1.resolveFileOperation)(tool2, metadata);
         if (operation) {
           const filePath = event.part.state.input?.filePath;
           if (filePath) {
             addFileChange(filePath, operation);
           }
-          const metadata = event.part.state.metadata;
           if (metadata?.diagnostics) {
             for (const diagPath of Object.keys(metadata.diagnostics)) {
               if (diagPath && !diagPath.includes("node_modules")) {
@@ -48912,7 +48994,7 @@ var require_parser = __commonJS({
         }
         if (tool2 === "task" && event.part.state.metadata?.summary) {
           for (const item of event.part.state.metadata.summary) {
-            const summaryOperation = OPENCODE_TOOL_TO_OPERATION[item.tool];
+            const summaryOperation = (0, constants_1.resolveFileOperation)(item.tool);
             if (summaryOperation && item.state?.status === "completed" && item.state?.title) {
               addFileChange(item.state.title, summaryOperation);
             }
@@ -48993,143 +49075,128 @@ var require_parser = __commonJS({
   }
 });
 
-// dist/opencode-integration/task-tracker.js
-var require_task_tracker = __commonJS({
-  "dist/opencode-integration/task-tracker.js"(exports2) {
+// dist/opencode-integration/task-tracker/extension-handler.js
+var require_extension_handler = __commonJS({
+  "dist/opencode-integration/task-tracker/extension-handler.js"(exports2) {
     "use strict";
     Object.defineProperty(exports2, "__esModule", { value: true });
-    exports2.OpenCodeTaskTracker = void 0;
-    var codeGenerationService_12 = require_codeGenerationService();
-    var job_context_storage_12 = require_job_context_storage();
+    exports2.ExtensionTaskHandler = void 0;
     var ditto_codegen_types_12 = require_dist4();
-    var types_1 = require_types5();
+    var codeGenerationService_12 = require_codeGenerationService();
     var logger_12 = require_logger();
-    var CODE_SEARCH_TOOLS = /* @__PURE__ */ new Set(["read", "grep", "glob", "list"]);
-    function isMcpTool(tool2) {
-      return tool2.startsWith("wix-mcp_");
-    }
-    var OpenCodeTaskTracker = class {
-      constructor(options) {
-        this.trackedTasks = /* @__PURE__ */ new Map();
-        this.taskCounter = 0;
-        this.codeSearchTaskCreated = false;
-        this.docSearchTaskCreated = false;
-        this.skillsUsed = /* @__PURE__ */ new Set();
-        this.trackToolCategories = options?.trackToolCategories ?? false;
-      }
-      getJobContext() {
-        const store = job_context_storage_12.jobContextStorage.getStore();
-        if (!store?.jobId || !store?.taskId) {
-          return null;
-        }
-        return { jobId: store.jobId, taskId: store.taskId };
+    var constants_1 = require_constants6();
+    var ExtensionTaskHandler = class {
+      constructor() {
+        this.extensionTasksCreated = /* @__PURE__ */ new Set();
       }
-      async processLine(line) {
-        if (!line.trim())
+      /** Inspects a write/edit event; if it targets an extension file, creates a tracked sub-task. */
+      async handle(tool2, state, ctx) {
+        const filePath = state.input?.filePath || state.title || "";
+        if (!(0, constants_1.isExtensionsFile)(filePath))
           return;
-        const event = (0, types_1.tryParseJson)(line);
-        if (event) {
-          await this.handleEvent(event);
-        }
+        const content = (0, constants_1.getWrittenContent)(tool2, state);
+        if (!content)
+          return;
+        const extensionType = (0, constants_1.detectExtensionType)(content);
+        if (!extensionType)
+          return;
+        const normalizedPath = (0, constants_1.normalizeFilePath)(filePath);
+        if (this.extensionTasksCreated.has(normalizedPath))
+          return;
+        await this.completeInitTask(ctx);
+        await this.createExtensionTask(normalizedPath, extensionType, filePath, ctx);
       }
-      async handleEvent(event) {
-        if (event.type !== "tool_use")
+      /** Completes the INIT_CODEGEN task once the extension tasks are created. */
+      async completeInitTask(ctx) {
+        const { store } = ctx;
+        if (store?.isCompleted || store?.kind !== ditto_codegen_types_12.TaskKind.INIT_CODEGEN)
           return;
-        const { tool: tool2, callID, state } = event.part || {};
-        if (!tool2 || !callID || !state)
+        if (!store.jobId || !store.taskId)
           return;
-        if (tool2 === "task") {
-          await this.handleTaskEvent(callID, state);
-        }
-        if (tool2 === "skill" && state.status === "completed") {
-          await this.handleSkillEvent(state);
-        }
-        if (this.trackToolCategories) {
-          await this.handleToolCategoryEvent(tool2);
+        store.isCompleted = true;
+        try {
+          await codeGenerationService_12.codeGenerationService.updateTask(store.jobId, store.taskId, ditto_codegen_types_12.Status.COMPLETED, {});
+          logger_12.logger.info("[OpenCode] Completed init task before extension tasks");
+        } catch (error48) {
+          logger_12.logger.error("[OpenCode] Failed to complete init task", {
+            error: error48 instanceof Error ? error48.message : String(error48)
+          });
         }
       }
-      async handleTaskEvent(callID, state) {
-        const jobContext = this.getJobContext();
+      /** Registers a new RUNNING sub-task in the job service for a detected extension. */
+      async createExtensionTask(normalizedPath, extensionType, originalPath, ctx) {
+        const jobContext = ctx.getJobContext();
         if (!jobContext)
           return;
-        const description = state.input?.description || state.title || "Running sub-agent";
-        if (!this.trackedTasks.has(callID)) {
-          this.taskCounter++;
-          const taskId = `${jobContext.taskId}-opencode-task-${this.taskCounter}`;
-          const initialStatus = state.status === "completed" ? ditto_codegen_types_12.Status.COMPLETED : state.status === "error" ? ditto_codegen_types_12.Status.FAILED : ditto_codegen_types_12.Status.RUNNING;
-          this.trackedTasks.set(callID, {
-            taskId,
+        const extensionName = (0, constants_1.extractExtensionName)(originalPath);
+        const description = `Creating ${extensionName} ${extensionType}`;
+        const extensionKey = `ext-${normalizedPath}`;
+        this.extensionTasksCreated.add(normalizedPath);
+        ctx.taskCounter++;
+        const taskId = `${jobContext.taskId}-ext-${ctx.taskCounter}`;
+        const extensionFolder = normalizedPath.substring(0, normalizedPath.lastIndexOf("/") + 1);
+        ctx.trackedTasks.set(extensionKey, {
+          taskId,
+          description,
+          status: ditto_codegen_types_12.Status.RUNNING,
+          extensionFolder
+        });
+        try {
+          await codeGenerationService_12.codeGenerationService.addTask(jobContext.jobId, {
+            id: taskId,
+            kind: ditto_codegen_types_12.TaskKind.RUN_AGENT,
+            status: ditto_codegen_types_12.Status.RUNNING,
+            name: "create_extension",
             description,
-            status: initialStatus
+            agentName: ""
+          });
+          logger_12.logger.info("[OpenCode] Created extension task", { description });
+        } catch (error48) {
+          logger_12.logger.error("[OpenCode] Failed to create extension task", {
+            error: error48 instanceof Error ? error48.message : String(error48)
           });
-          try {
-            await codeGenerationService_12.codeGenerationService.addTask(jobContext.jobId, {
-              id: taskId,
-              kind: ditto_codegen_types_12.TaskKind.RUN_AGENT,
-              status: initialStatus,
-              name: "run_agent",
-              description: `Run agent for ${description}`,
-              agentName: "OpenCodeSubAgent"
-            });
-            logger_12.logger.info("[OpenCode] Created sub-task", {
-              description,
-              status: state.status
-            });
-          } catch (error48) {
-            logger_12.logger.error("[OpenCode] Failed to create sub-task", {
-              error: error48 instanceof Error ? error48.message : String(error48)
-            });
-          }
-          return;
-        }
-        const tracked = this.trackedTasks.get(callID);
-        if (state.status === "completed" && tracked.status !== ditto_codegen_types_12.Status.COMPLETED) {
-          tracked.status = ditto_codegen_types_12.Status.COMPLETED;
-          try {
-            await codeGenerationService_12.codeGenerationService.updateTask(jobContext.jobId, tracked.taskId, ditto_codegen_types_12.Status.COMPLETED, {});
-            logger_12.logger.info("[OpenCode] Completed sub-task", {
-              description: tracked.description
-            });
-          } catch (error48) {
-            logger_12.logger.error("[OpenCode] Failed to update sub-task", {
-              error: error48 instanceof Error ? error48.message : String(error48)
-            });
-          }
-        } else if (state.status === "error" && tracked.status !== ditto_codegen_types_12.Status.FAILED) {
-          tracked.status = ditto_codegen_types_12.Status.FAILED;
-          try {
-            await codeGenerationService_12.codeGenerationService.updateTask(jobContext.jobId, tracked.taskId, ditto_codegen_types_12.Status.FAILED, { error: state.output || "Sub-agent failed" });
-            logger_12.logger.error("[OpenCode] Failed sub-task", {
-              description: tracked.description
-            });
-          } catch (error48) {
-            logger_12.logger.error("[OpenCode] Failed to update sub-task", {
-              error: error48 instanceof Error ? error48.message : String(error48)
-            });
-          }
         }
       }
-      async handleSkillEvent(state) {
-        const skillName = state.title?.replace("Loaded skill: ", "") || "unknown";
-        this.skillsUsed.add(skillName);
-        logger_12.logger.info("[OpenCode] Loaded skill", { skillName });
+    };
+    exports2.ExtensionTaskHandler = ExtensionTaskHandler;
+  }
+});
+
+// dist/opencode-integration/task-tracker/category-handler.js
+var require_category_handler = __commonJS({
+  "dist/opencode-integration/task-tracker/category-handler.js"(exports2) {
+    "use strict";
+    Object.defineProperty(exports2, "__esModule", { value: true });
+    exports2.CategoryTaskHandler = void 0;
+    var ditto_codegen_types_12 = require_dist4();
+    var codeGenerationService_12 = require_codeGenerationService();
+    var logger_12 = require_logger();
+    var constants_1 = require_constants6();
+    var CategoryTaskHandler = class {
+      constructor() {
+        this.codeSearchCreated = false;
+        this.docSearchCreated = false;
       }
-      async handleToolCategoryEvent(tool2) {
-        const jobContext = this.getJobContext();
+      /** Creates a search sub-task on the first use of code-search or MCP tools. */
+      async handle(tool2, ctx) {
+        const jobContext = ctx.getJobContext();
         if (!jobContext)
           return;
-        if (CODE_SEARCH_TOOLS.has(tool2) && !this.codeSearchTaskCreated) {
-          this.codeSearchTaskCreated = true;
-          await this.createCategoryTask(jobContext, "code-search", "Searching in code");
+        if (constants_1.CODE_SEARCH_TOOLS.has(tool2) && !this.codeSearchCreated) {
+          this.codeSearchCreated = true;
+          await this.createTask("code-search", "Searching in code", ctx);
         }
-        if (isMcpTool(tool2) && !this.docSearchTaskCreated) {
-          this.docSearchTaskCreated = true;
-          await this.createCategoryTask(jobContext, "doc-search", "Searching in docs");
+        if ((0, constants_1.isMcpTool)(tool2) && !this.docSearchCreated) {
+          this.docSearchCreated = true;
+          await this.createTask("doc-search", "Searching in docs", ctx);
         }
       }
-      async createCategoryTask(jobContext, suffix, description) {
+      async createTask(suffix, description, ctx) {
+        const jobContext = ctx.getJobContext();
+        if (!jobContext)
+          return;
         const taskId = `${jobContext.taskId}-${suffix}`;
-        this.trackedTasks.set(suffix, {
+        ctx.trackedTasks.set(suffix, {
           taskId,
           description,
           status: ditto_codegen_types_12.Status.RUNNING
@@ -49150,29 +49217,137 @@ var require_task_tracker = __commonJS({
           });
         }
       }
+    };
+    exports2.CategoryTaskHandler = CategoryTaskHandler;
+  }
+});
+
+// dist/opencode-integration/task-tracker/task-tracker.js
+var require_task_tracker = __commonJS({
+  "dist/opencode-integration/task-tracker/task-tracker.js"(exports2) {
+    "use strict";
+    Object.defineProperty(exports2, "__esModule", { value: true });
+    exports2.OpenCodeTaskTracker = void 0;
+    var ditto_codegen_types_12 = require_dist4();
+    var codeGenerationService_12 = require_codeGenerationService();
+    var job_context_storage_12 = require_job_context_storage();
+    var types_1 = require_types5();
+    var logger_12 = require_logger();
+    var constants_1 = require_constants6();
+    var extension_handler_1 = require_extension_handler();
+    var category_handler_1 = require_category_handler();
+    var OpenCodeTaskTracker = class {
+      constructor() {
+        this.trackedTasks = /* @__PURE__ */ new Map();
+        this.taskCounter = 0;
+        this.collectedFiles = [];
+        this.skillsUsed = /* @__PURE__ */ new Set();
+        this.extensionHandler = new extension_handler_1.ExtensionTaskHandler();
+        this.categoryHandler = new category_handler_1.CategoryTaskHandler();
+        this.store = job_context_storage_12.jobContextStorage.getStore();
+      }
       getSkillsUsed() {
         return Array.from(this.skillsUsed);
       }
+      getJobContext() {
+        if (!this.store?.jobId || !this.store?.taskId) {
+          return null;
+        }
+        return { jobId: this.store.jobId, taskId: this.store.taskId };
+      }
+      get handlerContext() {
+        return {
+          getJobContext: () => this.getJobContext(),
+          store: this.store,
+          trackedTasks: this.trackedTasks,
+          collectedFiles: this.collectedFiles,
+          taskCounter: this.taskCounter
+        };
+      }
+      async processLine(line) {
+        if (!line.trim())
+          return;
+        const event = (0, types_1.tryParseJson)(line);
+        if (event) {
+          await this.handleEvent(event);
+        }
+      }
+      async handleEvent(event) {
+        if (event.type !== "tool_use")
+          return;
+        const { tool: tool2, callID, state } = event.part || {};
+        if (!tool2 || !callID || !state)
+          return;
+        if (constants_1.WRITE_TOOLS.has(tool2)) {
+          this.collectFileChange(tool2, state);
+          const ctx = this.handlerContext;
+          await this.extensionHandler.handle(tool2, state, ctx);
+          this.taskCounter = ctx.taskCounter;
+        }
+        if (tool2 === "skill" && state.status === "completed") {
+          this.handleSkillEvent(state);
+        }
+        if (this.store?.kind === ditto_codegen_types_12.TaskKind.ASK_CODEGEN) {
+          const ctx = this.handlerContext;
+          await this.categoryHandler.handle(tool2, ctx);
+          this.taskCounter = ctx.taskCounter;
+        }
+      }
+      /** Records a unique file change (INSERT/EDIT) from a completed write/edit tool call. */
+      collectFileChange(tool2, state) {
+        if (state.status !== "completed")
+          return;
+        const filePath = state.input?.filePath || state.title;
+        const operation = (0, constants_1.resolveFileOperation)(tool2, state.metadata);
+        if (!filePath || !operation)
+          return;
+        const normalized = (0, constants_1.normalizeFilePath)(filePath);
+        if (!this.collectedFiles.some((f) => f.path === normalized)) {
+          this.collectedFiles.push({ path: normalized, operation });
+        }
+      }
+      async handleSkillEvent(state) {
+        const skillName = state.title?.replace("Loaded skill: ", "") || "unknown";
+        this.skillsUsed.add(skillName);
+        logger_12.logger.info("[OpenCode] Loaded skill", { skillName });
+      }
+      /** Builds the task completion payload — includes file list for extension tasks. */
+      buildTaskPayload(tracked) {
+        if (tracked.extensionFolder) {
+          const extensionFiles = this.collectedFiles.filter((f) => f.path.startsWith(tracked.extensionFolder));
+          return {
+            taskOutput: {
+              files: extensionFiles.map((f) => ({
+                path: f.path,
+                operation: f.operation
+              }))
+            }
+          };
+        }
+        return {};
+      }
+      /** Marks all still-running tracked tasks as COMPLETED (called on successful process exit). */
       async completeRemainingTasks() {
         const jobContext = this.getJobContext();
         if (!jobContext)
           return;
         for (const [, tracked] of this.trackedTasks) {
           if (tracked.status === ditto_codegen_types_12.Status.RUNNING) {
+            const payload = this.buildTaskPayload(tracked);
             try {
-              await codeGenerationService_12.codeGenerationService.updateTask(jobContext.jobId, tracked.taskId, ditto_codegen_types_12.Status.COMPLETED, {});
-              tracked.status = ditto_codegen_types_12.Status.COMPLETED;
-              logger_12.logger.info("[OpenCode] Auto-completed sub-task", {
+              await codeGenerationService_12.codeGenerationService.updateTask(jobContext.jobId, tracked.taskId, ditto_codegen_types_12.Status.COMPLETED, payload);
+              logger_12.logger.info("[OpenCode] Completed task", {
                 description: tracked.description
               });
             } catch (err) {
-              logger_12.logger.error("[OpenCode] Failed to auto-complete sub-task", {
+              logger_12.logger.error("[OpenCode] Failed to complete task", {
                 error: err instanceof Error ? err.message : String(err)
               });
             }
           }
         }
       }
+      /** Marks all still-running tracked tasks as FAILED (called on process error/timeout). */
       async failRemainingTasks(errorMessage) {
         const jobContext = this.getJobContext();
         if (!jobContext)
@@ -49181,12 +49356,11 @@ var require_task_tracker = __commonJS({
           if (tracked.status === ditto_codegen_types_12.Status.RUNNING) {
             try {
               await codeGenerationService_12.codeGenerationService.updateTask(jobContext.jobId, tracked.taskId, ditto_codegen_types_12.Status.FAILED, { error: errorMessage });
-              tracked.status = ditto_codegen_types_12.Status.FAILED;
               logger_12.logger.info("[OpenCode] Failed sub-task", {
                 description: tracked.description
               });
             } catch (err) {
-              logger_12.logger.error("[OpenCode] Failed to fail sub-task", {
+              logger_12.logger.error("[OpenCode] Failed to mark sub-task as failed", {
                 error: err instanceof Error ? err.message : String(err)
               });
             }
@@ -49198,6 +49372,19 @@ var require_task_tracker = __commonJS({
   }
 });
 
+// dist/opencode-integration/task-tracker/index.js
+var require_task_tracker2 = __commonJS({
+  "dist/opencode-integration/task-tracker/index.js"(exports2) {
+    "use strict";
+    Object.defineProperty(exports2, "__esModule", { value: true });
+    exports2.OpenCodeTaskTracker = void 0;
+    var task_tracker_1 = require_task_tracker();
+    Object.defineProperty(exports2, "OpenCodeTaskTracker", { enumerable: true, get: function() {
+      return task_tracker_1.OpenCodeTaskTracker;
+    } });
+  }
+});
+
 // dist/opencode-integration/process-manager.js
 var require_process_manager = __commonJS({
   "dist/opencode-integration/process-manager.js"(exports2) {
@@ -49406,11 +49593,11 @@ var require_executor = __commonJS({
     exports2.executeOpenCode = executeOpenCode;
     var child_process_1 = require("child_process");
     var ditto_codegen_types_12 = require_dist4();
-    var types_1 = require_types5();
     var config_1 = require_config();
+    var job_context_storage_12 = require_job_context_storage();
     var parser_1 = require_parser();
     var prompts_1 = require_prompts();
-    var task_tracker_1 = require_task_tracker();
+    var task_tracker_1 = require_task_tracker2();
     var process_manager_1 = require_process_manager();
     var process_handlers_1 = require_process_handlers();
     var logger_12 = require_logger();
@@ -49465,11 +49652,9 @@ var require_executor = __commonJS({
       return ["run", "--format", "json", prompt];
     }
     async function executeOpenCodeOnce(options, attempt) {
-      const { prompt, outputPath, projectId, mode, onStdout, onStderr } = options;
+      const { prompt, outputPath, projectId, onStdout, onStderr } = options;
       const startTime = Date.now();
-      const taskTracker = new task_tracker_1.OpenCodeTaskTracker({
-        trackToolCategories: mode === types_1.ASK_MODE
-      });
+      const taskTracker = new task_tracker_1.OpenCodeTaskTracker();
       return new Promise((resolve3) => {
         const state = {
           stdout: "",
@@ -49510,7 +49695,8 @@ var require_executor = __commonJS({
             maxRetries: MAX_RETRIES,
             workingDirectory: outputPath
           });
-          const env = mode === types_1.ASK_MODE ? (0, config_1.getOpenCodeAskEnv)(projectId) : (0, config_1.getOpenCodeEnv)(projectId);
+          const isAsk = job_context_storage_12.jobContextStorage.getStore()?.kind === ditto_codegen_types_12.TaskKind.ASK_CODEGEN;
+          const env = isAsk ? (0, config_1.getOpenCodeAskEnv)(projectId) : (0, config_1.getOpenCodeEnv)(projectId);
           ctx.child = (0, child_process_1.spawn)("opencode", args, {
             cwd: outputPath,
             env,
@@ -49553,7 +49739,6 @@ var require_opencode_runner = __commonJS({
     exports2.runOpenCodeInit = runOpenCodeInit;
     exports2.runOpenCodeIteration = runOpenCodeIteration;
     exports2.runOpenCodeAsk = runOpenCodeAsk;
-    var types_1 = require_types5();
     var prompts_1 = require_prompts();
     var executor_1 = require_executor();
     var parser_1 = require_parser();
@@ -49565,7 +49750,13 @@ var require_opencode_runner = __commonJS({
     async function runOpenCodeIteration(options) {
       const { outputPath, projectId, chatHistory, onStdout, onStderr } = options;
       const prompt = (0, prompts_1.buildIterationPrompt)(chatHistory);
-      return (0, executor_1.executeOpenCode)({ prompt, outputPath, projectId, onStdout, onStderr });
+      return (0, executor_1.executeOpenCode)({
+        prompt,
+        outputPath,
+        projectId,
+        onStdout,
+        onStderr
+      });
     }
     async function runOpenCodeAsk(options) {
       const { outputPath, projectId, chatHistory, onStdout, onStderr } = options;
@@ -49574,7 +49765,6 @@ var require_opencode_runner = __commonJS({
         prompt,
         outputPath,
         projectId,
-        mode: types_1.ASK_MODE,
         onStdout,
         onStderr
       });
@@ -49604,7 +49794,7 @@ var require_opencode_integration = __commonJS({
     Object.defineProperty(exports2, "runOpenCodeAsk", { enumerable: true, get: function() {
       return opencode_runner_1.runOpenCodeAsk;
     } });
-    var task_tracker_1 = require_task_tracker();
+    var task_tracker_1 = require_task_tracker2();
     Object.defineProperty(exports2, "OpenCodeTaskTracker", { enumerable: true, get: function() {
       return task_tracker_1.OpenCodeTaskTracker;
     } });
@@ -53274,7 +53464,7 @@ var require_expand = __commonJS({
 });
 
 // ../../node_modules/braces/lib/constants.js
-var require_constants6 = __commonJS({
+var require_constants7 = __commonJS({
   "../../node_modules/braces/lib/constants.js"(exports2, module2) {
     "use strict";
     module2.exports = {
@@ -53407,7 +53597,7 @@ var require_parse3 = __commonJS({
       /* ' */
       CHAR_NO_BREAK_SPACE,
       CHAR_ZERO_WIDTH_NOBREAK_SPACE
-    } = require_constants6();
+    } = require_constants7();
     var parse4 = (input, options = {}) => {
       if (typeof input !== "string") {
         throw new TypeError("Expected a string");
@@ -53676,7 +53866,7 @@ var require_braces = __commonJS({
 });
 
 // ../../node_modules/micromatch/node_modules/picomatch/lib/constants.js
-var require_constants7 = __commonJS({
+var require_constants8 = __commonJS({
   "../../node_modules/micromatch/node_modules/picomatch/lib/constants.js"(exports2, module2) {
     "use strict";
     var path4 = require("path");
@@ -53883,7 +54073,7 @@ var require_utils13 = __commonJS({
       REGEX_REMOVE_BACKSLASH,
       REGEX_SPECIAL_CHARS,
       REGEX_SPECIAL_CHARS_GLOBAL
-    } = require_constants7();
+    } = require_constants8();
     exports2.isObject = (val) => val !== null && typeof val === "object" && !Array.isArray(val);
     exports2.hasRegexChars = (str) => REGEX_SPECIAL_CHARS.test(str);
     exports2.isRegexChar = (str) => str.length === 1 && exports2.hasRegexChars(str);
@@ -53969,7 +54159,7 @@ var require_scan = __commonJS({
       /* ) */
       CHAR_RIGHT_SQUARE_BRACKET
       /* ] */
-    } = require_constants7();
+    } = require_constants8();
     var isPathSeparator = (code) => {
       return code === CHAR_FORWARD_SLASH || code === CHAR_BACKWARD_SLASH;
     };
@@ -54267,7 +54457,7 @@ var require_scan = __commonJS({
 var require_parse4 = __commonJS({
   "../../node_modules/micromatch/node_modules/picomatch/lib/parse.js"(exports2, module2) {
     "use strict";
-    var constants = require_constants7();
+    var constants = require_constants8();
     var utils = require_utils13();
     var {
       MAX_LENGTH,
@@ -55044,7 +55234,7 @@ var require_picomatch = __commonJS({
     var scan = require_scan();
     var parse4 = require_parse4();
     var utils = require_utils13();
-    var constants = require_constants7();
+    var constants = require_constants8();
     var isObject2 = (val) => val && typeof val === "object" && !Array.isArray(val);
     var picomatch = (glob, options, returnState = false) => {
       if (Array.isArray(glob)) {
@@ -55997,7 +56187,7 @@ var require_run_parallel = __commonJS({
 });
 
 // ../../node_modules/@nodelib/fs.scandir/out/constants.js
-var require_constants8 = __commonJS({
+var require_constants9 = __commonJS({
   "../../node_modules/@nodelib/fs.scandir/out/constants.js"(exports2) {
     "use strict";
     Object.defineProperty(exports2, "__esModule", { value: true });
@@ -56076,7 +56266,7 @@ var require_async3 = __commonJS({
     exports2.readdir = exports2.readdirWithFileTypes = exports2.read = void 0;
     var fsStat = require_out();
     var rpl = require_run_parallel();
-    var constants_1 = require_constants8();
+    var constants_1 = require_constants9();
     var utils = require_utils15();
     var common = require_common2();
     function read(directory, settings, callback) {
@@ -56185,7 +56375,7 @@ var require_sync2 = __commonJS({
     Object.defineProperty(exports2, "__esModule", { value: true });
     exports2.readdir = exports2.readdirWithFileTypes = exports2.read = void 0;
     var fsStat = require_out();
-    var constants_1 = require_constants8();
+    var constants_1 = require_constants9();
     var utils = require_utils15();
     var common = require_common2();
     function read(directory, settings) {
@@ -77432,7 +77622,7 @@ var require_tools2 = __commonJS({
 });
 
 // ../scaffolding/dist/constants.js
-var require_constants9 = __commonJS({
+var require_constants10 = __commonJS({
   "../scaffolding/dist/constants.js"(exports2) {
     "use strict";
     Object.defineProperty(exports2, "__esModule", { value: true });
@@ -77462,7 +77652,7 @@ var require_scaffolding = __commonJS({
     exports2.copyScaffoldingTemplate = copyScaffoldingTemplate;
     exports2.toKebabCase = toKebabCase;
     var tools_1 = require_tools2();
-    var constants_1 = require_constants9();
+    var constants_1 = require_constants10();
     var types_1 = require_types_impl2();
     var fs_1 = __importDefault2(require("fs"));
     var path_1 = __importDefault2(require("path"));
@@ -77584,7 +77774,7 @@ var require_dist13 = __commonJS({
     Object.defineProperty(exports2, "copyRecursive", { enumerable: true, get: function() {
       return tools_1.copyRecursive;
     } });
-    var constants_1 = require_constants9();
+    var constants_1 = require_constants10();
     Object.defineProperty(exports2, "spiToSubPath", { enumerable: true, get: function() {
       return constants_1.spiToSubPath;
     } });
@@ -91908,7 +92098,7 @@ var require_WixMCPClient = __commonJS({
           command: "npx",
           args: [
             "-y",
-            "@wix/mcp@1.0.22",
+            "@wix/mcp",
             "--experimental",
             "GET_TO_KNOW_WIX,WIX_API_THROUGH_FS"
           ]
@@ -96872,22 +97062,21 @@ var require_opencode_init = __commonJS({
           requiredPermissions: result.requiredPermissions,
           history
         });
-        const taskOutput = {
-          files: result.filesChanged.map((file2) => ({
-            path: file2.path,
-            operation: file2.operation
-          }))
-        };
-        await codeGenerationService_12.codeGenerationService.updateTask(jobId, taskId, ditto_codegen_types_12.Status.COMPLETED, {
-          taskOutput
-        });
+        if (!store.isCompleted) {
+          await codeGenerationService_12.codeGenerationService.updateTask(jobId, taskId, ditto_codegen_types_12.Status.COMPLETED, {});
+        }
         (0, biEvents_1.reportSessionEnd)({
           ...biBaseParams,
           Task_model: config_1.DEFAULT_MODEL,
           Task_type: kind,
           Task_cost: Math.round(result.usage.cost * 1e6),
           Task_status: ditto_codegen_types_12.Status.COMPLETED,
-          Agent_output: JSON.stringify(taskOutput),
+          Agent_output: JSON.stringify({
+            files: result.filesChanged.map((file2) => ({
+              path: file2.path,
+              operation: file2.operation
+            }))
+          }),
           Job_skills_used: result.skillsUsed.join(",")
         });
         jobLog.info(`[OpenCode Init] Completed opencode init task: jobId=${jobId}, taskId=${taskId}`);
@@ -345041,12 +345230,11 @@ async function processJob(job) {
       const payload = task.payload ?? {};
       const history = payload.history ?? [];
       const blueprint = payload.blueprint;
-      const taskKind = task.kind;
-      if (taskKind === "ASK_CODEGEN") {
+      if (task.kind === CodeGenService_1.TaskKind.ASK_CODEGEN) {
         await (0, opencode_ask_1.runOpencodeAskFlow)(history);
-      } else if (task.kind === "ITERATE_CODEGEN") {
+      } else if (task.kind === CodeGenService_1.TaskKind.ITERATE_CODEGEN) {
         await handleIteration(history);
-      } else if (task.kind === "INIT_CODEGEN" && blueprint) {
+      } else if (task.kind === CodeGenService_1.TaskKind.INIT_CODEGEN && blueprint) {
         await handleInit(blueprint, history);
       }
     });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@wix/ditto-codegen-public",
-  "version": "1.0.267",
+  "version": "1.0.269",
   "description": "AI-powered Wix CLI app generator - standalone executable",
   "scripts": {
     "build": "node build.mjs",
@@ -28,5 +28,5 @@
     "@wix/ditto-codegen": "1.0.0",
     "esbuild": "^0.27.2"
   },
-  "falconPackageHash": "b000bcde46855c5aba4f16f23ef55c340f744b534a5f75a38303ae1b"
+  "falconPackageHash": "ad1e291122bd10f8eddcaef9d712accb3c602ed227afff92970f532e"
 }