mirror of
https://github.com/getcompanion-ai/co-mono.git
synced 2026-04-15 23:01:30 +00:00
d6283f99dc
Major changes: - Replace monolithic SessionEvent with reason discriminator with individual event types: session_start, session_before_switch, session_switch, session_before_new, session_new, session_before_branch, session_branch, session_before_compact, session_compact, session_shutdown - Each event has dedicated result type (SessionBeforeSwitchResult, etc.) - HookHandler type now allows bare return statements (void in return type) - HookAPI.on() has proper overloads for each event with correct typing Additional fixes: - AgentSession now always subscribes to agent in constructor (was only subscribing when external subscribe() called, breaking internal handlers) - Standardize on undefined over null throughout codebase - HookUIContext methods return undefined instead of null - SessionManager methods return undefined instead of null - Simplify hook exports to 'export type * from types.js' - Add detailed JSDoc for skipConversationRestore vs cancel - Fix createBranchedSession to rebuild index in persist mode - newSession() now returns the session file path Updated all example hooks, tests, and emission sites to use new event types.
58 lines
1.6 KiB
TypeScript
58 lines
1.6 KiB
TypeScript
/**
|
|
* Confirm Destructive Actions Hook
|
|
*
|
|
* Prompts for confirmation before destructive session actions (clear, switch, branch).
|
|
* Demonstrates how to cancel session events using the before_* events.
|
|
*/
|
|
|
|
import type { SessionMessageEntry } from "@mariozechner/pi-coding-agent";
|
|
import type { HookAPI } from "@mariozechner/pi-coding-agent/hooks";
|
|
|
|
export default function (pi: HookAPI) {
|
|
pi.on("session_before_new", async (_event, ctx) => {
|
|
if (!ctx.hasUI) return;
|
|
|
|
const confirmed = await ctx.ui.confirm("Clear session?", "This will delete all messages in the current session.");
|
|
|
|
if (!confirmed) {
|
|
ctx.ui.notify("Clear cancelled", "info");
|
|
return { cancel: true };
|
|
}
|
|
});
|
|
|
|
pi.on("session_before_switch", async (_event, ctx) => {
|
|
if (!ctx.hasUI) return;
|
|
|
|
// Check if there are unsaved changes (messages since last assistant response)
|
|
const entries = ctx.sessionManager.getEntries();
|
|
const hasUnsavedWork = entries.some(
|
|
(e): e is SessionMessageEntry => e.type === "message" && e.message.role === "user",
|
|
);
|
|
|
|
if (hasUnsavedWork) {
|
|
const confirmed = await ctx.ui.confirm(
|
|
"Switch session?",
|
|
"You have messages in the current session. Switch anyway?",
|
|
);
|
|
|
|
if (!confirmed) {
|
|
ctx.ui.notify("Switch cancelled", "info");
|
|
return { cancel: true };
|
|
}
|
|
}
|
|
});
|
|
|
|
pi.on("session_before_branch", async (event, ctx) => {
|
|
if (!ctx.hasUI) return;
|
|
|
|
const choice = await ctx.ui.select(`Branch from turn ${event.entryIndex}?`, [
|
|
"Yes, create branch",
|
|
"No, stay in current session",
|
|
]);
|
|
|
|
if (choice !== "Yes, create branch") {
|
|
ctx.ui.notify("Branch cancelled", "info");
|
|
return { cancel: true };
|
|
}
|
|
});
|
|
}
|