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:
2101
proto/hostagent/gen/hostagent.pb.go
Normal file
2101
proto/hostagent/gen/hostagent.pb.go
Normal file
File diff suppressed because it is too large
Load Diff
517
proto/hostagent/gen/hostagentv1connect/hostagent.connect.go
Normal file
517
proto/hostagent/gen/hostagentv1connect/hostagent.connect.go
Normal 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"))
|
||||
}
|
||||
Reference in New Issue
Block a user