pluginhostsdk

type DispatchRequest struct {
	PackageID          string
	PackageVersion     string
	RouteGeneration    uint64
	RequestID          string
	IdempotencyKey     string
	RouteID            string
	Method             string
	RequestBody        []byte
	PrincipalJSON      []byte
	Metadata           RequestMetadata
	BridgeCapability   []byte
	DeadlineUnixMillis int64
}

type DispatchResponse struct {
	StatusCode   uint32
	ResponseBody []byte
	Headers      []Header
	OperationID  string
	FailureCode  string
}

type DrainResponse struct {
	Drained  bool
	InFlight uint64
}

type Header struct {
	Name  string
	Value string
}

type HealthResponse struct {
	Healthy     bool
	LeaseID     string
	DetailsJSON string
}

type MigrationRequest struct {
	PackageID          string
	PackageVersion     string
	MigrationID        string
	Checkpoint         string
	RouteGeneration    uint64
	BridgeCapability   []byte
	DeadlineUnixMillis int64
}

type MigrationResponse struct {
	Checkpoint       string
	ValidationDigest string
	Complete         bool
	FailureCode      string
}

type NativeHandler func(context.Context, NativeRequest) (NativeResponse, error)

type NativeRequest struct {
	RouteID string
	Method  string
	// Body is the request body. For the routes the kernel lists in
	// config/node-secret-fields.json, every node secret in it is a sealed
	// handle (v2compat.SealedHandlePrefix) that only the kernel resolves.
	Body      []byte
	Principal Principal
	Metadata  RequestMetadata
	// Binding is the request binding a KernelNodeOps call passes as
	// RequestBinding.bridge_capability: the kernel resolves the request's
	// sealed handles, and mints the handles its answer shows, only for it.
	// It is nil in a shadow run, which must not act, and whose handles are
	// never resolved.
	Binding []byte
}

type NativeResponse struct {
	StatusCode uint32
	Body       []byte
	Headers    []Header
}

type Package interface {
	Dispatch(context.Context, DispatchRequest) (DispatchResponse, error)
	Migrate(context.Context, MigrationRequest) (MigrationResponse, error)
	Health(context.Context) (HealthResponse, error)
	Drain(context.Context) (DrainResponse, error)
}

type Principal struct {
	ActorID   uint   `json:"actor_id"`
	Admin     bool   `json:"admin"`
	PackageID string `json:"package_id"`
}

type RequestMetadata struct {
	Path                             string              `json:"path,omitempty"`
	Query                            map[string][]string `json:"query,omitempty"`
	Headers                          map[string][]string `json:"headers,omitempty"`
	PathParams                       map[string]string   `json:"path_params,omitempty"`
	ClientIP                         string              `json:"client_ip,omitempty"`
	UserAgent                        string              `json:"user_agent,omitempty"`
	NodeID                           uint                `json:"node_id,omitempty"`
	TrustedAgentWebSocketAuth        bool                `json:"trusted_agent_websocket_auth,omitempty"`
	TrustedAgentWebSocketForwardNode bool                `json:"trusted_agent_websocket_forward_node,omitempty"`
	Scheme                           string              `json:"-"`
	Host                             string              `json:"-"`
}

type ResumablePackage interface {
	Resume(context.Context) error
}

type Router struct {
	// contains filtered or unexported fields
}

type RouterBridge interface {
	Invoke(ctx context.Context, capability []byte, operation string, payload []byte) (packagebridgesdk.Response, error)
	GetPackageConfig(ctx context.Context) (packagebridgesdk.PackageConfig, error)
}

type RouterConfig struct {
	PackageID string
	LeaseID   string
	Bridge    RouterBridge
	// Native maps route ids to package-native implementations. Routes without
	// one always run in legacy mode, whatever the configuration says.
	Native map[string]NativeHandler
	// AllowRoute, when set, rejects dispatch for route ids it returns false for.
	AllowRoute func(routeID string) bool
	// MigrationOperation maps a migration id to its bridge operation; nil uses
	// "migration.<package>.<id>". It returns false for unsupported migrations.
	MigrationOperation func(migrationID string) (string, bool)
	// IndexMigration runs the package's own migration index when the kernel
	// sends a migration id with IndexMigrationPrefix, which it does only for
	// packages that declare kernel.storage.v1. It needs no bridge capability;
	// packagestoresdk.IndexMigrator implements it.
	IndexMigration func(ctx context.Context, request MigrationRequest) (MigrationResponse, error)
	// Compare decides whether a shadow run matched; nil compares the status
	// code and v2compat-normalized bodies.
	Compare func(legacy, native NativeResponse) bool

	PollInterval      time.Duration
	ShadowTimeout     time.Duration
	ShadowConcurrency int
	Logf              func(format string, args ...any)
	Now               func() time.Time
}

type RouterWebSocketBridge interface {
	OpenWebSocket(ctx context.Context, capability []byte, operation string) (packagebridgesdk.WebSocketStream, error)
}

type Server struct {
	pluginhostv1.UnimplementedControlPackageHostServer
	// contains filtered or unexported fields
}

type ServerConfig struct {
	PackageID      string
	PackageVersion string
	// MaxResponseBytes bounds DispatchResponse.ResponseBody; the whole
	// encoded response may additionally use ResponseEnvelopeBytes. Hosts
	// should set it from MaxResponseBytesFromEnvironment. Zero selects
	// DefaultMaxResponseBytes.
	MaxResponseBytes int
}

type WebSocketClose struct {
	Code   uint32
	Reason string
}

type WebSocketFrame struct {
	Data  []byte
	Close *WebSocketClose
}

type WebSocketOpen struct {
	PackageID          string
	PackageVersion     string
	RouteGeneration    uint64
	RouteID            string
	PrincipalJSON      []byte
	Metadata           RequestMetadata
	RequestID          string
	IdempotencyKey     string
	DeadlineUnixMillis int64
	BridgeCapability   []byte
}

type WebSocketPackage interface {
	OpenWebSocket(context.Context, WebSocketOpen, WebSocketStream) error
}

type WebSocketStream interface {
	Recv() (WebSocketFrame, error)
	Send(WebSocketFrame) error
}