@wix/ditto-codegen-public 1.0.106 → 1.0.108

package/dist/out.js

@@ -76977,6 +76977,7 @@ var require_enums2 = __commonJS({
       ExtensionGenerationOperation2["INSERT"] = "insert";
       ExtensionGenerationOperation2["UPDATE"] = "update";
       ExtensionGenerationOperation2["DELETE"] = "delete";
+      ExtensionGenerationOperation2["EDIT"] = "edit";
     })(ExtensionGenerationOperation || (exports2.ExtensionGenerationOperation = ExtensionGenerationOperation = {}));
     var ErrorOrigin;
     (function(ErrorOrigin2) {

package/dist/runCodegenFlow.js

@@ -91,6 +91,7 @@ var require_enums = __commonJS({
       ExtensionGenerationOperation7["INSERT"] = "insert";
       ExtensionGenerationOperation7["UPDATE"] = "update";
       ExtensionGenerationOperation7["DELETE"] = "delete";
+      ExtensionGenerationOperation7["EDIT"] = "edit";
     })(ExtensionGenerationOperation6 || (exports2.ExtensionGenerationOperation = ExtensionGenerationOperation6 = {}));
     var ErrorOrigin4;
     (function(ErrorOrigin5) {
@@ -292,7 +293,7 @@ var require_FileSystemError = __commonJS({
     exports2.FileSystemError = void 0;
     var BaseCodegenError_1 = require_BaseCodegenError();
     var ErrorTypes_1 = require_ErrorTypes();
-    var FileSystemError2 = class extends BaseCodegenError_1.BaseCodegenError {
+    var FileSystemError3 = class extends BaseCodegenError_1.BaseCodegenError {
       constructor(message, options) {
         super(message);
         this.name = "FileSystemError";
@@ -314,7 +315,7 @@ var require_FileSystemError = __commonJS({
         return props;
       }
     };
-    exports2.FileSystemError = FileSystemError2;
+    exports2.FileSystemError = FileSystemError3;
   }
 });
 
@@ -59148,15 +59149,37 @@ ${hasEmbeddedScriptExtensions ? embeddedScriptDynamicParametersPrompt() : 'No em
 var import_fs = __toESM(require("fs"));
 var import_path2 = __toESM(require("path"));
 var import_ditto_codegen_types3 = __toESM(require_dist());
-var FileItemSchema = external_exports.object({
-  operation: external_exports.enum(import_ditto_codegen_types3.ExtensionGenerationOperation).describe(
-    "File operation: insert (new file), update (modify existing), delete (remove file)"
-  ).default(import_ditto_codegen_types3.ExtensionGenerationOperation.INSERT),
-  path: external_exports.string().describe("Relative file path from project root").optional(),
-  content: external_exports.string().describe(
-    "Complete file content as a string (for JSON files, stringify the object). Required for insert and update operations."
-  ).optional()
-});
+var FileItemSchema = external_exports.discriminatedUnion("operation", [
+  external_exports.object({
+    operation: external_exports.union([
+      external_exports.literal(import_ditto_codegen_types3.ExtensionGenerationOperation.INSERT),
+      external_exports.literal(import_ditto_codegen_types3.ExtensionGenerationOperation.UPDATE),
+      external_exports.literal(import_ditto_codegen_types3.ExtensionGenerationOperation.DELETE)
+    ]).describe(
+      "File operation: insert (new file), update (modify existing), delete (remove file)"
+    ).default(import_ditto_codegen_types3.ExtensionGenerationOperation.INSERT),
+    path: external_exports.string().describe("Relative file path from project root").optional(),
+    content: external_exports.string().describe(
+      "Complete file content as a string (for JSON files, stringify the object). Required for insert and update operations."
+    ).optional()
+  }),
+  external_exports.object({
+    operation: external_exports.literal(import_ditto_codegen_types3.ExtensionGenerationOperation.EDIT),
+    path: external_exports.string().describe("Relative file path from project root"),
+    replacements: external_exports.array(
+      external_exports.object({
+        oldString: external_exports.string().describe(
+          "The text to replace (must match the file contents exactly, including all whitespace and indentation)"
+        ),
+        newString: external_exports.string().describe("The edited text to replace the oldString"),
+        replaceAll: external_exports.boolean().optional().describe("Replace all occurrences of oldString (default false)")
+      })
+    ).min(1).describe("List of string replacements to apply")
+  })
+]).describe(`Representation of an operation that modifies a file on the local file system.
+Use the 'insert' operation to create a new file.
+Use the 'delete' operation to remove a file.
+To modify a file, you can either use the 'edit' operation to replace a specific strings with new strings (THIS IS THE PREFERRED WAY TO MODIFY A FILE), or use the 'update' operation to replace the entire file with new content.`);
 var FilesSchema = external_exports.object({
   files: external_exports.array(FileItemSchema).default([]).describe("An array of files")
 });
@@ -63372,12 +63395,12 @@ function attachOrchestratorListeners(eventEmitter) {
     timer.delete("PlannerAgent");
     console.log(`\u{1F916}\u2714\uFE0F  Generated plan in ${duration3 / 1e3}s`);
   });
-  eventEmitter.onEvent("iterationAgent:start", () => {
+  eventEmitter.onEvent("iterationPlanAgent:start", () => {
     timer.set("IterationAgent", Date.now());
     console.log("\u{1F916} Generating iteration plan...");
   });
   eventEmitter.onEvent(
-    "iterationAgent:done",
+    "iterationPlanAgent:done",
     ({ newExtensions, currentExtensions }) => {
       const duration3 = Date.now() - timer.get("IterationAgent");
       timer.delete("IterationAgent");
@@ -63854,10 +63877,6 @@ var DittoEventEmitter = class extends import_events.EventEmitter {
 // src/extensions/extensionProcessors.ts
 var import_types11 = __toESM(require_types_impl());
 
-// src/extensions/extensionHelpers.ts
-var import_path9 = __toESM(require("path"));
-var import_fs6 = __toESM(require("fs"));
-
 // src/extensions/extensionGenerators.ts
 var import_path7 = __toESM(require("path"));
 var import_fs_extra = __toESM(require_lib());
@@ -64266,9 +64285,137 @@ function collectAllCreatedFiles(options) {
   return uniqueFiles;
 }
 
-// src/extensions/extensionHelpers.ts
-var import_ditto_scaffolding4 = __toESM(require_dist5());
+// src/extensions/write-file.ts
 var import_ditto_codegen_types16 = __toESM(require_dist());
+var import_fs6 = __toESM(require("fs"));
+var import_path9 = __toESM(require("path"));
+
+// src/truncate.ts
+function truncate(str, maxLength = 50) {
+  if (str.length <= maxLength) {
+    return str;
+  }
+  return str.substring(0, maxLength) + "...";
+}
+
+// src/extensions/write-file.ts
+var handleInsert = (file2, fullPath) => {
+  if (!("content" in file2) || !file2.content) {
+    console.warn(
+      `\u26A0\uFE0F Skipping insert operation for ${file2.path}: no content provided`
+    );
+    return;
+  }
+  const dir = import_path9.default.dirname(fullPath);
+  if (!import_fs6.default.existsSync(dir)) {
+    import_fs6.default.mkdirSync(dir, { recursive: true });
+  }
+  writeFileInFileSystem(fullPath, file2.content.trim());
+  console.log(`\u{1F4DD} Inserted: ${file2.path}`);
+};
+var handleUpdate = (file2, fullPath) => {
+  if (!("content" in file2) || !file2.content) {
+    console.warn(
+      `\u26A0\uFE0F Skipping update operation for ${file2.path}: no content provided`
+    );
+    return;
+  }
+  if (!import_fs6.default.existsSync(fullPath)) {
+    console.warn(
+      `\u26A0\uFE0F Skipping update operation for ${file2.path}: file does not exist`
+    );
+    return;
+  }
+  writeFileInFileSystem(fullPath, file2.content.trim());
+  console.log(`\u{1F4DD} Updated: ${file2.path}`);
+};
+var handleDelete = (file2, fullPath) => {
+  if (import_fs6.default.existsSync(fullPath)) {
+    try {
+      import_fs6.default.unlinkSync(fullPath);
+      console.log(`\u{1F5D1}\uFE0F Deleted: ${file2.path}`);
+    } catch (error46) {
+      throw new import_ditto_codegen_types16.FileSystemError(`Failed to delete file: ${file2.path}`, {
+        filePath: fullPath,
+        operation: import_ditto_codegen_types16.FileSystemOperation.DELETE,
+        cause: error46
+      });
+    }
+  } else {
+    console.warn(
+      `\u26A0\uFE0F Skipping delete operation for ${file2.path}: file does not exist`
+    );
+  }
+};
+var handleEdit = (file2, fullPath) => {
+  if (!("replacements" in file2) || !file2.replacements || file2.replacements.length === 0) {
+    console.warn(
+      `\u26A0\uFE0F Skipping edit operation for ${file2.path}: no replacements provided`
+    );
+    return;
+  }
+  if (!import_fs6.default.existsSync(fullPath)) {
+    console.warn(
+      `\u26A0\uFE0F Skipping edit operation for ${file2.path}: file does not exist`
+    );
+    return;
+  }
+  let content = import_fs6.default.readFileSync(fullPath, "utf8");
+  for (const replacement of file2.replacements) {
+    if (!content.includes(replacement.oldString)) {
+      console.warn(
+        `\u26A0\uFE0F Replacement pattern not found in ${file2.path}: "${truncate(replacement.oldString)}"`
+      );
+      continue;
+    }
+    content = replacement.replaceAll ? content.replaceAll(replacement.oldString, replacement.newString) : content.replace(replacement.oldString, replacement.newString);
+  }
+  writeFileInFileSystem(fullPath, content);
+  console.log(`\u270F\uFE0F Edited: ${file2.path}`);
+};
+var operationHandlers = {
+  [import_ditto_codegen_types16.ExtensionGenerationOperation.INSERT]: handleInsert,
+  [import_ditto_codegen_types16.ExtensionGenerationOperation.UPDATE]: handleUpdate,
+  [import_ditto_codegen_types16.ExtensionGenerationOperation.DELETE]: handleDelete,
+  [import_ditto_codegen_types16.ExtensionGenerationOperation.EDIT]: handleEdit
+};
+function writeFile(files, outputPath) {
+  if (!files) {
+    console.warn("\u26A0\uFE0F Skipping file operation: no files provided");
+    return;
+  }
+  for (const file2 of files) {
+    if (!file2.path) {
+      console.warn("\u26A0\uFE0F Skipping file operation: no path specified");
+      continue;
+    }
+    const fullPath = import_path9.default.join(outputPath, file2.path);
+    const handler = operationHandlers[file2.operation];
+    if (!handler) {
+      throw new import_ditto_codegen_types16.ExtensionGenerationError(
+        `Unknown operation "${file2.operation}" for file ${file2.path || "unknown"}`,
+        {
+          operation: file2.operation,
+          filePath: file2.path || "unknown"
+        }
+      );
+    }
+    handler(file2, fullPath);
+  }
+}
+function writeFileInFileSystem(filePath, content) {
+  try {
+    import_fs6.default.writeFileSync(filePath, content);
+  } catch (error46) {
+    throw new import_ditto_codegen_types16.FileSystemError(`Failed to write file: ${filePath}`, {
+      filePath,
+      operation: import_ditto_codegen_types16.FileSystemOperation.WRITE,
+      cause: error46
+    });
+  }
+}
+
+// src/extensions/finalize-extension-generation.ts
 function finalizeExtensionGeneration(options, eventEmitter) {
   const {
     extension,
@@ -64306,101 +64453,16 @@ function finalizeExtensionGeneration(options, eventEmitter) {
     throw createExtensionErrorHandler(eventEmitter, extension)(error46);
   }
 }
-function writeFile(files, outputPath) {
-  if (!files) {
-    console.warn("\u26A0\uFE0F Skipping file operation: no files provided");
-    return;
-  }
-  for (const file2 of files) {
-    if (!file2.path) {
-      console.warn("\u26A0\uFE0F Skipping file operation: no path specified");
-      continue;
-    }
-    const fullPath = import_path9.default.join(outputPath, file2.path);
-    switch (file2.operation) {
-      case import_ditto_codegen_types16.ExtensionGenerationOperation.INSERT: {
-        if (!file2.content) {
-          console.warn(
-            `\u26A0\uFE0F Skipping insert operation for ${file2.path}: no content provided`
-          );
-          continue;
-        }
-        try {
-          const dir = import_path9.default.dirname(fullPath);
-          if (!import_fs6.default.existsSync(dir)) {
-            import_fs6.default.mkdirSync(dir, { recursive: true });
-          }
-          import_fs6.default.writeFileSync(fullPath, file2.content.trim());
-          console.log(`\u{1F4DD} Inserted: ${file2.path}`);
-        } catch (error46) {
-          throw new import_ditto_codegen_types16.FileSystemError(`Failed to write file: ${file2.path}`, {
-            filePath: fullPath,
-            operation: import_ditto_codegen_types16.FileSystemOperation.WRITE,
-            cause: error46
-          });
-        }
-        break;
-      }
-      case import_ditto_codegen_types16.ExtensionGenerationOperation.UPDATE:
-        if (!file2.content) {
-          console.warn(
-            `\u26A0\uFE0F Skipping update operation for ${file2.path}: no content provided`
-          );
-          continue;
-        }
-        if (!import_fs6.default.existsSync(fullPath)) {
-          console.warn(
-            `\u26A0\uFE0F Skipping update operation for ${file2.path}: file does not exist`
-          );
-          continue;
-        }
-        try {
-          import_fs6.default.writeFileSync(fullPath, file2.content.trim());
-          console.log(`\u{1F4DD} Updated: ${file2.path}`);
-        } catch (error46) {
-          throw new import_ditto_codegen_types16.FileSystemError(`Failed to update file: ${file2.path}`, {
-            filePath: fullPath,
-            operation: import_ditto_codegen_types16.FileSystemOperation.WRITE,
-            cause: error46
-          });
-        }
-        break;
-      case import_ditto_codegen_types16.ExtensionGenerationOperation.DELETE:
-        if (import_fs6.default.existsSync(fullPath)) {
-          try {
-            import_fs6.default.unlinkSync(fullPath);
-            console.log(`\u{1F5D1}\uFE0F Deleted: ${file2.path}`);
-          } catch (error46) {
-            throw new import_ditto_codegen_types16.FileSystemError(`Failed to delete file: ${file2.path}`, {
-              filePath: fullPath,
-              operation: import_ditto_codegen_types16.FileSystemOperation.DELETE,
-              cause: error46
-            });
-          }
-        } else {
-          console.warn(
-            `\u26A0\uFE0F Skipping delete operation for ${file2.path}: file does not exist`
-          );
-        }
-        break;
-      default: {
-        throw new import_ditto_codegen_types16.ExtensionGenerationError(
-          `Unknown operation "${file2.operation}" for file ${file2.path}`,
-          {
-            operation: file2.operation,
-            filePath: file2.path
-          }
-        );
-      }
-    }
-  }
-}
+
+// src/extensions/create-scaffolding.ts
+var import_ditto_codegen_types17 = __toESM(require_dist());
+var import_ditto_scaffolding4 = __toESM(require_dist5());
 async function createScaffolding(extension, outputPath, eventEmitter) {
   eventEmitter.emitEvent("scaffold:start", { extension });
   try {
     const scaffolds = await (0, import_ditto_scaffolding4.copyScaffoldingTemplate)(extension, outputPath);
     if (!scaffolds || scaffolds.length === 0) {
-      throw new import_ditto_codegen_types16.ScaffoldingError(`Failed to scaffold ${extension.type}`, {
+      throw new import_ditto_codegen_types17.ScaffoldingError(`Failed to scaffold ${extension.type}`, {
         extensionType: extension.type
       });
     }
@@ -64410,10 +64472,10 @@ async function createScaffolding(extension, outputPath, eventEmitter) {
     });
     return scaffolds;
   } catch (error46) {
-    if (error46 instanceof import_ditto_codegen_types16.FileSystemError || error46 instanceof import_ditto_codegen_types16.ScaffoldingError) {
+    if (error46 instanceof import_ditto_codegen_types17.FileSystemError || error46 instanceof import_ditto_codegen_types17.ScaffoldingError) {
       throw error46;
     }
-    throw (0, import_ditto_codegen_types16.toCodegenError)(error46);
+    throw (0, import_ditto_codegen_types17.toCodegenError)(error46);
   }
 }
 
@@ -64506,7 +64568,7 @@ async function processExistingExtension(params) {
 }
 
 // src/extensions/dashboardPage.ts
-var import_ditto_codegen_types17 = __toESM(require_dist());
+var import_ditto_codegen_types18 = __toESM(require_dist());
 async function processDashboardPage(params) {
   const {
     extension,
@@ -64544,10 +64606,10 @@ async function processDashboardPage(params) {
   console.log("\u{1F3AF} Using auto-patterns for dashboard generation");
   const relevantCollection = plan?.collections?.map((op) => op.data).find((c) => c.idSuffix === decision.relevantCollectionId);
   if (!relevantCollection) {
-    throw new import_ditto_codegen_types17.ResourceNotFoundError(
+    throw new import_ditto_codegen_types18.ResourceNotFoundError(
       `Collection with ID ${decision.relevantCollectionId} not found in created collections`,
       {
-        resourceType: import_ditto_codegen_types17.ResourceType.COLLECTION,
+        resourceType: import_ditto_codegen_types18.ResourceType.COLLECTION,
         resourceId: decision.relevantCollectionId
       }
     );
@@ -64579,7 +64641,7 @@ async function processDashboardPage(params) {
 
 // src/extensions/servicePlugin.ts
 var import_types12 = __toESM(require_types_impl());
-var import_ditto_codegen_types18 = __toESM(require_dist());
+var import_ditto_codegen_types19 = __toESM(require_dist());
 async function processMultipleServicePluginScaffolds(params) {
   const {
     extension,
@@ -64592,18 +64654,18 @@ async function processMultipleServicePluginScaffolds(params) {
     userRequestSummary
   } = params;
   if (!extension.relatedSpis || extension.relatedSpis.length === 0) {
-    throw new import_ditto_codegen_types18.UnsupportedExtensionTypeError(
+    throw new import_ditto_codegen_types19.UnsupportedExtensionTypeError(
       "Service plugin extension must have related SPIs",
       {
         extensionType: extension.type || import_types12.ExtensionType.SERVICE_PLUGIN,
-        origin: import_ditto_codegen_types18.ErrorOrigin.INITIAL
+        origin: import_ditto_codegen_types19.ErrorOrigin.INITIAL
       }
     );
   }
   const parallelTasks = extension.relatedSpis.map(async (spi, i) => {
     const scaffold = scaffolds[i];
     if (!scaffold) {
-      throw new import_ditto_codegen_types18.ScaffoldingError("No scaffold generated for extension", {
+      throw new import_ditto_codegen_types19.ScaffoldingError("No scaffold generated for extension", {
         extensionType: extension.type
       });
     }
@@ -64733,7 +64795,7 @@ async function processEmbeddedScriptExtension(params) {
 // src/cms/index.ts
 var import_path10 = __toESM(require("path"));
 var import_crypto2 = require("crypto");
-var import_ditto_codegen_types19 = __toESM(require_dist());
+var import_ditto_codegen_types20 = __toESM(require_dist());
 var getDataExtensionAndCollectionsFiles = ({
   plan
 }) => {
@@ -64767,7 +64829,7 @@ export const dataExtension = extensions.genericExtension(${JSON.stringify(
   )});
 `;
   files.push({
-    operation: import_ditto_codegen_types19.ExtensionGenerationOperation.INSERT,
+    operation: import_ditto_codegen_types20.ExtensionGenerationOperation.INSERT,
     path: import_path10.default.join("src", "data", "extensions.ts"),
     content: generatedCode
   });
@@ -64905,11 +64967,11 @@ var DittoOrchestrator = class extends DittoEventEmitter {
 };
 
 // src/flows/init-codegen.ts
-var import_ditto_codegen_types21 = __toESM(require_dist());
+var import_ditto_codegen_types22 = __toESM(require_dist());
 
 // src/flows/codegen-flow-helpers.ts
 var import_path11 = __toESM(require("path"));
-var import_ditto_codegen_types20 = __toESM(require_dist());
+var import_ditto_codegen_types21 = __toESM(require_dist());
 var import_http = __toESM(require_http_impl());
 var import_http_client = __toESM(require_index_node());
 var slugify = (str = "") => {
@@ -64925,7 +64987,7 @@ function setupAgentTaskTracking(eventEmitter, jobContext) {
     codeGenerationService.addTask(jobContext.jobId, {
       id: taskId,
       kind: "run_agent",
-      status: import_ditto_codegen_types20.Status.RUNNING,
+      status: import_ditto_codegen_types21.Status.RUNNING,
       name: "run_agent",
       description: `Run agent for ${extension.name}`,
       payload: { extension }
@@ -64937,7 +64999,7 @@ function setupAgentTaskTracking(eventEmitter, jobContext) {
     codeGenerationService.updateTask(
       jobContext.jobId,
       taskId,
-      import_ditto_codegen_types20.Status.COMPLETED,
+      import_ditto_codegen_types21.Status.COMPLETED,
       { taskOutput: { files } }
     );
   });
@@ -64947,7 +65009,7 @@ function setupAgentTaskTracking(eventEmitter, jobContext) {
     await codeGenerationService.updateTask(
       jobContext.jobId,
       taskId,
-      import_ditto_codegen_types20.Status.FAILED,
+      import_ditto_codegen_types21.Status.FAILED,
       {
         error: serializeError(error46)
       }
@@ -64999,7 +65061,7 @@ function getRelatedApis(extensions) {
 }
 
 // src/flows/init-codegen.ts
-var import_ditto_codegen_types22 = __toESM(require_dist());
+var import_ditto_codegen_types23 = __toESM(require_dist());
 var runInitCodegenFlow = async (blueprint) => {
   const localJobContext = jobContextStorage.getStore();
   console.log(
@@ -65008,7 +65070,7 @@ var runInitCodegenFlow = async (blueprint) => {
   await codeGenerationService.updateTask(
     localJobContext.jobId,
     localJobContext.taskId,
-    import_ditto_codegen_types21.Status.RUNNING,
+    import_ditto_codegen_types22.Status.RUNNING,
     {}
   );
   console.log(
@@ -65028,7 +65090,7 @@ var runInitCodegenFlow = async (blueprint) => {
         requiredPermissions: requiredPermissionsResponse?.data?.requiredPermissions ?? []
       });
     } catch (error46) {
-      const codegenError = (0, import_ditto_codegen_types22.toCodegenError)(error46);
+      const codegenError = (0, import_ditto_codegen_types23.toCodegenError)(error46);
       console.error(
         `\u274C [Init] Failed to get required permissions: jobId=${localJobContext.jobId}, taskId=${localJobContext.taskId}`,
         error46
@@ -65045,13 +65107,13 @@ var runInitCodegenFlow = async (blueprint) => {
       accessToken: ctx.accessToken,
       appNamespace: ctx.appNamespace
     });
-    await updateParentTaskStatus(localJobContext, import_ditto_codegen_types21.Status.COMPLETED);
+    await updateParentTaskStatus(localJobContext, import_ditto_codegen_types22.Status.COMPLETED);
     console.log(
       `[Init] Completed init codegen task: jobId=${localJobContext.jobId}, taskId=${localJobContext.taskId}`
     );
   } catch (error46) {
-    const codegenError = (0, import_ditto_codegen_types22.toCodegenError)(error46);
-    await updateParentTaskStatus(localJobContext, import_ditto_codegen_types21.Status.FAILED, codegenError);
+    const codegenError = (0, import_ditto_codegen_types23.toCodegenError)(error46);
+    await updateParentTaskStatus(localJobContext, import_ditto_codegen_types22.Status.FAILED, codegenError);
     console.error(
       `\u274C [Init] Failed init codegen task: jobId=${localJobContext.jobId}, taskId=${localJobContext.taskId}`,
       error46
@@ -65060,7 +65122,7 @@ var runInitCodegenFlow = async (blueprint) => {
 };
 
 // src/flows/iterate-codegen.ts
-var import_ditto_codegen_types23 = __toESM(require_dist());
+var import_ditto_codegen_types24 = __toESM(require_dist());
 
 // src/IterationOrchestrator.ts
 var import_path12 = __toESM(require("path"));
@@ -65072,22 +65134,27 @@ var IterationOrchestrator = class extends DittoEventEmitter {
     this.agentsFactory = agentsFactory;
     this.apiKey = apiKey;
   }
-  /**
-   * Main entry point for iteration code generation
-   */
-  async generateIterationCode(request) {
+  async getIterationPlan(request) {
     this.emitEvent("start:iteration", {});
-    const { outputPath, chatHistory, appNamespace } = request;
-    const start = Date.now();
-    this.emitEvent("iterationAgent:start", {});
+    const { outputPath, chatHistory } = request;
+    this.emitEvent("iterationPlanAgent:start", {});
     const iterationPlan = await this.runIterationPlanning(
       outputPath,
       chatHistory
     );
-    this.emitEvent("iterationAgent:done", {
+    this.emitEvent("iterationPlanAgent:done", {
       newExtensions: iterationPlan?.additionalExtensions?.length || 0,
       currentExtensions: iterationPlan?.currentExtensions?.length || 0
     });
+    return iterationPlan;
+  }
+  /**
+   * Main entry point for iteration code generation
+   */
+  async generateIterationCode(request) {
+    const startTime = Date.now();
+    const { outputPath, appNamespace, iterationPlan } = request;
+    this.emitEvent("start:code-generation", {});
     const newExtensions = iterationPlan?.additionalExtensions?.map((ext) => ext.extension) || [];
     const currentExtensions = iterationPlan?.currentExtensions?.map((ext) => ext.extension) || [];
     const plan = await this.generatePlan(
@@ -65143,7 +65210,7 @@ var IterationOrchestrator = class extends DittoEventEmitter {
     const results = await Promise.allSettled(parallelTasks.filter(Boolean));
     throwIfFailures(results);
     await generateMainExtensionsFile(outputPath);
-    const durationMsCodeGeneration = Date.now() - start;
+    const durationMsCodeGeneration = Date.now() - startTime;
     this.emitEvent("finish:code-generation", {
       outputPath,
       durationMs: durationMsCodeGeneration
@@ -65155,9 +65222,8 @@ var IterationOrchestrator = class extends DittoEventEmitter {
       },
       this
     );
-    const durationMs = Date.now() - start;
+    const durationMs = Date.now() - startTime;
     this.emitEvent("finish:iteration", { outputPath, durationMs });
-    return iterationPlan;
   }
   /**
    * Run iteration planning to determine which extensions to create/update
@@ -65290,7 +65356,7 @@ var IterationOrchestrator = class extends DittoEventEmitter {
 };
 
 // src/flows/iterate-codegen.ts
-var import_ditto_codegen_types24 = __toESM(require_dist());
+var import_ditto_codegen_types25 = __toESM(require_dist());
 var runIterateCodegenFlow = async (chatHistory) => {
   const localJobContext = jobContextStorage.getStore();
   console.log(
@@ -65299,7 +65365,7 @@ var runIterateCodegenFlow = async (chatHistory) => {
   await codeGenerationService.updateTask(
     localJobContext.jobId,
     localJobContext.taskId,
-    import_ditto_codegen_types23.Status.RUNNING,
+    import_ditto_codegen_types24.Status.RUNNING,
     {}
   );
   console.log(
@@ -65314,49 +65380,56 @@ var runIterateCodegenFlow = async (chatHistory) => {
     );
     setupAgentTaskTracking(iterationOrchestrator, localJobContext);
     attachOrchestratorListeners(iterationOrchestrator);
-    const iterationPlan = await iterationOrchestrator.generateIterationCode({
-      chatHistory,
+    const iterationPlan = await iterationOrchestrator.getIterationPlan({
+      outputPath,
+      chatHistory
+    });
+    await updateRequiredPermissions(iterationPlan, localJobContext);
+    await iterationOrchestrator.generateIterationCode({
       outputPath,
       siteId: ctx.siteId,
       accessToken: ctx.accessToken,
-      appNamespace: ctx.appNamespace
+      appNamespace: ctx.appNamespace,
+      iterationPlan
     });
-    try {
-      const requiredPermissionsResponse = await getRequiredPermissions([
-        ...iterationPlan.additionalExtensions.map(
-          (extension) => extension.extension
-        ),
-        ...iterationPlan.currentExtensions.map(
-          (extension) => extension.extension
-        )
-      ]);
-      await updateJobPayload(localJobContext, {
-        requiredPermissions: requiredPermissionsResponse?.data?.requiredPermissions ?? []
-      });
-    } catch (error46) {
-      const codegenError = (0, import_ditto_codegen_types24.toCodegenError)(error46);
-      console.error(
-        `\u274C [Iterate] Failed to get required permissions: jobId=${localJobContext.jobId}, taskId=${localJobContext.taskId}`,
-        error46
-      );
-      await updateJobPayload(localJobContext, {
-        requiredPermissions: [],
-        requiredPermissionsErrors: codegenError
-      });
-    }
-    await updateParentTaskStatus(localJobContext, import_ditto_codegen_types23.Status.COMPLETED);
+    await updateParentTaskStatus(localJobContext, import_ditto_codegen_types24.Status.COMPLETED);
     console.log(
       `[Init] Completed iterate codegen task: jobId=${localJobContext.jobId}, taskId=${localJobContext.taskId}`
     );
   } catch (error46) {
-    const codegenError = (0, import_ditto_codegen_types24.toCodegenError)(error46);
-    await updateParentTaskStatus(localJobContext, import_ditto_codegen_types23.Status.FAILED, codegenError);
+    const codegenError = (0, import_ditto_codegen_types25.toCodegenError)(error46);
+    await updateParentTaskStatus(localJobContext, import_ditto_codegen_types24.Status.FAILED, codegenError);
     console.error(
       `\u274C [Iterate] Failed iterate codegen task: jobId=${localJobContext.jobId}, taskId=${localJobContext.taskId}`,
       error46
     );
   }
 };
+async function updateRequiredPermissions(iterationPlan, localJobContext) {
+  try {
+    const requiredPermissionsResponse = await getRequiredPermissions([
+      ...iterationPlan.additionalExtensions.map(
+        (extension) => extension.extension
+      ),
+      ...iterationPlan.currentExtensions.map(
+        (extension) => extension.extension
+      )
+    ]);
+    await updateJobPayload(localJobContext, {
+      requiredPermissions: requiredPermissionsResponse?.data?.requiredPermissions ?? []
+    });
+  } catch (error46) {
+    const codegenError = (0, import_ditto_codegen_types25.toCodegenError)(error46);
+    console.error(
+      `\u274C [Iterate] Failed to get required permissions: jobId=${localJobContext.jobId}, taskId=${localJobContext.taskId}`,
+      error46
+    );
+    await updateJobPayload(localJobContext, {
+      requiredPermissions: [],
+      requiredPermissionsErrors: codegenError
+    });
+  }
+}
 
 // src/runCodegenFlow.ts
 async function runCodegenFlow() {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@wix/ditto-codegen-public",
-  "version": "1.0.106",
+  "version": "1.0.108",
   "description": "AI-powered Wix CLI app generator - standalone executable",
   "scripts": {
     "build": "node build.mjs",
@@ -24,5 +24,5 @@
     "@wix/ditto-codegen": "1.0.0",
     "esbuild": "^0.25.9"
   },
-  "falconPackageHash": "bd5e3a8ebc7655e138d99a8a9832e011637e7344963d6886d5840e88"
+  "falconPackageHash": "d308a26ac94d9e52b29937f76e7837685cfd0ba752c295ffc1e14a9d"
 }