feat: add CLI core commands (status, prime, init)
- status: Show town status with rig/polecat/witness counts (--json supported) - prime: Output role context based on current directory - init: Initialize a git repo as a Gas Town rig Closes gt-u1j.10 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
This commit is contained in:
171
internal/cmd/status.go
Normal file
171
internal/cmd/status.go
Normal file
@@ -0,0 +1,171 @@
|
||||
package cmd
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"os"
|
||||
"path/filepath"
|
||||
|
||||
"github.com/spf13/cobra"
|
||||
"github.com/steveyegge/gastown/internal/config"
|
||||
"github.com/steveyegge/gastown/internal/git"
|
||||
"github.com/steveyegge/gastown/internal/rig"
|
||||
"github.com/steveyegge/gastown/internal/style"
|
||||
"github.com/steveyegge/gastown/internal/workspace"
|
||||
)
|
||||
|
||||
var statusJSON bool
|
||||
|
||||
var statusCmd = &cobra.Command{
|
||||
Use: "status",
|
||||
Short: "Show overall town status",
|
||||
Long: `Display the current status of the Gas Town workspace.
|
||||
|
||||
Shows town name, registered rigs, active polecats, and witness status.`,
|
||||
RunE: runStatus,
|
||||
}
|
||||
|
||||
func init() {
|
||||
statusCmd.Flags().BoolVar(&statusJSON, "json", false, "Output as JSON")
|
||||
rootCmd.AddCommand(statusCmd)
|
||||
}
|
||||
|
||||
// TownStatus represents the overall status of the workspace.
|
||||
type TownStatus struct {
|
||||
Name string `json:"name"`
|
||||
Location string `json:"location"`
|
||||
Rigs []RigStatus `json:"rigs"`
|
||||
Summary StatusSum `json:"summary"`
|
||||
}
|
||||
|
||||
// RigStatus represents status of a single rig.
|
||||
type RigStatus struct {
|
||||
Name string `json:"name"`
|
||||
Polecats []string `json:"polecats"`
|
||||
PolecatCount int `json:"polecat_count"`
|
||||
HasWitness bool `json:"has_witness"`
|
||||
HasRefinery bool `json:"has_refinery"`
|
||||
}
|
||||
|
||||
// StatusSum provides summary counts.
|
||||
type StatusSum struct {
|
||||
RigCount int `json:"rig_count"`
|
||||
PolecatCount int `json:"polecat_count"`
|
||||
WitnessCount int `json:"witness_count"`
|
||||
RefineryCount int `json:"refinery_count"`
|
||||
}
|
||||
|
||||
func runStatus(cmd *cobra.Command, args []string) error {
|
||||
// Find town root
|
||||
townRoot, err := workspace.FindFromCwdOrError()
|
||||
if err != nil {
|
||||
return fmt.Errorf("not in a Gas Town workspace: %w", err)
|
||||
}
|
||||
|
||||
// Load town config
|
||||
townConfigPath := filepath.Join(townRoot, "config", "town.json")
|
||||
townConfig, err := config.LoadTownConfig(townConfigPath)
|
||||
if err != nil {
|
||||
// Try to continue without config
|
||||
townConfig = &config.TownConfig{Name: filepath.Base(townRoot)}
|
||||
}
|
||||
|
||||
// Load rigs config
|
||||
rigsConfigPath := filepath.Join(townRoot, "config", "rigs.json")
|
||||
rigsConfig, err := config.LoadRigsConfig(rigsConfigPath)
|
||||
if err != nil {
|
||||
// Empty config if file doesn't exist
|
||||
rigsConfig = &config.RigsConfig{Rigs: make(map[string]config.RigEntry)}
|
||||
}
|
||||
|
||||
// Create rig manager
|
||||
g := git.NewGit(townRoot)
|
||||
mgr := rig.NewManager(townRoot, rigsConfig, g)
|
||||
|
||||
// Discover rigs
|
||||
rigs, err := mgr.DiscoverRigs()
|
||||
if err != nil {
|
||||
return fmt.Errorf("discovering rigs: %w", err)
|
||||
}
|
||||
|
||||
// Build status
|
||||
status := TownStatus{
|
||||
Name: townConfig.Name,
|
||||
Location: townRoot,
|
||||
Rigs: make([]RigStatus, 0, len(rigs)),
|
||||
}
|
||||
|
||||
for _, r := range rigs {
|
||||
rs := RigStatus{
|
||||
Name: r.Name,
|
||||
Polecats: r.Polecats,
|
||||
PolecatCount: len(r.Polecats),
|
||||
HasWitness: r.HasWitness,
|
||||
HasRefinery: r.HasRefinery,
|
||||
}
|
||||
status.Rigs = append(status.Rigs, rs)
|
||||
|
||||
// Update summary
|
||||
status.Summary.PolecatCount += len(r.Polecats)
|
||||
if r.HasWitness {
|
||||
status.Summary.WitnessCount++
|
||||
}
|
||||
if r.HasRefinery {
|
||||
status.Summary.RefineryCount++
|
||||
}
|
||||
}
|
||||
status.Summary.RigCount = len(rigs)
|
||||
|
||||
// Output
|
||||
if statusJSON {
|
||||
return outputStatusJSON(status)
|
||||
}
|
||||
return outputStatusText(status)
|
||||
}
|
||||
|
||||
func outputStatusJSON(status TownStatus) error {
|
||||
enc := json.NewEncoder(os.Stdout)
|
||||
enc.SetIndent("", " ")
|
||||
return enc.Encode(status)
|
||||
}
|
||||
|
||||
func outputStatusText(status TownStatus) error {
|
||||
// Header
|
||||
fmt.Printf("%s %s\n", style.Bold.Render("⚙️ Gas Town:"), status.Name)
|
||||
fmt.Printf(" Location: %s\n\n", style.Dim.Render(status.Location))
|
||||
|
||||
// Summary
|
||||
fmt.Printf("%s\n", style.Bold.Render("Summary"))
|
||||
fmt.Printf(" Rigs: %d\n", status.Summary.RigCount)
|
||||
fmt.Printf(" Polecats: %d\n", status.Summary.PolecatCount)
|
||||
fmt.Printf(" Witnesses: %d\n", status.Summary.WitnessCount)
|
||||
fmt.Printf(" Refineries: %d\n", status.Summary.RefineryCount)
|
||||
|
||||
if len(status.Rigs) == 0 {
|
||||
fmt.Printf("\n%s\n", style.Dim.Render("No rigs registered. Use 'gt rig add' to add one."))
|
||||
return nil
|
||||
}
|
||||
|
||||
// Rigs detail
|
||||
fmt.Printf("\n%s\n", style.Bold.Render("Rigs"))
|
||||
for _, r := range status.Rigs {
|
||||
// Rig name with indicators
|
||||
indicators := ""
|
||||
if r.HasWitness {
|
||||
indicators += " 👁"
|
||||
}
|
||||
if r.HasRefinery {
|
||||
indicators += " 🏭"
|
||||
}
|
||||
|
||||
fmt.Printf(" %s%s\n", style.Bold.Render(r.Name), indicators)
|
||||
|
||||
if len(r.Polecats) > 0 {
|
||||
fmt.Printf(" Polecats: %v\n", r.Polecats)
|
||||
} else {
|
||||
fmt.Printf(" %s\n", style.Dim.Render("No polecats"))
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
Reference in New Issue
Block a user