mirror of
https://github.com/mark3labs/kit.git
synced 2026-06-14 03:30:26 +00:00
Ed Zynda
23c16bb197
Phase 2+3 extension API additions:
- Tool management: GetAllTools, SetActiveTools (plan-mode support)
- Model management: SetModel, GetAvailableModels, ModelChangedEvent
- Extension options: RegisterOption, GetOption, SetOption (env/config/default)
- Inter-extension event bus: OnCustomEvent, EmitCustomEvent
- Direct LLM completion: ctx.Complete with streaming/blocking modes
- Steer delivery mode: CancelAndSend for interrupt-and-redirect
New example extensions (10):
- plan-mode.go: read-only exploration with /plan toggle
- summarize.go: conversation summarization via ctx.Complete
- bookmark.go: persistent bookmarks via AppendEntry/GetEntries
- auto-commit.go: auto-commit on exit using last assistant message
- permission-gate.go: confirm dangerous bash commands
- protected-paths.go: block writes to .env, .git/, secrets/
- notify.go: desktop notifications on agent completion
- inline-bash.go: !{cmd} expansion in prompts
- pirate.go: system prompt persona injection
- project-rules.go: load .kit/rules/*.md into system prompt
Always-wrap tools through runner for SetActiveTools disabled-tool checking.
Removed phase1/phase2 test extensions from examples.
73 lines
1.9 KiB
Go
73 lines
1.9 KiB
Go
//go:build ignore
|
|
|
|
package main
|
|
|
|
import (
|
|
"os/exec"
|
|
"strings"
|
|
|
|
"kit/ext"
|
|
)
|
|
|
|
// Init automatically commits staged changes when the session shuts down,
|
|
// using the last assistant message as the commit message. Inspired by
|
|
// Pi's auto-commit-on-exit.ts.
|
|
//
|
|
// Only commits if:
|
|
// - There are staged changes (git diff --cached is non-empty)
|
|
// - There is at least one assistant message to use as commit message
|
|
//
|
|
// The commit message is derived from the last assistant response, trimmed
|
|
// to the first paragraph (max 72 chars for the subject line).
|
|
//
|
|
// Usage: kit -e examples/extensions/auto-commit.go
|
|
func Init(api ext.API) {
|
|
api.OnSessionShutdown(func(_ ext.SessionShutdownEvent, ctx ext.Context) {
|
|
// Check for staged changes.
|
|
diff, err := exec.Command("git", "diff", "--cached", "--quiet").CombinedOutput()
|
|
_ = diff
|
|
if err == nil {
|
|
return // exit code 0 means no staged changes
|
|
}
|
|
|
|
// Get the last assistant message.
|
|
msgs := ctx.GetMessages()
|
|
var lastAssistant string
|
|
for i := len(msgs) - 1; i >= 0; i-- {
|
|
if msgs[i].Role == "assistant" {
|
|
lastAssistant = msgs[i].Content
|
|
break
|
|
}
|
|
}
|
|
if lastAssistant == "" {
|
|
return
|
|
}
|
|
|
|
// Build commit message: first paragraph, subject line max 72 chars.
|
|
subject := firstParagraph(lastAssistant)
|
|
if len(subject) > 72 {
|
|
subject = subject[:69] + "..."
|
|
}
|
|
|
|
// Commit.
|
|
cmd := exec.Command("git", "commit", "-m", subject)
|
|
output, err := cmd.CombinedOutput()
|
|
if err != nil {
|
|
ctx.PrintError("Auto-commit failed: " + string(output))
|
|
return
|
|
}
|
|
ctx.PrintInfo("Auto-committed: " + subject)
|
|
})
|
|
}
|
|
|
|
// firstParagraph returns the first non-empty paragraph of text.
|
|
func firstParagraph(text string) string {
|
|
text = strings.TrimSpace(text)
|
|
// Split on double newlines (paragraph breaks).
|
|
parts := strings.SplitN(text, "\n\n", 2)
|
|
line := strings.TrimSpace(parts[0])
|
|
// Collapse to single line.
|
|
line = strings.ReplaceAll(line, "\n", " ")
|
|
return line
|
|
}
|