package models

import (
	"time"

	"github.com/beego/beego/v2/client/orm"
)

// Knowledge 知识库模型
type Knowledge struct {
	Id           int       `orm:"column(knowledge_id);pk;auto" json:"id"`
	Title        string    `orm:"column(title);size(200)" json:"title"`
	CategoryId   int       `orm:"column(category_id);default(0);null" json:"categoryId"`
	CategoryName string    `orm:"-" json:"categoryName"` // 不映射到数据库，从联查获取
	Tags         string    `orm:"column(tags);type(text);null" json:"tags"`
	Author       string    `orm:"column(author);size(50)" json:"author"`
	Content      string    `orm:"column(content);type(longtext)" json:"content"`
	Summary      string    `orm:"column(summary);size(500);null" json:"summary"`
	CoverUrl     string    `orm:"column(cover_url);size(500);null" json:"coverUrl"`
	Status       int8      `orm:"column(status);default(0)" json:"status"`
	ViewCount    int       `orm:"column(view_count);default(0)" json:"viewCount"`
	LikeCount    int       `orm:"column(like_count);default(0)" json:"likeCount"`
	IsRecommend  int8      `orm:"column(is_recommend);default(0)" json:"isRecommend"`
	IsTop        int8      `orm:"column(is_top);default(0)" json:"isTop"`
	CreateTime   time.Time `orm:"column(create_time);auto_now_add;type(datetime)" json:"createTime"`
	UpdateTime   time.Time `orm:"column(update_time);auto_now;type(datetime)" json:"updateTime"`
	CreateBy     string    `orm:"column(create_by);size(50);null" json:"createBy"`
	UpdateBy     string    `orm:"column(update_by);size(50);null" json:"updateBy"`
	DeleteTime   time.Time `orm:"column(delete_time);type(datetime);null" json:"deleteTime"`
	DeleteBy     string    `orm:"column(delete_by);size(50);null" json:"deleteBy"`
}

// TableName 设置表名
func (k *Knowledge) TableName() string {
	return "yz_knowledge"
}

// KnowledgeCategory 知识库分类模型
type KnowledgeCategory struct {
	CategoryId   int       `orm:"column(category_id);pk;auto" json:"categoryId"`
	CategoryName string    `orm:"column(category_name);size(100)" json:"categoryName"`
	CategoryDesc string    `orm:"column(category_desc);size(500);null" json:"categoryDesc"`
	ParentId     int       `orm:"column(parent_id);default(0)" json:"parentId"`
	SortOrder    int       `orm:"column(sort_order);default(0)" json:"sortOrder"`
	CreateTime   time.Time `orm:"column(create_time);auto_now_add;type(datetime)" json:"createTime"`
	UpdateTime   time.Time `orm:"column(update_time);auto_now;type(datetime)" json:"updateTime"`
}

// TableName 设置表名
func (kc *KnowledgeCategory) TableName() string {
	return "yz_knowledge_category"
}

// KnowledgeTag 知识库标签模型
type KnowledgeTag struct {
	TagId      int       `orm:"column(tag_id);pk;auto" json:"tagId"`
	TagName    string    `orm:"column(tag_name);size(50);unique" json:"tagName"`
	TagColor   string    `orm:"column(tag_color);size(20);null" json:"tagColor"`
	UsageCount int       `orm:"column(usage_count);default(0)" json:"usageCount"`
	CreateTime time.Time `orm:"column(create_time);auto_now_add;type(datetime)" json:"createTime"`
	UpdateTime time.Time `orm:"column(update_time);auto_now;type(datetime)" json:"updateTime"`
}

// TableName 设置表名
func (kt *KnowledgeTag) TableName() string {
	return "yz_knowledge_tags"
}

// AddKnowledge 创建知识
func AddKnowledge(k *Knowledge) (int64, error) {
	o := orm.NewOrm()
	id, err := o.Insert(k)
	return id, err
}

