- stripe_session_id -> stripe_invoice_id

- Add priceToPlan
- Add /api/subscription
- Add HandleInvoicePaid

Author: BretRen

cmd/web/main.go

@@ -15,7 +15,14 @@ import (
 	"github.com/stripe/stripe-go/v84"
 )
 
-// TODO: 提供升级订阅的选择
+// TODO: 禁止重复订阅
+// TODO: 防止 /checkout/success 被滥用
+// TODO: 日志
+// TODO: 创建订阅集合迁移文件
+// TODO: 创建用户集合迁移文件
+// TODO: 用.env初始化SMTP和设置
+// TODO: 添加限速
+// TODO: 发送各种邮件
 
 const version string = "v1.0.0-alpha"
 

config/config.go

@@ -8,18 +8,31 @@ type Config struct {
 	StripeKey     string
 	StripeSignKey string
 	PlanToPrice   map[string]string
+
+	PriceToPlan map[string]string
 }
 
 func New() *Config {
+	// 原始映射
+	planToPrice := map[string]string{
+		"starter": getEnv("STRIPE_PLAN_STARTER", ""),
+		"pro":     getEnv("STRIPE_PLAN_PRO", ""),
+		"plus":    getEnv("STRIPE_PLAN_PLUS", ""),
+	}
+
+	// 自动生成反向映射
+	priceToPlan := make(map[string]string)
+	for plan, price := range planToPrice {
+		if price != "" {
+			priceToPlan[price] = plan
+		}
+	}
+
 	return &Config{
 		StripeKey:     getEnv("STRIPE_KEY", ""),
 		StripeSignKey: getEnv("STRIPE_SIGN_KEY", ""),
-		// 动态从环境变量读取价格 ID
-		PlanToPrice: map[string]string{
-			"starter": os.Getenv("STRIPE_PLAN_STARTER"),
-			"pro":     os.Getenv("STRIPE_PLAN_PRO"),
-			"plus":    os.Getenv("STRIPE_PLAN_PLUS"),
-		},
+		PlanToPrice:   planToPrice,
+		PriceToPlan:   priceToPlan,
 	}
 }
 
@@ -28,5 +41,9 @@ func getEnv(key, defaultValue string) string {
 		return value
 	}
 
+	if defaultValue == "" {
+		panic("environment variable '" + key + "' has not been set, and is required")
+	}
+
 	return defaultValue
 }

internal/subscriptions/handler.go

@@ -1,12 +1,16 @@
 package subscriptions
 
 import (
+	"database/sql"
+	"encoding/json"
+	"errors"
 	"fmt"
 	"io"
 	"net/http"
 
 	"github.com/pocketbase/pocketbase/apis"
 	"github.com/pocketbase/pocketbase/core"
+	"github.com/stripe/stripe-go/v84"
 	"github.com/stripe/stripe-go/v84/webhook"
 )
 
@@ -50,14 +54,67 @@ func (h *SubscriptionHandler) StripeWebhook(e *core.RequestEvent) error {
 		return e.BadRequestError("Read body failed", nil)
 	}
 
-	event, err := webhook.ConstructEvent(payload, e.Request.Header.Get("Stripe-Signature"), h.service.cfg.StripeSignKey)
+	event, err := webhook.ConstructEventWithOptions(
+		payload,
+		e.Request.Header.Get("Stripe-Signature"),
+		h.service.cfg.StripeSignKey,
+		webhook.ConstructEventOptions{
+			IgnoreAPIVersionMismatch: true, // 忽略版本不一致报错
+		},
+	)
 	if err != nil {
+		fmt.Println(err)
 		return e.BadRequestError("Invalid signature", nil)
 	}
 
-	if event.Type == "checkout.session.completed" {
-		// ... 反序列化并调用 h.service.HandleCheckoutCompleted(sess)
+	switch event.Type {
+	case "checkout.session.completed":
+		var session stripe.CheckoutSession
+		err := json.Unmarshal(event.Data.Raw, &session)
+		if err != nil {
+			fmt.Println(err)
+			return e.BadRequestError("JSON unmarshal failed", nil)
+		}
+
+		// 🌟 调用 Service 层处理业务（如更新用户订阅状态、发货等）
+		// 传入 e.App (PocketBase 实例) 以便在 Service 里操作数据库
+		if err := h.service.HandleCheckoutCompleted(session); err != nil {
+			fmt.Println(err)
+
+			return e.InternalServerError("Handle checkout failed", err)
+		}
+	case "invoice.paid":
+
+		var inv stripe.Invoice
+		err := json.Unmarshal(event.Data.Raw, &inv)
+		if err != nil {
+			fmt.Println(err)
+			return e.BadRequestError("Parsing invoice failed", err)
+		}
+
+		err = h.service.HandleInvoicePaid(inv)
+		if err != nil {
+			fmt.Println(err)
+			return e.InternalServerError("Handle checkout failed", err)
+		}
+
 	}
 
 	return e.NoContent(http.StatusOK)
 }
