1
0
forked from wrenn/wrenn

Prototype with single host server and no admin panel (#2)

Reviewed-on: wrenn/sandbox#2
Co-authored-by: pptx704 <rafeed@omukk.dev>
Co-committed-by: pptx704 <rafeed@omukk.dev>
This commit is contained in:
2026-03-22 21:01:23 +00:00
committed by Rafeed M. Bhuiyan
parent bd78cc068c
commit 32e5a5a715
293 changed files with 46885 additions and 1033 deletions

View File

@ -0,0 +1,13 @@
version: v2
plugins:
- protoc_builtin: go
out: gen
opt: paths=source_relative
- local: protoc-gen-connect-go
out: gen
opt: paths=source_relative
managed:
enabled: true
override:
- file_option: go_package_prefix
value: git.omukk.dev/wrenn/sandbox/proto/hostagent/gen

3
proto/hostagent/buf.yaml Normal file
View File

@ -0,0 +1,3 @@
version: v2
modules:
- path: .

File diff suppressed because it is too large Load Diff

View File

@ -0,0 +1,517 @@
// Code generated by protoc-gen-connect-go. DO NOT EDIT.
//
// Source: hostagent.proto
package hostagentv1connect
import (
connect "connectrpc.com/connect"
context "context"
errors "errors"
gen "git.omukk.dev/wrenn/sandbox/proto/hostagent/gen"
http "net/http"
strings "strings"
)
// This is a compile-time assertion to ensure that this generated file and the connect package are
// compatible. If you get a compiler error that this constant is not defined, this code was
// generated with a version of connect newer than the one compiled into your binary. You can fix the
// problem by either regenerating this code with an older version of connect or updating the connect
// version compiled into your binary.
const _ = connect.IsAtLeastVersion1_13_0
const (
// HostAgentServiceName is the fully-qualified name of the HostAgentService service.
HostAgentServiceName = "hostagent.v1.HostAgentService"
)
// These constants are the fully-qualified names of the RPCs defined in this package. They're
// exposed at runtime as Spec.Procedure and as the final two segments of the HTTP route.
//
// Note that these are different from the fully-qualified method names used by
// google.golang.org/protobuf/reflect/protoreflect. To convert from these constants to
// reflection-formatted method names, remove the leading slash and convert the remaining slash to a
// period.
const (
// HostAgentServiceCreateSandboxProcedure is the fully-qualified name of the HostAgentService's
// CreateSandbox RPC.
HostAgentServiceCreateSandboxProcedure = "/hostagent.v1.HostAgentService/CreateSandbox"
// HostAgentServiceDestroySandboxProcedure is the fully-qualified name of the HostAgentService's
// DestroySandbox RPC.
HostAgentServiceDestroySandboxProcedure = "/hostagent.v1.HostAgentService/DestroySandbox"
// HostAgentServicePauseSandboxProcedure is the fully-qualified name of the HostAgentService's
// PauseSandbox RPC.
HostAgentServicePauseSandboxProcedure = "/hostagent.v1.HostAgentService/PauseSandbox"
// HostAgentServiceResumeSandboxProcedure is the fully-qualified name of the HostAgentService's
// ResumeSandbox RPC.
HostAgentServiceResumeSandboxProcedure = "/hostagent.v1.HostAgentService/ResumeSandbox"
// HostAgentServiceExecProcedure is the fully-qualified name of the HostAgentService's Exec RPC.
HostAgentServiceExecProcedure = "/hostagent.v1.HostAgentService/Exec"
// HostAgentServiceListSandboxesProcedure is the fully-qualified name of the HostAgentService's
// ListSandboxes RPC.
HostAgentServiceListSandboxesProcedure = "/hostagent.v1.HostAgentService/ListSandboxes"
// HostAgentServiceWriteFileProcedure is the fully-qualified name of the HostAgentService's
// WriteFile RPC.
HostAgentServiceWriteFileProcedure = "/hostagent.v1.HostAgentService/WriteFile"
// HostAgentServiceReadFileProcedure is the fully-qualified name of the HostAgentService's ReadFile
// RPC.
HostAgentServiceReadFileProcedure = "/hostagent.v1.HostAgentService/ReadFile"
// HostAgentServiceCreateSnapshotProcedure is the fully-qualified name of the HostAgentService's
// CreateSnapshot RPC.
HostAgentServiceCreateSnapshotProcedure = "/hostagent.v1.HostAgentService/CreateSnapshot"
// HostAgentServiceDeleteSnapshotProcedure is the fully-qualified name of the HostAgentService's
// DeleteSnapshot RPC.
HostAgentServiceDeleteSnapshotProcedure = "/hostagent.v1.HostAgentService/DeleteSnapshot"
// HostAgentServiceExecStreamProcedure is the fully-qualified name of the HostAgentService's
// ExecStream RPC.
HostAgentServiceExecStreamProcedure = "/hostagent.v1.HostAgentService/ExecStream"
// HostAgentServiceWriteFileStreamProcedure is the fully-qualified name of the HostAgentService's
// WriteFileStream RPC.
HostAgentServiceWriteFileStreamProcedure = "/hostagent.v1.HostAgentService/WriteFileStream"
// HostAgentServiceReadFileStreamProcedure is the fully-qualified name of the HostAgentService's
// ReadFileStream RPC.
HostAgentServiceReadFileStreamProcedure = "/hostagent.v1.HostAgentService/ReadFileStream"
// HostAgentServicePingSandboxProcedure is the fully-qualified name of the HostAgentService's
// PingSandbox RPC.
HostAgentServicePingSandboxProcedure = "/hostagent.v1.HostAgentService/PingSandbox"
)
// HostAgentServiceClient is a client for the hostagent.v1.HostAgentService service.
type HostAgentServiceClient interface {
// CreateSandbox boots a new microVM with the given configuration.
CreateSandbox(context.Context, *connect.Request[gen.CreateSandboxRequest]) (*connect.Response[gen.CreateSandboxResponse], error)
// DestroySandbox stops and cleans up a sandbox (VM, network, rootfs).
DestroySandbox(context.Context, *connect.Request[gen.DestroySandboxRequest]) (*connect.Response[gen.DestroySandboxResponse], error)
// PauseSandbox pauses a running sandbox's VM.
PauseSandbox(context.Context, *connect.Request[gen.PauseSandboxRequest]) (*connect.Response[gen.PauseSandboxResponse], error)
// ResumeSandbox resumes a paused sandbox's VM.
ResumeSandbox(context.Context, *connect.Request[gen.ResumeSandboxRequest]) (*connect.Response[gen.ResumeSandboxResponse], error)
// Exec runs a command inside a sandbox and returns the collected output.
Exec(context.Context, *connect.Request[gen.ExecRequest]) (*connect.Response[gen.ExecResponse], error)
// ListSandboxes returns all sandboxes managed by this host agent.
ListSandboxes(context.Context, *connect.Request[gen.ListSandboxesRequest]) (*connect.Response[gen.ListSandboxesResponse], error)
// WriteFile writes content to a file inside a sandbox.
WriteFile(context.Context, *connect.Request[gen.WriteFileRequest]) (*connect.Response[gen.WriteFileResponse], error)
// ReadFile reads a file from inside a sandbox.
ReadFile(context.Context, *connect.Request[gen.ReadFileRequest]) (*connect.Response[gen.ReadFileResponse], error)
// CreateSnapshot pauses a sandbox, takes a snapshot, stores it as a reusable
// template, and destroys the sandbox.
CreateSnapshot(context.Context, *connect.Request[gen.CreateSnapshotRequest]) (*connect.Response[gen.CreateSnapshotResponse], error)
// DeleteSnapshot removes a snapshot template from disk.
DeleteSnapshot(context.Context, *connect.Request[gen.DeleteSnapshotRequest]) (*connect.Response[gen.DeleteSnapshotResponse], error)
// ExecStream runs a command inside a sandbox and streams output events as they arrive.
ExecStream(context.Context, *connect.Request[gen.ExecStreamRequest]) (*connect.ServerStreamForClient[gen.ExecStreamResponse], error)
// WriteFileStream writes a file to a sandbox using chunked streaming.
// First message must contain metadata (sandbox_id, path). Subsequent messages contain data chunks.
WriteFileStream(context.Context) *connect.ClientStreamForClient[gen.WriteFileStreamRequest, gen.WriteFileStreamResponse]
// ReadFileStream reads a file from a sandbox and streams it back in chunks.
ReadFileStream(context.Context, *connect.Request[gen.ReadFileStreamRequest]) (*connect.ServerStreamForClient[gen.ReadFileStreamResponse], error)
// PingSandbox resets the inactivity timer for a running sandbox.
PingSandbox(context.Context, *connect.Request[gen.PingSandboxRequest]) (*connect.Response[gen.PingSandboxResponse], error)
}
// NewHostAgentServiceClient constructs a client for the hostagent.v1.HostAgentService service. By
// default, it uses the Connect protocol with the binary Protobuf Codec, asks for gzipped responses,
// and sends uncompressed requests. To use the gRPC or gRPC-Web protocols, supply the
// connect.WithGRPC() or connect.WithGRPCWeb() options.
//
// The URL supplied here should be the base URL for the Connect or gRPC server (for example,
// http://api.acme.com or https://acme.com/grpc).
func NewHostAgentServiceClient(httpClient connect.HTTPClient, baseURL string, opts ...connect.ClientOption) HostAgentServiceClient {
baseURL = strings.TrimRight(baseURL, "/")
hostAgentServiceMethods := gen.File_hostagent_proto.Services().ByName("HostAgentService").Methods()
return &hostAgentServiceClient{
createSandbox: connect.NewClient[gen.CreateSandboxRequest, gen.CreateSandboxResponse](
httpClient,
baseURL+HostAgentServiceCreateSandboxProcedure,
connect.WithSchema(hostAgentServiceMethods.ByName("CreateSandbox")),
connect.WithClientOptions(opts...),
),
destroySandbox: connect.NewClient[gen.DestroySandboxRequest, gen.DestroySandboxResponse](
httpClient,
baseURL+HostAgentServiceDestroySandboxProcedure,
connect.WithSchema(hostAgentServiceMethods.ByName("DestroySandbox")),
connect.WithClientOptions(opts...),
),
pauseSandbox: connect.NewClient[gen.PauseSandboxRequest, gen.PauseSandboxResponse](
httpClient,
baseURL+HostAgentServicePauseSandboxProcedure,
connect.WithSchema(hostAgentServiceMethods.ByName("PauseSandbox")),
connect.WithClientOptions(opts...),
),
resumeSandbox: connect.NewClient[gen.ResumeSandboxRequest, gen.ResumeSandboxResponse](
httpClient,
baseURL+HostAgentServiceResumeSandboxProcedure,
connect.WithSchema(hostAgentServiceMethods.ByName("ResumeSandbox")),
connect.WithClientOptions(opts...),
),
exec: connect.NewClient[gen.ExecRequest, gen.ExecResponse](
httpClient,
baseURL+HostAgentServiceExecProcedure,
connect.WithSchema(hostAgentServiceMethods.ByName("Exec")),
connect.WithClientOptions(opts...),
),
listSandboxes: connect.NewClient[gen.ListSandboxesRequest, gen.ListSandboxesResponse](
httpClient,
baseURL+HostAgentServiceListSandboxesProcedure,
connect.WithSchema(hostAgentServiceMethods.ByName("ListSandboxes")),
connect.WithClientOptions(opts...),
),
writeFile: connect.NewClient[gen.WriteFileRequest, gen.WriteFileResponse](
httpClient,
baseURL+HostAgentServiceWriteFileProcedure,
connect.WithSchema(hostAgentServiceMethods.ByName("WriteFile")),
connect.WithClientOptions(opts...),
),
readFile: connect.NewClient[gen.ReadFileRequest, gen.ReadFileResponse](
httpClient,
baseURL+HostAgentServiceReadFileProcedure,
connect.WithSchema(hostAgentServiceMethods.ByName("ReadFile")),
connect.WithClientOptions(opts...),
),
createSnapshot: connect.NewClient[gen.CreateSnapshotRequest, gen.CreateSnapshotResponse](
httpClient,
baseURL+HostAgentServiceCreateSnapshotProcedure,
connect.WithSchema(hostAgentServiceMethods.ByName("CreateSnapshot")),
connect.WithClientOptions(opts...),
),
deleteSnapshot: connect.NewClient[gen.DeleteSnapshotRequest, gen.DeleteSnapshotResponse](
httpClient,
baseURL+HostAgentServiceDeleteSnapshotProcedure,
connect.WithSchema(hostAgentServiceMethods.ByName("DeleteSnapshot")),
connect.WithClientOptions(opts...),
),
execStream: connect.NewClient[gen.ExecStreamRequest, gen.ExecStreamResponse](
httpClient,
baseURL+HostAgentServiceExecStreamProcedure,
connect.WithSchema(hostAgentServiceMethods.ByName("ExecStream")),
connect.WithClientOptions(opts...),
),
writeFileStream: connect.NewClient[gen.WriteFileStreamRequest, gen.WriteFileStreamResponse](
httpClient,
baseURL+HostAgentServiceWriteFileStreamProcedure,
connect.WithSchema(hostAgentServiceMethods.ByName("WriteFileStream")),
connect.WithClientOptions(opts...),
),
readFileStream: connect.NewClient[gen.ReadFileStreamRequest, gen.ReadFileStreamResponse](
httpClient,
baseURL+HostAgentServiceReadFileStreamProcedure,
connect.WithSchema(hostAgentServiceMethods.ByName("ReadFileStream")),
connect.WithClientOptions(opts...),
),
pingSandbox: connect.NewClient[gen.PingSandboxRequest, gen.PingSandboxResponse](
httpClient,
baseURL+HostAgentServicePingSandboxProcedure,
connect.WithSchema(hostAgentServiceMethods.ByName("PingSandbox")),
connect.WithClientOptions(opts...),
),
}
}
// hostAgentServiceClient implements HostAgentServiceClient.
type hostAgentServiceClient struct {
createSandbox *connect.Client[gen.CreateSandboxRequest, gen.CreateSandboxResponse]
destroySandbox *connect.Client[gen.DestroySandboxRequest, gen.DestroySandboxResponse]
pauseSandbox *connect.Client[gen.PauseSandboxRequest, gen.PauseSandboxResponse]
resumeSandbox *connect.Client[gen.ResumeSandboxRequest, gen.ResumeSandboxResponse]
exec *connect.Client[gen.ExecRequest, gen.ExecResponse]
listSandboxes *connect.Client[gen.ListSandboxesRequest, gen.ListSandboxesResponse]
writeFile *connect.Client[gen.WriteFileRequest, gen.WriteFileResponse]
readFile *connect.Client[gen.ReadFileRequest, gen.ReadFileResponse]
createSnapshot *connect.Client[gen.CreateSnapshotRequest, gen.CreateSnapshotResponse]
deleteSnapshot *connect.Client[gen.DeleteSnapshotRequest, gen.DeleteSnapshotResponse]
execStream *connect.Client[gen.ExecStreamRequest, gen.ExecStreamResponse]
writeFileStream *connect.Client[gen.WriteFileStreamRequest, gen.WriteFileStreamResponse]
readFileStream *connect.Client[gen.ReadFileStreamRequest, gen.ReadFileStreamResponse]
pingSandbox *connect.Client[gen.PingSandboxRequest, gen.PingSandboxResponse]
}
// CreateSandbox calls hostagent.v1.HostAgentService.CreateSandbox.
func (c *hostAgentServiceClient) CreateSandbox(ctx context.Context, req *connect.Request[gen.CreateSandboxRequest]) (*connect.Response[gen.CreateSandboxResponse], error) {
return c.createSandbox.CallUnary(ctx, req)
}
// DestroySandbox calls hostagent.v1.HostAgentService.DestroySandbox.
func (c *hostAgentServiceClient) DestroySandbox(ctx context.Context, req *connect.Request[gen.DestroySandboxRequest]) (*connect.Response[gen.DestroySandboxResponse], error) {
return c.destroySandbox.CallUnary(ctx, req)
}
// PauseSandbox calls hostagent.v1.HostAgentService.PauseSandbox.
func (c *hostAgentServiceClient) PauseSandbox(ctx context.Context, req *connect.Request[gen.PauseSandboxRequest]) (*connect.Response[gen.PauseSandboxResponse], error) {
return c.pauseSandbox.CallUnary(ctx, req)
}
// ResumeSandbox calls hostagent.v1.HostAgentService.ResumeSandbox.
func (c *hostAgentServiceClient) ResumeSandbox(ctx context.Context, req *connect.Request[gen.ResumeSandboxRequest]) (*connect.Response[gen.ResumeSandboxResponse], error) {
return c.resumeSandbox.CallUnary(ctx, req)
}
// Exec calls hostagent.v1.HostAgentService.Exec.
func (c *hostAgentServiceClient) Exec(ctx context.Context, req *connect.Request[gen.ExecRequest]) (*connect.Response[gen.ExecResponse], error) {
return c.exec.CallUnary(ctx, req)
}
// ListSandboxes calls hostagent.v1.HostAgentService.ListSandboxes.
func (c *hostAgentServiceClient) ListSandboxes(ctx context.Context, req *connect.Request[gen.ListSandboxesRequest]) (*connect.Response[gen.ListSandboxesResponse], error) {
return c.listSandboxes.CallUnary(ctx, req)
}
// WriteFile calls hostagent.v1.HostAgentService.WriteFile.
func (c *hostAgentServiceClient) WriteFile(ctx context.Context, req *connect.Request[gen.WriteFileRequest]) (*connect.Response[gen.WriteFileResponse], error) {
return c.writeFile.CallUnary(ctx, req)
}
// ReadFile calls hostagent.v1.HostAgentService.ReadFile.
func (c *hostAgentServiceClient) ReadFile(ctx context.Context, req *connect.Request[gen.ReadFileRequest]) (*connect.Response[gen.ReadFileResponse], error) {
return c.readFile.CallUnary(ctx, req)
}
// CreateSnapshot calls hostagent.v1.HostAgentService.CreateSnapshot.
func (c *hostAgentServiceClient) CreateSnapshot(ctx context.Context, req *connect.Request[gen.CreateSnapshotRequest]) (*connect.Response[gen.CreateSnapshotResponse], error) {
return c.createSnapshot.CallUnary(ctx, req)
}
// DeleteSnapshot calls hostagent.v1.HostAgentService.DeleteSnapshot.
func (c *hostAgentServiceClient) DeleteSnapshot(ctx context.Context, req *connect.Request[gen.DeleteSnapshotRequest]) (*connect.Response[gen.DeleteSnapshotResponse], error) {
return c.deleteSnapshot.CallUnary(ctx, req)
}
// ExecStream calls hostagent.v1.HostAgentService.ExecStream.
func (c *hostAgentServiceClient) ExecStream(ctx context.Context, req *connect.Request[gen.ExecStreamRequest]) (*connect.ServerStreamForClient[gen.ExecStreamResponse], error) {
return c.execStream.CallServerStream(ctx, req)
}
// WriteFileStream calls hostagent.v1.HostAgentService.WriteFileStream.
func (c *hostAgentServiceClient) WriteFileStream(ctx context.Context) *connect.ClientStreamForClient[gen.WriteFileStreamRequest, gen.WriteFileStreamResponse] {
return c.writeFileStream.CallClientStream(ctx)
}
// ReadFileStream calls hostagent.v1.HostAgentService.ReadFileStream.
func (c *hostAgentServiceClient) ReadFileStream(ctx context.Context, req *connect.Request[gen.ReadFileStreamRequest]) (*connect.ServerStreamForClient[gen.ReadFileStreamResponse], error) {
return c.readFileStream.CallServerStream(ctx, req)
}
// PingSandbox calls hostagent.v1.HostAgentService.PingSandbox.
func (c *hostAgentServiceClient) PingSandbox(ctx context.Context, req *connect.Request[gen.PingSandboxRequest]) (*connect.Response[gen.PingSandboxResponse], error) {
return c.pingSandbox.CallUnary(ctx, req)
}
// HostAgentServiceHandler is an implementation of the hostagent.v1.HostAgentService service.
type HostAgentServiceHandler interface {
// CreateSandbox boots a new microVM with the given configuration.
CreateSandbox(context.Context, *connect.Request[gen.CreateSandboxRequest]) (*connect.Response[gen.CreateSandboxResponse], error)
// DestroySandbox stops and cleans up a sandbox (VM, network, rootfs).
DestroySandbox(context.Context, *connect.Request[gen.DestroySandboxRequest]) (*connect.Response[gen.DestroySandboxResponse], error)
// PauseSandbox pauses a running sandbox's VM.
PauseSandbox(context.Context, *connect.Request[gen.PauseSandboxRequest]) (*connect.Response[gen.PauseSandboxResponse], error)
// ResumeSandbox resumes a paused sandbox's VM.
ResumeSandbox(context.Context, *connect.Request[gen.ResumeSandboxRequest]) (*connect.Response[gen.ResumeSandboxResponse], error)
// Exec runs a command inside a sandbox and returns the collected output.
Exec(context.Context, *connect.Request[gen.ExecRequest]) (*connect.Response[gen.ExecResponse], error)
// ListSandboxes returns all sandboxes managed by this host agent.
ListSandboxes(context.Context, *connect.Request[gen.ListSandboxesRequest]) (*connect.Response[gen.ListSandboxesResponse], error)
// WriteFile writes content to a file inside a sandbox.
WriteFile(context.Context, *connect.Request[gen.WriteFileRequest]) (*connect.Response[gen.WriteFileResponse], error)
// ReadFile reads a file from inside a sandbox.
ReadFile(context.Context, *connect.Request[gen.ReadFileRequest]) (*connect.Response[gen.ReadFileResponse], error)
// CreateSnapshot pauses a sandbox, takes a snapshot, stores it as a reusable
// template, and destroys the sandbox.
CreateSnapshot(context.Context, *connect.Request[gen.CreateSnapshotRequest]) (*connect.Response[gen.CreateSnapshotResponse], error)
// DeleteSnapshot removes a snapshot template from disk.
DeleteSnapshot(context.Context, *connect.Request[gen.DeleteSnapshotRequest]) (*connect.Response[gen.DeleteSnapshotResponse], error)
// ExecStream runs a command inside a sandbox and streams output events as they arrive.
ExecStream(context.Context, *connect.Request[gen.ExecStreamRequest], *connect.ServerStream[gen.ExecStreamResponse]) error
// WriteFileStream writes a file to a sandbox using chunked streaming.
// First message must contain metadata (sandbox_id, path). Subsequent messages contain data chunks.
WriteFileStream(context.Context, *connect.ClientStream[gen.WriteFileStreamRequest]) (*connect.Response[gen.WriteFileStreamResponse], error)
// ReadFileStream reads a file from a sandbox and streams it back in chunks.
ReadFileStream(context.Context, *connect.Request[gen.ReadFileStreamRequest], *connect.ServerStream[gen.ReadFileStreamResponse]) error
// PingSandbox resets the inactivity timer for a running sandbox.
PingSandbox(context.Context, *connect.Request[gen.PingSandboxRequest]) (*connect.Response[gen.PingSandboxResponse], error)
}
// NewHostAgentServiceHandler builds an HTTP handler from the service implementation. It returns the
// path on which to mount the handler and the handler itself.
//
// By default, handlers support the Connect, gRPC, and gRPC-Web protocols with the binary Protobuf
// and JSON codecs. They also support gzip compression.
func NewHostAgentServiceHandler(svc HostAgentServiceHandler, opts ...connect.HandlerOption) (string, http.Handler) {
hostAgentServiceMethods := gen.File_hostagent_proto.Services().ByName("HostAgentService").Methods()
hostAgentServiceCreateSandboxHandler := connect.NewUnaryHandler(
HostAgentServiceCreateSandboxProcedure,
svc.CreateSandbox,
connect.WithSchema(hostAgentServiceMethods.ByName("CreateSandbox")),
connect.WithHandlerOptions(opts...),
)
hostAgentServiceDestroySandboxHandler := connect.NewUnaryHandler(
HostAgentServiceDestroySandboxProcedure,
svc.DestroySandbox,
connect.WithSchema(hostAgentServiceMethods.ByName("DestroySandbox")),
connect.WithHandlerOptions(opts...),
)
hostAgentServicePauseSandboxHandler := connect.NewUnaryHandler(
HostAgentServicePauseSandboxProcedure,
svc.PauseSandbox,
connect.WithSchema(hostAgentServiceMethods.ByName("PauseSandbox")),
connect.WithHandlerOptions(opts...),
)
hostAgentServiceResumeSandboxHandler := connect.NewUnaryHandler(
HostAgentServiceResumeSandboxProcedure,
svc.ResumeSandbox,
connect.WithSchema(hostAgentServiceMethods.ByName("ResumeSandbox")),
connect.WithHandlerOptions(opts...),
)
hostAgentServiceExecHandler := connect.NewUnaryHandler(
HostAgentServiceExecProcedure,
svc.Exec,
connect.WithSchema(hostAgentServiceMethods.ByName("Exec")),
connect.WithHandlerOptions(opts...),
)
hostAgentServiceListSandboxesHandler := connect.NewUnaryHandler(
HostAgentServiceListSandboxesProcedure,
svc.ListSandboxes,
connect.WithSchema(hostAgentServiceMethods.ByName("ListSandboxes")),
connect.WithHandlerOptions(opts...),
)
hostAgentServiceWriteFileHandler := connect.NewUnaryHandler(
HostAgentServiceWriteFileProcedure,
svc.WriteFile,
connect.WithSchema(hostAgentServiceMethods.ByName("WriteFile")),
connect.WithHandlerOptions(opts...),
)
hostAgentServiceReadFileHandler := connect.NewUnaryHandler(
HostAgentServiceReadFileProcedure,
svc.ReadFile,
connect.WithSchema(hostAgentServiceMethods.ByName("ReadFile")),
connect.WithHandlerOptions(opts...),
)
hostAgentServiceCreateSnapshotHandler := connect.NewUnaryHandler(
HostAgentServiceCreateSnapshotProcedure,
svc.CreateSnapshot,
connect.WithSchema(hostAgentServiceMethods.ByName("CreateSnapshot")),
connect.WithHandlerOptions(opts...),
)
hostAgentServiceDeleteSnapshotHandler := connect.NewUnaryHandler(
HostAgentServiceDeleteSnapshotProcedure,
svc.DeleteSnapshot,
connect.WithSchema(hostAgentServiceMethods.ByName("DeleteSnapshot")),
connect.WithHandlerOptions(opts...),
)
hostAgentServiceExecStreamHandler := connect.NewServerStreamHandler(
HostAgentServiceExecStreamProcedure,
svc.ExecStream,
connect.WithSchema(hostAgentServiceMethods.ByName("ExecStream")),
connect.WithHandlerOptions(opts...),
)
hostAgentServiceWriteFileStreamHandler := connect.NewClientStreamHandler(
HostAgentServiceWriteFileStreamProcedure,
svc.WriteFileStream,
connect.WithSchema(hostAgentServiceMethods.ByName("WriteFileStream")),
connect.WithHandlerOptions(opts...),
)
hostAgentServiceReadFileStreamHandler := connect.NewServerStreamHandler(
HostAgentServiceReadFileStreamProcedure,
svc.ReadFileStream,
connect.WithSchema(hostAgentServiceMethods.ByName("ReadFileStream")),
connect.WithHandlerOptions(opts...),
)
hostAgentServicePingSandboxHandler := connect.NewUnaryHandler(
HostAgentServicePingSandboxProcedure,
svc.PingSandbox,
connect.WithSchema(hostAgentServiceMethods.ByName("PingSandbox")),
connect.WithHandlerOptions(opts...),
)
return "/hostagent.v1.HostAgentService/", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
switch r.URL.Path {
case HostAgentServiceCreateSandboxProcedure:
hostAgentServiceCreateSandboxHandler.ServeHTTP(w, r)
case HostAgentServiceDestroySandboxProcedure:
hostAgentServiceDestroySandboxHandler.ServeHTTP(w, r)
case HostAgentServicePauseSandboxProcedure:
hostAgentServicePauseSandboxHandler.ServeHTTP(w, r)
case HostAgentServiceResumeSandboxProcedure:
hostAgentServiceResumeSandboxHandler.ServeHTTP(w, r)
case HostAgentServiceExecProcedure:
hostAgentServiceExecHandler.ServeHTTP(w, r)
case HostAgentServiceListSandboxesProcedure:
hostAgentServiceListSandboxesHandler.ServeHTTP(w, r)
case HostAgentServiceWriteFileProcedure:
hostAgentServiceWriteFileHandler.ServeHTTP(w, r)
case HostAgentServiceReadFileProcedure:
hostAgentServiceReadFileHandler.ServeHTTP(w, r)
case HostAgentServiceCreateSnapshotProcedure:
hostAgentServiceCreateSnapshotHandler.ServeHTTP(w, r)
case HostAgentServiceDeleteSnapshotProcedure:
hostAgentServiceDeleteSnapshotHandler.ServeHTTP(w, r)
case HostAgentServiceExecStreamProcedure:
hostAgentServiceExecStreamHandler.ServeHTTP(w, r)
case HostAgentServiceWriteFileStreamProcedure:
hostAgentServiceWriteFileStreamHandler.ServeHTTP(w, r)
case HostAgentServiceReadFileStreamProcedure:
hostAgentServiceReadFileStreamHandler.ServeHTTP(w, r)
case HostAgentServicePingSandboxProcedure:
hostAgentServicePingSandboxHandler.ServeHTTP(w, r)
default:
http.NotFound(w, r)
}
})
}
// UnimplementedHostAgentServiceHandler returns CodeUnimplemented from all methods.
type UnimplementedHostAgentServiceHandler struct{}
func (UnimplementedHostAgentServiceHandler) CreateSandbox(context.Context, *connect.Request[gen.CreateSandboxRequest]) (*connect.Response[gen.CreateSandboxResponse], error) {
return nil, connect.NewError(connect.CodeUnimplemented, errors.New("hostagent.v1.HostAgentService.CreateSandbox is not implemented"))
}
func (UnimplementedHostAgentServiceHandler) DestroySandbox(context.Context, *connect.Request[gen.DestroySandboxRequest]) (*connect.Response[gen.DestroySandboxResponse], error) {
return nil, connect.NewError(connect.CodeUnimplemented, errors.New("hostagent.v1.HostAgentService.DestroySandbox is not implemented"))
}
func (UnimplementedHostAgentServiceHandler) PauseSandbox(context.Context, *connect.Request[gen.PauseSandboxRequest]) (*connect.Response[gen.PauseSandboxResponse], error) {
return nil, connect.NewError(connect.CodeUnimplemented, errors.New("hostagent.v1.HostAgentService.PauseSandbox is not implemented"))
}
func (UnimplementedHostAgentServiceHandler) ResumeSandbox(context.Context, *connect.Request[gen.ResumeSandboxRequest]) (*connect.Response[gen.ResumeSandboxResponse], error) {
return nil, connect.NewError(connect.CodeUnimplemented, errors.New("hostagent.v1.HostAgentService.ResumeSandbox is not implemented"))
}
func (UnimplementedHostAgentServiceHandler) Exec(context.Context, *connect.Request[gen.ExecRequest]) (*connect.Response[gen.ExecResponse], error) {
return nil, connect.NewError(connect.CodeUnimplemented, errors.New("hostagent.v1.HostAgentService.Exec is not implemented"))
}
func (UnimplementedHostAgentServiceHandler) ListSandboxes(context.Context, *connect.Request[gen.ListSandboxesRequest]) (*connect.Response[gen.ListSandboxesResponse], error) {
return nil, connect.NewError(connect.CodeUnimplemented, errors.New("hostagent.v1.HostAgentService.ListSandboxes is not implemented"))
}
func (UnimplementedHostAgentServiceHandler) WriteFile(context.Context, *connect.Request[gen.WriteFileRequest]) (*connect.Response[gen.WriteFileResponse], error) {
return nil, connect.NewError(connect.CodeUnimplemented, errors.New("hostagent.v1.HostAgentService.WriteFile is not implemented"))
}
func (UnimplementedHostAgentServiceHandler) ReadFile(context.Context, *connect.Request[gen.ReadFileRequest]) (*connect.Response[gen.ReadFileResponse], error) {
return nil, connect.NewError(connect.CodeUnimplemented, errors.New("hostagent.v1.HostAgentService.ReadFile is not implemented"))
}
func (UnimplementedHostAgentServiceHandler) CreateSnapshot(context.Context, *connect.Request[gen.CreateSnapshotRequest]) (*connect.Response[gen.CreateSnapshotResponse], error) {
return nil, connect.NewError(connect.CodeUnimplemented, errors.New("hostagent.v1.HostAgentService.CreateSnapshot is not implemented"))
}
func (UnimplementedHostAgentServiceHandler) DeleteSnapshot(context.Context, *connect.Request[gen.DeleteSnapshotRequest]) (*connect.Response[gen.DeleteSnapshotResponse], error) {
return nil, connect.NewError(connect.CodeUnimplemented, errors.New("hostagent.v1.HostAgentService.DeleteSnapshot is not implemented"))
}
func (UnimplementedHostAgentServiceHandler) ExecStream(context.Context, *connect.Request[gen.ExecStreamRequest], *connect.ServerStream[gen.ExecStreamResponse]) error {
return connect.NewError(connect.CodeUnimplemented, errors.New("hostagent.v1.HostAgentService.ExecStream is not implemented"))
}
func (UnimplementedHostAgentServiceHandler) WriteFileStream(context.Context, *connect.ClientStream[gen.WriteFileStreamRequest]) (*connect.Response[gen.WriteFileStreamResponse], error) {
return nil, connect.NewError(connect.CodeUnimplemented, errors.New("hostagent.v1.HostAgentService.WriteFileStream is not implemented"))
}
func (UnimplementedHostAgentServiceHandler) ReadFileStream(context.Context, *connect.Request[gen.ReadFileStreamRequest], *connect.ServerStream[gen.ReadFileStreamResponse]) error {
return connect.NewError(connect.CodeUnimplemented, errors.New("hostagent.v1.HostAgentService.ReadFileStream is not implemented"))
}
func (UnimplementedHostAgentServiceHandler) PingSandbox(context.Context, *connect.Request[gen.PingSandboxRequest]) (*connect.Response[gen.PingSandboxResponse], error) {
return nil, connect.NewError(connect.CodeUnimplemented, errors.New("hostagent.v1.HostAgentService.PingSandbox is not implemented"))
}

