pi-ask-user 0.5.1 → 0.6.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-ask-user",
-  "version": "0.5.1",
+  "version": "0.6.0",
   "description": "Interactive ask_user tool for pi-coding-agent with searchable split-pane selection UI, multi-select, and freeform input",
   "type": "module",
   "keywords": [

package/single-select-layout.ts

@@ -3,11 +3,18 @@ export interface QuestionOption {
 	description?: string;
 }
 
+export interface AnnotatedRow {
+	line: string;
+	selected: boolean;
+}
+
 export interface RenderSingleSelectRowsParams {
 	options: QuestionOption[];
 	selectedIndex: number;
 	width: number;
 	allowFreeform: boolean;
+	allowComment?: boolean;
+	commentEnabled?: boolean;
 	maxRows?: number;
 	hideDescriptions?: boolean;
 }
@@ -65,25 +72,36 @@ interface ItemBlock {
 	lines: string[];
 }
 
+type ListItem =
+	| { type: "option"; option: QuestionOption }
+	| { type: "comment-toggle"; option: QuestionOption }
+	| { type: "freeform"; option: QuestionOption };
+
 function buildItemBlocks(
 	options: QuestionOption[],
 	width: number,
 	allowFreeform: boolean,
+	allowComment: boolean,
+	commentEnabled: boolean,
 	selectedIndex: number,
 	hideDescriptions = false,
 ): ItemBlock[] {
 	const normalizedWidth = Math.max(12, width);
 	const freeformLabel = "Type something. — Enter a custom response";
-	const allItems = options.map((option) => ({ type: "option" as const, option }));
+	const commentToggleLabel = `${commentEnabled ? "[✓]" : "[ ]"} Add extra context after selection`;
+	const allItems: ListItem[] = options.map((option) => ({ type: "option", option }));
+	if (allowComment) {
+		allItems.push({ type: "comment-toggle", option: { title: commentToggleLabel } });
+	}
 	if (allowFreeform) {
-		allItems.push({ type: "freeform" as const, option: { title: freeformLabel } });
+		allItems.push({ type: "freeform", option: { title: freeformLabel } });
 	}
 
 	return allItems.map((item, itemIndex) => {
 		const pointer = itemIndex === selectedIndex ? "→" : " ";
 		const lines: string[] = [];
 
-		if (item.type === "freeform") {
+		if (item.type === "comment-toggle" || item.type === "freeform") {
 			const prefix = `${pointer}   `;
 			const wrapped = wrapText(item.option.title, Math.max(8, normalizedWidth - prefix.length));
 			wrapped.forEach((line, lineIndex) => {
@@ -114,8 +132,13 @@ function buildItemBlocks(
 	});
 }
 
-function flatten(blocks: ItemBlock[]): string[] {
-	return blocks.flatMap((block) => block.lines);
+function flatten(blocks: ItemBlock[], selectedIndex: number): AnnotatedRow[] {
+	return blocks.flatMap((block) =>
+		block.lines.map((line) => ({
+			line,
+			selected: block.itemIndex === selectedIndex,
+		})),
+	);
 }
 
 export function renderSingleSelectRows({
@@ -123,12 +146,14 @@ export function renderSingleSelectRows({
 	selectedIndex,
 	width,
 	allowFreeform,
+	allowComment = false,
+	commentEnabled = false,
 	maxRows,
 	hideDescriptions,
-}: RenderSingleSelectRowsParams): string[] {
-	const itemCount = options.length + (allowFreeform ? 1 : 0);
-	const blocks = buildItemBlocks(options, width, allowFreeform, selectedIndex, hideDescriptions);
-	const allRows = flatten(blocks);
+}: RenderSingleSelectRowsParams): AnnotatedRow[] {
+	const itemCount = options.length + (allowComment ? 1 : 0) + (allowFreeform ? 1 : 0);
+	const blocks = buildItemBlocks(options, width, allowFreeform, allowComment, commentEnabled, selectedIndex, hideDescriptions);
+	const allRows = flatten(blocks, selectedIndex);
 
 	if (!Number.isFinite(maxRows) || !maxRows || maxRows <= 0 || allRows.length <= maxRows) {
 		return allRows;
@@ -142,8 +167,11 @@ export function renderSingleSelectRows({
 	const availableRows = safeMaxRows > 1 ? safeMaxRows - 1 : 1;
 
 	if (selectedBlock.lines.length >= availableRows) {
-		const visible = selectedBlock.lines.slice(0, availableRows);
-		if (safeMaxRows > 1) visible.push(indicator);
+		const visible = selectedBlock.lines.slice(0, availableRows).map((line) => ({
+			line,
+			selected: true,
+		}));
+		if (safeMaxRows > 1) visible.push({ line: indicator, selected: false });
 		return visible.slice(0, safeMaxRows);
 	}
 
@@ -169,7 +197,7 @@ export function renderSingleSelectRows({
 		break;
 	}
 
-	const visible = flatten(blocks.slice(start, end));
-	visible.push(indicator);
+	const visible = flatten(blocks.slice(start, end), selectedIndex);
+	visible.push({ line: indicator, selected: false });
 	return visible.slice(0, safeMaxRows);
 }