package entities

import (
	"time"

	"gorm.io/gorm"
)

// SubscriptionStatus 订阅状态枚举
type SubscriptionStatus string

const (
	SubscriptionStatusActive   SubscriptionStatus = "active"   // 活跃
	SubscriptionStatusInactive SubscriptionStatus = "inactive" // 非活跃
	SubscriptionStatusExpired  SubscriptionStatus = "expired"  // 已过期
	SubscriptionStatusCanceled SubscriptionStatus = "canceled" // 已取消
)

// Subscription 订阅实体
type Subscription struct {
	ID        string             `gorm:"primaryKey;type:varchar(36)" comment:"订阅ID"`
	UserID    string             `gorm:"type:varchar(36);not null;index" comment:"用户ID"`
	ProductID string             `gorm:"type:varchar(36);not null;index" comment:"产品ID"`
	Status    SubscriptionStatus `gorm:"type:varchar(20);not null;default:'active'" comment:"订阅状态"`
	Price     float64            `gorm:"type:decimal(10,2);not null" comment:"订阅价格"`
	APIUsed   int64              `gorm:"default:0" comment:"已使用API调用次数"`

	// 关联关系
	Product *Product `gorm:"foreignKey:ProductID" comment:"产品"`

	CreatedAt time.Time      `gorm:"autoCreateTime" comment:"创建时间"`
	UpdatedAt time.Time      `gorm:"autoUpdateTime" comment:"更新时间"`
	DeletedAt gorm.DeletedAt `gorm:"index" comment:"软删除时间"`
}

// IsValid 检查订阅是否有效
func (s *Subscription) IsValid() bool {
	return s.DeletedAt.Time.IsZero()
}

// IncrementAPIUsage 增加API使用次数
func (s *Subscription) IncrementAPIUsage(count int64) {
	s.APIUsed += count
}

// Activate 激活订阅
func (s *Subscription) Activate() {
	s.Status = SubscriptionStatusActive
}

// Deactivate 停用订阅
func (s *Subscription) Deactivate() {
	s.Status = SubscriptionStatusInactive
}

// ResetAPIUsage 重置API使用次数
func (s *Subscription) ResetAPIUsage() {
	s.APIUsed = 0
}