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:
84
envd/internal/services/process/service.go
Normal file
84
envd/internal/services/process/service.go
Normal file
@ -0,0 +1,84 @@
|
||||
package process
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
|
||||
"connectrpc.com/connect"
|
||||
"github.com/go-chi/chi/v5"
|
||||
"github.com/rs/zerolog"
|
||||
|
||||
"git.omukk.dev/wrenn/sandbox/envd/internal/execcontext"
|
||||
"git.omukk.dev/wrenn/sandbox/envd/internal/logs"
|
||||
"git.omukk.dev/wrenn/sandbox/envd/internal/services/cgroups"
|
||||
"git.omukk.dev/wrenn/sandbox/envd/internal/services/process/handler"
|
||||
rpc "git.omukk.dev/wrenn/sandbox/envd/internal/services/spec/process"
|
||||
spec "git.omukk.dev/wrenn/sandbox/envd/internal/services/spec/process/processconnect"
|
||||
"git.omukk.dev/wrenn/sandbox/envd/internal/utils"
|
||||
)
|
||||
|
||||
type Service struct {
|
||||
processes *utils.Map[uint32, *handler.Handler]
|
||||
logger *zerolog.Logger
|
||||
defaults *execcontext.Defaults
|
||||
cgroupManager cgroups.Manager
|
||||
}
|
||||
|
||||
func newService(l *zerolog.Logger, defaults *execcontext.Defaults, cgroupManager cgroups.Manager) *Service {
|
||||
return &Service{
|
||||
logger: l,
|
||||
processes: utils.NewMap[uint32, *handler.Handler](),
|
||||
defaults: defaults,
|
||||
cgroupManager: cgroupManager,
|
||||
}
|
||||
}
|
||||
|
||||
func Handle(server *chi.Mux, l *zerolog.Logger, defaults *execcontext.Defaults, cgroupManager cgroups.Manager) *Service {
|
||||
service := newService(l, defaults, cgroupManager)
|
||||
|
||||
interceptors := connect.WithInterceptors(logs.NewUnaryLogInterceptor(l))
|
||||
|
||||
path, h := spec.NewProcessHandler(service, interceptors)
|
||||
|
||||
server.Mount(path, h)
|
||||
|
||||
return service
|
||||
}
|
||||
|
||||
func (s *Service) getProcess(selector *rpc.ProcessSelector) (*handler.Handler, error) {
|
||||
var proc *handler.Handler
|
||||
|
||||
switch selector.GetSelector().(type) {
|
||||
case *rpc.ProcessSelector_Pid:
|
||||
p, ok := s.processes.Load(selector.GetPid())
|
||||
if !ok {
|
||||
return nil, connect.NewError(connect.CodeNotFound, fmt.Errorf("process with pid %d not found", selector.GetPid()))
|
||||
}
|
||||
|
||||
proc = p
|
||||
case *rpc.ProcessSelector_Tag:
|
||||
tag := selector.GetTag()
|
||||
|
||||
s.processes.Range(func(_ uint32, value *handler.Handler) bool {
|
||||
if value.Tag == nil {
|
||||
return true
|
||||
}
|
||||
|
||||
if *value.Tag == tag {
|
||||
proc = value
|
||||
|
||||
return true
|
||||
}
|
||||
|
||||
return false
|
||||
})
|
||||
|
||||
if proc == nil {
|
||||
return nil, connect.NewError(connect.CodeNotFound, fmt.Errorf("process with tag %s not found", tag))
|
||||
}
|
||||
|
||||
default:
|
||||
return nil, connect.NewError(connect.CodeUnimplemented, fmt.Errorf("invalid input type %T", selector))
|
||||
}
|
||||
|
||||
return proc, nil
|
||||
}
|
||||
Reference in New Issue
Block a user