feat(hooks): add hook registry and install command (bd-qj9nc)
- Add hooks_registry.go: LoadRegistry(), HookRegistry/HookDefinition types - Add hooks_install.go: gt hooks install command with --role and --all-rigs flags - gt hooks list now reads from ~/gt/hooks/registry.toml - Supports dry-run, deduplication, and creates .claude dirs as needed Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
This commit is contained in:
beads/crew/lizzy
committed by
beads/crew/emma
beads/crew/emma
parent
6616a4726c
commit
d0e49a216a
267
internal/cmd/hooks_install.go
Normal file
267
internal/cmd/hooks_install.go
Normal file
@@ -0,0 +1,267 @@
|
||||
package cmd
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
|
||||
"github.com/spf13/cobra"
|
||||
"github.com/steveyegge/gastown/internal/style"
|
||||
"github.com/steveyegge/gastown/internal/workspace"
|
||||
)
|
||||
|
||||
var (
|
||||
installRole string
|
||||
installAllRigs bool
|
||||
installDryRun bool
|
||||
)
|
||||
|
||||
var hooksInstallCmd = &cobra.Command{
|
||||
Use: "install <hook-name>",
|
||||
Short: "Install a hook from the registry",
|
||||
Long: `Install a hook from the registry to worktrees.
|
||||
|
||||
By default, installs to the current worktree. Use --role to install
|
||||
to all worktrees of a specific role in the current rig.
|
||||
|
||||
Examples:
|
||||
gt hooks install pr-workflow-guard # Install to current worktree
|
||||
gt hooks install pr-workflow-guard --role crew # Install to all crew in current rig
|
||||
gt hooks install session-prime --role crew --all-rigs # Install to all crew everywhere
|
||||
gt hooks install pr-workflow-guard --dry-run # Preview what would be installed`,
|
||||
Args: cobra.ExactArgs(1),
|
||||
RunE: runHooksInstall,
|
||||
}
|
||||
|
||||
func init() {
|
||||
hooksCmd.AddCommand(hooksInstallCmd)
|
||||
hooksInstallCmd.Flags().StringVar(&installRole, "role", "", "Install to all worktrees of this role (crew, polecat, witness, refinery)")
|
||||
hooksInstallCmd.Flags().BoolVar(&installAllRigs, "all-rigs", false, "Install across all rigs (requires --role)")
|
||||
hooksInstallCmd.Flags().BoolVar(&installDryRun, "dry-run", false, "Preview changes without writing files")
|
||||
}
|
||||
|
||||
func runHooksInstall(cmd *cobra.Command, args []string) error {
|
||||
hookName := args[0]
|
||||
|
||||
townRoot, err := workspace.FindFromCwd()
|
||||
if err != nil {
|
||||
return fmt.Errorf("not in a Gas Town workspace: %w", err)
|
||||
}
|
||||
|
||||
// Load registry
|
||||
registry, err := LoadRegistry(townRoot)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// Find the hook
|
||||
hookDef, ok := registry.Hooks[hookName]
|
||||
if !ok {
|
||||
return fmt.Errorf("hook %q not found in registry", hookName)
|
||||
}
|
||||
|
||||
if !hookDef.Enabled {
|
||||
fmt.Printf("%s Hook %q is disabled in registry. Use --force to install anyway.\n",
|
||||
style.Warning.Render("Warning:"), hookName)
|
||||
}
|
||||
|
||||
// Determine target worktrees
|
||||
targets, err := determineTargets(townRoot, installRole, installAllRigs, hookDef.Roles)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if len(targets) == 0 {
|
||||
// No role specified, install to current worktree
|
||||
cwd, err := os.Getwd()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
targets = []string{cwd}
|
||||
}
|
||||
|
||||
// Install to each target
|
||||
installed := 0
|
||||
for _, target := range targets {
|
||||
if err := installHookTo(target, hookName, hookDef, installDryRun); err != nil {
|
||||
fmt.Printf("%s Failed to install to %s: %v\n", style.Error.Render("Error:"), target, err)
|
||||
continue
|
||||
}
|
||||
installed++
|
||||
}
|
||||
|
||||
if installDryRun {
|
||||
fmt.Printf("\n%s Would install %q to %d worktree(s)\n", style.Dim.Render("Dry run:"), hookName, installed)
|
||||
} else {
|
||||
fmt.Printf("\n%s Installed %q to %d worktree(s)\n", style.Success.Render("Done:"), hookName, installed)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// determineTargets finds all worktree paths matching the role criteria.
|
||||
func determineTargets(townRoot, role string, allRigs bool, allowedRoles []string) ([]string, error) {
|
||||
if role == "" {
|
||||
return nil, nil // Will use current directory
|
||||
}
|
||||
|
||||
// Check if role is allowed for this hook
|
||||
roleAllowed := false
|
||||
for _, r := range allowedRoles {
|
||||
if r == role {
|
||||
roleAllowed = true
|
||||
break
|
||||
}
|
||||
}
|
||||
if !roleAllowed {
|
||||
return nil, fmt.Errorf("hook is not applicable to role %q (allowed: %s)", role, strings.Join(allowedRoles, ", "))
|
||||
}
|
||||
|
||||
var targets []string
|
||||
|
||||
// Find rigs to scan
|
||||
var rigs []string
|
||||
if allRigs {
|
||||
entries, err := os.ReadDir(townRoot)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
for _, e := range entries {
|
||||
if e.IsDir() && !strings.HasPrefix(e.Name(), ".") && e.Name() != "mayor" && e.Name() != "deacon" && e.Name() != "hooks" {
|
||||
rigs = append(rigs, e.Name())
|
||||
}
|
||||
}
|
||||
} else {
|
||||
// Find current rig from cwd
|
||||
cwd, err := os.Getwd()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
relPath, err := filepath.Rel(townRoot, cwd)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
parts := strings.Split(relPath, string(filepath.Separator))
|
||||
if len(parts) > 0 {
|
||||
rigs = []string{parts[0]}
|
||||
}
|
||||
}
|
||||
|
||||
// Find worktrees for the role in each rig
|
||||
for _, rig := range rigs {
|
||||
rigPath := filepath.Join(townRoot, rig)
|
||||
|
||||
switch role {
|
||||
case "crew":
|
||||
crewDir := filepath.Join(rigPath, "crew")
|
||||
if entries, err := os.ReadDir(crewDir); err == nil {
|
||||
for _, e := range entries {
|
||||
if e.IsDir() && !strings.HasPrefix(e.Name(), ".") {
|
||||
targets = append(targets, filepath.Join(crewDir, e.Name()))
|
||||
}
|
||||
}
|
||||
}
|
||||
case "polecat":
|
||||
polecatsDir := filepath.Join(rigPath, "polecats")
|
||||
if entries, err := os.ReadDir(polecatsDir); err == nil {
|
||||
for _, e := range entries {
|
||||
if e.IsDir() && !strings.HasPrefix(e.Name(), ".") {
|
||||
targets = append(targets, filepath.Join(polecatsDir, e.Name()))
|
||||
}
|
||||
}
|
||||
}
|
||||
case "witness":
|
||||
witnessPath := filepath.Join(rigPath, "witness")
|
||||
if _, err := os.Stat(witnessPath); err == nil {
|
||||
targets = append(targets, witnessPath)
|
||||
}
|
||||
case "refinery":
|
||||
refineryPath := filepath.Join(rigPath, "refinery")
|
||||
if _, err := os.Stat(refineryPath); err == nil {
|
||||
targets = append(targets, refineryPath)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return targets, nil
|
||||
}
|
||||
|
||||
// installHookTo installs a hook to a specific worktree.
|
||||
func installHookTo(worktreePath, hookName string, hookDef HookDefinition, dryRun bool) error {
|
||||
settingsPath := filepath.Join(worktreePath, ".claude", "settings.json")
|
||||
|
||||
// Load existing settings or create new
|
||||
var settings ClaudeSettings
|
||||
if data, err := os.ReadFile(settingsPath); err == nil {
|
||||
if err := json.Unmarshal(data, &settings); err != nil {
|
||||
return fmt.Errorf("parsing existing settings: %w", err)
|
||||
}
|
||||
}
|
||||
|
||||
// Initialize maps if needed
|
||||
if settings.Hooks == nil {
|
||||
settings.Hooks = make(map[string][]ClaudeHookMatcher)
|
||||
}
|
||||
if settings.EnabledPlugins == nil {
|
||||
settings.EnabledPlugins = make(map[string]bool)
|
||||
}
|
||||
|
||||
// Build the hook entries
|
||||
for _, matcher := range hookDef.Matchers {
|
||||
hookEntry := ClaudeHookMatcher{
|
||||
Matcher: matcher,
|
||||
Hooks: []ClaudeHook{
|
||||
{Type: "command", Command: hookDef.Command},
|
||||
},
|
||||
}
|
||||
|
||||
// Check if this exact matcher already exists
|
||||
exists := false
|
||||
for _, existing := range settings.Hooks[hookDef.Event] {
|
||||
if existing.Matcher == matcher {
|
||||
exists = true
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
if !exists {
|
||||
settings.Hooks[hookDef.Event] = append(settings.Hooks[hookDef.Event], hookEntry)
|
||||
}
|
||||
}
|
||||
|
||||
// Ensure beads plugin is disabled (standard for Gas Town)
|
||||
settings.EnabledPlugins["beads@beads-marketplace"] = false
|
||||
|
||||
// Pretty print relative path
|
||||
relPath := worktreePath
|
||||
if home, err := os.UserHomeDir(); err == nil {
|
||||
if rel, err := filepath.Rel(home, worktreePath); err == nil && !strings.HasPrefix(rel, "..") {
|
||||
relPath = "~/" + rel
|
||||
}
|
||||
}
|
||||
|
||||
if dryRun {
|
||||
fmt.Printf(" %s %s\n", style.Dim.Render("Would install to:"), relPath)
|
||||
return nil
|
||||
}
|
||||
|
||||
// Create directory if needed
|
||||
if err := os.MkdirAll(filepath.Dir(settingsPath), 0755); err != nil {
|
||||
return fmt.Errorf("creating .claude directory: %w", err)
|
||||
}
|
||||
|
||||
// Write settings
|
||||
data, err := json.MarshalIndent(settings, "", " ")
|
||||
if err != nil {
|
||||
return fmt.Errorf("marshaling settings: %w", err)
|
||||
}
|
||||
|
||||
if err := os.WriteFile(settingsPath, data, 0600); err != nil {
|
||||
return fmt.Errorf("writing settings: %w", err)
|
||||
}
|
||||
|
||||
fmt.Printf(" %s %s\n", style.Success.Render("Installed to:"), relPath)
|
||||
return nil
|
||||
}
|
||||
165
internal/cmd/hooks_registry.go
Normal file
165
internal/cmd/hooks_registry.go
Normal file
@@ -0,0 +1,165 @@
|
||||
package cmd
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
|
||||
"github.com/BurntSushi/toml"
|
||||
"github.com/spf13/cobra"
|
||||
"github.com/steveyegge/gastown/internal/style"
|
||||
"github.com/steveyegge/gastown/internal/workspace"
|
||||
)
|
||||
|
||||
// HookRegistry represents the hooks/registry.toml structure.
|
||||
type HookRegistry struct {
|
||||
Hooks map[string]HookDefinition `toml:"hooks"`
|
||||
}
|
||||
|
||||
// HookDefinition represents a single hook definition in the registry.
|
||||
type HookDefinition struct {
|
||||
Description string `toml:"description"`
|
||||
Event string `toml:"event"`
|
||||
Matchers []string `toml:"matchers"`
|
||||
Command string `toml:"command"`
|
||||
Roles []string `toml:"roles"`
|
||||
Scope string `toml:"scope"`
|
||||
Enabled bool `toml:"enabled"`
|
||||
}
|
||||
|
||||
var (
|
||||
hooksListAll bool
|
||||
)
|
||||
|
||||
var hooksListCmd = &cobra.Command{
|
||||
Use: "list",
|
||||
Short: "List available hooks from the registry",
|
||||
Long: `List all hooks defined in the hook registry.
|
||||
|
||||
The registry is at ~/gt/hooks/registry.toml and defines hooks that can be
|
||||
installed for different roles (crew, polecat, witness, etc.).
|
||||
|
||||
Examples:
|
||||
gt hooks list # Show enabled hooks
|
||||
gt hooks list --all # Show all hooks including disabled`,
|
||||
RunE: runHooksList,
|
||||
}
|
||||
|
||||
func init() {
|
||||
hooksCmd.AddCommand(hooksListCmd)
|
||||
hooksListCmd.Flags().BoolVarP(&hooksListAll, "all", "a", false, "Show all hooks including disabled")
|
||||
hooksListCmd.Flags().BoolVarP(&hooksVerbose, "verbose", "v", false, "Show hook commands and matchers")
|
||||
}
|
||||
|
||||
// LoadRegistry loads the hook registry from the town's hooks directory.
|
||||
func LoadRegistry(townRoot string) (*HookRegistry, error) {
|
||||
registryPath := filepath.Join(townRoot, "hooks", "registry.toml")
|
||||
|
||||
data, err := os.ReadFile(registryPath)
|
||||
if err != nil {
|
||||
if os.IsNotExist(err) {
|
||||
return nil, fmt.Errorf("hook registry not found at %s", registryPath)
|
||||
}
|
||||
return nil, fmt.Errorf("reading registry: %w", err)
|
||||
}
|
||||
|
||||
var registry HookRegistry
|
||||
if _, err := toml.Decode(string(data), ®istry); err != nil {
|
||||
return nil, fmt.Errorf("parsing registry: %w", err)
|
||||
}
|
||||
|
||||
return ®istry, nil
|
||||
}
|
||||
|
||||
func runHooksList(cmd *cobra.Command, args []string) error {
|
||||
townRoot, err := workspace.FindFromCwd()
|
||||
if err != nil {
|
||||
return fmt.Errorf("not in a Gas Town workspace: %w", err)
|
||||
}
|
||||
|
||||
registry, err := LoadRegistry(townRoot)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if len(registry.Hooks) == 0 {
|
||||
fmt.Println(style.Dim.Render("No hooks defined in registry"))
|
||||
return nil
|
||||
}
|
||||
|
||||
fmt.Printf("\n%s Hook Registry\n", style.Bold.Render("📋"))
|
||||
fmt.Printf("Source: %s\n\n", style.Dim.Render(filepath.Join(townRoot, "hooks", "registry.toml")))
|
||||
|
||||
// Group by event type
|
||||
byEvent := make(map[string][]struct {
|
||||
name string
|
||||
def HookDefinition
|
||||
})
|
||||
eventOrder := []string{"PreToolUse", "PostToolUse", "SessionStart", "PreCompact", "UserPromptSubmit", "Stop"}
|
||||
|
||||
for name, def := range registry.Hooks {
|
||||
if !hooksListAll && !def.Enabled {
|
||||
continue
|
||||
}
|
||||
byEvent[def.Event] = append(byEvent[def.Event], struct {
|
||||
name string
|
||||
def HookDefinition
|
||||
}{name, def})
|
||||
}
|
||||
|
||||
// Add any events not in the predefined order
|
||||
for event := range byEvent {
|
||||
found := false
|
||||
for _, o := range eventOrder {
|
||||
if event == o {
|
||||
found = true
|
||||
break
|
||||
}
|
||||
}
|
||||
if !found {
|
||||
eventOrder = append(eventOrder, event)
|
||||
}
|
||||
}
|
||||
|
||||
count := 0
|
||||
for _, event := range eventOrder {
|
||||
hooks := byEvent[event]
|
||||
if len(hooks) == 0 {
|
||||
continue
|
||||
}
|
||||
|
||||
fmt.Printf("%s %s\n", style.Bold.Render("▸"), event)
|
||||
|
||||
for _, h := range hooks {
|
||||
count++
|
||||
statusIcon := "●"
|
||||
statusColor := style.Success
|
||||
if !h.def.Enabled {
|
||||
statusIcon = "○"
|
||||
statusColor = style.Dim
|
||||
}
|
||||
|
||||
rolesStr := strings.Join(h.def.Roles, ", ")
|
||||
scopeStr := h.def.Scope
|
||||
|
||||
fmt.Printf(" %s %s\n", statusColor.Render(statusIcon), style.Bold.Render(h.name))
|
||||
fmt.Printf(" %s\n", h.def.Description)
|
||||
fmt.Printf(" %s %s %s %s\n",
|
||||
style.Dim.Render("roles:"), rolesStr,
|
||||
style.Dim.Render("scope:"), scopeStr)
|
||||
|
||||
if hooksVerbose {
|
||||
fmt.Printf(" %s %s\n", style.Dim.Render("command:"), h.def.Command)
|
||||
for _, m := range h.def.Matchers {
|
||||
fmt.Printf(" %s %s\n", style.Dim.Render("matcher:"), m)
|
||||
}
|
||||
}
|
||||
}
|
||||
fmt.Println()
|
||||
}
|
||||
|
||||
fmt.Printf("%s %d hooks in registry\n", style.Dim.Render("Total:"), count)
|
||||
|
||||
return nil
|
||||
}
|
||||
Reference in New Issue
Block a user