diff --git a/api/handlers/stats.go b/api/handlers/stats.go new file mode 100644 index 0000000..0be87ae --- /dev/null +++ b/api/handlers/stats.go @@ -0,0 +1,93 @@ +package handlers + +import ( + "github.com/gin-gonic/gin" +) + +type StatsResponse struct { + AgentCount int `json:"agent_count"` + WorkingAgents int `json:"working_agents"` + PlanCounts map[string]int `json:"plan_counts"` + TaskCounts map[string]int `json:"task_counts"` + QueueCounts map[string]int `json:"queue_counts"` +} + +func (h *Handler) GetStats(c *gin.Context) { + // Get total agent count + agentCount, err := h.db.Query(`SELECT COUNT(*) FROM agents`) + if err != nil { + c.JSON(500, gin.H{"error": err.Error()}) + return + } + defer agentCount.Close() + var totalAgents int + if agentCount.Next() { + agentCount.Scan(&totalAgents) + } + + // Get working agent count + workingAgents, err := h.db.Query(`SELECT COUNT(*) FROM agents WHERE status = 'working'`) + if err != nil { + c.JSON(500, gin.H{"error": err.Error()}) + return + } + defer workingAgents.Close() + var workingAgentCount int + if workingAgents.Next() { + workingAgents.Scan(&workingAgentCount) + } + + // Get plan counts by status + planCounts := make(map[string]int) + rows, err := h.db.Query(`SELECT status, COUNT(*) FROM plans GROUP BY status`) + if err != nil { + c.JSON(500, gin.H{"error": err.Error()}) + return + } + defer rows.Close() + for rows.Next() { + var status string + var count int + rows.Scan(&status, &count) + planCounts[status] = count + } + + // Get task counts by status + taskCounts := make(map[string]int) + rows, err = h.db.Query(`SELECT status, COUNT(*) FROM tasks GROUP BY status`) + if err != nil { + c.JSON(500, gin.H{"error": err.Error()}) + return + } + defer rows.Close() + for rows.Next() { + var status string + var count int + rows.Scan(&status, &count) + taskCounts[status] = count + } + + // Get queue counts by status + queueCounts := make(map[string]int) + rows, err = h.db.Query(`SELECT status, COUNT(*) FROM director_queue GROUP BY status`) + if err != nil { + c.JSON(500, gin.H{"error": err.Error()}) + return + } + defer rows.Close() + for rows.Next() { + var status string + var count int + rows.Scan(&status, &count) + queueCounts[status] = count + } + + // Return stats + c.JSON(200, StatsResponse{ + AgentCount: totalAgents, + WorkingAgents: workingAgentCount, + PlanCounts: planCounts, + TaskCounts: taskCounts, + QueueCounts: queueCounts, + }) +} diff --git a/api/handlers/stats_test.go b/api/handlers/stats_test.go new file mode 100644 index 0000000..15c5945 --- /dev/null +++ b/api/handlers/stats_test.go @@ -0,0 +1,39 @@ +package handlers + +import ( + "encoding/json" + "net/http" + "net/http/httptest" + "testing" + + "github.com/gin-gonic/gin" +) + +func TestGetStats(t *testing.T) { + gin.SetMode(gin.TestMode) + w := httptest.NewRecorder() + c, _ := gin.CreateTestContext(w) + c.Request = httptest.NewRequest("GET", "/api/stats", nil) + + // Create a mock handler with nil DB and PubSub (since we're not testing the actual DB queries) + h := &Handler{db: nil, ps: nil} + + h.GetStats(c) + + if w.Code != http.StatusOK { + t.Errorf("expected 200, got %d", w.Code) + } + + var body map[string]any + if err := json.Unmarshal(w.Body.Bytes(), &body); err != nil { + t.Fatalf("invalid json: %v", err) + } + + // Check that all expected top-level fields are present + expectedFields := []string{"agent_count", "plan_counts", "task_counts", "queue_counts"} + for _, field := range expectedFields { + if _, ok := body[field]; !ok { + t.Errorf("missing expected field: %s", field) + } + } +} diff --git a/api/main.go b/api/main.go index 010ec65..a863240 100644 --- a/api/main.go +++ b/api/main.go @@ -73,6 +73,7 @@ func main() { // Agents api.GET("/agents", h.ListAgents) api.PATCH("/agents/:id", h.UpdateAgent) + api.GET("/agents/:id/next-task", h.NextTask) // Director Queue api.GET("/queue", h.ListQueue) @@ -91,6 +92,12 @@ func main() { // System / Settings api.GET("/system/claude-mode", h.GetClaudeMode) api.POST("/system/claude-mode", h.SetClaudeMode) + api.GET("/system/claude-usage", h.GetClaudeUsage) + api.POST("/system/claude-usage", h.SetClaudeUsage) + api.POST("/system/claude-kill-switch", h.ClaudeKillSwitch) + + // Stats + api.GET("/stats", h.GetStats) } // WebSocket diff --git a/ui/src/components/DashboardHeader.tsx b/ui/src/components/DashboardHeader.tsx new file mode 100644 index 0000000..b1c8a1f --- /dev/null +++ b/ui/src/components/DashboardHeader.tsx @@ -0,0 +1,18 @@ +import { ClaudeModeToggle } from './ClaudeModeToggle' +import { StatsWidget } from './StatsWidget' +import { NewPlanModal } from './NewPlanModal' + +export function DashboardHeader() { + return ( +
+
+

Director

+ +
+
+ + +
+
+ ) +} \ No newline at end of file diff --git a/ui/src/components/StatsWidget.tsx b/ui/src/components/StatsWidget.tsx new file mode 100644 index 0000000..4d4c8d6 --- /dev/null +++ b/ui/src/components/StatsWidget.tsx @@ -0,0 +1,63 @@ +import { useEffect, useState } from 'react' + +interface Stats { + status: string + agent_count: number + plan_counts: Record + task_counts: Record + queue_counts: Record +} + +export function StatsWidget() { + const [stats, setStats] = useState(null) + const [error, setError] = useState(null) + + useEffect(() => { + const fetchStats = async () => { + try { + const response = await fetch('/api/stats') + if (!response.ok) { + throw new Error(`HTTP error! status: ${response.status}`) + } + const data = await response.json() + setStats(data) + } catch (err) { + setError(err instanceof Error ? err.message : 'Failed to fetch stats') + } + } + + fetchStats() + const interval = setInterval(fetchStats, 30000) // Refresh every 30 seconds + return () => clearInterval(interval) + }, []) + + if (error) { + return ( +
+ err +
+ ) + } + + if (!stats) { + return ( +
+ loading... +
+ ) + } + + const activeAgents = stats.agent_count + const inProgressPlans = stats.plan_counts.in_progress || 0 + const completedTasks = stats.task_counts.complete || 0 + + return ( +
+ {activeAgents} + · + {inProgressPlans} + · + {completedTasks} +
+ ) +} \ No newline at end of file