can1357
779ca4872b
- Removed Prettier configuration files (.prettierignore and .prettierrc) and migrated formatting to Biome. - Updated Biome configuration from version 2.3.11 to 2.3.12 and changed arrowParentheses rule from 'always' to 'asNeeded'. - Pinned @biomejs/biome dependency to exact version 2.3.12 in package.json and bun.lock. - Applied consistent arrow function formatting across 489 files by removing unnecessary parentheses around single parameters. - Removed blank lines after comment blocks and reorganized imports for consistency across the codebase.
53 lines
1.4 KiB
TypeScript
53 lines
1.4 KiB
TypeScript
/**
|
|
* Git Checkpoint Hook
|
|
*
|
|
* Creates git stash checkpoints at each turn so /branch can restore code state.
|
|
* When branching, offers to restore code to that point in history.
|
|
*/
|
|
import type { HookAPI } from "@oh-my-pi/pi-coding-agent";
|
|
|
|
export default function (pi: HookAPI) {
|
|
const checkpoints = new Map<string, string>();
|
|
let currentEntryId: string | undefined;
|
|
|
|
// Track the current entry ID when user messages are saved
|
|
pi.on("tool_result", async (_event, ctx) => {
|
|
const leaf = ctx.sessionManager.getLeafEntry();
|
|
if (leaf) currentEntryId = leaf.id;
|
|
});
|
|
|
|
pi.on("turn_start", async () => {
|
|
// Create a git stash entry before LLM makes changes
|
|
const { stdout } = await pi.exec("git", ["stash", "create"]);
|
|
const ref = stdout.trim();
|
|
if (ref && currentEntryId) {
|
|
checkpoints.set(currentEntryId, ref);
|
|
}
|
|
});
|
|
|
|
pi.on("session_before_branch", async (event, ctx) => {
|
|
const ref = checkpoints.get(event.entryId);
|
|
if (!ref) return;
|
|
|
|
if (!ctx.hasUI) {
|
|
// In non-interactive mode, don't restore automatically
|
|
return;
|
|
}
|
|
|
|
const choice = await ctx.ui.select("Restore code state?", [
|
|
"Yes, restore code to that point",
|
|
"No, keep current code",
|
|
]);
|
|
|
|
if (choice?.startsWith("Yes")) {
|
|
await pi.exec("git", ["stash", "apply", ref]);
|
|
ctx.ui.notify("Code restored to checkpoint", "info");
|
|
}
|
|
});
|
|
|
|
pi.on("agent_end", async () => {
|
|
// Clear checkpoints after agent completes
|
|
checkpoints.clear();
|
|
});
|
|
}
|