mirror of
https://github.com/getcompanion-ai/co-mono.git
synced 2026-04-15 23:01:30 +00:00
6f7c10e323
- Add setEditorText() and getEditorText() to HookUIContext for prompt generator pattern - custom() now accepts async factories for fire-and-forget work - Add CancellableLoader component to tui package - Add BorderedLoader component for hooks with cancel UI - Export HookAPI, HookContext, HookFactory from main package - Update all examples to import from packages instead of relative paths - Update hooks.md and custom-tools.md documentation fixes #350
49 lines
1.5 KiB
TypeScript
49 lines
1.5 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";
|
|
|
|
export default function (pi: HookAPI) {
|
|
pi.on("session_shutdown", async (_event, ctx) => {
|
|
// Check for uncommitted changes
|
|
const { stdout: status, code } = await pi.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 = ctx.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 pi.exec("git", ["add", "-A"]);
|
|
const { code: commitCode } = await pi.exec("git", ["commit", "-m", commitMessage]);
|
|
|
|
if (commitCode === 0 && ctx.hasUI) {
|
|
ctx.ui.notify(`Auto-committed: ${commitMessage}`, "info");
|
|
}
|
|
});
|
|
}
|