File size: 2,027 Bytes
d6f631f | 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 | package addondiff
import (
"github.com/samber/lo"
"github.com/openmeterio/openmeter/openmeter/subscription"
subscriptionaddon "github.com/openmeterio/openmeter/openmeter/subscription/addon"
)
func GetDiffableFromAddon(
_ subscription.SubscriptionView,
addon subscriptionaddon.SubscriptionAddon,
) (Diffable, error) {
instances := addon.GetInstances()
if len(instances) == 0 {
// no-op
return nil, nil
}
// As there's no overlap between the instances (no two instance would affect the same items), we can just apply them in (any) sequence
diffs := lo.Map(instances, func(instance subscriptionaddon.SubscriptionAddonInstance, _ int) Diffable {
return &diffable{
addon: instance,
}
})
return &someDiffable{
ApplyFn: func(spec *subscription.SubscriptionSpec, actx subscription.ApplyContext) error {
applieses := lo.Map(diffs, func(diff Diffable, _ int) subscription.AppliesToSpec {
return diff.GetApplies()
})
agg := subscription.NewAggregateAppliesToSpec(applieses)
return spec.Apply(agg, actx)
},
RestoreFn: func(spec *subscription.SubscriptionSpec, actx subscription.ApplyContext) error {
applieses := lo.Map(diffs, func(diff Diffable, _ int) subscription.AppliesToSpec {
return diff.GetRestores()
})
agg := subscription.NewAggregateAppliesToSpec(applieses)
return spec.Apply(agg, actx)
},
}, nil
}
var _ Diffable = &diffable{}
type diffable struct {
addon subscriptionaddon.SubscriptionAddonInstance
}
func (d *diffable) GetApplies() subscription.AppliesToSpec {
if d.addon.Quantity == 0 {
return subscription.NewAppliesToSpec(func(spec *subscription.SubscriptionSpec, actx subscription.ApplyContext) error {
return nil
})
}
applieses := lo.Map(d.addon.RateCards, func(rc subscriptionaddon.SubscriptionAddonRateCard, _ int) subscription.AppliesToSpec {
return d.getApplyForRateCard(rc)
})
return subscription.NewAggregateAppliesToSpec(applieses)
}
func (d *diffable) GetRestores() subscription.AppliesToSpec {
return d.restore()
}
|