ktx/packages/cli/src/prompt-navigation.ts

/** @internal */
export const MULTISELECT_NAVIGATION_FRAGMENTS = {
  move: 'Up/Down to move',
  expand: 'Right/Left to expand or collapse',
  select: 'Tab to select or unselect',
  search: 'Type to search',
  confirm: 'Enter to confirm',
  back: 'Escape to go back',
  backSearchableTree: 'Escape to clear search or go back',
  exit: 'Ctrl+C to exit',
} as const;

function composeNavigationHint(fragments: readonly string[]): string {
  return `${fragments.join(', ')}.`;
}

const fragment = MULTISELECT_NAVIGATION_FRAGMENTS;

/** @internal */
export const FLAT_MULTISELECT_NAVIGATION_HINT = composeNavigationHint([
  fragment.move,
  fragment.select,
  fragment.confirm,
  fragment.back,
  fragment.exit,
]);

/** @internal */
export const SEARCHABLE_MULTISELECT_NAVIGATION_HINT = composeNavigationHint([
  fragment.move,
  fragment.select,
  fragment.search,
  fragment.confirm,
  fragment.back,
  fragment.exit,
]);

export const TREE_PICKER_NAVIGATION_HINT = composeNavigationHint([
  fragment.move,
  fragment.expand,
  fragment.select,
  fragment.search,
  fragment.confirm,
  fragment.backSearchableTree,
  fragment.exit,
]);

const TEXT_INPUT_NAVIGATION_HINT = 'Press Escape to go back.';

function removeTrailingBlankLines(message: string): string {
  return message.replace(/\n+$/, '');
}

function prefixContinuationLines(message: string): string {
  const lines = message.split('\n');
  if (lines.length <= 1) return message;
  const [title, ...body] = lines;
  let trailingEmptyCount = 0;
  while (trailingEmptyCount < body.length && body[body.length - 1 - trailingEmptyCount] === '') {
    trailingEmptyCount++;
  }
  const contentBody = trailingEmptyCount > 0 ? body.slice(0, -trailingEmptyCount) : body;
  const trailingBody = trailingEmptyCount > 0 ? body.slice(-trailingEmptyCount) : [];
  return [
    title,
    ...contentBody.map((line) => {
      const stripped = line.replace(/^│\s*/, '');
      return stripped === '' ? '│' : `│  ${stripped}`;
    }),
    ...trailingBody,
  ].join('\n');
}

function withTextInputBodySpacing(message: string): string {
  const normalized = removeTrailingBlankLines(message);
  if (!normalized.includes('\n')) {
    return normalized;
  }
  const [title, ...bodyLines] = normalized.split('\n');
  if (bodyLines[0] === '') {
    return normalized;
  }
  return `${title}\n\n${bodyLines.join('\n')}`;
}

/** @internal */
export function withMenuOptionSpacing(message: string): string {
  if (!message.includes('\n') || message.endsWith('\n')) {
    return message;
  }
  return `${message}\n`;
}

export function withMenuOptionsSpacing<T extends { message: string }>(options: T): T {
  return { ...options, message: withMenuOptionSpacing(options.message) };
}

export function withMultiselectNavigation(message: string): string {
  if (message.includes(FLAT_MULTISELECT_NAVIGATION_HINT)) {
    return message;
  }
  return `${message}\n${FLAT_MULTISELECT_NAVIGATION_HINT}`;
}

export function withSearchableMultiselectNavigation(message: string): string {
  if (message.includes(SEARCHABLE_MULTISELECT_NAVIGATION_HINT)) {
    return message;
  }
  return `${message}\n${SEARCHABLE_MULTISELECT_NAVIGATION_HINT}`;
}

export function withTextInputNavigation(message: string): string {
  const messageWithoutHint = removeTrailingBlankLines(message)
    .split('\n')
    .filter((line) => !line.includes(TEXT_INPUT_NAVIGATION_HINT))
    .map((line) => line.replace(/^│\s*/, ''))
    .join('\n');
  const full = `${withTextInputBodySpacing(messageWithoutHint)}\n${TEXT_INPUT_NAVIGATION_HINT}`;
  return `${prefixContinuationLines(full)}\n│`;
}