Port envd from e2b with internalized shared packages and Connect RPC
- Copy envd source from e2b-dev/infra, internalize shared dependencies
into envd/internal/shared/ (keys, filesystem, id, smap, utils)
- Switch from gRPC to Connect RPC for all envd services
- Update module paths to git.omukk.dev/wrenn/{sandbox,sandbox/envd}
- Add proto specs (process, filesystem) with buf-based code generation
- Implement full envd: process exec, filesystem ops, port forwarding,
cgroup management, MMDS integration, and HTTP API
- Update main module dependencies (firecracker SDK, pgx, goose, etc.)
- Remove placeholder .gitkeep files replaced by real implementations
This commit is contained in:
107
envd/internal/services/process/input.go
Normal file
107
envd/internal/services/process/input.go
Normal file
@ -0,0 +1,107 @@
|
||||
package process
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
|
||||
"connectrpc.com/connect"
|
||||
"github.com/rs/zerolog"
|
||||
|
||||
"git.omukk.dev/wrenn/sandbox/envd/internal/logs"
|
||||
"git.omukk.dev/wrenn/sandbox/envd/internal/services/process/handler"
|
||||
rpc "git.omukk.dev/wrenn/sandbox/envd/internal/services/spec/process"
|
||||
)
|
||||
|
||||
func handleInput(ctx context.Context, process *handler.Handler, in *rpc.ProcessInput, logger *zerolog.Logger) error {
|
||||
switch in.GetInput().(type) {
|
||||
case *rpc.ProcessInput_Pty:
|
||||
err := process.WriteTty(in.GetPty())
|
||||
if err != nil {
|
||||
return connect.NewError(connect.CodeInternal, fmt.Errorf("error writing to tty: %w", err))
|
||||
}
|
||||
|
||||
case *rpc.ProcessInput_Stdin:
|
||||
err := process.WriteStdin(in.GetStdin())
|
||||
if err != nil {
|
||||
return connect.NewError(connect.CodeInternal, fmt.Errorf("error writing to stdin: %w", err))
|
||||
}
|
||||
|
||||
logger.Debug().
|
||||
Str("event_type", "stdin").
|
||||
Interface("stdin", in.GetStdin()).
|
||||
Str(string(logs.OperationIDKey), ctx.Value(logs.OperationIDKey).(string)).
|
||||
Msg("Streaming input to process")
|
||||
|
||||
default:
|
||||
return connect.NewError(connect.CodeUnimplemented, fmt.Errorf("invalid input type %T", in.GetInput()))
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *Service) SendInput(ctx context.Context, req *connect.Request[rpc.SendInputRequest]) (*connect.Response[rpc.SendInputResponse], error) {
|
||||
proc, err := s.getProcess(req.Msg.GetProcess())
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
err = handleInput(ctx, proc, req.Msg.GetInput(), s.logger)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return connect.NewResponse(&rpc.SendInputResponse{}), nil
|
||||
}
|
||||
|
||||
func (s *Service) StreamInput(ctx context.Context, stream *connect.ClientStream[rpc.StreamInputRequest]) (*connect.Response[rpc.StreamInputResponse], error) {
|
||||
return logs.LogClientStreamWithoutEvents(ctx, s.logger, stream, s.streamInputHandler)
|
||||
}
|
||||
|
||||
func (s *Service) streamInputHandler(ctx context.Context, stream *connect.ClientStream[rpc.StreamInputRequest]) (*connect.Response[rpc.StreamInputResponse], error) {
|
||||
var proc *handler.Handler
|
||||
|
||||
for stream.Receive() {
|
||||
req := stream.Msg()
|
||||
|
||||
switch req.GetEvent().(type) {
|
||||
case *rpc.StreamInputRequest_Start:
|
||||
p, err := s.getProcess(req.GetStart().GetProcess())
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
proc = p
|
||||
case *rpc.StreamInputRequest_Data:
|
||||
err := handleInput(ctx, proc, req.GetData().GetInput(), s.logger)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
case *rpc.StreamInputRequest_Keepalive:
|
||||
default:
|
||||
return nil, connect.NewError(connect.CodeUnimplemented, fmt.Errorf("invalid event type %T", req.GetEvent()))
|
||||
}
|
||||
}
|
||||
|
||||
err := stream.Err()
|
||||
if err != nil {
|
||||
return nil, connect.NewError(connect.CodeUnknown, fmt.Errorf("error streaming input: %w", err))
|
||||
}
|
||||
|
||||
return connect.NewResponse(&rpc.StreamInputResponse{}), nil
|
||||
}
|
||||
|
||||
func (s *Service) CloseStdin(
|
||||
_ context.Context,
|
||||
req *connect.Request[rpc.CloseStdinRequest],
|
||||
) (*connect.Response[rpc.CloseStdinResponse], error) {
|
||||
handler, err := s.getProcess(req.Msg.GetProcess())
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
if err := handler.CloseStdin(); err != nil {
|
||||
return nil, connect.NewError(connect.CodeUnknown, fmt.Errorf("error closing stdin: %w", err))
|
||||
}
|
||||
|
||||
return connect.NewResponse(&rpc.CloseStdinResponse{}), nil
|
||||
}
|
||||
Reference in New Issue
Block a user