File size: 3,784 Bytes
16cdcb7 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 | package customer
import (
"context"
"fmt"
"github.com/samber/lo"
"github.com/openmeterio/openmeter/openmeter/customer"
"github.com/openmeterio/openmeter/openmeter/subscription"
"github.com/openmeterio/openmeter/pkg/clock"
"github.com/openmeterio/openmeter/pkg/filter"
"github.com/openmeterio/openmeter/pkg/models"
)
var _ customer.RequestValidator = (*Validator)(nil)
func NewValidator(subscriptionService subscription.Service, customerService customer.Service) (*Validator, error) {
if subscriptionService == nil {
return nil, fmt.Errorf("subscription service is required")
}
if customerService == nil {
return nil, fmt.Errorf("customer service is required")
}
return &Validator{
subscriptionService: subscriptionService,
customerService: customerService,
}, nil
}
type Validator struct {
customer.NoopRequestValidator
subscriptionService subscription.Service
customerService customer.Service
}
func (v *Validator) ValidateUpdateCustomer(ctx context.Context, input customer.UpdateCustomerInput) error {
if err := input.Validate(); err != nil {
return err
}
// The subject association can only be changed if the customer doesn't have a subscription
subscriptions, err := v.subscriptionService.List(ctx, subscription.ListSubscriptionsInput{
Namespaces: []string{input.CustomerID.Namespace},
CustomerID: &filter.FilterULID{FilterString: filter.FilterString{Eq: &input.CustomerID.ID}},
ActiveAt: lo.ToPtr(clock.Now()),
})
if err != nil {
return err
}
hasSub := len(subscriptions.Items) > 0
// If there's an update to the subject keys we need additional checks
if input.CustomerMutate.UsageAttribution != nil && input.CustomerMutate.UsageAttribution.SubjectKeys != nil {
currentCustomer, err := v.customerService.GetCustomer(ctx, customer.GetCustomerInput{
CustomerID: &customer.CustomerID{
Namespace: input.CustomerID.Namespace,
ID: input.CustomerID.ID,
},
})
if err != nil {
return err
}
if currentCustomer != nil && currentCustomer.IsDeleted() {
return models.NewGenericPreConditionFailedError(
fmt.Errorf("customer is deleted [namespace=%s customer.id=%s]", currentCustomer.Namespace, currentCustomer.ID),
)
}
if currentCustomer == nil {
return models.NewGenericNotFoundError(
fmt.Errorf("customer [namespace=%s customer.id=%s]", input.CustomerID.Namespace, input.CustomerID.ID),
)
}
// Let's check the two subjectKey arrays are the same
if hasSub {
var currentSubjectKeys []string
if currentCustomer.UsageAttribution != nil {
currentSubjectKeys = currentCustomer.UsageAttribution.SubjectKeys
}
if len(currentSubjectKeys) != len(input.CustomerMutate.UsageAttribution.SubjectKeys) {
return fmt.Errorf("cannot change subject keys for customer with active subscriptions")
}
for i, key := range currentSubjectKeys {
if key != input.CustomerMutate.UsageAttribution.SubjectKeys[i] {
return fmt.Errorf("cannot change subject keys for customer with active subscriptions")
}
}
}
}
return nil
}
func (v *Validator) ValidateDeleteCustomer(ctx context.Context, input customer.DeleteCustomerInput) error {
// A customer can only be deleted if all of his invocies are in final state
if err := input.Validate(); err != nil {
return err
}
subscriptions, err := v.subscriptionService.List(ctx, subscription.ListSubscriptionsInput{
Namespaces: []string{input.Namespace},
CustomerID: &filter.FilterULID{FilterString: filter.FilterString{Eq: &input.ID}},
ActiveAt: lo.ToPtr(clock.Now()),
})
if err != nil {
return err
}
if len(subscriptions.Items) > 0 {
return fmt.Errorf("customer %s still have active subscriptions, please cancel them before deleting the customer", input.ID)
}
return nil
}
|