+
+func (h *SubscriptionHandler) CheckSubscription(e *core.RequestEvent) error {
+	subscription, err := h.service.CheckValidSubscription(e.Auth.Original())
+
+	if errors.Is(err, sql.ErrNoRows) {
+
+		return e.BadRequestError("No subscription", nil)
+	}
+
+	if err != nil {
+
+		return e.InternalServerError("Check subscription failed", err)
+	}
+
+	return e.JSON(http.StatusOK, subscription)
+}

internal/subscriptions/router.go

@@ -15,4 +15,5 @@ func RegisterRoutes(app *pocketbase.PocketBase, se *core.ServeEvent, cfg *config
 	// 路由注册
 	se.Router.POST("/api/webhook/stripe", handler.StripeWebhook)
 	se.Router.POST("/api/checkout/subscription", handler.Checkout).Bind(apis.RequireAuth())
+	se.Router.GET("/api/subscription", handler.CheckSubscription).Bind(apis.RequireAuth())
 }

internal/subscriptions/service.go

@@ -1,15 +1,16 @@
 package subscriptions
 
 import (
+	"errors"
 	"fmt"
 	"time"
 	"website-pb/config"
 
+	"github.com/pocketbase/dbx"
 	"github.com/pocketbase/pocketbase"
 	"github.com/pocketbase/pocketbase/core"
 	"github.com/stripe/stripe-go/v84"
 	"github.com/stripe/stripe-go/v84/checkout/session"
-	"github.com/stripe/stripe-go/v84/subscription"
 )
 
 type SubscriptionService struct {
@@ -21,6 +22,31 @@ func NewService(app *pocketbase.PocketBase, cfg *config.Config) *SubscriptionSer
 	return &SubscriptionService{app: app, cfg: cfg}
 }
 
