feat(cust): add EP to check customer entitlement value

api/spec/src/customer.tsp

@@ -103,6 +103,19 @@ interface Customers {
   deleteAppData(@path customerId: ULID, @path appId: ULID): {
     @statusCode _: 204;
   } | NotFoundError | CommonErrors;
+
+  /**
+   * Checks customer access to a given feature (by key). All entitlement types share the hasAccess property in their value response, but multiple other properties are returned based on the entitlement type.
+   */
+  @get
+  @operationId("getCustomerEntitlementValue")
+  @route("/{customerId}/entitlements/{featureKey}/value")
+  @summary("Get entitlement value")
+  getCustomerEntitlementValue(
+    @path customerId: string,
+    @path featureKey: string,
+    @query(#{ explode: true }) time?: DateTime,
+  ): Entitlements.EntitlementValue | OpenMeter.CommonErrors | OpenMeter.NotFoundError;
 }
 
 /**

api/spec/src/entitlements/subjects.tsp

@@ -6,7 +6,6 @@ using TypeSpec.Http;
 using TypeSpec.Rest;
 using TypeSpec.OpenAPI;
 
-// TODO: does this have to be in a separate namespace?
 namespace OpenMeter.Entitlements;
 
 @route("/api/v1/subjects/{subjectIdOrKey}/entitlements")

app/common/customer.go

@@ -10,13 +10,14 @@ import (
 	customeradapter "github.com/openmeterio/openmeter/openmeter/customer/adapter"
 	customerservice "github.com/openmeterio/openmeter/openmeter/customer/service"
 	entdb "github.com/openmeterio/openmeter/openmeter/ent/db"
+	"github.com/openmeterio/openmeter/openmeter/registry"
 )
 
 var Customer = wire.NewSet(
 	NewCustomerService,
 )
 
-func NewCustomerService(logger *slog.Logger, db *entdb.Client) (customer.Service, error) {
+func NewCustomerService(logger *slog.Logger, db *entdb.Client, entRegistry *registry.Entitlement) (customer.Service, error) {
 	customerAdapter, err := customeradapter.New(customeradapter.Config{
 		Client: db,
 		Logger: logger.WithGroup("customer.postgres"),
@@ -26,6 +27,7 @@ func NewCustomerService(logger *slog.Logger, db *entdb.Client) (customer.Service
 	}
 
 	return customerservice.New(customerservice.Config{
-		Adapter: customerAdapter,
+		Adapter:              customerAdapter,
+		EntitlementConnector: entRegistry.Entitlement,
 	})
 }

cmd/jobs/billing/advance/advance.go

@@ -15,6 +15,7 @@ import (
 	billingadapter "github.com/openmeterio/openmeter/openmeter/billing/adapter"
 	billingservice "github.com/openmeterio/openmeter/openmeter/billing/service"
 	billingworkerautoadvance "github.com/openmeterio/openmeter/openmeter/billing/worker/advance"
+	registrybuilder "github.com/openmeterio/openmeter/openmeter/registry/builder"
 	"github.com/openmeterio/openmeter/openmeter/watermill/driver/kafka"
 	"github.com/openmeterio/openmeter/pkg/framework/entutils/entdriver"
 	"github.com/openmeterio/openmeter/pkg/framework/pgdriver"
@@ -57,40 +58,8 @@ func NewAutoAdvancer(ctx context.Context, conf appconfig.Configuration, logger *
 	// Initialize Ent driver
 	entPostgresDriver := entdriver.NewEntPostgresDriver(postgresDriver.DB())
 
-	customerService, err := common.NewCustomerService(logger, entPostgresDriver.Client())
-	if err != nil {
-		return nil, fmt.Errorf("failed to initialize customer service: %w", err)
-	}
-
-	secretService, err := common.NewUnsafeSecretService(logger, entPostgresDriver.Client())
-	if err != nil {
-		return nil, fmt.Errorf("failed to initialize secret service: %w", err)
-	}
-
-	appService, err := common.NewAppService(logger, entPostgresDriver.Client(), conf.Apps)
-	if err != nil {
-		return nil, fmt.Errorf("failed to initialize app service: %w", err)
-	}
-
-	_, err = common.NewAppStripeService(logger, entPostgresDriver.Client(), conf.Apps, appService, customerService, secretService)
-	if err != nil {
-		return nil, fmt.Errorf("failed to initialize stripe app service: %w", err)
-	}
-
-	namespaceManager, err := common.NewNamespaceManager(nil, conf.Namespace)
-	if err != nil {
-		return nil, fmt.Errorf("failed to initialize namespace manager: %w", err)
-	}
-
-	_, err = common.NewAppSandboxProvisioner(ctx, logger, conf.Apps, appService, namespaceManager)
-	if err != nil {
-		return nil, fmt.Errorf("failed to initialize sandbox app provisioner: %w", err)
-	}
-
 	meterRepository := common.NewInMemoryRepository(conf.Meters)
 
-	featureService := common.NewFeatureConnector(logger, entPostgresDriver.Client(), meterRepository)
-
 	clickhouseConn, err := common.NewClickHouse(conf.Aggregation.ClickHouse)
 	if err != nil {
 		return nil, fmt.Errorf("failed to initialize clickhouse connection: %w", err)
@@ -129,6 +98,44 @@ func NewAutoAdvancer(ctx context.Context, conf appconfig.Configuration, logger *
 		return nil, fmt.Errorf("failed to initialize event bus publisher: %w", err)
 	}
 
+	entitlementRegistry := registrybuilder.GetEntitlementRegistry(registrybuilder.EntitlementOptions{
+		DatabaseClient:     entPostgresDriver.Client(),
+		StreamingConnector: streamingConnector,
+		Logger:             logger,
+		MeterRepository:    meterRepository,
+		Publisher:          ebPublisher,
+	})
+
+	customerService, err := common.NewCustomerService(logger, entPostgresDriver.Client(), entitlementRegistry)
+	if err != nil {
+		return nil, fmt.Errorf("failed to initialize customer service: %w", err)
+	}
+
+	secretService, err := common.NewUnsafeSecretService(logger, entPostgresDriver.Client())
+	if err != nil {
+		return nil, fmt.Errorf("failed to initialize secret service: %w", err)
+	}
+
+	appService, err := common.NewAppService(logger, entPostgresDriver.Client(), conf.Apps)
+	if err != nil {
+		return nil, fmt.Errorf("failed to initialize app service: %w", err)
+	}
+
+	_, err = common.NewAppStripeService(logger, entPostgresDriver.Client(), conf.Apps, appService, customerService, secretService)
+	if err != nil {
+		return nil, fmt.Errorf("failed to initialize stripe app service: %w", err)
+	}
+
+	namespaceManager, err := common.NewNamespaceManager(nil, conf.Namespace)
+	if err != nil {
+		return nil, fmt.Errorf("failed to initialize namespace manager: %w", err)
+	}
+
+	_, err = common.NewAppSandboxProvisioner(ctx, logger, conf.Apps, appService, namespaceManager)
+	if err != nil {
+		return nil, fmt.Errorf("failed to initialize sandbox app provisioner: %w", err)
+	}
+
 	billingAdapter, err := billingadapter.New(billingadapter.Config{
 		Client: entPostgresDriver.Client(),
 		Logger: logger,
@@ -142,7 +149,7 @@ func NewAutoAdvancer(ctx context.Context, conf appconfig.Configuration, logger *
 		CustomerService:    customerService,
 		AppService:         appService,
 		Logger:             logger,
-		FeatureService:     featureService,
+		FeatureService:     entitlementRegistry.Feature,
 		MeterRepo:          meterRepository,
 		StreamingConnector: streamingConnector,
 		Publisher:          ebPublisher,