localstorage

type AppCtx struct {
	EnsureLoggedIn func(ctx context.Context) error
	Features       featurev1connect.FeatureServiceClient
	Config         *config.Config
	State          *state.State
}

type Cursor interface {
	IDs() (machineID, sessionID int64)
	Next(context.Context) bool
	Event(*typesv1.LogEvent) error
	Err() error
	Close() error
}

type OTLPLogger interface {
	ExportLogs(ctx context.Context, req *otlplogssvcpb.ExportLogsServiceRequest) (*otlplogssvcpb.ExportLogsServiceResponse, error)
}

type OTLPMeter interface {
	ExportMetrics(ctx context.Context, req *otlpmetricssvcpb.ExportMetricsServiceRequest) (*otlpmetricssvcpb.ExportMetricsServiceResponse, error)
}

type OTLPTracer interface {
	ExportTraces(ctx context.Context, req *otlptracesvcpb.ExportTraceServiceRequest) (*otlptracesvcpb.ExportTraceServiceResponse, error)
}

type Queryable interface {
	Parse(ctx context.Context, q string) (*typesv1.Query, error)
	Format(ctx context.Context, q *typesv1.Query) (string, error)

	Query(ctx context.Context, q *typesv1.Query, c *typesv1.Cursor, limit int) (*typesv1.Data, *typesv1.Cursor, error)
	ResolveQueryType(ctx context.Context, query *typesv1.Query) (*typesv1.DataStreamType, error)
	ListSymbols(ctx context.Context, query *typesv1.Query, c *typesv1.Cursor, limit int) ([]*typesv1.Symbol, *typesv1.Cursor, error)

	Stream(ctx context.Context, q *typesv1.Query, cb func(context.Context, *typesv1.Data) (bool, error), opts *StreamOption) error

	GetTraceByID(ctx context.Context, traceID string) (*typesv1.Trace, error)
	GetTraceBySpanID(ctx context.Context, spanID string) (*typesv1.Trace, error)
	GetSpanByID(ctx context.Context, spanID string) (*typesv1.Span, error)
}

type Storage interface {
	Queryable
	SinkFor(ctx context.Context, machineID, sessionID int64) (_ sink.Sink, heartbeatIn time.Duration, _ error)
	Heartbeat(ctx context.Context, machineID, sessionID int64) (time.Duration, error)
	Stats(ctx context.Context) (*typesv1.Val, error)
	Close() error

	OTLPLogger
	OTLPTracer
	OTLPMeter
}

type StorageBuilder func(
	ctx context.Context,
	ll *slog.Logger,
	cfg map[string]interface{},
	app *AppCtx,
) (Storage, error)

type StreamOption struct {
	BatchSize             int
	BatchTrigger          <-chan time.Time
	NotifyStreamListening func(ctx context.Context)
}

type Symbol struct {
	Name string
	Type *typesv1.VarType
}