View File

@ -0,0 +1,238 @@
syntax = "proto3";
package hostagent.v1;
// HostAgentService manages sandbox VMs on a single physical host.
// The control plane calls these RPCs to orchestrate sandbox lifecycle.
service HostAgentService {
// CreateSandbox boots a new microVM with the given configuration.
rpc CreateSandbox(CreateSandboxRequest) returns (CreateSandboxResponse);
// DestroySandbox stops and cleans up a sandbox (VM, network, rootfs).
rpc DestroySandbox(DestroySandboxRequest) returns (DestroySandboxResponse);
// PauseSandbox pauses a running sandbox's VM.
rpc PauseSandbox(PauseSandboxRequest) returns (PauseSandboxResponse);
// ResumeSandbox resumes a paused sandbox's VM.
rpc ResumeSandbox(ResumeSandboxRequest) returns (ResumeSandboxResponse);
// Exec runs a command inside a sandbox and returns the collected output.
rpc Exec(ExecRequest) returns (ExecResponse);
// ListSandboxes returns all sandboxes managed by this host agent.
rpc ListSandboxes(ListSandboxesRequest) returns (ListSandboxesResponse);
// WriteFile writes content to a file inside a sandbox.
rpc WriteFile(WriteFileRequest) returns (WriteFileResponse);
// ReadFile reads a file from inside a sandbox.
rpc ReadFile(ReadFileRequest) returns (ReadFileResponse);
// CreateSnapshot pauses a sandbox, takes a snapshot, stores it as a reusable
// template, and destroys the sandbox.
rpc CreateSnapshot(CreateSnapshotRequest) returns (CreateSnapshotResponse);
// DeleteSnapshot removes a snapshot template from disk.
rpc DeleteSnapshot(DeleteSnapshotRequest) returns (DeleteSnapshotResponse);
// ExecStream runs a command inside a sandbox and streams output events as they arrive.
rpc ExecStream(ExecStreamRequest) returns (stream ExecStreamResponse);
// WriteFileStream writes a file to a sandbox using chunked streaming.
// First message must contain metadata (sandbox_id, path). Subsequent messages contain data chunks.
rpc WriteFileStream(stream WriteFileStreamRequest) returns (WriteFileStreamResponse);
// ReadFileStream reads a file from a sandbox and streams it back in chunks.
rpc ReadFileStream(ReadFileStreamRequest) returns (stream ReadFileStreamResponse);
// PingSandbox resets the inactivity timer for a running sandbox.
rpc PingSandbox(PingSandboxRequest) returns (PingSandboxResponse);
}
message CreateSandboxRequest {
// Sandbox ID assigned by the control plane. If empty, the host agent generates one.
string sandbox_id = 5;
// Template name (e.g., "minimal", "python311"). Determines base rootfs.
string template = 1;
// Number of virtual CPUs (default: 1).
int32 vcpus = 2;
// Memory in MB (default: 512).
int32 memory_mb = 3;
// TTL in seconds. Sandbox is auto-paused after this duration of
// inactivity. 0 means no auto-pause.
int32 timeout_sec = 4;
}
message CreateSandboxResponse {
string sandbox_id = 1;
string status = 2;
string host_ip = 3;
}
message DestroySandboxRequest {
string sandbox_id = 1;
}
message DestroySandboxResponse {}
message PauseSandboxRequest {
string sandbox_id = 1;
}
message PauseSandboxResponse {}
message ResumeSandboxRequest {
string sandbox_id = 1;
// TTL in seconds restored from the DB so the reaper can auto-pause
// the sandbox again after inactivity. 0 means no auto-pause.
int32 timeout_sec = 2;
}
message ResumeSandboxResponse {
string sandbox_id = 1;
string status = 2;
string host_ip = 3;
}
message CreateSnapshotRequest {
string sandbox_id = 1;
string name = 2;
}
message CreateSnapshotResponse {
string name = 1;
int64 size_bytes = 2;
}
message DeleteSnapshotRequest {
string name = 1;
}
message DeleteSnapshotResponse {}
message ExecRequest {
string sandbox_id = 1;
string cmd = 2;
repeated string args = 3;
// Timeout for the command in seconds (default: 30).
int32 timeout_sec = 4;
}
message ExecResponse {
bytes stdout = 1;
bytes stderr = 2;
int32 exit_code = 3;
}
message ListSandboxesRequest {}
message ListSandboxesResponse {
repeated SandboxInfo sandboxes = 1;
// IDs of sandboxes that were automatically paused by the TTL reaper
// since the last call. Drained on read.
repeated string auto_paused_sandbox_ids = 2;
}
message SandboxInfo {
string sandbox_id = 1;
string status = 2;
string template = 3;
int32 vcpus = 4;
int32 memory_mb = 5;
string host_ip = 6;
int64 created_at_unix = 7;
int64 last_active_at_unix = 8;
int32 timeout_sec = 9;
}
message WriteFileRequest {
string sandbox_id = 1;
string path = 2;
bytes content = 3;
}
message WriteFileResponse {}
message ReadFileRequest {
string sandbox_id = 1;
string path = 2;
}
message ReadFileResponse {
bytes content = 1;
}
// ── Streaming Exec ──────────────────────────────────────────────────
message ExecStreamRequest {
string sandbox_id = 1;
string cmd = 2;
repeated string args = 3;
int32 timeout_sec = 4;
}
message ExecStreamResponse {
oneof event {
ExecStreamStart start = 1;
ExecStreamData data = 2;
ExecStreamEnd end = 3;
}
}
message ExecStreamStart {
uint32 pid = 1;
}
message ExecStreamData {
oneof output {
bytes stdout = 1;
bytes stderr = 2;
}
}
message ExecStreamEnd {
int32 exit_code = 1;
string error = 2;
}
// ── Streaming File Transfer ─────────────────────────────────────────
message WriteFileStreamRequest {
oneof content {
WriteFileStreamMeta meta = 1;
bytes chunk = 2;
}
}
message WriteFileStreamMeta {
string sandbox_id = 1;
string path = 2;
}
message WriteFileStreamResponse {}
message ReadFileStreamRequest {
string sandbox_id = 1;
string path = 2;
}
message ReadFileStreamResponse {
bytes chunk = 1;
}
// ── Ping ────────────────────────────────────────────────────────────
message PingSandboxRequest {
string sandbox_id = 1;
}
message PingSandboxResponse {}