casdoor/pp/stripe.go

// Copyright 2023 The Casdoor Authors. All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package pp

import (
	"fmt"
	"time"

	"github.com/casdoor/casdoor/conf"
	"github.com/stripe/stripe-go/v74"
	stripeCheckout "github.com/stripe/stripe-go/v74/checkout/session"
	stripeIntent "github.com/stripe/stripe-go/v74/paymentintent"
	stripePrice "github.com/stripe/stripe-go/v74/price"
	stripeProduct "github.com/stripe/stripe-go/v74/product"
)

type StripePaymentProvider struct {
	PublishableKey string
	SecretKey      string
	isProd         bool
}

func NewStripePaymentProvider(PublishableKey, SecretKey string) (*StripePaymentProvider, error) {
	isProd := false
	if conf.GetConfigString("runmode") == "prod" {
		isProd = true
	}
	pp := &StripePaymentProvider{
		PublishableKey: PublishableKey,
		SecretKey:      SecretKey,
		isProd:         isProd,
	}
	stripe.Key = pp.SecretKey
	return pp, nil
}

func (pp *StripePaymentProvider) Pay(r *PayReq) (*PayResp, error) {
	// Create a temp product
	description := joinAttachString([]string{r.ProductName, r.ProductDisplayName, r.ProviderName})
	productParams := &stripe.ProductParams{
		Name:        stripe.String(r.ProductDisplayName),
		Description: stripe.String(description),
		DefaultPriceData: &stripe.ProductDefaultPriceDataParams{
			UnitAmount: stripe.Int64(priceFloat64ToInt64(r.Price)),
			Currency:   stripe.String(r.Currency),
		},
	}
	sProduct, err := stripeProduct.New(productParams)
	if err != nil {
		return nil, err
	}
	// Create a price for an existing product
	priceParams := &stripe.PriceParams{
		Currency:   stripe.String(r.Currency),
		UnitAmount: stripe.Int64(priceFloat64ToInt64(r.Price)),
		Product:    stripe.String(sProduct.ID),
	}
	sPrice, err := stripePrice.New(priceParams)
	if err != nil {
		return nil, err
	}
	// Create a Checkout Session
	checkoutParams := &stripe.CheckoutSessionParams{
		LineItems: []*stripe.CheckoutSessionLineItemParams{
			{
				Price:    stripe.String(sPrice.ID),
				Quantity: stripe.Int64(1),
			},
		},
		Mode:              stripe.String(string(stripe.CheckoutSessionModePayment)),
		SuccessURL:        stripe.String(r.ReturnUrl),
		CancelURL:         stripe.String(r.ReturnUrl),
		ClientReferenceID: stripe.String(r.PaymentName),
		ExpiresAt:         stripe.Int64(time.Now().Add(30 * time.Minute).Unix()),
	}
	checkoutParams.AddMetadata("product_description", description)
	sCheckout, err := stripeCheckout.New(checkoutParams)
	if err != nil {
		return nil, err
	}
	payResp := &PayResp{
		PayUrl:  sCheckout.URL,
		OrderId: sCheckout.ID,
	}
	return payResp, nil
}

func (pp *StripePaymentProvider) Notify(body []byte, orderId string) (*NotifyResult, error) {
	notifyResult := &NotifyResult{}
	sCheckout, err := stripeCheckout.Get(orderId, nil)
	if err != nil {
		return nil, err
	}
	switch sCheckout.Status {
	case "open":
		// The checkout session is still in progress. Payment processing has not started
		notifyResult.PaymentStatus = PaymentStateCreated
		return notifyResult, nil
	case "complete":
		// The checkout session is complete. Payment processing may still be in progress
	case "expired":
		// The checkout session has expired. No further processing will occur
		notifyResult.PaymentStatus = PaymentStateTimeout
		return notifyResult, nil
	default:
		notifyResult.PaymentStatus = PaymentStateError
		notifyResult.NotifyMessage = fmt.Sprintf("unexpected stripe checkout status: %v", sCheckout.Status)
		return notifyResult, nil
	}
	switch sCheckout.PaymentStatus {
	case "paid":
		// Skip
	case "unpaid":
		notifyResult.PaymentStatus = PaymentStateCreated
		return notifyResult, nil
	default:
		notifyResult.PaymentStatus = PaymentStateError
		notifyResult.NotifyMessage = fmt.Sprintf("unexpected stripe checkout payment status: %v", sCheckout.PaymentStatus)
		return notifyResult, nil
	}
	// Once payment is successful, the Checkout Session will contain a reference to the successful `PaymentIntent`
	sIntent, err := stripeIntent.Get(sCheckout.PaymentIntent.ID, nil)
	var (
		productName        string
		productDisplayName string
		providerName       string
	)
	if description, ok := sCheckout.Metadata["product_description"]; ok {
		productName, productDisplayName, providerName, _ = parseAttachString(description)
	}
	notifyResult = &NotifyResult{
		PaymentName:   sCheckout.ClientReferenceID,
		PaymentStatus: PaymentStatePaid,

		ProductName:        productName,
		ProductDisplayName: productDisplayName,
		ProviderName:       providerName,

		Price:    priceInt64ToFloat64(sIntent.Amount),
		Currency: string(sIntent.Currency),

		OrderId: orderId,
	}
	return notifyResult, nil
}

func (pp *StripePaymentProvider) GetInvoice(paymentName string, personName string, personIdCard string, personEmail string, personPhone string, invoiceType string, invoiceTitle string, invoiceTaxId string) (string, error) {
	return "", nil
}

func (pp *StripePaymentProvider) GetResponseError(err error) string {
	if err == nil {
		return "success"
	} else {
		return "fail"
	}
}