chore: expose balance and notification types

cmd/balance-worker/main.go

@@ -55,7 +55,7 @@ import (
 
 const (
 	defaultShutdownTimeout = 5 * time.Second
-	otelName               = "openmeter.io/backend"
+	otelName               = "openmeter.io/balance-worker"
 )
 
 func main() {

cmd/notification-service/main.go

@@ -55,7 +55,7 @@ import (
 
 const (
 	defaultShutdownTimeout = 5 * time.Second
-	otelName               = "openmeter.io/backend"
+	otelName               = "openmeter.io/notification-service"
 )
 
 func main() {

openmeter/entitlement/balanceworker/worker.go

@@ -0,0 +1,13 @@
+package balanceworker
+
+import "github.com/openmeterio/openmeter/internal/entitlement/balanceworker"
+
+type (
+	Worker           = balanceworker.Worker
+	WorkerOptions    = balanceworker.WorkerOptions
+	WorkerDLQOptions = balanceworker.WorkerDLQOptions
+)
+
+func New(opts WorkerOptions) (*Worker, error) {
+	return balanceworker.New(opts)
+}

openmeter/event/publisher/publisher.go

@@ -3,11 +3,20 @@ package publisher
 import "github.com/openmeterio/openmeter/internal/event/publisher"
 
 type (
-	Publisher        = publisher.Publisher
-	PublisherOptions = publisher.PublisherOptions
-	TopicPublisher   = publisher.TopicPublisher
+	Publisher           = publisher.Publisher
+	PublisherOptions    = publisher.PublisherOptions
+	TopicPublisher      = publisher.TopicPublisher
+	CloudEventMarshaler = publisher.CloudEventMarshaler
+)
+
+type (
+	TransformFunc = publisher.TransformFunc
 )
 
 func NewPublisher(options PublisherOptions) (Publisher, error) {
 	return publisher.NewPublisher(options)
 }
+
+func NewCloudEventMarshaler(transform TransformFunc) CloudEventMarshaler {
+	return publisher.NewCloudEventMarshaler(transform)
+}

openmeter/notification/consumer/consumer.go

@@ -0,0 +1,13 @@
+package consumer
+
+import "github.com/openmeterio/openmeter/internal/notification/consumer"
+
+type (
+	Options    = consumer.Options
+	DLQOptions = consumer.DLQOptions
+	Consumer   = consumer.Consumer
+)
+
+func New(opts Options) (*Consumer, error) {
+	return consumer.New(opts)
+}

openmeter/registry/entitlement.go

@@ -0,0 +1,12 @@
+package registry
+
+import "github.com/openmeterio/openmeter/internal/registry"
+
+type (
+	Entitlement        = registry.Entitlement
+	EntitlementOptions = registry.EntitlementOptions
+)
+
+func GetEntitlementRegistry(opts EntitlementOptions) *Entitlement {
+	return registry.GetEntitlementRegistry(opts)
+}