feat: add bd setup gemini for Gemini CLI integration (#845)
Add support for Gemini CLI hook-based integration, similar to Claude Code: - bd setup gemini: Install SessionStart/PreCompress hooks - bd setup gemini --check: Verify installation - bd setup gemini --remove: Remove hooks - bd setup gemini --project: Project-only installation - bd setup gemini --stealth: Use bd prime --stealth Also adds Gemini CLI integration check to bd doctor. Gemini CLI's hook system is nearly identical to Claude Code's, making this a clean, low-maintenance addition. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
This commit is contained in:
beads/crew/wolf
committed by
Steve Yegge
Steve Yegge
parent
6572654cdc
commit
a06b40bd48
270
cmd/bd/setup/gemini.go
Normal file
270
cmd/bd/setup/gemini.go
Normal file
@@ -0,0 +1,270 @@
|
||||
package setup
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"fmt"
|
||||
"io"
|
||||
"os"
|
||||
"path/filepath"
|
||||
)
|
||||
|
||||
var (
|
||||
geminiEnvProvider = defaultGeminiEnv
|
||||
errGeminiHooksMissing = errors.New("gemini hooks not installed")
|
||||
)
|
||||
|
||||
type geminiEnv struct {
|
||||
stdout io.Writer
|
||||
stderr io.Writer
|
||||
homeDir string
|
||||
projectDir string
|
||||
ensureDir func(string, os.FileMode) error
|
||||
readFile func(string) ([]byte, error)
|
||||
writeFile func(string, []byte) error
|
||||
}
|
||||
|
||||
func defaultGeminiEnv() (geminiEnv, error) {
|
||||
home, err := os.UserHomeDir()
|
||||
if err != nil {
|
||||
return geminiEnv{}, fmt.Errorf("home directory: %w", err)
|
||||
}
|
||||
workDir, err := os.Getwd()
|
||||
if err != nil {
|
||||
return geminiEnv{}, fmt.Errorf("working directory: %w", err)
|
||||
}
|
||||
return geminiEnv{
|
||||
stdout: os.Stdout,
|
||||
stderr: os.Stderr,
|
||||
homeDir: home,
|
||||
projectDir: workDir,
|
||||
ensureDir: EnsureDir,
|
||||
readFile: os.ReadFile,
|
||||
writeFile: func(path string, data []byte) error {
|
||||
return atomicWriteFile(path, data)
|
||||
},
|
||||
}, nil
|
||||
}
|
||||
|
||||
func geminiProjectSettingsPath(base string) string {
|
||||
return filepath.Join(base, ".gemini", "settings.json")
|
||||
}
|
||||
|
||||
func geminiGlobalSettingsPath(home string) string {
|
||||
return filepath.Join(home, ".gemini", "settings.json")
|
||||
}
|
||||
|
||||
// InstallGemini installs Gemini CLI hooks
|
||||
func InstallGemini(project bool, stealth bool) {
|
||||
env, err := geminiEnvProvider()
|
||||
if err != nil {
|
||||
fmt.Fprintf(os.Stderr, "Error: %v\n", err)
|
||||
setupExit(1)
|
||||
return
|
||||
}
|
||||
if err := installGemini(env, project, stealth); err != nil {
|
||||
setupExit(1)
|
||||
}
|
||||
}
|
||||
|
||||
func installGemini(env geminiEnv, project bool, stealth bool) error {
|
||||
var settingsPath string
|
||||
if project {
|
||||
settingsPath = geminiProjectSettingsPath(env.projectDir)
|
||||
_, _ = fmt.Fprintln(env.stdout, "Installing Gemini CLI hooks for this project...")
|
||||
} else {
|
||||
settingsPath = geminiGlobalSettingsPath(env.homeDir)
|
||||
_, _ = fmt.Fprintln(env.stdout, "Installing Gemini CLI hooks globally...")
|
||||
}
|
||||
|
||||
if err := env.ensureDir(filepath.Dir(settingsPath), 0o755); err != nil {
|
||||
_, _ = fmt.Fprintf(env.stderr, "Error: %v\n", err)
|
||||
return err
|
||||
}
|
||||
|
||||
settings := make(map[string]interface{})
|
||||
if data, err := env.readFile(settingsPath); err == nil {
|
||||
if err := json.Unmarshal(data, &settings); err != nil {
|
||||
_, _ = fmt.Fprintf(env.stderr, "Error: failed to parse settings.json: %v\n", err)
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
hooks, ok := settings["hooks"].(map[string]interface{})
|
||||
if !ok {
|
||||
hooks = make(map[string]interface{})
|
||||
settings["hooks"] = hooks
|
||||
}
|
||||
|
||||
command := "bd prime"
|
||||
if stealth {
|
||||
command = "bd prime --stealth"
|
||||
}
|
||||
|
||||
// Gemini CLI uses "PreCompress" instead of Claude's "PreCompact"
|
||||
if addHookCommand(hooks, "SessionStart", command) {
|
||||
_, _ = fmt.Fprintln(env.stdout, "✓ Registered SessionStart hook")
|
||||
}
|
||||
if addHookCommand(hooks, "PreCompress", command) {
|
||||
_, _ = fmt.Fprintln(env.stdout, "✓ Registered PreCompress hook")
|
||||
}
|
||||
|
||||
data, err := json.MarshalIndent(settings, "", " ")
|
||||
if err != nil {
|
||||
_, _ = fmt.Fprintf(env.stderr, "Error: marshal settings: %v\n", err)
|
||||
return err
|
||||
}
|
||||
|
||||
if err := env.writeFile(settingsPath, data); err != nil {
|
||||
_, _ = fmt.Fprintf(env.stderr, "Error: write settings: %v\n", err)
|
||||
return err
|
||||
}
|
||||
|
||||
_, _ = fmt.Fprintln(env.stdout, "\n✓ Gemini CLI integration installed")
|
||||
_, _ = fmt.Fprintf(env.stdout, " Settings: %s\n", settingsPath)
|
||||
_, _ = fmt.Fprintln(env.stdout, "\nRestart Gemini CLI for changes to take effect.")
|
||||
return nil
|
||||
}
|
||||
|
||||
// CheckGemini checks if Gemini integration is installed
|
||||
func CheckGemini() {
|
||||
env, err := geminiEnvProvider()
|
||||
if err != nil {
|
||||
fmt.Fprintf(os.Stderr, "Error: %v\n", err)
|
||||
setupExit(1)
|
||||
return
|
||||
}
|
||||
if err := checkGemini(env); err != nil {
|
||||
setupExit(1)
|
||||
}
|
||||
}
|
||||
|
||||
func checkGemini(env geminiEnv) error {
|
||||
globalSettings := geminiGlobalSettingsPath(env.homeDir)
|
||||
projectSettings := geminiProjectSettingsPath(env.projectDir)
|
||||
|
||||
switch {
|
||||
case hasGeminiBeadsHooks(globalSettings):
|
||||
_, _ = fmt.Fprintf(env.stdout, "✓ Global hooks installed: %s\n", globalSettings)
|
||||
return nil
|
||||
case hasGeminiBeadsHooks(projectSettings):
|
||||
_, _ = fmt.Fprintf(env.stdout, "✓ Project hooks installed: %s\n", projectSettings)
|
||||
return nil
|
||||
default:
|
||||
_, _ = fmt.Fprintln(env.stdout, "✗ No hooks installed")
|
||||
_, _ = fmt.Fprintln(env.stdout, " Run: bd setup gemini")
|
||||
return errGeminiHooksMissing
|
||||
}
|
||||
}
|
||||
|
||||
// RemoveGemini removes Gemini CLI hooks
|
||||
func RemoveGemini(project bool) {
|
||||
env, err := geminiEnvProvider()
|
||||
if err != nil {
|
||||
fmt.Fprintf(os.Stderr, "Error: %v\n", err)
|
||||
setupExit(1)
|
||||
return
|
||||
}
|
||||
if err := removeGemini(env, project); err != nil {
|
||||
setupExit(1)
|
||||
}
|
||||
}
|
||||
|
||||
func removeGemini(env geminiEnv, project bool) error {
|
||||
var settingsPath string
|
||||
if project {
|
||||
settingsPath = geminiProjectSettingsPath(env.projectDir)
|
||||
_, _ = fmt.Fprintln(env.stdout, "Removing Gemini CLI hooks from project...")
|
||||
} else {
|
||||
settingsPath = geminiGlobalSettingsPath(env.homeDir)
|
||||
_, _ = fmt.Fprintln(env.stdout, "Removing Gemini CLI hooks globally...")
|
||||
}
|
||||
|
||||
data, err := env.readFile(settingsPath)
|
||||
if err != nil {
|
||||
_, _ = fmt.Fprintln(env.stdout, "No settings file found")
|
||||
return nil
|
||||
}
|
||||
|
||||
var settings map[string]interface{}
|
||||
if err := json.Unmarshal(data, &settings); err != nil {
|
||||
_, _ = fmt.Fprintf(env.stderr, "Error: failed to parse settings.json: %v\n", err)
|
||||
return err
|
||||
}
|
||||
|
||||
hooks, ok := settings["hooks"].(map[string]interface{})
|
||||
if !ok {
|
||||
_, _ = fmt.Fprintln(env.stdout, "No hooks found")
|
||||
return nil
|
||||
}
|
||||
|
||||
// Remove both variants from both events
|
||||
removeHookCommand(hooks, "SessionStart", "bd prime")
|
||||
removeHookCommand(hooks, "PreCompress", "bd prime")
|
||||
removeHookCommand(hooks, "SessionStart", "bd prime --stealth")
|
||||
removeHookCommand(hooks, "PreCompress", "bd prime --stealth")
|
||||
|
||||
data, err = json.MarshalIndent(settings, "", " ")
|
||||
if err != nil {
|
||||
_, _ = fmt.Fprintf(env.stderr, "Error: marshal settings: %v\n", err)
|
||||
return err
|
||||
}
|
||||
|
||||
if err := env.writeFile(settingsPath, data); err != nil {
|
||||
_, _ = fmt.Fprintf(env.stderr, "Error: write settings: %v\n", err)
|
||||
return err
|
||||
}
|
||||
|
||||
_, _ = fmt.Fprintln(env.stdout, "✓ Gemini CLI hooks removed")
|
||||
return nil
|
||||
}
|
||||
|
||||
// hasGeminiBeadsHooks checks if a settings file has bd prime hooks for Gemini CLI
|
||||
func hasGeminiBeadsHooks(settingsPath string) bool {
|
||||
data, err := os.ReadFile(settingsPath) // #nosec G304 -- settingsPath is constructed from known safe locations (user home/.gemini), not user input
|
||||
if err != nil {
|
||||
return false
|
||||
}
|
||||
|
||||
var settings map[string]interface{}
|
||||
if err := json.Unmarshal(data, &settings); err != nil {
|
||||
return false
|
||||
}
|
||||
|
||||
hooks, ok := settings["hooks"].(map[string]interface{})
|
||||
if !ok {
|
||||
return false
|
||||
}
|
||||
|
||||
// Check SessionStart and PreCompress for "bd prime"
|
||||
for _, event := range []string{"SessionStart", "PreCompress"} {
|
||||
eventHooks, ok := hooks[event].([]interface{})
|
||||
if !ok {
|
||||
continue
|
||||
}
|
||||
|
||||
for _, hook := range eventHooks {
|
||||
hookMap, ok := hook.(map[string]interface{})
|
||||
if !ok {
|
||||
continue
|
||||
}
|
||||
commands, ok := hookMap["hooks"].([]interface{})
|
||||
if !ok {
|
||||
continue
|
||||
}
|
||||
for _, cmd := range commands {
|
||||
cmdMap, ok := cmd.(map[string]interface{})
|
||||
if !ok {
|
||||
continue
|
||||
}
|
||||
// Check for either variant
|
||||
cmdStr := cmdMap["command"]
|
||||
if cmdStr == "bd prime" || cmdStr == "bd prime --stealth" {
|
||||
return true
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return false
|
||||
}
|
||||
Reference in New Issue
Block a user