// GetKnowledgeById 根据ID获取知识详情（使用联查获取分类名称）
func GetKnowledgeById(id int) (*Knowledge, error) {
	o := orm.NewOrm()

	// 使用联查获取分类名称（只查询未删除的记录）
	querySQL := `
		SELECT k.*, c.category_name 
		FROM yz_knowledge k 
		LEFT JOIN yz_knowledge_category c ON k.category_id = c.category_id 
		WHERE k.knowledge_id = ? AND k.delete_time IS NULL
	`

	var result struct {
		Id           int       `orm:"column(knowledge_id)"`
		Title        string    `orm:"column(title)"`
		CategoryId   int       `orm:"column(category_id)"`
		CategoryName string    `orm:"column(category_name)"`
		Tags         string    `orm:"column(tags)"`
		Author       string    `orm:"column(author)"`
		Content      string    `orm:"column(content)"`
		Summary      string    `orm:"column(summary)"`
		CoverUrl     string    `orm:"column(cover_url)"`
		Status       int8      `orm:"column(status)"`
		ViewCount    int       `orm:"column(view_count)"`
		LikeCount    int       `orm:"column(like_count)"`
		IsRecommend  int8      `orm:"column(is_recommend)"`
		IsTop        int8      `orm:"column(is_top)"`
		CreateTime   time.Time `orm:"column(create_time)"`
		UpdateTime   time.Time `orm:"column(update_time)"`
		CreateBy     string    `orm:"column(create_by)"`
		UpdateBy     string    `orm:"column(update_by)"`
	}

	err := o.Raw(querySQL, id).QueryRow(&result)
	if err != nil {
		return nil, err
	}

	knowledge := &Knowledge{
		Id:           result.Id,
		Title:        result.Title,
		CategoryId:   result.CategoryId,
		CategoryName: result.CategoryName,
		Tags:         result.Tags,
		Author:       result.Author,
		Content:      result.Content,
		Summary:      result.Summary,
		CoverUrl:     result.CoverUrl,
		Status:       result.Status,
		ViewCount:    result.ViewCount,
		LikeCount:    result.LikeCount,
		IsRecommend:  result.IsRecommend,
		IsTop:        result.IsTop,
		CreateTime:   result.CreateTime,
		UpdateTime:   result.UpdateTime,
		CreateBy:     result.CreateBy,
		UpdateBy:     result.UpdateBy,
	}

	// 增加查看次数
	knowledge.ViewCount++
	_, err = o.Raw("UPDATE yz_knowledge SET view_count = ? WHERE knowledge_id = ?", knowledge.ViewCount, id).Exec()

	return knowledge, err
}

// GetAllKnowledge 获取所有知识（支持分页和筛选）
func GetAllKnowledge(page, pageSize int, status int8, categoryId int, keyword string) ([]*Knowledge, int64, error) {
	o := orm.NewOrm()

	// 构建 WHERE 条件（只查询未删除的记录）
	whereSQL := "delete_time IS NULL"
	params := []interface{}{}

	// 状态筛选
	if status >= 0 {
		whereSQL += " AND status = ?"
		params = append(params, status)
	}

	// 分类筛选
	if categoryId > 0 {
		whereSQL += " AND category_id = ?"
		params = append(params, categoryId)
	}

	// 关键词搜索
	if keyword != "" {
		whereSQL += " AND title LIKE ?"
		params = append(params, "%"+keyword+"%")
	}

	// 获取总数
	var total int64
	countSQL := "SELECT COUNT(*) FROM yz_knowledge WHERE " + whereSQL
	err := o.Raw(countSQL, params...).QueryRow(&total)
	if err != nil {
		return nil, 0, err
	}

	// 联查获取分类名称
	querySQL := `
		SELECT k.*, c.category_name 
		FROM yz_knowledge k 
		LEFT JOIN yz_knowledge_category c ON k.category_id = c.category_id 
		WHERE ` + whereSQL + ` 
		ORDER BY k.is_top DESC, k.create_time DESC 
		LIMIT ? OFFSET ?
	`
	params = append(params, pageSize, (page-1)*pageSize)

	var knowledges []*Knowledge
	var results []struct {
		Id           int       `orm:"column(knowledge_id)"`
		Title        string    `orm:"column(title)"`
		CategoryId   int       `orm:"column(category_id)"`
		CategoryName string    `orm:"column(category_name)"`
		Tags         string    `orm:"column(tags)"`
		Author       string    `orm:"column(author)"`
		Content      string    `orm:"column(content)"`
		Summary      string    `orm:"column(summary)"`
		CoverUrl     string    `orm:"column(cover_url)"`
		Status       int8      `orm:"column(status)"`
		ViewCount    int       `orm:"column(view_count)"`
		LikeCount    int       `orm:"column(like_count)"`
		IsRecommend  int8      `orm:"column(is_recommend)"`
		IsTop        int8      `orm:"column(is_top)"`
		CreateTime   time.Time `orm:"column(create_time)"`
		UpdateTime   time.Time `orm:"column(update_time)"`
		CreateBy     string    `orm:"column(create_by)"`
		UpdateBy     string    `orm:"column(update_by)"`
	}

	_, err = o.Raw(querySQL, params...).QueryRows(&results)
	if err != nil {
		return nil, 0, err
	}

	// 转换结果
	for _, r := range results {
		k := &Knowledge{
			Id:           r.Id,
			Title:        r.Title,
			CategoryId:   r.CategoryId,
			CategoryName: r.CategoryName,
			Tags:         r.Tags,
			Author:       r.Author,
			Content:      r.Content,
			Summary:      r.Summary,
			CoverUrl:     r.CoverUrl,
			Status:       r.Status,
			ViewCount:    r.ViewCount,
			LikeCount:    r.LikeCount,
			IsRecommend:  r.IsRecommend,
			IsTop:        r.IsTop,
			CreateTime:   r.CreateTime,
			UpdateTime:   r.UpdateTime,
			CreateBy:     r.CreateBy,
			UpdateBy:     r.UpdateBy,
		}
		knowledges = append(knowledges, k)
	}

	return knowledges, total, nil
}

