Function undoByCount

apps/kimi-code/src/tui/commands/undo.ts:80–125

(host: SlashCommandHost, count: number)

}

async function undoByCount(host: SlashCommandHost, count: number): Promise<boolean> {
  const session = host.session;
  if (session === undefined) {
    host.showError(NO_ACTIVE_SESSION_MESSAGE);
    return false;
  }

  const entries = host.state.transcriptEntries;
  const lastUserIndex = findUndoAnchorEntryIndex(entries, count);
  if (lastUserIndex === undefined) {
    showUndoLimitStatus(host, 'Nothing to undo.');
    return false;
  }

  try {
    await session.undoHistory(count);
  } catch (error) {
    const limit = undoLimitFromError(error);
    if (limit !== undefined) {
      showUndoLimitStatus(host, formatUndoLimitMessage(limit.requestedCount, limit));
      return false;
    }
    const message = formatErrorMessage(error);
    host.showError(`Failed to undo: ${message}`);
    return false;
  }

  const children = host.state.transcriptContainer.children;
  const lastUserComponentIndex = findUndoAnchorComponentIndex(children, count);
  if (lastUserComponentIndex !== undefined) {
    removeUndoContextComponents(children, lastUserComponentIndex);
    host.state.transcriptContainer.invalidate();
  }

  const preservedEntries = entries.slice(lastUserIndex).filter(
    (entry) => !isUndoContextEntry(entry),
  );
  entries.splice(lastUserIndex, entries.length - lastUserIndex, ...preservedEntries);

  if (entries.length === 0) {
    renderWelcome(host);
  }

  host.state.ui.requestRender();
  return true;
}

async function showUndoSelector(host: SlashCommandHost): Promise<void> {
  if (host.session === undefined) {