rpc

type BidiStream[TIn, TOut any] interface {
	// Send sends a message to the client.
	Send(*TOut) error
	// Recv receives a message from the client.
	Recv() (*TIn, error)
	// Context returns the context for this stream.
	Context() context.Context
}

type BidiStreamHandler[TIn, TOut any] func(context.Context, BidiStream[TIn, TOut]) error

type BrotliCompressor struct{}

type ClientStream[T any] interface {
	// Recv receives a message from the client.
	Recv() (*T, error)
	// Context returns the context for this stream.
	Context() context.Context
}

type ClientStreamHandler[TIn, TOut any] func(context.Context, ClientStream[TIn]) (*TOut, error)

type Code string

type Compressor interface {
	Compress(data []byte) ([]byte, error)
	Decompress(data []byte) ([]byte, error)
	Name() string
}

type Error struct {
	Code    Code           `json:"code"`
	Message string         `json:"message"`
	Details map[string]any `json:"details,omitempty"`
}

type ErrorDetail struct {
	Type  string `json:"type"`
	Value any    `json:"value"`
}

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

type GzipCompressor struct{}

type Handler[TIn, TOut any] func(context.Context, *TIn) (*TOut, error)

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

type HandlerFunc func(context.Context, any) (any, error)

type HandlerInfo struct {
	HandlerValue reflect.Value
	HandlerType  reflect.Type
	InputType    reflect.Type
	OutputType   reflect.Type
	IsPointer    bool // Whether handler expects pointer input
}

type Interceptor interface {
	// Intercept wraps the handler call.
	Intercept(ctx context.Context, method string, req any, handler func(context.Context, any) (any, error)) (any, error)
}

type JSONRPCError struct {
	Code    int         `json:"code"`
	Message string      `json:"message"`
	Data    interface{} `json:"data,omitempty"`
}

type JSONRPCRequest struct {
	JSONRPC string          `json:"jsonrpc"`
	Method  string          `json:"method"`
	Params  json.RawMessage `json:"params,omitempty"`
	ID      interface{}     `json:"id,omitempty"` // Can be string, number, or null
}

type JSONRPCResponse struct {
	JSONRPC string          `json:"jsonrpc"`
	Result  json.RawMessage `json:"result,omitempty"`
	Error   *JSONRPCError   `json:"error,omitempty"`
	ID      interface{}     `json:"id"` // Must match the request ID
}

type LoggingInterceptor struct {
	Logger *log.Logger
}

type Method struct {
	Name       string
	Handler    any
	InputType  reflect.Type
	OutputType reflect.Type
	Options    MethodOptions
	StreamType StreamType // Type of streaming RPC

	// Protobuf type support - these are set if the types implement proto.Message
	ProtoInput  proto.Message // Optional: set if input type is a protobuf message
	ProtoOutput proto.Message // Optional: set if output type is a protobuf message
}

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

type MethodConfig struct {
	// Name identifies the methods to which this configuration applies.
	Name []MethodName `json:"name"`

	// Timeout for the method. Format: "1s", "100ms", etc.
	Timeout string `json:"timeout,omitempty"`

	// RetryPolicy for the method.
	RetryPolicy *RetryPolicy `json:"retryPolicy,omitempty"`
}

type MethodName struct {
	// Service name including proto package name. Required.
	Service string `json:"service"`

	// Method name. If empty, applies to all methods in the service.
	Method string `json:"method,omitempty"`
}

type MethodOptions struct {
	// Validate enables input validation for this method
	Validate *bool
	// Interceptors specific to this method
	Interceptors []Interceptor
	// Description is the method-level documentation
	Description string
}

type MetricsInterceptor struct {
	RequestCount  int64
	SuccessCount  int64
	FailureCount  int64
	TotalDuration time.Duration
}

type RecoveryInterceptor struct{}

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

type RetryPolicy struct {
	// MaxAttempts is the maximum number of attempts including the original request.
	// Must be greater than 1. Required.
	MaxAttempts int `json:"maxAttempts"`

	// InitialBackoff is the initial delay before the first retry.
	// Format: "0.1s", "100ms", etc.
	InitialBackoff string `json:"initialBackoff"`

	// MaxBackoff is the maximum delay between retries.
	// Format: "30s", "1m", etc.
	MaxBackoff string `json:"maxBackoff"`

	// BackoffMultiplier is the multiplier for exponential backoff.
	// Default: 2.0
	BackoffMultiplier float64 `json:"backoffMultiplier"`

	// RetryableStatusCodes defines which status codes should trigger a retry.
	// Common values: ["UNAVAILABLE"], ["DEADLINE_EXCEEDED"], ["UNAVAILABLE", "RESOURCE_EXHAUSTED"]
	RetryableStatusCodes []string `json:"retryableStatusCodes"`
}

type RetryThrottling struct {
	// MaxTokens is the maximum number of tokens in the bucket.
	// Must be in range (0, 1000]. Required.
	MaxTokens int `json:"maxTokens"`

	// TokenRatio is the ratio of tokens to add on each successful RPC.
	// Must be greater than 0. Decimal places beyond 3 are ignored.
	TokenRatio float64 `json:"tokenRatio"`
}

type ServerStream[T any] interface {
	// Send sends a message to the client.
	Send(*T) error
	// Context returns the context for this stream.
	Context() context.Context
}

type ServerStreamHandler[TIn, TOut any] func(context.Context, *TIn, ServerStream[TOut]) error

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

type ServiceConfig struct {
	// MethodConfig contains per-method configuration.
	MethodConfig []MethodConfig `json:"methodConfig,omitempty"`

	// RetryThrottling controls client-side retry throttling.
	RetryThrottling *RetryThrottling `json:"retryThrottling,omitempty"`
}

type ServiceOption func(*ServiceOptions)

type ServiceOptions struct {
	// Package sets the protobuf package name
	Package string
	// EnableValidation enables input validation by default
	EnableValidation bool
	// EnableReflection enables gRPC reflection
	EnableReflection bool
	// Interceptors to apply to all methods
	Interceptors []Interceptor
	// Edition sets the Protobuf edition (e.g., "2023", "2024")
	Edition string
	// UseEditions enables Protobuf Editions mode instead of proto3
	UseEditions bool
	// ServiceConfig is the gRPC service configuration (JSON string)
	ServiceConfig string
	// Description is the service-level documentation
	Description string
	// EnableJSONRPC enables JSON-RPC 2.0 support
	EnableJSONRPC bool
	// JSONRPCPath is the path to serve JSON-RPC requests (default: "/jsonrpc")
	JSONRPCPath string
	// JSONRPCBatchLimit is the maximum number of requests in a batch (default: 100)
	JSONRPCBatchLimit int
	// MaxReceiveMessageSize is the maximum size of a message that can be received (default: 4MB)
	MaxReceiveMessageSize int
	// MaxSendMessageSize is the maximum size of a message that can be sent (default: 4MB)
	MaxSendMessageSize int
}

type StreamType int

type TimeoutInterceptor struct {
	Timeout time.Duration
}

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