feat: add config to set trial sub end behaviour

raystack · Jan 6, 2025 · 731eb60 · 731eb60
1 parent 83bd68f
commit 731eb60
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 1 deletion.
diff --git a/billing/config.go b/billing/config.go
@@ -55,6 +55,8 @@ type PlanChangeConfig struct {
 }
 
 type SubscriptionConfig struct {
+	// CancelAfterTrial will cancel the subscription after trial end and will not generate invoice.
+	CancelAfterTrial bool `yaml:"cancel_after_trial" mapstructure:"cancel_after_trial" default:"false"`
 }
 
 type ProductConfig struct {

diff --git a/billing/subscription/service.go b/billing/subscription/service.go
@@ -990,7 +990,7 @@ func (s *Service) CancelUpcomingPhase(ctx context.Context, sub Subscription) err
 
 	var endBehavior = stripe.SubscriptionScheduleEndBehaviorRelease
 
-	if stripeSub.Status == stripe.SubscriptionStatusTrialing {
+	if stripeSub.Status == stripe.SubscriptionStatusTrialing && s.config.SubscriptionConfig.CancelAfterTrial {
 		endBehavior = stripe.SubscriptionScheduleEndBehaviorCancel
 	}
 

diff --git a/config/sample.config.yaml b/config/sample.config.yaml
@@ -227,3 +227,5 @@ billing:
     subscription: 1m
     invoice: 5m
     checkout: 1m
+  subscription:
+    cancel_after_trial: true