web_huangguo2_h5/app/service/authser/vip.go

package authser

import (
	"strconv"
	"strings"
	"time"

	"github.com/gin-gonic/gin"
)

func parseVipExpireMs(raw string) int64 {
	raw = strings.TrimSpace(raw)
	if raw == "" {
		return 0
	}
	if n, err := strconv.ParseInt(raw, 10, 64); err == nil {
		if n > 0 && n < 1e12 {
			return n * 1000
		}
		return n
	}
	layouts := []string{
		time.RFC3339Nano,
		time.RFC3339,
		"2006-01-02T15:04:05Z07:00",
		"2006-01-02 15:04:05",
	}
	for _, layout := range layouts {
		if t, err := time.Parse(layout, raw); err == nil {
			return t.UnixMilli()
		}
	}
	return 0
}

// UserIDFromCookies 读取 cookie userId。
func UserIDFromCookies(c *gin.Context) int64 {
	if c == nil {
		return 0
	}
	n, _ := strconv.ParseInt(strings.TrimSpace(cookieValue(c, "userId")), 10, 64)
	if n < 0 {
		return 0
	}
	return n
}

// VipTypeFromCookies 读取 cookie vipType。
func VipTypeFromCookies(c *gin.Context) int {
	if c == nil {
		return 0
	}
	n, _ := strconv.Atoi(strings.TrimSpace(cookieValue(c, "vipType")))
	return n
}

// VipTypeLabel vipType 文案：0普通 1月卡 2年卡（与 91ly / 编辑资料注释一致）。
func VipTypeLabel(vipType int) string {
	switch vipType {
	case 1:
		return "月卡会员"
	case 2:
		return "年卡会员"
	default:
		return "普通会员"
	}
}

// ShowCardNameFromCookies 读取 cookie showCardName。
func ShowCardNameFromCookies(c *gin.Context) string {
	if c == nil {
		return ""
	}
	return strings.TrimSpace(cookieValue(c, "showCardName"))
}

// ShowCardIDFromCookies 读取 cookie showCardId。
func ShowCardIDFromCookies(c *gin.Context) int64 {
	if c == nil {
		return 0
	}
	n, _ := strconv.ParseInt(strings.TrimSpace(cookieValue(c, "showCardId")), 10, 64)
	if n < 0 {
		return 0
	}
	return n
}

// VipDisplayLabel 优先使用 getInfo.showCardName，否则回退 vipType 文案。
func VipDisplayLabel(showCardName string, vipType int) string {
	if s := strings.TrimSpace(showCardName); s != "" {
		return s
	}
	return VipTypeLabel(vipType)
}

// VipDisplayLabelFromCookies 从 cookie 组装会员展示名。
func VipDisplayLabelFromCookies(c *gin.Context) string {
	return VipDisplayLabel(ShowCardNameFromCookies(c), VipTypeFromCookies(c))
}

// IsActiveVipUser 根据 vipType + currentVipExpire 判断是否有效会员。
func IsActiveVipUser(vipType int, currentVipExpire string) bool {
	if vipType == 0 {
		return false
	}
	expireMs := parseVipExpireMs(currentVipExpire)
	if expireMs <= 0 {
		return false
	}
	return expireMs > time.Now().UnixMilli()
}

// FormatVipExpireText 格式化到期时间展示。
func FormatVipExpireText(currentVipExpire string) string {
	ms := parseVipExpireMs(currentVipExpire)
	if ms <= 0 {
		return ""
	}
	return time.UnixMilli(ms).In(time.Local).Format("有效期至 2006/01/02")
}

// VipExpireTextFromCookies 格式化到期时间展示。
func VipExpireTextFromCookies(c *gin.Context) string {
	if c == nil {
		return ""
	}
	return FormatVipExpireText(cookieValue(c, "currentVipExpire"))
}

// IsActiveVipFromCookies vipType≠0 且 currentVipExpire 未过期（与 91ly 一致）。
func IsActiveVipFromCookies(c *gin.Context) bool {
	if c == nil {
		return false
	}
	return IsActiveVipUser(VipTypeFromCookies(c), cookieValue(c, "currentVipExpire"))
}

// RechargeFromCookies 读取 cookie recharge（金币余额）。
func RechargeFromCookies(c *gin.Context) int64 {
	if c == nil {
		return 0
	}
	n, _ := strconv.ParseInt(strings.TrimSpace(cookieValue(c, "recharge")), 10, 64)
	if n < 0 {
		return 0
	}
	return n
}

// SetRechargeCookie 更新金币余额 cookie（解锁/充值后同步）。
func SetRechargeCookie(c *gin.Context, balance int64) {
	if c == nil {
		return
	}
	if balance < 0 {
		balance = 0
	}
	secure := c.Request != nil && c.Request.TLS != nil
	setCookie(c, "recharge", strconv.FormatInt(balance, 10), secure)
}

// FormatRecharge 金币余额千分位展示。
func FormatRecharge(n int64) string {
	if n < 0 {
		n = 0
	}
	s := strconv.FormatInt(n, 10)
	if len(s) <= 3 {
		return s
	}
	var b strings.Builder
	rem := len(s) % 3
	if rem > 0 {
		b.WriteString(s[:rem])
		if len(s) > rem {
			b.WriteByte(',')
		}
	}
	for i := rem; i < len(s); i += 3 {
		b.WriteString(s[i : i+3])
		if i+3 < len(s) {
			b.WriteByte(',')
		}
	}
	return b.String()
}