can1357
f11641d5a8
- Implemented session stores and metadata converters to import Claude and Codex sessions into OMP. - Added `--from-claude` and `--from-codex` CLI flags and `/resume` command arguments for foreign session resolution. - Updated session selector components and controllers to support listing and picking external agent sessions. - Added comprehensive unit tests and documentation covering foreign session import functionality.
111 lines
3.4 KiB
TypeScript
111 lines
3.4 KiB
TypeScript
import { ProcessTerminal, TUI } from "@oh-my-pi/pi-tui";
|
|
import { logger } from "@oh-my-pi/pi-utils";
|
|
import { SessionSelectorComponent } from "../modes/components/session-selector";
|
|
import { HistoryStorage } from "../session/history-storage";
|
|
import type { SessionInfo } from "../session/session-listing";
|
|
import { SessionManager } from "../session/session-manager";
|
|
import { FileSessionStorage } from "../session/session-storage";
|
|
|
|
/** Presentation and capability controls for the standalone session picker. */
|
|
export interface SessionPickerOptions {
|
|
allSessions?: SessionInfo[];
|
|
title?: string;
|
|
scopeLabel?: string | false;
|
|
showCwd?: boolean;
|
|
allowDelete?: boolean;
|
|
allowGlobalScope?: boolean;
|
|
historySearch?: boolean;
|
|
}
|
|
|
|
/**
|
|
* Show the TUI session selector and return the selected session, or null if
|
|
* cancelled. The default OMP picker supports deletion, transcript-history
|
|
* search, and an all-projects scope; foreign import pickers disable those
|
|
* source-owned capabilities.
|
|
*/
|
|
export async function selectSession(
|
|
sessions: SessionInfo[],
|
|
options: SessionPickerOptions = {},
|
|
): Promise<SessionInfo | null> {
|
|
const { promise, resolve } = Promise.withResolvers<SessionInfo | null>();
|
|
const ui = new TUI(new ProcessTerminal());
|
|
let resolved = false;
|
|
const storage = new FileSessionStorage();
|
|
|
|
// Rank sessions with prompt-history matches too, recovering prompts the 4KB
|
|
// session-list prefix never sees. Best-effort: a missing/locked history.db
|
|
// must not break the picker.
|
|
let historyMatcher: ((query: string) => string[]) | undefined;
|
|
if (options.historySearch !== false) {
|
|
try {
|
|
const history = HistoryStorage.open();
|
|
historyMatcher = (query: string) => history.matchingSessionIds(query);
|
|
} catch (error) {
|
|
logger.warn("History storage unavailable for session ranking", { error: String(error) });
|
|
}
|
|
}
|
|
|
|
const showSelector = () => {
|
|
const selector = new SessionSelectorComponent(
|
|
sessions,
|
|
(session: SessionInfo) => {
|
|
if (!resolved) {
|
|
resolved = true;
|
|
ui.stop();
|
|
resolve(session);
|
|
}
|
|
},
|
|
() => {
|
|
if (!resolved) {
|
|
resolved = true;
|
|
ui.stop();
|
|
resolve(null);
|
|
}
|
|
},
|
|
() => {
|
|
if (!resolved) {
|
|
resolved = true;
|
|
ui.stop();
|
|
process.exit(0);
|
|
}
|
|
},
|
|
{
|
|
onDelete:
|
|
options.allowDelete === false
|
|
? undefined
|
|
: async (session: SessionInfo) => {
|
|
await storage.deleteSessionWithArtifacts(session.path);
|
|
return true;
|
|
},
|
|
historyMatcher,
|
|
loadAllSessions: options.allowGlobalScope === false ? undefined : () => SessionManager.listAll(storage),
|
|
allSessions: options.allSessions,
|
|
getTerminalRows: () => ui.terminal.rows,
|
|
fillHeight: true,
|
|
title: options.title,
|
|
scopeLabel: options.scopeLabel,
|
|
showCwd: options.showCwd,
|
|
},
|
|
);
|
|
return selector;
|
|
};
|
|
|
|
const selector = showSelector();
|
|
selector.setOnRequestRender(() => ui.requestRender());
|
|
// Present as a fullscreen overlay so the picker borrows the terminal's
|
|
// alternate screen buffer (vim/less idiom): the list scrolls and rows are
|
|
// clickable via the mouse tracking the overlay enables for its lifetime.
|
|
// Anchored top-left at full size so a mouse row maps directly to a rendered
|
|
// line (the overlay paints from screen row 0).
|
|
ui.showOverlay(selector, {
|
|
anchor: "top-left",
|
|
width: "100%",
|
|
maxHeight: "100%",
|
|
margin: 0,
|
|
fullscreen: true,
|
|
});
|
|
ui.setFocus(selector);
|
|
ui.start();
|
|
return promise;
|
|
}
|