feat(service): expose subscribe, input, exit status for ws handler
Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
@@ -23,6 +23,8 @@ type Manager interface {
|
||||
Restart(workspaceID string, workspaceRoot string) error
|
||||
Status(workspaceID string) Status
|
||||
Subscribe(workspaceID string) (Subscription, error)
|
||||
Stdin(workspaceID string) (io.WriteCloser, error)
|
||||
ExitStatus(workspaceID string) (ExitInfo, error)
|
||||
}
|
||||
|
||||
// LocalManager implements Manager using local OS processes.
|
||||
@@ -237,6 +239,41 @@ func (m *LocalManager) Subscribe(workspaceID string) (Subscription, error) {
|
||||
return sub, nil
|
||||
}
|
||||
|
||||
// Stdin returns the stdin writer for the workspace process.
|
||||
func (m *LocalManager) Stdin(workspaceID string) (io.WriteCloser, error) {
|
||||
return m.stdinOf(workspaceID)
|
||||
}
|
||||
|
||||
// ExitStatus returns the exit status for the workspace process.
|
||||
func (m *LocalManager) ExitStatus(workspaceID string) (ExitInfo, error) {
|
||||
return m.exitOf(workspaceID)
|
||||
}
|
||||
|
||||
// stdinOf returns the session's stdin writer or CodeNotFound.
|
||||
func (m *LocalManager) stdinOf(workspaceID string) (io.WriteCloser, error) {
|
||||
m.mu.Lock()
|
||||
defer m.mu.Unlock()
|
||||
sess, ok := m.sessions[workspaceID]
|
||||
if !ok {
|
||||
return nil, util.New(util.CodeNotFound, "no running process")
|
||||
}
|
||||
return sess.Stdin, nil
|
||||
}
|
||||
|
||||
// exitOf returns the session's recorded exit info or CodeNotFound.
|
||||
func (m *LocalManager) exitOf(workspaceID string) (ExitInfo, error) {
|
||||
m.mu.Lock()
|
||||
sess, ok := m.sessions[workspaceID]
|
||||
m.mu.Unlock()
|
||||
if !ok {
|
||||
return ExitInfo{}, util.New(util.CodeNotFound, "no running process")
|
||||
}
|
||||
|
||||
sess.mu.Lock()
|
||||
defer sess.mu.Unlock()
|
||||
return sess.Exit, nil
|
||||
}
|
||||
|
||||
// ringBuffer is a fixed-size byte buffer that drops the oldest bytes when full.
|
||||
type ringBuffer struct {
|
||||
mu sync.Mutex
|
||||
|
||||
@@ -3,6 +3,8 @@ package service
|
||||
import (
|
||||
"log/slog"
|
||||
|
||||
"io"
|
||||
|
||||
"codespace/internal/process"
|
||||
"codespace/internal/workspace"
|
||||
)
|
||||
@@ -70,3 +72,27 @@ func (s *ProcessService) Status(workspaceID string) (process.Status, error) {
|
||||
}
|
||||
return s.processes.Status(workspaceID), nil
|
||||
}
|
||||
|
||||
// Subscribe subscribes to output events for the workspace process.
|
||||
func (s *ProcessService) Subscribe(workspaceID string) (process.Subscription, error) {
|
||||
if _, err := s.workspaces.Get(workspaceID); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return s.processes.Subscribe(workspaceID)
|
||||
}
|
||||
|
||||
// Input returns the stdin writer for the workspace process.
|
||||
func (s *ProcessService) Input(workspaceID string) (io.WriteCloser, error) {
|
||||
if _, err := s.workspaces.Get(workspaceID); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return s.processes.Stdin(workspaceID)
|
||||
}
|
||||
|
||||
// ExitStatus returns the exit status for the workspace process.
|
||||
func (s *ProcessService) ExitStatus(workspaceID string) (process.ExitInfo, error) {
|
||||
if _, err := s.workspaces.Get(workspaceID); err != nil {
|
||||
return process.ExitInfo{}, err
|
||||
}
|
||||
return s.processes.ExitStatus(workspaceID)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user