package semantic_analysis

import (
	"fmt"
	"github.com/beego/beego/v2/client/orm"
	"strings"
	"time"
)

// SaCompareSearchKeyword 文档比对搜索关键词表
type SaCompareSearchKeyword struct {
	SaCompareSearchKeywordId int       `orm:"column(sa_compare_search_keyword_id);pk"`
	CompareId                int       `description:"文档比对ID"`
	Keyword                  string    `description:"关键词"`
	CreateTime               time.Time `description:"创建时间"`
}

var SaCompareSearchKeywordColumns = struct {
	SaCompareSearchKeywordId string
	CompareId                string
	Keyword                  string
	CreateTime               string
}{
	SaCompareSearchKeywordId: "sa_compare_search_keyword_id",
	CompareId:                "compare_id",
	Keyword:                  "keyword",
	CreateTime:               "create_time",
}

func (m *SaCompareSearchKeyword) TableName() string {
	return "sa_compare_search_keyword"
}

func (m *SaCompareSearchKeyword) Create() (err error) {
	o := orm.NewOrm()
	id, err := o.Insert(m)
	if err != nil {
		return
	}
	m.SaCompareSearchKeywordId = int(id)
	return
}

func (m *SaCompareSearchKeyword) Update(cols []string) (err error) {
	o := orm.NewOrm()
	_, err = o.Update(m, cols...)
	return
}

func (m *SaCompareSearchKeyword) Del() (err error) {
	o := orm.NewOrm()
	sql := `DELETE FROM sa_compare_search_keyword WHERE sa_compare_search_keyword_id = ? LIMIT 1`
	_, err = o.Raw(sql, m.SaCompareSearchKeywordId).Exec()
	return
}

func (m *SaCompareSearchKeyword) GetItemById(id int) (err error) {
	o := orm.NewOrm()
	sql := `SELECT * FROM sa_compare_search_keyword WHERE sa_compare_search_keyword_id = ? LIMIT 1`
	err = o.Raw(sql, id).QueryRow(&m)
	return
}

func (m *SaCompareSearchKeyword) GetItemByCondition(condition string, pars []interface{}) (err error) {
	o := orm.NewOrm()
	sql := `SELECT * FROM sa_compare_search_keyword WHERE 1=1 `
	sql += condition
	sql += ` LIMIT 1`
	err = o.Raw(sql, pars).QueryRow(&m)
	return
}

func (m *SaCompareSearchKeyword) GetCountByCondition(condition string, pars []interface{}) (count int, err error) {
	o := orm.NewOrm()
	sql := fmt.Sprintf(`SELECT COUNT(1) FROM %s WHERE 1=1 %s`, m.TableName(), condition)
	err = o.Raw(sql, pars).QueryRow(&count)
	return
}

func (m *SaCompareSearchKeyword) GetItemsByCondition(condition string, pars []interface{}, fieldArr []string, orderRule string) (items []*SaCompareSearchKeyword, err error) {
	o := orm.NewOrm()
	fields := strings.Join(fieldArr, ",")
	if len(fieldArr) == 0 {
		fields = `*`
	}
	order := `ORDER BY create_time DESC`
	if orderRule != "" {
		order = ` ORDER BY ` + orderRule
	}
	sql := fmt.Sprintf(`SELECT %s FROM %s WHERE 1=1 %s %s`, fields, m.TableName(), condition, order)
	_, err = o.Raw(sql, pars).QueryRows(&items)
	return
}

func (m *SaCompareSearchKeyword) GetPageItemsByCondition(startSize, pageSize int, condition string, pars []interface{}, fieldArr []string, orderRule string) (total int, items []*SaCompareSearchKeyword, err error) {
	o := orm.NewOrm()
	fields := strings.Join(fieldArr, ",")
	if len(fieldArr) == 0 {
		fields = `*`
	}
	order := `ORDER BY create_time DESC`
	if orderRule != "" {
		order = ` ORDER BY ` + orderRule
	}
	sql := fmt.Sprintf(`SELECT %s FROM %s WHERE 1=1 %s %s`, fields, m.TableName(), condition, order)
	totalSql := `SELECT COUNT(1) total FROM (` + sql + `) z`
	if err = o.Raw(totalSql, pars).QueryRow(&total); err != nil {
		return
	}
	sql += ` LIMIT ?,?`
	_, err = o.Raw(sql, pars, startSize, pageSize).QueryRows(&items)
	return
}

type SaCompareSearchKeywordSaveReq struct {
	SaCompareId int      `description:"文档比对ID"`
	Keywords    []string `description:"关键词"`
}

func (m *SaCompareSearchKeyword) DelAndCreateSearchKeywords(compareId int, keywords []string) (err error) {
	o := orm.NewOrm()
	tx, err := o.Begin()
	if err != nil {
		return
	}
	defer func() {
		if err != nil {
			_ = tx.Rollback()
			return
		}
		_ = tx.Commit()
	}()

	sql := fmt.Sprintf(`DELETE FROM %s WHERE %s = ?`, m.TableName(), SaCompareSearchKeywordColumns.CompareId)
	if _, e := o.Raw(sql, compareId).Exec(); e != nil {
		return e
	}
	if len(keywords) > 0 {
		items := make([]*SaCompareSearchKeyword, 0)
		for _, v := range keywords {
			items = append(items, &SaCompareSearchKeyword{
				CompareId:  compareId,
				Keyword:    v,
				CreateTime: time.Now().Local(),
			})
		}
		if _, e := o.InsertMulti(len(items), items); e != nil {
			return e
		}
	}
	return
}