@algolia/wizard 0.43.0 → 0.44.0

package/dist/main.js

@@ -3472,7 +3472,7 @@ async function runAnalysis(mode, extraInstructions = []) {
 // package.json
 var package_default = {
   name: "@algolia/wizard",
-  version: "0.43.0",
+  version: "0.44.0",
   description: "Magically implement Algolia functionality in your codebase",
   type: "module",
   engines: {
@@ -3727,6 +3727,7 @@ async function promptUser(ctx, params) {
     table,
     secondary,
     cancelable,
+    error,
     onSubmit
   } = params;
   const question = promptContent.find((c) => c.type === "question")?.value ?? "";
@@ -3741,7 +3742,8 @@ async function promptUser(ctx, params) {
     table,
     secondary,
     cancelable,
-    helpText
+    helpText,
+    error
   });
   ctx.setUserInput(workflowStateKey, selection);
   onSubmit();
@@ -3755,68 +3757,118 @@ var confirmEntitiesSchema = z26.object({
   ingestionAnalysis: ingestionAnalysisSchema.shape.ingestionAnalysis.optional(),
   confirmedEntities: confirmedEntitiesFieldSchema
 });
+var SUGGEST_ENTITY = "Suggest a different entity\u2026";
+async function askEntityName(ctx, opts) {
+  const blankHint = opts?.blankHint ?? "cancel";
+  const declared = await promptUser(ctx, {
+    workflowStateKey: "confirmedEntities",
+    promptType: "textInput",
+    error: opts?.error,
+    promptContent: [
+      ...opts?.message ? [{ type: "message", value: opts.message }] : [],
+      {
+        type: "question",
+        value: `Name an entity to search for (leave blank to ${blankHint}):`
+      }
+    ],
+    onSubmit: () => {
+    }
+  });
+  if (typeof declared !== "string" || declared.trim() === "") return null;
+  return declared.trim();
+}
+function sameEntityName(a, b) {
+  return a.toLowerCase() === b.toLowerCase();
+}
+async function resolveSuggestedEntity(ctx, declaredName) {
+  ctx.setUserInput("confirmedEntities", [declaredName]);
+  ctx.clearNotices();
+  ctx.notify({
+    messages: [
+      `Scanning the codebase for the entity "${declaredName}" and other domain models`
+    ]
+  });
+  const rerun = await runAnalysis("ingestion", [
+    `The user has confirmed the codebase contains an entity named "${declaredName}". Focus your search on locating this entity, the file path(s) where it is defined, and its indexable attribute keys.`
+  ]);
+  const detected = rerun.ingestionAnalysis ?? [];
+  const found = detected.some((e) => sameEntityName(e.name, declaredName));
+  track("AI Wizard Suggest Different Entity", {
+    entity_name: declaredName,
+    found
+  });
+  if (!found) return { found: false };
+  ctx.setUserInput("ingestionAnalysis", detected);
+  return { found: true, entities: detected };
+}
 async function confirmEntities(ctx) {
   const scan = ctx.getStepOutput("project-scan");
   let entities = scan.ingestionAnalysis;
   if (!entities?.length) {
-    const declared = await promptUser(ctx, {
+    let nameError = void 0;
+    for (; ; ) {
+      const declaredName = await askEntityName(ctx, {
+        message: nameError ? void 0 : "We couldn't detect any entities to ingest in this codebase.",
+        error: nameError,
+        blankHint: "cancel"
+      });
+      if (declaredName === null) {
+        throw new Error("User declined to specify an entity \u2014 analysis halted.");
+      }
+      const resolved2 = await resolveSuggestedEntity(ctx, declaredName);
+      if (resolved2.found) {
+        entities = resolved2.entities;
+        break;
+      }
+      nameError = `We couldn't find "${declaredName}"`;
+    }
+  }
+  let listError = void 0;
+  for (; ; ) {
+    const selection = await promptUser(ctx, {
       workflowStateKey: "confirmedEntities",
-      promptType: "textInput",
+      promptType: "multipleChoice",
+      cancelable: true,
+      error: listError,
+      secondary: [
+        ...entities.map((e) => ({
+          kind: "text",
+          value: e.attributes.join(", ") || "\u2014"
+        })),
+        { kind: "text", value: "\u2014" }
+      ],
       promptContent: [
-        {
-          type: "message",
-          value: "We couldn't detect any entities to ingest in this codebase."
-        },
-        {
-          type: "question",
-          value: "Name an entity to search for (leave blank to cancel):"
-        }
+        { type: "question", value: "Pick the entity to index" },
+        ...entities.map((e) => ({ type: "option", value: e.name })),
+        { type: "option", value: SUGGEST_ENTITY }
       ],
       onSubmit: () => {
       }
     });
-    if (typeof declared !== "string" || declared.trim() === "") {
-      throw new Error("User declined to specify an entity \u2014 analysis halted.");
-    }
-    const declaredName = declared.trim();
-    ctx.setUserInput("confirmedEntities", [declaredName]);
-    const rerun = await runAnalysis("ingestion", [
-      `The user has confirmed the codebase contains an entity named "${declaredName}". Focus your search on locating this entity, the file path(s) where it is defined, and its indexable attribute keys.`
-    ]);
-    entities = rerun.ingestionAnalysis;
-    ctx.setUserInput("ingestionAnalysis", entities);
-    if (!entities?.length) {
-      throw new Error(
-        `Could not locate the "${declaredName}" entity in this codebase \u2014 analysis halted.`
-      );
+    listError = void 0;
+    if (selection === SUGGEST_ENTITY) {
+      const declaredName = await askEntityName(ctx, {
+        blankHint: "go back"
+      });
+      if (declaredName === null) continue;
+      const resolved2 = await resolveSuggestedEntity(ctx, declaredName);
+      if (!resolved2.found) {
+        listError = `We couldn't find "${declaredName}"`;
+        continue;
+      }
+      entities = resolved2.entities;
+      continue;
     }
-  }
-  const selection = await promptUser(ctx, {
-    workflowStateKey: "confirmedEntities",
-    promptType: "multipleChoice",
-    // A Cancel row lets the user halt here (the single-select equivalent of the
-    // old multi-select Cancel); its 'cancel' sentinel matches no entity below.
-    cancelable: true,
-    secondary: entities.map((e) => ({
-      kind: "text",
-      value: e.attributes.join(", ") || "\u2014"
-    })),
-    promptContent: [
-      { type: "question", value: "Pick the entity to index" },
-      ...entities.map((e) => ({ type: "option", value: e.name }))
-    ],
-    onSubmit: () => {
+    const confirmed2 = typeof selection === "string" ? entities.filter((e) => e.name === selection) : [];
+    if (confirmed2.length === 0) {
+      throw new Error("User cancelled entity selection \u2014 analysis halted.");
     }
-  });
-  const confirmed2 = typeof selection === "string" ? entities.filter((e) => e.name === selection) : [];
-  if (confirmed2.length === 0) {
-    throw new Error("User cancelled entity selection \u2014 analysis halted.");
+    ctx.setUserInput("confirmedEntities", confirmed2);
+    track("AI Wizard Entities Confirmed", {
+      entities: toEntitySummary(confirmed2)
+    });
+    return { ingestionAnalysis: entities, confirmedEntities: confirmed2 };
   }
-  ctx.setUserInput("confirmedEntities", confirmed2);
-  track("AI Wizard Entities Confirmed", {
-    entities: toEntitySummary(confirmed2)
-  });
-  return { ingestionAnalysis: entities, confirmedEntities: confirmed2 };
 }
 
 // src/actions/review.ts

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@algolia/wizard",
-  "version": "0.43.0",
+  "version": "0.44.0",
   "description": "Magically implement Algolia functionality in your codebase",
   "type": "module",
   "engines": {