@wix/ditto-codegen-public 1.0.294 → 1.0.295

package/dist/out.js

@@ -12321,31 +12321,66 @@ var require_category_handler = __commonJS({
     var codeGenerationService_12 = require_codeGenerationService();
     var logger_12 = require_logger();
     var constants_1 = require_constants5();
+    var CategoryKey;
+    (function(CategoryKey2) {
+      CategoryKey2["CodeSearch"] = "code-search";
+      CategoryKey2["DocSearch"] = "doc-search";
+      CategoryKey2["LoadSkill"] = "load-skill";
+      CategoryKey2["RunCommands"] = "run-commands";
+      CategoryKey2["WriteCode"] = "write-code";
+      CategoryKey2["Validate"] = "validate";
+    })(CategoryKey || (CategoryKey = {}));
+    var CATEGORY_DESCRIPTIONS = {
+      [CategoryKey.CodeSearch]: "Searching in code",
+      [CategoryKey.DocSearch]: "Searching in docs",
+      [CategoryKey.LoadSkill]: "Loading agent skills",
+      [CategoryKey.RunCommands]: "Executing commands",
+      [CategoryKey.WriteCode]: "Writing code",
+      [CategoryKey.Validate]: "Validating code"
+    };
     var CategoryTaskHandler = class {
       constructor() {
-        this.codeSearchCreated = false;
-        this.docSearchCreated = false;
+        this.created = /* @__PURE__ */ new Set();
+        this.completed = /* @__PURE__ */ new Set();
       }
-      /** Creates a search sub-task on the first use of code-search or MCP tools. */
-      async handle(tool, ctx) {
-        const jobContext = ctx.getJobContext();
-        if (!jobContext)
+      async handle(tool, ctx, state) {
+        if (!ctx.getJobContext())
           return;
-        if (constants_1.CODE_SEARCH_TOOLS.has(tool) && !this.codeSearchCreated) {
-          this.codeSearchCreated = true;
-          await this.createTask("code-search", "Searching in code", ctx);
+        if ((0, constants_1.isMcpTool)(tool)) {
+          await this.track(CategoryKey.DocSearch, ctx, state);
+        }
+        if (constants_1.CODE_SEARCH_TOOLS.has(tool)) {
+          await this.track(CategoryKey.CodeSearch, ctx, state);
         }
-        if ((0, constants_1.isMcpTool)(tool) && !this.docSearchCreated) {
-          this.docSearchCreated = true;
-          await this.createTask("doc-search", "Searching in docs", ctx);
+        if (tool === "bash") {
+          await this.track(CategoryKey.RunCommands, ctx, state);
+        }
+        if (tool === "validate") {
+          await this.track(CategoryKey.Validate, ctx, state);
+        }
+        if (constants_1.WRITE_TOOLS.has(tool)) {
+          await this.track(CategoryKey.WriteCode, ctx);
         }
       }
-      async createTask(suffix, description, ctx) {
+      async trackLoadSkill(ctx, state) {
+        await this.track(CategoryKey.LoadSkill, ctx, state);
+      }
+      async track(key, ctx, state) {
+        if (!this.created.has(key)) {
+          await this.createTask(key, ctx);
+        }
+        if (state?.status === "completed" && !this.completed.has(key)) {
+          await this.completeTask(key, ctx);
+        }
+      }
+      async createTask(key, ctx) {
+        this.created.add(key);
         const jobContext = ctx.getJobContext();
         if (!jobContext)
           return;
-        const taskId = `${jobContext.taskId}-${suffix}`;
-        ctx.trackedTasks.set(suffix, {
+        const description = CATEGORY_DESCRIPTIONS[key];
+        const taskId = `${jobContext.taskId}-${key}`;
+        ctx.trackedTasks.set(key, {
           taskId,
           description,
           status: ditto_codegen_types_12.Status.RUNNING
@@ -12366,6 +12401,26 @@ var require_category_handler = __commonJS({
           });
         }
       }
+      async completeTask(key, ctx) {
+        this.completed.add(key);
+        const jobContext = ctx.getJobContext();
+        if (!jobContext)
+          return;
+        const tracked = ctx.trackedTasks.get(key);
+        if (!tracked)
+          return;
+        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 category task", {
+            description: tracked.description
+          });
+        } catch (error) {
+          logger_12.logger.error("[OpenCode] Failed to mark category task as completed", {
+            error: error instanceof Error ? error.message : String(error)
+          });
+        }
+      }
     };
     exports2.CategoryTaskHandler = CategoryTaskHandler;
   }
@@ -12436,13 +12491,11 @@ var require_task_tracker = __commonJS({
           await this.handleWriteEvent(tool, state);
         }
         if (tool === "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(tool, ctx);
-          this.taskCounter = ctx.taskCounter;
+          await this.handleSkillEvent(state);
         }
+        const ctx = this.handlerContext;
+        await this.categoryHandler.handle(tool, ctx, state);
+        this.taskCounter = ctx.taskCounter;
       }
       async handleWriteEvent(tool, state) {
         if (state.status !== "completed")
@@ -12477,6 +12530,7 @@ var require_task_tracker = __commonJS({
         const skillName = state.title?.replace("Loaded skill: ", "") || "unknown";
         this.skillsUsed.add(skillName);
         logger_12.logger.info("[OpenCode] Loaded skill", { skillName });
+        await this.categoryHandler.trackLoadSkill(this.handlerContext, state);
       }
       /** Builds the task completion payload — includes file list for extension tasks. */
       buildTaskPayload(tracked) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@wix/ditto-codegen-public",
-  "version": "1.0.294",
+  "version": "1.0.295",
   "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": "6125c4c39a2c5d06b6cfdaadc601ecddcaab97b50fc7b8524f2102b8"
+  "falconPackageHash": "573c9d342dec6ea685b4bc2443eadc778217c63ed0bfb2289d23f8a8"
 }