api

type ABIContext = api.ABIContext

type BinaryModule interface {
	Get() ([]byte, error)
}

type Closer interface {
	Close()
}

type Context interface {
	PropertyHolder

	ID() int32
	NewContextID() int32
	GetOrCreateContext(rootID string) Context
}

type ContextHandler = api.ContextHandler

type DataSource interface {
	Get() ([]byte, error)
}

type FilterContext interface {
	api.ContextHandler

	Logger() logr.Logger
	Lock()
	Unlock()
	Close() error
	ID() int32
	RootContext() Context
	GetABIContext() api.ContextHandler
}

type HTTPRequest interface {
	URL() *url.URL
	Header() HeaderMap
	Trailer() HeaderMap
	Body() io.ReadCloser
	SetBody(io.ReadCloser)

	HTTPProtocol() string
	Host() string
	Method() string

	Connection() network.Connection
}

type HTTPResponse interface {
	Header() HeaderMap
	Trailer() HeaderMap
	Body() io.ReadCloser
	SetBody(io.ReadCloser)

	Connection() network.Connection

	ContentLength() int64
	StatusCode() int
}

type HeaderMap interface {
	Get(key string) (string, bool)
	Set(key, value string)
	Add(key, value string)
	Del(key string)
	Keys() []string
	Len() int
	Raw() map[string][]string
	Clone() HeaderMap
}

type IoBuffer = api.IoBuffer

type JsonnableMap map[string]interface{}

type ListenerDirection uint64

type Marshallable interface {
	Marshal() ([]byte, error)
}

type MetricHandler interface {
	HTTPHandler() http.Handler
	DefineMetric(metricType int32, name string) int32
	RecordMetric(metricID int32, value int64) error
	IncrementMetric(metricID int32, offset int64) error
}

type PropertyHolder interface {
	Get(key string) (interface{}, bool)
	Set(key string, value interface{})
}

type RuntimeCreatorStore interface {
	Get(name string) (WasmRuntimeCreator, bool)
	Set(name string, creator WasmRuntimeCreator)
}

type Stream interface {
	PropertyHolder

	Logger() logr.Logger
	Close() error
	Direction() ListenerDirection

	HandleHTTPRequest(req HTTPRequest) error
	HandleHTTPResponse(resp HTTPResponse) error

	HandleTCPNewConnection(conn net.Conn) error
	HandleTCPCloseConnection(conn net.Conn) error
	HandleDownstreamData(conn net.Conn, n int) error
	HandleUpstreamData(conn net.Conn, n int) error
}

type StreamHandler interface {
	NewStream(direction ListenerDirection, options ...StreamOption) (Stream, error)
	Logger() logr.Logger
}

type StreamOption func(instance interface{})

type VMKey string

type VMStore interface {
	GetOrCreateVM(vmConfig WasmVMConfig) (WasmVM, error)
}

type WasmInstance = api.WasmInstance

type WasmInstanceContext interface {
	GetInstance() WasmInstance
	GetProperties() PropertyHolder
	GetABIContext() ContextHandler
}

type WasmModule = api.WasmModule

type WasmPlugin interface {
	Name() string
	ID() string
	GetInstance() (WasmInstance, error)
	ReleaseInstance(instance WasmInstance)
	EnsureInstances(desired uint32) uint32
	Exec(f func(instance WasmInstance) bool)
	Close()
	Report()
	VM() WasmVM
	Context() Context
	Logger() logr.Logger

	RegisterFilterContext(instance WasmInstance, filterContext FilterContext)
	UnregisterFilterContext(instance WasmInstance, filterContext FilterContext)
	GetFilterContext(instance WasmInstance, id int32) (FilterContext, bool)

	GetWasmInstanceContext(instance WasmInstance) WasmInstanceContext
}

type WasmPluginConfig struct {
	Name          string       `json:"name,omitempty"`
	RootID        string       `json:"rootID,omitempty"`
	VMConfig      WasmVMConfig `json:"vmConfig,omitempty"`
	Configuration Marshallable `json:"configuration,omitempty"`
	InstanceCount uint32       `json:"instanceCount,omitempty"`
}

type WasmPluginManager interface {
	Delete(config WasmPluginConfig) error
	Get(config WasmPluginConfig) (WasmPlugin, error)
	GetOrCreate(config WasmPluginConfig) (WasmPlugin, error)
	Add(config WasmPluginConfig) (WasmPlugin, error)
	Logger() logr.Logger
	GetBaseContext() Context
}

type WasmResult = api.WasmResult

type WasmRuntime interface {
	Name() string
	Init()
	NewModule(wasmBytes []byte) (WasmModule, error)
}

type WasmRuntimeCreator func() WasmRuntime

type WasmVM interface {
	WasmRuntime
	Acquire(user interface{})
	Release(user interface{})
	Close()
}

type WasmVMConfig struct {
	ID            string                 `json:"id,omitempty"`
	Runtime       string                 `json:"runtime,omitempty"`
	Code          DataSource             `json:"code,omitempty"`
	Configuration map[string]interface{} `json:"configuration,omitempty"`
}

type WrappedPropertyHolder interface {
	PropertyHolder
	Properties() PropertyHolder
	ParentProperties() PropertyHolder
}