e5b07cc1d8
Go API server + Preact UI + Claude Code adapter. - App-centric model (ideas, not repos) - AgentProvider interface for multi-agent support - K8s pod lifecycle for sandboxed agent sessions - Gitea integration (create repos, push branches) - WebSocket streaming for live session output - Woodpecker CI/CD pipelines (kaniko build + kubectl deploy)
75 lines
1.8 KiB
Go
75 lines
1.8 KiB
Go
package api
|
|
|
|
import (
|
|
"bufio"
|
|
"io"
|
|
"net/http"
|
|
|
|
"github.com/go-chi/chi/v5"
|
|
"github.com/google/uuid"
|
|
"nhooyr.io/websocket"
|
|
|
|
"github.com/agentsphere/agent-mgr/internal/provider"
|
|
"github.com/agentsphere/agent-mgr/internal/provider/claudecode"
|
|
)
|
|
|
|
func (s *Server) streamSession(w http.ResponseWriter, r *http.Request) {
|
|
sessID, err := uuid.Parse(chi.URLParam(r, "sessionID"))
|
|
if err != nil {
|
|
http.Error(w, "invalid session id", http.StatusBadRequest)
|
|
return
|
|
}
|
|
|
|
sess, err := s.store.GetSession(r.Context(), sessID)
|
|
if err != nil || sess == nil {
|
|
http.Error(w, "session not found", http.StatusNotFound)
|
|
return
|
|
}
|
|
|
|
p, err := s.registry.Get(sess.Provider)
|
|
if err != nil {
|
|
http.Error(w, "provider not found", http.StatusInternalServerError)
|
|
return
|
|
}
|
|
|
|
handle := &provider.SessionHandle{SessionID: sess.ID, PodName: sess.PodName}
|
|
|
|
conn, err := websocket.Accept(w, r, &websocket.AcceptOptions{
|
|
InsecureSkipVerify: true, // Traefik handles TLS/origin
|
|
})
|
|
if err != nil {
|
|
s.log.Error("websocket accept failed", "err", err)
|
|
return
|
|
}
|
|
defer conn.CloseNow()
|
|
|
|
ctx := conn.CloseRead(r.Context())
|
|
|
|
stream, err := p.StreamOutput(ctx, handle)
|
|
if err != nil {
|
|
s.log.Error("stream output failed", "err", err, "pod", sess.PodName)
|
|
conn.Close(websocket.StatusInternalError, "failed to stream logs")
|
|
return
|
|
}
|
|
defer stream.Close()
|
|
|
|
tracker := claudecode.NewProgressTracker()
|
|
scanner := bufio.NewScanner(stream)
|
|
scanner.Buffer(make([]byte, 64*1024), 64*1024)
|
|
|
|
for scanner.Scan() {
|
|
line := scanner.Text()
|
|
tracker.ProcessLine(line)
|
|
|
|
if err := conn.Write(ctx, websocket.MessageText, []byte(line)); err != nil {
|
|
break
|
|
}
|
|
}
|
|
|
|
if err := scanner.Err(); err != nil && err != io.EOF {
|
|
s.log.Warn("stream scanner error", "err", err)
|
|
}
|
|
|
|
conn.Close(websocket.StatusNormalClosure, "stream ended")
|
|
}
|