package models

import (
	"eta/eta_task/global"
	"eta/eta_task/utils"
)

type WechatArticleTag struct {
	TagName     string
	ArticleId   string
	TextContent string
}

type WechatArticle struct {
	TextContent string
}

func GetWechatArticleTags(startTime, endTime string) (list []*WechatArticleTag, err error) {
	sql := `SELECT a.tag_name,GROUP_CONCAT(a.article_id) AS article_id
		FROM article_dfa_tag_mapping AS a
		INNER JOIN wechat_article AS b ON a.article_id=b.wechat_article_id
		WHERE a.tag_name<>''
		AND b.article_create_time >= ?
		AND b.article_create_time <= ?
		AND b.is_summary=0
		AND a.source='wechat'
		GROUP BY a.tag_name `

	err = global.DbMap[utils.DbNameAI].Raw(sql, startTime, endTime).Find(&list).Error
	return list, err
}

func GetWechatArticle(articleId, startTime, endTime string) (list []*WechatArticle, err error) {
	sql := `SELECT a.text_content
		FROM wechat_article AS a
		WHERE a.wechat_article_id IN(` + articleId + `) 
		AND a.article_create_time >= ?
		AND a.article_create_time <= ?
        AND a.is_summary=0`

	err = global.DbMap[utils.DbNameAI].Raw(sql, startTime, endTime).Find(&list).Error
	return list, err
}

func ModifyWechatArticleSummary(articleId string, isSummary int) (err error) {
	sql := ` UPDATE wechat_article SET is_summary=? WHERE wechat_article_id IN(` + articleId + `)`
	err = global.DbMap[utils.DbNameAI].Exec(sql, isSummary).Error
	return err
}