pi-ask-tool-extension 0.2.5 → 0.2.6

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-ask-tool-extension",
-  "version": "0.2.5",
+  "version": "0.2.6",
   "description": "Ask tool extension for pi with tabbed questioning and inline note editing",
   "repository": {
     "type": "git",

package/src/ask-inline-note.ts

@@ -1,4 +1,4 @@
-import { wrapTextWithAnsi } from "@mariozechner/pi-tui";
+import { CURSOR_MARKER, wrapTextWithAnsi } from "@mariozechner/pi-tui";
 
 const INLINE_NOTE_SEPARATOR = " — note: ";
 const INLINE_EDIT_CURSOR_INVERT_ON = "\u001b[7m";
@@ -17,14 +17,19 @@ function clampCursorIndex(index: number, rawTextLength: number): number {
 	return Math.floor(index);
 }
 
-function buildEditingInlineNote(rawNote: string, editingCursorIndex?: number): string {
+function buildEditingInlineNote(
+	rawNote: string,
+	editingCursorIndex?: number,
+	includeHardwareCursorMarker = false,
+): string {
 	const cursorIndex = clampCursorIndex(editingCursorIndex ?? rawNote.length, rawNote.length);
 	const beforeCursor = sanitizeNoteForInlineDisplay(rawNote.slice(0, cursorIndex));
 	const rawCharAtCursor = rawNote.slice(cursorIndex, cursorIndex + 1);
 	const charAtCursor = sanitizeNoteForInlineDisplay(rawCharAtCursor) || " ";
 	const afterCursorStartIndex = rawCharAtCursor.length > 0 ? cursorIndex + 1 : cursorIndex;
 	const afterCursor = sanitizeNoteForInlineDisplay(rawNote.slice(afterCursorStartIndex));
-	const cursorCell = `${INLINE_EDIT_CURSOR_INVERT_ON}${charAtCursor}${INLINE_EDIT_CURSOR_INVERT_OFF}`;
+	const cursorMarker = includeHardwareCursorMarker ? CURSOR_MARKER : "";
+	const cursorCell = `${cursorMarker}${INLINE_EDIT_CURSOR_INVERT_ON}${charAtCursor}${INLINE_EDIT_CURSOR_INVERT_OFF}`;
 	return `${beforeCursor}${cursorCell}${afterCursor}`;
 }
 
@@ -48,6 +53,7 @@ export function buildOptionLabelWithInlineNote(
 	isEditingNote: boolean,
 	maxInlineLabelLength?: number,
 	editingCursorIndex?: number,
+	includeHardwareCursorMarker = false,
 ): string {
 	const sanitizedNote = sanitizeNoteForInlineDisplay(rawNote);
 	if (!isEditingNote && sanitizedNote.trim().length === 0) {
@@ -55,7 +61,9 @@ export function buildOptionLabelWithInlineNote(
 	}
 
 	const labelPrefix = `${baseOptionLabel}${INLINE_NOTE_SEPARATOR}`;
-	const inlineNote = isEditingNote ? buildEditingInlineNote(rawNote, editingCursorIndex) : sanitizedNote.trim();
+	const inlineNote = isEditingNote
+		? buildEditingInlineNote(rawNote, editingCursorIndex, includeHardwareCursorMarker)
+		: sanitizedNote.trim();
 	const inlineLabel = `${labelPrefix}${inlineNote}`;
 
 	if (maxInlineLabelLength == null) {
@@ -74,6 +82,7 @@ export function buildWrappedOptionLabelWithInlineNote(
 	maxInlineLabelLength: number,
 	wrapPadding = INLINE_NOTE_WRAP_PADDING,
 	editingCursorIndex?: number,
+	includeHardwareCursorMarker = false,
 ): string[] {
 	const inlineLabel = buildOptionLabelWithInlineNote(
 		baseOptionLabel,
@@ -81,6 +90,7 @@ export function buildWrappedOptionLabelWithInlineNote(
 		isEditingNote,
 		undefined,
 		editingCursorIndex,
+		includeHardwareCursorMarker,
 	);
 	const sanitizedWrapPadding = Number.isFinite(wrapPadding) ? Math.max(0, Math.floor(wrapPadding)) : 0;
 	const sanitizedMaxInlineLabelLength = Number.isFinite(maxInlineLabelLength)

package/src/ask-inline-ui.ts

@@ -187,6 +187,7 @@ export async function askSingleQuestionWithInlineNote(
 					Math.max(1, width - prefixWidth),
 					INLINE_NOTE_WRAP_PADDING,
 					isEditingThisOption ? activeEditingCursorIndex : undefined,
+					isEditingThisOption,
 				);
 				const continuationPrefix = " ".repeat(prefixWidth);
 				addLine(`${cursorPrefix}${theme.fg(optionColor, `${markerText}${wrappedInlineLabelLines[0] ?? ""}`)}`);
@@ -288,6 +289,7 @@ export async function askSingleQuestionWithInlineNote(
 		};
 
 		return {
+			focused: true,
 			render,
 			invalidate: () => {
 				cachedRenderedLines = undefined;

package/src/ask-tabs-ui.ts

@@ -388,6 +388,7 @@ export async function askQuestionsWithTabs(
 					Math.max(1, width - prefixWidth),
 					INLINE_NOTE_WRAP_PADDING,
 					isEditingThisOption ? activeEditingCursorIndex : undefined,
+					isEditingThisOption,
 				);
 				const continuationPrefix = " ".repeat(prefixWidth);
 				addLine(`${cursorPrefix}${theme.fg(optionColor, `${markerText}${wrappedInlineLabelLines[0] ?? ""}`)}`);
@@ -404,12 +405,12 @@ export async function askQuestionsWithTabs(
 					addLine(
 						theme.fg(
 							"dim",
-							" ↑↓ move • Enter toggle/select • Tab add note • ←/→ switch tabs • Esc cancel",
+							" ↑↓ move • Space toggle/select • Enter next • Tab add note • ←/→ switch tabs • Esc cancel",
 						),
 					);
 				} else {
 					addLine(
-						theme.fg("dim", " ↑↓ move • Enter select • Tab add note • ←/→ switch tabs • Esc cancel"),
+						theme.fg("dim", " ↑↓ move • Space/Enter select • Tab add note • ←/→ switch tabs • Esc cancel"),
 					);
 				}
 			}
@@ -533,7 +534,7 @@ export async function askQuestionsWithTabs(
 				return;
 			}
 
-			if (matchesKey(data, Key.enter)) {
+			if (matchesKey(data, Key.space)) {
 				const cursorOptionIndex = cursorOptionIndexByQuestion[questionIndex];
 
 				if (preparedQuestion.multi) {
@@ -566,6 +567,37 @@ export async function askQuestionsWithTabs(
 					return;
 				}
 
+				requestUiRerender();
+				return;
+			}
+
+			if (matchesKey(data, Key.enter)) {
+				const cursorOptionIndex = cursorOptionIndexByQuestion[questionIndex];
+
+				if (preparedQuestion.multi) {
+					if (
+						cursorOptionIndex === preparedQuestion.otherOptionIndex &&
+						selectedOptionIndexesByQuestion[questionIndex].includes(cursorOptionIndex) &&
+						getTrimmedQuestionNote(questionIndex, cursorOptionIndex).length === 0
+					) {
+						openNoteEditorForActiveOption();
+						return;
+					}
+
+					advanceToNextTabOrSubmit();
+					requestUiRerender();
+					return;
+				}
+
+				selectedOptionIndexesByQuestion[questionIndex] = [cursorOptionIndex];
+				if (
+					cursorOptionIndex === preparedQuestion.otherOptionIndex &&
+					getTrimmedQuestionNote(questionIndex, cursorOptionIndex).length === 0
+				) {
+					openNoteEditorForActiveOption();
+					return;
+				}
+
 				advanceToNextTabOrSubmit();
 				requestUiRerender();
 				return;
@@ -585,6 +617,7 @@ export async function askQuestionsWithTabs(
 		};
 
 		return {
+			focused: true,
 			render,
 			invalidate: () => {
 				cachedRenderedLines = undefined;