diff --git a/pkg/pay/stripe.go b/pkg/pay/stripe.go index ec807f85f04..faa419d025b 100644 --- a/pkg/pay/stripe.go +++ b/pkg/pay/stripe.go @@ -17,6 +17,7 @@ package pay import ( "fmt" "os" + "strings" "github.com/labring/sealos/controllers/pkg/utils" @@ -28,16 +29,24 @@ var DefaultURL = fmt.Sprintf("https://%s", utils.GetEnvWithDefault("DOMAIN", Def const ( stripeSuccessPostfix = "STRIPE_SUCCESS_POSTFIX" stripeCancelPostfix = "STRIPE_CANCEL_POSTFIX" + stripeCurrency = "STRIPE_CURRENCY" ) +var Currency string + func init() { if port := os.Getenv("PORT"); port != "" { DefaultURL = fmt.Sprintf("%s:%s", DefaultURL, port) } + currency := strings.ToLower(strings.TrimSpace(os.Getenv(stripeCurrency))) + if currency != USD { + currency = CNY + } + Currency = currency } func (s StripePayment) CreatePayment(amount int64, _ string) (string, string, error) { - session, err := CreateCheckoutSession(amount, CNY, DefaultURL+os.Getenv(stripeSuccessPostfix), DefaultURL+os.Getenv(stripeCancelPostfix)) + session, err := CreateCheckoutSession(amount, Currency, DefaultURL+os.Getenv(stripeSuccessPostfix), DefaultURL+os.Getenv(stripeCancelPostfix)) if err != nil { return "", "", err }