package service import ( "time" "gorm.io/gorm" "github.com/sundynix/pets-be/internal/model" ) // ProFeatures Pro 权益列表 var ProFeatures = []string{ "365 天养宠计划", "多宠物管理", "月度成长报告", "PDF 健康档案", "年度养宠账单", } // ProInfo Pro 状态 + 权益 type ProInfo struct { Status string `json:"status"` PlanType string `json:"plan_type"` Price float64 `json:"price"` EndDate *time.Time `json:"end_date"` Features []string `json:"features"` } // GetPro 取会员信息 func (s *Service) GetPro(userID string) (*ProInfo, error) { var m model.ProMembership err := s.db.Where("user_id = ?", userID).First(&m).Error if err == gorm.ErrRecordNotFound { return &ProInfo{Status: model.ProNone, Features: ProFeatures}, nil } if err != nil { return nil, err } return &ProInfo{Status: m.Status, PlanType: m.PlanType, Price: m.Price, EndDate: m.EndDate, Features: ProFeatures}, nil } // ActivatePro 开通年费会员 // ActivatePro 开通会员。 // ⚠️ 支付尚未接入:在接上微信支付、且校验到真实支付成功之前,这里必须直接拒绝, // 否则任何人调一次接口就白拿一年会员。接支付时把下面这段换成「校验支付单据后再写库」。 func (s *Service) ActivatePro(userID string) (*ProInfo, error) { return nil, ErrPayNotReady } // activateProAfterPaid 支付成功后真正写入会员状态(当前无人调用,等接支付时使用) func (s *Service) activateProAfterPaid(userID string) (*ProInfo, error) { now := time.Now() end := now.AddDate(1, 0, 0) var m model.ProMembership err := s.db.Where("user_id = ?", userID).First(&m).Error if err == gorm.ErrRecordNotFound { m = model.ProMembership{UserID: userID} } else if err != nil { return nil, err } m.Status = model.ProActive m.PlanType = "yearly" m.Price = 29.9 m.StartDate = &now m.EndDate = &end if err := s.db.Save(&m).Error; err != nil { return nil, err } return &ProInfo{Status: m.Status, PlanType: m.PlanType, Price: m.Price, EndDate: m.EndDate, Features: ProFeatures}, nil }