Something went wrong. Try again.
fork of prime agent with changes i want but are not upstreamable probqbly (and some fixes)
Something went wrong. Try again.
3.9 kB · 125 lines
TypeScript
at main
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126import { type Component, Container, getKeybindings, Spacer, Text, truncateToWidth } from "@earendil-works/pi-tui";import { theme } from "../theme/theme.js";import { DynamicBorder } from "./dynamic-border.js";
interface UserMessageItem { id: string; text: string; timestamp?: string;}
class UserMessageList implements Component { private messages: UserMessageItem[] = []; private selectedIndex: number = 0; public onSelect?: (entryId: string) => void; public onCancel?: () => void; private maxVisible: number = 10;
constructor(messages: UserMessageItem[], initialSelectedId?: string) { // Session history is chronological; default to the latest fork point. this.messages = messages; const initialIndex = initialSelectedId ? messages.findIndex((message) => message.id === initialSelectedId) : -1; this.selectedIndex = initialIndex >= 0 ? initialIndex : Math.max(0, messages.length - 1); }
invalidate(): void {}
render(width: number): string[] { const lines: string[] = [];
if (this.messages.length === 0) { lines.push(theme.fg("muted", " No user messages found")); return lines; }
const startIndex = Math.max( 0, Math.min(this.selectedIndex - Math.floor(this.maxVisible / 2), this.messages.length - this.maxVisible), ); const endIndex = Math.min(startIndex + this.maxVisible, this.messages.length);
for (let i = startIndex; i < endIndex; i++) { const message = this.messages[i]; const isSelected = i === this.selectedIndex;
const normalizedMessage = message.text.replace(/\n/g, " ").trim();
const cursor = isSelected ? theme.fg("accent", "› ") : " "; const maxMsgWidth = width - 2; const truncatedMsg = truncateToWidth(normalizedMessage, maxMsgWidth); const messageLine = cursor + (isSelected ? theme.bold(truncatedMsg) : truncatedMsg);
lines.push(messageLine);
const position = i + 1; const metadata = ` Message ${position} of ${this.messages.length}`; const metadataLine = theme.fg("muted", metadata); lines.push(metadataLine); lines.push(""); }
if (startIndex > 0 || endIndex < this.messages.length) { const scrollInfo = theme.fg("muted", ` (${this.selectedIndex + 1}/${this.messages.length})`); lines.push(scrollInfo); }
return lines; }
handleInput(keyData: string): void { const kb = getKeybindings(); if (kb.matches(keyData, "tui.select.up")) { this.selectedIndex = this.selectedIndex === 0 ? this.messages.length - 1 : this.selectedIndex - 1; } else if (kb.matches(keyData, "tui.select.down")) { this.selectedIndex = this.selectedIndex === this.messages.length - 1 ? 0 : this.selectedIndex + 1; } else if (kb.matches(keyData, "tui.select.confirm")) { const selected = this.messages[this.selectedIndex]; if (selected && this.onSelect) { this.onSelect(selected.id); } } else if (kb.matches(keyData, "tui.select.cancel")) { if (this.onCancel) { this.onCancel(); } } }}
export class UserMessageSelectorComponent extends Container { private messageList: UserMessageList;
constructor( messages: UserMessageItem[], onSelect: (entryId: string) => void, onCancel: () => void, initialSelectedId?: string, ) { super();
this.addChild(new Spacer(1)); this.addChild(new Text("Fork from Message", 1, 0)); this.addChild( new Text( theme.fg("muted", "Select a user message to copy the active path up to that point into a new session"), 1, 0, ), ); this.addChild(new Spacer(1)); this.addChild(new DynamicBorder()); this.addChild(new Spacer(1));
this.messageList = new UserMessageList(messages, initialSelectedId); this.messageList.onSelect = onSelect; this.messageList.onCancel = onCancel;
this.addChild(this.messageList);
this.addChild(new Spacer(1)); this.addChild(new DynamicBorder()); }
getMessageList(): UserMessageList { return this.messageList; }}