package models

import (
	"strings"
	"time"

	"gorm.io/gorm"
)

// Tag 表示支持多语言的博客文章标签。
type Tag struct {
	ID        uint      `gorm:"primarykey" json:"id"`
	NameZh    string    `gorm:"size:50;not null" json:"name_zh"`
	NameEn    string    `gorm:"size:50;not null" json:"name_en"`
	Slug      string    `gorm:"size:100;uniqueIndex;not null" json:"slug"`
	Count     int       `gorm:"default:0" json:"count"`
	CreatedAt time.Time `json:"created_at"`
	UpdatedAt time.Time `json:"updated_at"`
}

// TableName 覆盖 GORM 默认的表名。
func (Tag) TableName() string {
	return "tags"
}

// Name 返回指定语言下的标签名称。
func (t *Tag) Name(lang string) string {
	if lang == "zh" {
		return t.NameZh
	}
	return t.NameEn
}

// generateTagSlug 根据标签名称生成对 URL 友好的 slug。
func generateTagSlug(name string) string {
	slug := strings.ToLower(strings.TrimSpace(name))
	slug = strings.ReplaceAll(slug, " ", "-")
	slug = strings.ReplaceAll(slug, "_", "-")
	return slug
}

// FindOrCreateTag 按名称查找标签，不存在则创建。
// 若同时提供 nameZh 与 nameEn 则分别使用；否则两种语言使用相同的名称。
func FindOrCreateTag(db *gorm.DB, nameZh, nameEn string) (*Tag, error) {
	nameZh = strings.TrimSpace(nameZh)
	nameEn = strings.TrimSpace(nameEn)

	// 若只提供其中一个名称，两种语言都使用它
	if nameZh == "" && nameEn != "" {
		nameZh = nameEn
	} else if nameEn == "" && nameZh != "" {
		nameEn = nameZh
	}

	if nameZh == "" {
		return nil, nil
	}

	slug := generateTagSlug(nameZh)

	var tag Tag
	err := db.Where("slug = ?", slug).First(&tag).Error
	if err == nil {
		return &tag, nil
	}

	if err != gorm.ErrRecordNotFound {
		return nil, err
	}

	// 创建新标签
	tag = Tag{
		NameZh: nameZh,
		NameEn: nameEn,
		Slug:   slug,
		Count:  0,
	}

	if err := db.Create(&tag).Error; err != nil {
		return nil, err
	}

	return &tag, nil
}

// GetAllTags 按数量降序返回所有标签。
func GetAllTags(db *gorm.DB) ([]Tag, error) {
	var tags []Tag
	err := db.Order("count DESC, name_zh ASC").Find(&tags).Error
	return tags, err
}

// GetTagBySlug 根据 slug 返回标签。
func GetTagBySlug(db *gorm.DB, slug string) (*Tag, error) {
	var tag Tag
	err := db.Where("slug = ?", slug).First(&tag).Error
	if err != nil {
		return nil, err
	}
	return &tag, nil
}

// UpdateTagCount 重新计算某个标签的文章数量。
func UpdateTagCount(db *gorm.DB, tagID uint) error {
	var count int64
	db.Table("article_tags").Where("tag_id = ?", tagID).Count(&count)
	return db.Model(&Tag{}).Where("id = ?", tagID).Update("count", count).Error
}

// UpdateAllTagCounts 重新计算所有标签的文章数量。
func UpdateAllTagCounts(db *gorm.DB) error {
	// 获取所有标签
	var tags []Tag
	if err := db.Find(&tags).Error; err != nil {
		return err
	}

	// 为每个标签更新数量
	for _, tag := range tags {
		if err := UpdateTagCount(db, tag.ID); err != nil {
			return err
		}
	}

	return nil
}