feat(costs): Wire up gt costs record to Stop hook
Add Claude Code Stop hook that automatically records session costs when sessions end. The hook calls `gt costs record` which now can derive the session name from GT_* environment variables (GT_RIG, GT_POLECAT, GT_CREW, GT_ROLE). Changes: - Add deriveSessionName() to infer tmux session name from environment - Add Stop hook to settings-autonomous.json and settings-interactive.json - Add unit tests for deriveSessionName function When a Gas Town session ends, the Stop hook fires and records the session cost as a bead event. Costs then appear in `gt costs --today` output. Closes: gt-ntzhc 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
This commit is contained in:
@@ -35,6 +35,17 @@
|
||||
}
|
||||
]
|
||||
}
|
||||
],
|
||||
"Stop": [
|
||||
{
|
||||
"matcher": "",
|
||||
"hooks": [
|
||||
{
|
||||
"type": "command",
|
||||
"command": "gt costs record"
|
||||
}
|
||||
]
|
||||
}
|
||||
]
|
||||
}
|
||||
}
|
||||
|
||||
@@ -35,6 +35,17 @@
|
||||
}
|
||||
]
|
||||
}
|
||||
],
|
||||
"Stop": [
|
||||
{
|
||||
"matcher": "",
|
||||
"hooks": [
|
||||
{
|
||||
"type": "command",
|
||||
"command": "gt costs record"
|
||||
}
|
||||
]
|
||||
}
|
||||
]
|
||||
}
|
||||
}
|
||||
|
||||
@@ -35,6 +35,17 @@
|
||||
}
|
||||
]
|
||||
}
|
||||
],
|
||||
"Stop": [
|
||||
{
|
||||
"matcher": "",
|
||||
"hooks": [
|
||||
{
|
||||
"type": "command",
|
||||
"command": "gt costs record"
|
||||
}
|
||||
]
|
||||
}
|
||||
]
|
||||
}
|
||||
}
|
||||
|
||||
@@ -519,11 +519,14 @@ func runCostsRecord(cmd *cobra.Command, args []string) error {
|
||||
// Get session from flag or try to detect from environment
|
||||
session := recordSession
|
||||
if session == "" {
|
||||
// Try to get from TMUX_PANE or tmux environment
|
||||
session = os.Getenv("GT_SESSION")
|
||||
}
|
||||
if session == "" {
|
||||
return fmt.Errorf("--session flag required (or set GT_SESSION env var)")
|
||||
// Derive session name from GT_* environment variables
|
||||
session = deriveSessionName()
|
||||
}
|
||||
if session == "" {
|
||||
return fmt.Errorf("--session flag required (or set GT_SESSION env var, or GT_RIG/GT_ROLE)")
|
||||
}
|
||||
|
||||
t := tmux.NewTmux()
|
||||
@@ -610,6 +613,41 @@ func runCostsRecord(cmd *cobra.Command, args []string) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
// deriveSessionName derives the tmux session name from GT_* environment variables.
|
||||
// Session naming patterns:
|
||||
// - Polecats: gt-{rig}-{polecat} (e.g., gt-gastown-toast)
|
||||
// - Crew: gt-{rig}-crew-{crew} (e.g., gt-gastown-crew-max)
|
||||
// - Witness/Refinery: gt-{rig}-{role} (e.g., gt-gastown-witness)
|
||||
// - Mayor/Deacon: gt-{role} (e.g., gt-mayor)
|
||||
func deriveSessionName() string {
|
||||
role := os.Getenv("GT_ROLE")
|
||||
rig := os.Getenv("GT_RIG")
|
||||
polecat := os.Getenv("GT_POLECAT")
|
||||
crew := os.Getenv("GT_CREW")
|
||||
|
||||
// Polecat: gt-{rig}-{polecat}
|
||||
if polecat != "" && rig != "" {
|
||||
return fmt.Sprintf("gt-%s-%s", rig, polecat)
|
||||
}
|
||||
|
||||
// Crew: gt-{rig}-crew-{crew}
|
||||
if crew != "" && rig != "" {
|
||||
return fmt.Sprintf("gt-%s-crew-%s", rig, crew)
|
||||
}
|
||||
|
||||
// Global roles without rig: gt-{role}
|
||||
if role != "" && rig == "" {
|
||||
return fmt.Sprintf("gt-%s", role)
|
||||
}
|
||||
|
||||
// Rig-based roles (witness, refinery): gt-{rig}-{role}
|
||||
if role != "" && rig != "" {
|
||||
return fmt.Sprintf("gt-%s-%s", rig, role)
|
||||
}
|
||||
|
||||
return ""
|
||||
}
|
||||
|
||||
// buildAgentPath builds the agent path from role, rig, and worker.
|
||||
// Examples: "mayor", "gastown/witness", "gastown/polecats/toast"
|
||||
func buildAgentPath(role, rig, worker string) string {
|
||||
|
||||
98
internal/cmd/costs_test.go
Normal file
98
internal/cmd/costs_test.go
Normal file
@@ -0,0 +1,98 @@
|
||||
package cmd
|
||||
|
||||
import (
|
||||
"os"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestDeriveSessionName(t *testing.T) {
|
||||
tests := []struct {
|
||||
name string
|
||||
envVars map[string]string
|
||||
expected string
|
||||
}{
|
||||
{
|
||||
name: "polecat session",
|
||||
envVars: map[string]string{
|
||||
"GT_ROLE": "polecat",
|
||||
"GT_RIG": "gastown",
|
||||
"GT_POLECAT": "toast",
|
||||
},
|
||||
expected: "gt-gastown-toast",
|
||||
},
|
||||
{
|
||||
name: "crew session",
|
||||
envVars: map[string]string{
|
||||
"GT_ROLE": "crew",
|
||||
"GT_RIG": "gastown",
|
||||
"GT_CREW": "max",
|
||||
},
|
||||
expected: "gt-gastown-crew-max",
|
||||
},
|
||||
{
|
||||
name: "witness session",
|
||||
envVars: map[string]string{
|
||||
"GT_ROLE": "witness",
|
||||
"GT_RIG": "gastown",
|
||||
},
|
||||
expected: "gt-gastown-witness",
|
||||
},
|
||||
{
|
||||
name: "refinery session",
|
||||
envVars: map[string]string{
|
||||
"GT_ROLE": "refinery",
|
||||
"GT_RIG": "gastown",
|
||||
},
|
||||
expected: "gt-gastown-refinery",
|
||||
},
|
||||
{
|
||||
name: "mayor session",
|
||||
envVars: map[string]string{
|
||||
"GT_ROLE": "mayor",
|
||||
},
|
||||
expected: "gt-mayor",
|
||||
},
|
||||
{
|
||||
name: "deacon session",
|
||||
envVars: map[string]string{
|
||||
"GT_ROLE": "deacon",
|
||||
},
|
||||
expected: "gt-deacon",
|
||||
},
|
||||
{
|
||||
name: "no env vars",
|
||||
envVars: map[string]string{},
|
||||
expected: "",
|
||||
},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
// Save and clear relevant env vars
|
||||
saved := make(map[string]string)
|
||||
envKeys := []string{"GT_ROLE", "GT_RIG", "GT_POLECAT", "GT_CREW"}
|
||||
for _, key := range envKeys {
|
||||
saved[key] = os.Getenv(key)
|
||||
os.Unsetenv(key)
|
||||
}
|
||||
defer func() {
|
||||
// Restore env vars
|
||||
for key, val := range saved {
|
||||
if val != "" {
|
||||
os.Setenv(key, val)
|
||||
}
|
||||
}
|
||||
}()
|
||||
|
||||
// Set test env vars
|
||||
for key, val := range tt.envVars {
|
||||
os.Setenv(key, val)
|
||||
}
|
||||
|
||||
result := deriveSessionName()
|
||||
if result != tt.expected {
|
||||
t.Errorf("deriveSessionName() = %q, want %q", result, tt.expected)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user