add stripe custom currency

labring · Aug 9, 2023 · 00b0dc0 · 00b0dc0
1 parent 624ecac
commit 00b0dc0
Showing 1 changed file with 10 additions and 1 deletion.
diff --git 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
 	}