+func (s *SubscriptionService) CheckValidSubscription(user *core.Record) (*core.Record, error) {
+
+	now := time.Now().UTC().Format("2006-01-02 15:04:05.000Z")
+
+	record := &core.Record{}
+
+	err := s.app.RecordQuery("subscriptions").
+		// 1. 基础过滤：用户 ID
+		AndWhere(dbx.HashExp{"user_id": user.Id}).
+		// 2. 时间过滤：未过期
+		AndWhere(dbx.NewExp("expires_at > {:now}", dbx.Params{"now": now})).
+		// 3. 排序：将到期时间最远的排在最前面
+		OrderBy("expires_at DESC").
+		// 4. 只取一条
+		Limit(1).
+		// 5. 将结果映射到 record 对象
+		One(record)
+
+	if err != nil {
+		return nil, err // 没找到会返回 sql.ErrNoRows
+	}
+
+	return record, nil
+}
+
 // CreateCheckoutSession 处理 Stripe 会话创建
 func (s *SubscriptionService) CreateCheckoutSession(user *core.Record, plan string, frontendURL string) (string, error) {
 	priceID, exists := s.cfg.PlanToPrice[plan]
@@ -33,8 +59,11 @@ func (s *SubscriptionService) CreateCheckoutSession(user *core.Record, plan stri
 		CancelURL:         stripe.String(frontendURL),
 		Mode:              stripe.String(string(stripe.CheckoutSessionModeSubscription)),
 		ClientReferenceID: stripe.String(user.Id),
-		Metadata:          map[string]string{"plan": plan},
-		LineItems:         []*stripe.CheckoutSessionLineItemParams{{Price: stripe.String(priceID), Quantity: stripe.Int64(1)}},
+		SubscriptionData: &stripe.CheckoutSessionSubscriptionDataParams{
+			Metadata: map[string]string{"plan": "pro"}, // 存到订阅对象里
+		},
+		Metadata:  map[string]string{"plan": plan},
+		LineItems: []*stripe.CheckoutSessionLineItemParams{{Price: stripe.String(priceID), Quantity: stripe.Int64(1)}},
 	}
 
 	// 关联已有 Stripe Customer ID
@@ -48,28 +77,63 @@ func (s *SubscriptionService) CreateCheckoutSession(user *core.Record, plan stri
 	return sess.URL, err
 }
 
-// HandleCheckoutCompleted 处理支付成功后的数据更新
-func (s *SubscriptionService) HandleCheckoutCompleted(sess stripe.CheckoutSession) error {
-	// 1. 获取用户
-	user, err := s.app.FindRecordById("users", sess.ClientReferenceID)
+func (s *SubscriptionService) HandleInvoicePaid(inv stripe.Invoice) error {
+
+	if inv.Customer == nil {
+		fmt.Println("invoice customer is nil")
+		return errors.New("invoice customer is nil")
+	}
+
+	if len(inv.Lines.Data) == 0 {
+		fmt.Println("invoice has no lines")
+
+		return errors.New("invoice has no lines")
+	}
+
+	stripeCustomerID := inv.Customer.ID
+	user, err := s.app.FindFirstRecordByFilter("users", "stripe_customer_id = {:id}", map[string]any{"id": stripeCustomerID})
 	if err != nil {
 		return err
 	}
 
-	// 2. 更新用户信息
-	user.Set("stripe_customer_id", sess.Customer.ID)
-	user.Set("plan", sess.Metadata["plan"])
-	if err := s.app.Save(user); err != nil {
-		return err
+	collection, err := s.app.FindCollectionByNameOrId("subscriptions")
+	if err != nil {
+		return errors.New("subscriptions collection not found")
 	}
 
-	// 3. 创建订阅记录
-	sub, _ := subscription.Get(sess.Subscription.ID, nil)
-	collection, _ := s.app.FindCollectionByNameOrId("subscriptions")
 	record := core.NewRecord(collection)
+
+	fmt.Println(inv.Lines.Data[0].Pricing.PriceDetails.Price)
+
+	priceID := inv.Lines.Data[0].Pricing.PriceDetails.Price
+
+	priceIDMap := s.cfg.PriceToPlan[priceID]
+	fmt.Println(priceID)
+
+	if priceIDMap == "" {
+		return errors.New("invalid price")
+	}
+
+	expiresAt := time.Unix(inv.Lines.Data[0].Period.End, 0).UTC()
+	fmt.Println("Expires at:", expiresAt)
+	fmt.Println(inv.Lines.Data[0].Period.End)
+
 	record.Set("user_id", user.Id)
-	record.Set("plan", sess.Metadata["plan"])
-	record.Set("expires_at", time.Unix(sub.Items.Data[0].CurrentPeriodEnd, 0).UTC())
+	record.Set("plan", priceIDMap)
+	record.Set("stripe_invoice_id", inv.ID)
+	record.Set("expires_at", expiresAt)
 
 	return s.app.Save(record)
 }
+
+func (s *SubscriptionService) HandleCheckoutCompleted(sess stripe.CheckoutSession) error {
+	user, err := s.app.FindRecordById("users", sess.ClientReferenceID)
+	if err != nil {
+		return err
+	}
+
+	// 2. 更新用户信息
+	user.Set("stripe_customer_id", sess.Customer.ID)
+
+	return s.app.Save(user)
+}

migrations/1769317614_collections_snapshot.go

@@ -848,7 +848,7 @@ func init() {
 						"id": "text439437911",
 						"max": 0,
 						"min": 0,
-						"name": "stripe_session_id",
+						"name": "stripe_invoice_id",
 						"pattern": "",
 						"presentable": false,
 						"primaryKey": false,
@@ -890,7 +890,7 @@ func init() {
 				],
 				"id": "pbc_4099809493",
 				"indexes": [
-					"CREATE UNIQUE INDEX ` + "`" + `idx_Ey9oa0Co7S` + "`" + ` ON ` + "`" + `subscriptions` + "`" + ` (` + "`" + `stripe_session_id` + "`" + `)"
+					"CREATE UNIQUE INDEX ` + "`" + `idx_Ey9oa0Co7S` + "`" + ` ON ` + "`" + `subscriptions` + "`" + ` (` + "`" + `stripe_invoice_id` + "`" + `)"
 				],
 				"listRule": null,
 				"name": "subscriptions",