28f82a66c8
- Wire all goroutines in main.go: watcher, quota monitor, account switcher, dispatcher, janitor, and 10s state flush loop - Add missing sections to config.example.yaml: notifications, dispatcher, watcher, janitor - Add scripts/claude-failover.service systemd unit Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
120 lines
3.4 KiB
Go
120 lines
3.4 KiB
Go
// Package main is the entrypoint for the claude-failover daemon.
|
|
package main
|
|
|
|
import (
|
|
"context"
|
|
"flag"
|
|
"log"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
"time"
|
|
|
|
"forge.secuaas.ovh/olivier/claude-failover/internal/api"
|
|
"forge.secuaas.ovh/olivier/claude-failover/internal/config"
|
|
"forge.secuaas.ovh/olivier/claude-failover/internal/dispatcher"
|
|
"forge.secuaas.ovh/olivier/claude-failover/internal/janitor"
|
|
"forge.secuaas.ovh/olivier/claude-failover/internal/lifecycle"
|
|
"forge.secuaas.ovh/olivier/claude-failover/internal/notify"
|
|
"forge.secuaas.ovh/olivier/claude-failover/internal/quota"
|
|
"forge.secuaas.ovh/olivier/claude-failover/internal/state"
|
|
"forge.secuaas.ovh/olivier/claude-failover/internal/switcher"
|
|
"forge.secuaas.ovh/olivier/claude-failover/internal/tmux"
|
|
"forge.secuaas.ovh/olivier/claude-failover/internal/watcher"
|
|
)
|
|
|
|
const version = "0.1.0"
|
|
|
|
func main() {
|
|
var cfgPath string
|
|
flag.StringVar(&cfgPath, "config", "config.yaml", "path to YAML config file")
|
|
flag.Parse()
|
|
|
|
log.SetFlags(log.LstdFlags | log.Lmicroseconds | log.LUTC)
|
|
log.Printf("claude-failover v%s starting (config=%s)", version, cfgPath)
|
|
|
|
cfg, err := config.Load(cfgPath)
|
|
if err != nil {
|
|
log.Fatalf("config load failed: %v", err)
|
|
}
|
|
log.Printf("config loaded: %d account(s), pool min=%d max=%d",
|
|
len(cfg.Accounts), cfg.Pool.Autonomous.Min, cfg.Pool.Autonomous.Max)
|
|
|
|
// Initialise state — reload from disk if a snapshot exists.
|
|
stateFile := cfg.Checkpoint.Dir + "/state.json"
|
|
s, err := state.LoadFromFile(stateFile)
|
|
if err != nil {
|
|
log.Fatalf("state init failed: %v", err)
|
|
}
|
|
log.Printf("state loaded (%d sessions tracked)", len(s.Sessions))
|
|
|
|
// Initialise tmux client and lifecycle manager.
|
|
tmuxClient := tmux.NewExecClient()
|
|
lm := lifecycle.New(tmuxClient, s, cfg)
|
|
lm.EnsureAllSessions()
|
|
|
|
// Block until SIGINT or SIGTERM.
|
|
ctx, cancel := signal.NotifyContext(context.Background(),
|
|
syscall.SIGINT, syscall.SIGTERM)
|
|
defer cancel()
|
|
|
|
go lm.Run(ctx)
|
|
|
|
// Notifier — reads credentials from environment variables.
|
|
notifier := notify.New(cfg)
|
|
|
|
// Session Watcher — detects when sessions finish their tasks.
|
|
sw := watcher.New(tmuxClient, s, cfg)
|
|
go sw.Run(ctx)
|
|
|
|
// Quota Monitor — polls panes for quota exhaustion signals.
|
|
qm := quota.New(tmuxClient, s, cfg)
|
|
go qm.Run(ctx)
|
|
|
|
// Account Switcher — orchestrates account failover on quota exhaustion.
|
|
as := switcher.New(tmuxClient, s, cfg, qm.SwitchChan(), notifier)
|
|
go as.Run(ctx)
|
|
|
|
// Dispatcher — assigns inbox tasks to idle sessions.
|
|
disp := dispatcher.New(tmuxClient, s, cfg, sw.DoneChan())
|
|
go disp.Run(ctx)
|
|
|
|
// Janitor — periodic cleanup of orphaned files and stale status.json.
|
|
jan := janitor.New(s, cfg.Dispatcher.ProjectsDir)
|
|
go jan.Run(ctx)
|
|
|
|
// State flush loop — persists state to disk every 10 seconds.
|
|
go func() {
|
|
ticker := time.NewTicker(10 * time.Second)
|
|
defer ticker.Stop()
|
|
for {
|
|
select {
|
|
case <-ctx.Done():
|
|
return
|
|
case <-ticker.C:
|
|
s.Flush() //nolint:errcheck
|
|
}
|
|
}
|
|
}()
|
|
|
|
// Start HTTP API server.
|
|
listenAddr := cfg.MCPHTTP.Listen
|
|
if listenAddr == "" {
|
|
listenAddr = "127.0.0.1:9090"
|
|
}
|
|
srv := api.New(listenAddr, s)
|
|
go func() {
|
|
if err := srv.Start(); err != nil {
|
|
log.Printf("API server error: %v", err)
|
|
os.Exit(1)
|
|
}
|
|
}()
|
|
|
|
log.Printf("claude-failover v%s — all goroutines running", version)
|
|
|
|
<-ctx.Done()
|
|
log.Printf("shutdown signal received — flushing state and exiting")
|
|
if err := s.Flush(); err != nil {
|
|
log.Printf("state flush warning: %v", err)
|
|
}
|
|
}
|