feat: keepalive signal from gt commands

Every gt command now touches .gastown/keepalive.json with the last
command and timestamp. This enables smarter daemon backoff:
- Fresh (< 2 min): agent is working, skip heartbeat
- Stale (2-5 min): might be thinking, gentle poke
- Very stale (> 5 min): likely idle, safe to interrupt

Uses PersistentPreRun hook to capture all commands including subcommands.

Closes gt-bfd

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
This commit is contained in:
Steve Yegge
2025-12-18 20:03:37 -08:00
parent 884045f031
commit 62bd8739d6
3 changed files with 226 additions and 0 deletions

View File

@@ -0,0 +1,111 @@
// Package keepalive provides agent activity signaling via file touch.
package keepalive
import (
"encoding/json"
"os"
"path/filepath"
"strings"
"time"
"github.com/steveyegge/gastown/internal/workspace"
)
// State represents the keepalive file contents.
type State struct {
LastCommand string `json:"last_command"`
Timestamp time.Time `json:"timestamp"`
}
// Touch updates the keepalive file in the workspace's .gastown directory.
// It silently ignores errors (best-effort signaling).
func Touch(command string) {
TouchWithArgs(command, nil)
}
// TouchWithArgs updates the keepalive file with the full command including args.
// It silently ignores errors (best-effort signaling).
func TouchWithArgs(command string, args []string) {
root, err := workspace.FindFromCwd()
if err != nil || root == "" {
return // Not in a workspace, nothing to do
}
// Build full command string
fullCmd := command
if len(args) > 0 {
fullCmd = command + " " + strings.Join(args, " ")
}
TouchInWorkspace(root, fullCmd)
}
// TouchInWorkspace updates the keepalive file in a specific workspace.
// It silently ignores errors (best-effort signaling).
func TouchInWorkspace(workspaceRoot, command string) {
gastown := filepath.Join(workspaceRoot, ".gastown")
// Ensure .gastown directory exists
if err := os.MkdirAll(gastown, 0755); err != nil {
return
}
state := State{
LastCommand: command,
Timestamp: time.Now().UTC(),
}
data, err := json.Marshal(state)
if err != nil {
return
}
keepalivePath := filepath.Join(gastown, "keepalive.json")
_ = os.WriteFile(keepalivePath, data, 0644)
}
// Read returns the current keepalive state for the workspace.
// Returns nil if the file doesn't exist or can't be read.
func Read(workspaceRoot string) *State {
keepalivePath := filepath.Join(workspaceRoot, ".gastown", "keepalive.json")
data, err := os.ReadFile(keepalivePath)
if err != nil {
return nil
}
var state State
if err := json.Unmarshal(data, &state); err != nil {
return nil
}
return &state
}
// Age returns how old the keepalive signal is.
// Returns a very large duration if the state is nil.
func (s *State) Age() time.Duration {
if s == nil {
return 24 * time.Hour * 365 // Very stale
}
return time.Since(s.Timestamp)
}
// IsFresh returns true if the keepalive is less than 2 minutes old.
func (s *State) IsFresh() bool {
return s != nil && s.Age() < 2*time.Minute
}
// IsStale returns true if the keepalive is 2-5 minutes old.
func (s *State) IsStale() bool {
if s == nil {
return false
}
age := s.Age()
return age >= 2*time.Minute && age < 5*time.Minute
}
// IsVeryStale returns true if the keepalive is more than 5 minutes old.
func (s *State) IsVeryStale() bool {
return s == nil || s.Age() >= 5*time.Minute
}

View File

@@ -0,0 +1,97 @@
package keepalive
import (
"os"
"path/filepath"
"testing"
"time"
)
func TestTouchInWorkspace(t *testing.T) {
// Create temp directory
tmpDir := t.TempDir()
// Touch the keepalive
TouchInWorkspace(tmpDir, "gt status")
// Read back
state := Read(tmpDir)
if state == nil {
t.Fatal("expected state to be non-nil")
}
if state.LastCommand != "gt status" {
t.Errorf("expected last_command 'gt status', got %q", state.LastCommand)
}
// Check timestamp is recent
if time.Since(state.Timestamp) > time.Minute {
t.Errorf("timestamp too old: %v", state.Timestamp)
}
}
func TestReadNonExistent(t *testing.T) {
tmpDir := t.TempDir()
state := Read(tmpDir)
if state != nil {
t.Error("expected nil state for non-existent file")
}
}
func TestStateAge(t *testing.T) {
// Test nil state
var nilState *State
if nilState.Age() < 24*time.Hour {
t.Error("nil state should have very large age")
}
// Test fresh state
freshState := &State{Timestamp: time.Now().Add(-30 * time.Second)}
if !freshState.IsFresh() {
t.Error("30-second-old state should be fresh")
}
if freshState.IsStale() {
t.Error("30-second-old state should not be stale")
}
if freshState.IsVeryStale() {
t.Error("30-second-old state should not be very stale")
}
// Test stale state (3 minutes)
staleState := &State{Timestamp: time.Now().Add(-3 * time.Minute)}
if staleState.IsFresh() {
t.Error("3-minute-old state should not be fresh")
}
if !staleState.IsStale() {
t.Error("3-minute-old state should be stale")
}
if staleState.IsVeryStale() {
t.Error("3-minute-old state should not be very stale")
}
// Test very stale state (10 minutes)
veryStaleState := &State{Timestamp: time.Now().Add(-10 * time.Minute)}
if veryStaleState.IsFresh() {
t.Error("10-minute-old state should not be fresh")
}
if veryStaleState.IsStale() {
t.Error("10-minute-old state should not be stale (it's very stale)")
}
if !veryStaleState.IsVeryStale() {
t.Error("10-minute-old state should be very stale")
}
}
func TestDirectoryCreation(t *testing.T) {
tmpDir := t.TempDir()
workDir := filepath.Join(tmpDir, "some", "nested", "workspace")
// Touch should create .gastown directory
TouchInWorkspace(workDir, "gt test")
// Verify directory was created
gastown := filepath.Join(workDir, ".gastown")
if _, err := os.Stat(gastown); os.IsNotExist(err) {
t.Error("expected .gastown directory to be created")
}
}