Method Check

api/pkg/services/entitlement_service.go:60–111

Check verifies if the user can create another instance of the given entity.

(
	ctx context.Context,
	userID entities.UserID,
	entityName string,
	countFunc func() (int, error),
)

// Check verifies if the user can create another instance of the given entity.
func (service *EntitlementService) Check(
	ctx context.Context,
	userID entities.UserID,
	entityName string,
	countFunc func() (int, error),
) (*EntitlementCheckResult, error) {
	ctx, span := service.tracer.Start(ctx)
	defer span.End()

	if !service.enabled {
		return &EntitlementCheckResult{Allowed: true}, nil
	}

	limits, exists := entityLimits[entityName]
	if !exists {
		return &EntitlementCheckResult{Allowed: true}, nil
	}

	user, err := service.userRepository.Load(ctx, userID)
	if err != nil {
		return nil, service.tracer.WrapErrorSpan(
			span,
			stacktrace.Propagate(err, fmt.Sprintf("cannot load user [%s] for entitlement check", userID)),
		)
	}

	limit, hasLimit := limits[user.SubscriptionName]
	if !hasLimit || limit == 0 {
		return &EntitlementCheckResult{Allowed: true}, nil
	}

	currentCount, err := countFunc()
	if err != nil {
		return nil, service.tracer.WrapErrorSpan(
			span,
			stacktrace.Propagate(err, fmt.Sprintf("cannot count entities [%s] for user [%s]", entityName, userID)),
		)
	}

	if currentCount >= limit {
		return &EntitlementCheckResult{
			Allowed: false,
			Message: fmt.Sprintf(
				"Upgrade to a paid plan to create more than [%d] %s. Visit https://httpsms.com/pricing for details.",
				limit,
				formatEntityName(entityName, true),
			),
		}, nil
	}

	return &EntitlementCheckResult{Allowed: true}, nil
}

// formatEntityName converts a PascalCase entity name to lowercase words and optionally pluralizes it.
// Consecutive uppercase letters (acronyms like API) are kept together as a single word.