sundynix-pets/pets-be/internal/service/pro.go

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 uint) (*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 开通年费会员
func (s *Service) ActivatePro(userID uint) (*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
}