Add script to create 2025 plans

ref DEV-2327

cmd/portal/cmd/cmdpricing/create_stripe_plans_2025.go

@@ -0,0 +1,292 @@
+package cmdpricing
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/spf13/cobra"
+	"github.com/stripe/stripe-go/v72"
+	"github.com/stripe/stripe-go/v72/client"
+
+	portalcmd "github.com/authgear/authgear-server/cmd/portal/cmd"
+	portalconfig "github.com/authgear/authgear-server/pkg/portal/config"
+	"github.com/authgear/authgear-server/pkg/util/cobrasentry"
+)
+
+var Dollar int64 = 100
+var Cent int64 = 1
+
+func createPlanProduct(api *client.API, logger Logger, planName string, planDisplayName string, price int64) error {
+	params := &stripe.PriceParams{
+		Currency:    stripe.String(string(stripe.CurrencyUSD)),
+		UnitAmount:  stripe.Int64(price),
+		TaxBehavior: stripe.String(string(stripe.PriceTaxBehaviorInclusive)),
+		Recurring: &stripe.PriceRecurringParams{
+			Interval: stripe.String(string(stripe.PriceRecurringIntervalMonth)),
+		},
+		ProductData: &stripe.PriceProductDataParams{
+			Name: stripe.String(planDisplayName),
+			Metadata: map[string]string{
+				"plan_name":              planName,
+				"price_type":             "fixed",
+				"subscription_item_type": "plan",
+				"version":                "2025",
+			},
+		},
+	}
+	result, err := api.Prices.New(params)
+	if err != nil {
+		return err
+	}
+
+	updateProductParams := &stripe.ProductParams{
+		DefaultPrice: stripe.String(result.ID),
+	}
+	_, err = api.Products.Update(result.Product.ID, updateProductParams)
+	if err != nil {
+		return err
+	}
+
+	logger.
+		WithField("name", planDisplayName).
+		WithField("product_id", result.Product.ID).
+		Info("Created product")
+	return nil
+}
+
+func createSMSProduct(api *client.API, logger Logger, displayName, itemType string, smsRegion string, perUnitPrice int64) error {
+	params := &stripe.PriceParams{
+		Currency:    stripe.String(string(stripe.CurrencyUSD)),
+		TaxBehavior: stripe.String(string(stripe.PriceTaxBehaviorInclusive)),
+		Recurring: &stripe.PriceRecurringParams{
+			Interval:       stripe.String(string(stripe.PriceRecurringIntervalMonth)),
+			UsageType:      stripe.String(string(stripe.PriceRecurringUsageTypeMetered)),
+			AggregateUsage: stripe.String(string(stripe.PriceRecurringAggregateUsageSum)),
+		},
+		BillingScheme: stripe.String(string(stripe.PlanBillingSchemeTiered)),
+		TiersMode:     stripe.String(string(stripe.PlanTiersModeGraduated)),
+		Tiers: []*stripe.PriceTierParams{
+			{
+				UnitAmount: stripe.Int64(perUnitPrice),
+				UpToInf:    stripe.Bool(true),
+			},
+		},
+		ProductData: &stripe.PriceProductDataParams{
+			Name: stripe.String(displayName),
+			Metadata: map[string]string{
+				"price_type":             "usage",
+				"subscription_item_type": itemType,
+				"sms_region":             smsRegion,
+				"usage_type":             "sms",
+				"version":                "2025",
+			},
+		},
+	}
+	result, err := api.Prices.New(params)
+	if err != nil {
+		return err
+	}
+
+	updateProductParams := &stripe.ProductParams{
+		DefaultPrice: stripe.String(result.ID),
+	}
+	_, err = api.Products.Update(result.Product.ID, updateProductParams)
+	if err != nil {
+		return err
+	}
+
+	logger.
+		WithField("name", displayName).
+		WithField("product_id", result.Product.ID).
+		Info("Created product")
+	return nil
+}
+
+func createWhatsappProduct(api *client.API, logger Logger, displayName, itemType string, whatsappRegion string, perUnitPrice int64) error {
+	params := &stripe.PriceParams{
+		Currency:    stripe.String(string(stripe.CurrencyUSD)),
+		TaxBehavior: stripe.String(string(stripe.PriceTaxBehaviorInclusive)),
+		Recurring: &stripe.PriceRecurringParams{
+			Interval:       stripe.String(string(stripe.PriceRecurringIntervalMonth)),
+			UsageType:      stripe.String(string(stripe.PriceRecurringUsageTypeMetered)),
+			AggregateUsage: stripe.String(string(stripe.PriceRecurringAggregateUsageSum)),
+		},
+		BillingScheme: stripe.String(string(stripe.PlanBillingSchemeTiered)),
+		TiersMode:     stripe.String(string(stripe.PlanTiersModeGraduated)),
+		Tiers: []*stripe.PriceTierParams{
+			{
+				UnitAmount: stripe.Int64(perUnitPrice),
+				UpToInf:    stripe.Bool(true),
+			},
+		},
+		ProductData: &stripe.PriceProductDataParams{
+			Name: stripe.String(displayName),
+			Metadata: map[string]string{
+				"price_type":             "usage",
+				"subscription_item_type": itemType,
+				"whatsapp_region":        whatsappRegion,
+				"usage_type":             "whatsapp",
+				"version":                "2025",
+			},
+		},
+	}
+	result, err := api.Prices.New(params)
+	if err != nil {
+		return err
+	}
+
+	updateProductParams := &stripe.ProductParams{
+		DefaultPrice: stripe.String(result.ID),
+	}
+	_, err = api.Products.Update(result.Product.ID, updateProductParams)
+	if err != nil {
+		return err
+	}
+
+	logger.
+		WithField("name", displayName).
+		WithField("product_id", result.Product.ID).
+		Info("Created product")
+	return nil
+}
+
+func createMAUProduct(api *client.API, logger Logger, displayName, planName string, perGroupPrice, groupUnit, freeQuantity int64) error {
+	params := &stripe.PriceParams{
+		Params: stripe.Params{
+			Metadata: map[string]string{
+				"free_quantity": fmt.Sprint(freeQuantity),
+			},
+		},
+		Currency:    stripe.String(string(stripe.CurrencyUSD)),
+		TaxBehavior: stripe.String(string(stripe.PriceTaxBehaviorInclusive)),
+		Recurring: &stripe.PriceRecurringParams{
+			Interval:       stripe.String(string(stripe.PriceRecurringIntervalMonth)),
+			UsageType:      stripe.String(string(stripe.PriceRecurringUsageTypeMetered)),
+			AggregateUsage: stripe.String(string(stripe.PriceRecurringAggregateUsageLastDuringPeriod)),
+		},
+		BillingScheme: stripe.String(string(stripe.PlanBillingSchemePerUnit)),
+		UnitAmount:    stripe.Int64(perGroupPrice),
+		TransformQuantity: &stripe.PriceTransformQuantityParams{
+			DivideBy: &groupUnit,
+			Round:    stripe.String(string(stripe.PriceTransformQuantityRoundUp)),
+		},
+		ProductData: &stripe.PriceProductDataParams{
+			Name: stripe.String(displayName),
+			Metadata: map[string]string{
+				"plan_name":              planName,
+				"price_type":             "usage",
+				"subscription_item_type": "mau",
+				"usage_type":             "mau",
+				"version":                "2025",
+			},
+		},
+	}
+	result, err := api.Prices.New(params)
+	if err != nil {
+		return err
+	}
+
+	updateProductParams := &stripe.ProductParams{
+		DefaultPrice: stripe.String(result.ID),
+	}
+	_, err = api.Products.Update(result.Product.ID, updateProductParams)
+	if err != nil {
+		return err
+	}
+
+	logger.
+		WithField("name", displayName).
+		WithField("product_id", result.Product.ID).
+		Info("Created product")
+	return nil
+}
+
+var cmdPricingCreateStripePlans2025 = &cobra.Command{
+	Use: "create-stripe-plans-2025",
+	RunE: cobrasentry.RunEWrap(portalcmd.GetBinder, func(ctx context.Context, cmd *cobra.Command, args []string) (err error) {
+		binder := portalcmd.GetBinder()
+
+		stripeSecretKey, err := binder.GetRequiredString(cmd, portalcmd.ArgStripeSecretKey)
+		if err != nil {
+			return
+		}
+
+		stripeConfig := &portalconfig.StripeConfig{
+			SecretKey: stripeSecretKey,
+		}
+
+		hub := cobrasentry.GetHub(ctx)
+		factory := cobrasentry.NewLoggerFactory(hub)
+		logger := NewLogger(factory)
+
+		api := NewClientAPI(stripeConfig, logger)
+
+		err = createPlanProduct(api, logger,
+			"developers2025", "Developers (2025)",
+			50*Dollar,
+		)
+		if err != nil {
+			return err
+		}
+
+		err = createPlanProduct(api, logger,
+			"business2025", "Business (2025)",
+			500*Dollar,
+		)
+		if err != nil {
+			return err
+		}
+
+		err = createSMSProduct(api, logger,
+			"SMS usage (North America) (2025)",
+			"sms-north-america",
+			"north-america",
+			2*Cent,
+		)
+		if err != nil {
+			return err
+		}
+
+		err = createSMSProduct(api, logger,
+			"SMS usage (Other regions) (2025)",
+			"sms-other-region",
+			"other-regions",
+			10*Cent,
+		)
+		if err != nil {
+			return err
+		}
+
+		err = createWhatsappProduct(api, logger,
+			"Whatsapp Usage (North America) (2025)",
+			"whatsapp-north-america",
+			"north-america",
+			2*Cent,
+		)
+		if err != nil {
+			return err
+		}
+
+		err = createWhatsappProduct(api, logger,
+			"Whatsapp Usage (Other regions) (2025)",
+			"whatsapp-other-region",
+			"other-regions",
+			10*Cent,
+		)
+		if err != nil {
+			return err
+		}
+
+		err = createMAUProduct(api, logger,
+			"Business Plan Additional MAU (2025)",
+			"business2025",
+			50*Dollar, 5000,
+			25000,
+		)
+		if err != nil {
+			return err
+		}
+
+		return
+	}),
+}

cmd/portal/cmd/cmdpricing/pricing.go

@@ -55,6 +55,9 @@ func init() {
 	binder.BindString(cmdPricingAppUpdate.Flags(), portalcmd.ArgFeatureConfigFilePath)
 	binder.BindString(cmdPricingAppUpdate.Flags(), portalcmd.ArgPlanNameForAppUpdate)
 
+	cmdPricing.AddCommand(cmdPricingCreateStripePlans2025)
+	binder.BindString(cmdPricingCreateStripePlans2025.Flags(), portalcmd.ArgStripeSecretKey)
+
 	portalcmd.Root.AddCommand(cmdPricing)
 }