// UpdateKnowledge 更新知识
func UpdateKnowledge(id int, k *Knowledge) error {
	o := orm.NewOrm()
	knowledge := &Knowledge{Id: id}
	err := o.Read(knowledge)
	if err != nil {
		return err
	}

	// 更新字段（不包含category_name，因为它是从联查获取的）
	knowledge.Title = k.Title
	knowledge.CategoryId = k.CategoryId
	knowledge.Tags = k.Tags
	knowledge.Author = k.Author
	knowledge.Content = k.Content
	knowledge.Summary = k.Summary
	knowledge.CoverUrl = k.CoverUrl
	knowledge.Status = k.Status
	knowledge.IsRecommend = k.IsRecommend
	knowledge.IsTop = k.IsTop
	knowledge.UpdateBy = k.UpdateBy

	_, err = o.Update(knowledge, "title", "category_id", "tags", "author", "content", "summary", "cover_url", "status", "is_recommend", "is_top", "update_by", "update_time")
	return err
}

// DeleteKnowledge 软删除知识
func DeleteKnowledge(id int, deleteBy string) error {
	o := orm.NewOrm()
	knowledge := &Knowledge{Id: id}
	err := o.Read(knowledge)
	if err != nil {
		return err
	}

	// 执行软删除：设置 delete_time 和 delete_by
	_, err = o.Raw("UPDATE yz_knowledge SET delete_time = ?, delete_by = ? WHERE knowledge_id = ?", time.Now(), deleteBy, id).Exec()
	return err
}

// GetAllCategories 获取所有分类
func GetAllCategories() ([]*KnowledgeCategory, error) {
	o := orm.NewOrm()
	var categories []*KnowledgeCategory
	_, err := o.QueryTable("yz_knowledge_category").OrderBy("sort_order").All(&categories)
	return categories, err
}

// GetCategoryById 根据ID获取分类
func GetCategoryById(id int) (*KnowledgeCategory, error) {
	o := orm.NewOrm()
	category := &KnowledgeCategory{CategoryId: id}
	err := o.Read(category)
	return category, err
}

// GetAllTags 获取所有标签
func GetAllTags() ([]*KnowledgeTag, error) {
	o := orm.NewOrm()
	var tags []*KnowledgeTag
	_, err := o.QueryTable("yz_knowledge_tags").All(&tags)
	return tags, err
}

// AddCategory 添加分类
func AddCategory(category *KnowledgeCategory) (int64, error) {
	o := orm.NewOrm()
	id, err := o.Insert(category)
	return id, err
}

// AddTag 添加标签
func AddTag(tag *KnowledgeTag) (int64, error) {
	o := orm.NewOrm()
	id, err := o.Insert(tag)
	return id, err
}