@tduyng/pi-ask 0.0.1 → 0.0.2

package/README.md

@@ -49,5 +49,5 @@ Local checkout: `pi install /path/to/extensions/pi-ask`
 ## Tests
 
 ```bash
-pnpm vitest run extensions/pi-ask
+bun test extensions/pi-ask
 ```

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@tduyng/pi-ask",
-  "version": "0.0.1",
+  "version": "0.0.2",
   "description": "One tool that stops and waits: ask the user up to 4 questions with options and a stated guess, answered in the terminal.",
   "keywords": [
     "pi",

package/src/index.ts

@@ -29,24 +29,43 @@ function title(question: AskParams['questions'][number]): string {
 export async function askUser(params: AskParams, ctx: Dialog): Promise<string> {
     if (!ctx.hasUI) return NO_UI_ANSWER
     const lines: string[] = []
-    for (const question of params.questions.slice(0, MAX_QUESTIONS)) {
+    const questions = params.questions.slice(0, MAX_QUESTIONS)
+    for (let index = 0; index < questions.length; index++) {
+        const question = questions[index]!
         let answer: string
         if (question.options && question.options.length > 0) {
             const pick = await ctx.ui.select(title(question), [...question.options, OTHER])
-            answer =
-                pick === undefined
-                    ? '(skipped)'
-                    : pick === OTHER
-                      ? ((await ctx.ui.input(title(question), 'your answer')) ?? '(skipped)')
-                      : pick
+            if (pick === undefined) {
+                lines.push(...cancelled(questions.slice(index)))
+                break
+            }
+            if (pick === OTHER) {
+                const typed = await ctx.ui.input(title(question), 'your answer')
+                if (typed === undefined) {
+                    lines.push(...cancelled(questions.slice(index)))
+                    break
+                }
+                answer = typed.trim() || '(skipped)'
+            } else {
+                answer = pick
+            }
         } else {
-            answer = (await ctx.ui.input(title(question))) ?? '(skipped)'
+            const typed = await ctx.ui.input(title(question))
+            if (typed === undefined) {
+                lines.push(...cancelled(questions.slice(index)))
+                break
+            }
+            answer = typed.trim() || '(skipped)'
         }
         lines.push(`Q: ${question.question}\nA: ${answer}`)
     }
     return lines.join('\n\n') || 'No well-formed questions were provided.'
 }
 
+function cancelled(questions: AskParams['questions']): string[] {
+    return questions.map((question) => `Q: ${question.question}\nA: (cancelled)`)
+}
+
 export default function askUserExtension(pi: Pick<ExtensionAPI, 'registerTool'>): void {
     pi.registerTool({
         name: 'ask_user',