- Add daemon discovery mechanism with socket scanning - Implement depth-limited filesystem walk to avoid hangs - Add DaemonInfo struct with metadata collection - Create 'bd daemons list' command with table and JSON output - Add FindDaemonByWorkspace and CleanupStaleSockets utilities - Fix workspace path to be parent of .beads directory - Add comprehensive tests for discovery functionality Closes bd-146 Closes bd-147
126 lines
3.1 KiB
Go
126 lines
3.1 KiB
Go
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"os"
|
|
"text/tabwriter"
|
|
"time"
|
|
|
|
"github.com/spf13/cobra"
|
|
"github.com/steveyegge/beads/internal/daemon"
|
|
)
|
|
|
|
var daemonsCmd = &cobra.Command{
|
|
Use: "daemons",
|
|
Short: "Manage multiple bd daemons",
|
|
Long: `Manage bd daemon processes across all repositories and worktrees.
|
|
|
|
Subcommands:
|
|
list - Show all running daemons
|
|
health - Check health of all daemons
|
|
killall - Stop all running daemons`,
|
|
}
|
|
|
|
var daemonsListCmd = &cobra.Command{
|
|
Use: "list",
|
|
Short: "List all running bd daemons",
|
|
Long: `List all running bd daemons with metadata including workspace path, PID, version,
|
|
uptime, last activity, and exclusive lock status.`,
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
searchRoots, _ := cmd.Flags().GetStringSlice("search")
|
|
jsonOutput, _ := cmd.Flags().GetBool("json")
|
|
|
|
// Discover daemons
|
|
daemons, err := daemon.DiscoverDaemons(searchRoots)
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "Error discovering daemons: %v\n", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
// Filter to only alive daemons
|
|
var aliveDaemons []daemon.DaemonInfo
|
|
for _, d := range daemons {
|
|
if d.Alive {
|
|
aliveDaemons = append(aliveDaemons, d)
|
|
}
|
|
}
|
|
|
|
if jsonOutput {
|
|
data, _ := json.MarshalIndent(aliveDaemons, "", " ")
|
|
fmt.Println(string(data))
|
|
return
|
|
}
|
|
|
|
// Human-readable table output
|
|
if len(aliveDaemons) == 0 {
|
|
fmt.Println("No running daemons found")
|
|
return
|
|
}
|
|
|
|
w := tabwriter.NewWriter(os.Stdout, 0, 0, 2, ' ', 0)
|
|
fmt.Fprintln(w, "WORKSPACE\tPID\tVERSION\tUPTIME\tLAST ACTIVITY\tLOCK")
|
|
|
|
for _, d := range aliveDaemons {
|
|
workspace := d.WorkspacePath
|
|
if workspace == "" {
|
|
workspace = "(unknown)"
|
|
}
|
|
|
|
uptime := formatDaemonDuration(d.UptimeSeconds)
|
|
|
|
lastActivity := "(unknown)"
|
|
if d.LastActivityTime != "" {
|
|
if t, err := time.Parse(time.RFC3339, d.LastActivityTime); err == nil {
|
|
lastActivity = formatDaemonRelativeTime(t)
|
|
}
|
|
}
|
|
|
|
lock := "-"
|
|
if d.ExclusiveLockActive {
|
|
lock = fmt.Sprintf("🔒 %s", d.ExclusiveLockHolder)
|
|
}
|
|
|
|
fmt.Fprintf(w, "%s\t%d\t%s\t%s\t%s\t%s\n",
|
|
workspace, d.PID, d.Version, uptime, lastActivity, lock)
|
|
}
|
|
|
|
w.Flush()
|
|
},
|
|
}
|
|
|
|
func formatDaemonDuration(seconds float64) string {
|
|
d := time.Duration(seconds * float64(time.Second))
|
|
if d < time.Minute {
|
|
return fmt.Sprintf("%.0fs", d.Seconds())
|
|
} else if d < time.Hour {
|
|
return fmt.Sprintf("%.0fm", d.Minutes())
|
|
} else if d < 24*time.Hour {
|
|
return fmt.Sprintf("%.1fh", d.Hours())
|
|
}
|
|
return fmt.Sprintf("%.1fd", d.Hours()/24)
|
|
}
|
|
|
|
func formatDaemonRelativeTime(t time.Time) string {
|
|
d := time.Since(t)
|
|
if d < time.Minute {
|
|
return "just now"
|
|
} else if d < time.Hour {
|
|
return fmt.Sprintf("%.0fm ago", d.Minutes())
|
|
} else if d < 24*time.Hour {
|
|
return fmt.Sprintf("%.1fh ago", d.Hours())
|
|
}
|
|
return fmt.Sprintf("%.1fd ago", d.Hours()/24)
|
|
}
|
|
|
|
func init() {
|
|
rootCmd.AddCommand(daemonsCmd)
|
|
|
|
// Add subcommands
|
|
daemonsCmd.AddCommand(daemonsListCmd)
|
|
|
|
// Flags for list command
|
|
daemonsListCmd.Flags().StringSlice("search", nil, "Directories to search for daemons (default: home, /tmp, cwd)")
|
|
daemonsListCmd.Flags().Bool("json", false, "Output in JSON format")
|
|
}
|