mirror of
https://github.com/getcompanion-ai/co-mono.git
synced 2026-04-15 14:03:49 +00:00
Breaking changes to hook types: - SessionEventBase now passes sessionManager and modelRegistry directly - before_compact: passes preparation, previousCompactions (newest first) - before_switch: has targetSessionFile; switch: has previousSessionFile - Removed resolveApiKey (use modelRegistry.getApiKey()) - getSessionFile() returns string | undefined for in-memory sessions Updated: - All session event emissions in agent-session.ts - Hook examples (custom-compaction.ts, auto-commit-on-exit.ts, confirm-destructive.ts) - Tests (compaction-hooks.test.ts, compaction-hooks-example.test.ts) - export-html.ts guards for in-memory sessions
51 lines
1.6 KiB
TypeScript
51 lines
1.6 KiB
TypeScript
/**
|
|
* Auto-Commit on Exit Hook
|
|
*
|
|
* Automatically commits changes when the agent exits.
|
|
* Uses the last assistant message to generate a commit message.
|
|
*/
|
|
|
|
import type { HookAPI } from "@mariozechner/pi-coding-agent/hooks";
|
|
|
|
export default function (pi: HookAPI) {
|
|
pi.on("session", async (event, ctx) => {
|
|
if (event.reason !== "shutdown") return;
|
|
|
|
// Check for uncommitted changes
|
|
const { stdout: status, code } = await ctx.exec("git", ["status", "--porcelain"]);
|
|
|
|
if (code !== 0 || status.trim().length === 0) {
|
|
// Not a git repo or no changes
|
|
return;
|
|
}
|
|
|
|
// Find the last assistant message for commit context
|
|
const entries = event.sessionManager.getEntries();
|
|
let lastAssistantText = "";
|
|
for (let i = entries.length - 1; i >= 0; i--) {
|
|
const entry = entries[i];
|
|
if (entry.type === "message" && entry.message.role === "assistant") {
|
|
const content = entry.message.content;
|
|
if (Array.isArray(content)) {
|
|
lastAssistantText = content
|
|
.filter((c): c is { type: "text"; text: string } => c.type === "text")
|
|
.map((c) => c.text)
|
|
.join("\n");
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
|
|
// Generate a simple commit message
|
|
const firstLine = lastAssistantText.split("\n")[0] || "Work in progress";
|
|
const commitMessage = `[pi] ${firstLine.slice(0, 50)}${firstLine.length > 50 ? "..." : ""}`;
|
|
|
|
// Stage and commit
|
|
await ctx.exec("git", ["add", "-A"]);
|
|
const { code: commitCode } = await ctx.exec("git", ["commit", "-m", commitMessage]);
|
|
|
|
if (commitCode === 0 && ctx.hasUI) {
|
|
ctx.ui.notify(`Auto-committed: ${commitMessage}`, "info");
|
|
}
|
|
});
|
|
}
|