package semantic_analysis

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

type SaCompareLabel struct {
	SaCompareLabelId int       `orm:"column(sa_compare_label_id);pk" description:"段落标签ID"`
	CompareId        int       `description:"比对ID"`
	DocId            int       `description:"文档ID"`
	SectionId        int       `description:"段落ID"`
	LabelId          int       `description:"标签ID"`
	LabelName        string    `description:"标签名称"`
	IsPart           int       `description:"是否为片段: 0-整段; 1-片段"`
	StartIndex       int       `description:"片段开始下标"`
	EndIndex         int       `description:"片段结束下标"`
	Content          string    `description:"内容"`
	SysAdminId       int       `description:"创建人ID"`
	SysAdminName     string    `description:"创建人姓名"`
	CreateTime       time.Time `description:"创建时间"`
}

var SaCompareLabelColumns = struct {
	SaCompareLabelId string
	CompareId        string
	DocId            string
	SectionId        string
	LabelId          string
	LabelName        string
	IsPart           string
	StartIndex       string
	EndIndex         string
	Content          string
	SysAdminId       string
	SysAdminName     string
	CreateTime       string
}{
	SaCompareLabelId: "sa_compare_label_id",
	CompareId:        "compare_id",
	DocId:            "doc_id",
	SectionId:        "section_id",
	LabelId:          "label_id",
	LabelName:        "label_name",
	IsPart:           "is_part",
	StartIndex:       "start_index",
	EndIndex:         "end_index",
	SysAdminId:       "sys_admin_id",
	SysAdminName:     "sys_admin_name",
	CreateTime:       "create_time",
}

func (m *SaCompareLabel) TableName() string {
	return "sa_compare_label"
}

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

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

func (m *SaCompareLabel) Del() (err error) {
	o := orm.NewOrm()
	sql := `DELETE FROM sa_compare_label WHERE sa_compare_label_id = ? LIMIT 1`
	_, err = o.Raw(sql, m.SaCompareLabelId).Exec()
	return
}

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

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

func (m *SaCompareLabel) 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
}

type SaCompareLabelGroupCount struct {
	DocId     int `description:"文档ID"`
	SectionId int `description:"段落ID"`
	LabelId   int `description:"标签ID"`
	UseNum    int `description:"引用数"`
}

func (m *SaCompareLabel) GetGroupCountByCondition(condition string, pars []interface{}, groupRule string) (items []*SaCompareLabelGroupCount, err error) {
	o := orm.NewOrm()
	sql := fmt.Sprintf(`SELECT COUNT(1) AS use_num, %s FROM %s WHERE 1=1 %s GROUP BY %s`, groupRule, m.TableName(), condition, groupRule)
	_, err = o.Raw(sql, pars).QueryRows(&items)
	return
}

func (m *SaCompareLabel) GetItemsByCondition(condition string, pars []interface{}, fieldArr []string, orderRule string) (items []*SaCompareLabel, 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 *SaCompareLabel) GetPageItemsByCondition(startSize, pageSize int, condition string, pars []interface{}, fieldArr []string, orderRule string) (total int, items []*SaCompareLabel, 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
}

// UpdateSaCompareLabelByLabelId 更新比对标签信息
func UpdateSaCompareLabelByLabelId(labelId int, labelName string) (err error) {
	o := orm.NewOrm()
	sql := `UPDATE sa_compare_label SET label_name = ? WHERE label_id = ?`
	_, err = o.Raw(sql, labelName, labelId).Exec()
	return
}

// SaCompareLabelItem 比对标签
type SaCompareLabelItem struct {
	SaCompareLabelId int    `description:"比对标签ID"`
	LabelId          int    `description:"标签ID"`
	LabelName        string `description:"标签名称"`
	CompareId        int    `description:"比对ID"`
	CompareContent   string `description:"比对内容(片段)"`
	DocId            int    `description:"文档ID"`
	Title            string `description:"文档标题"`
	SectionId        int    `description:"段落ID"`
	SectionContent   string `description:"段落内容"`
	Sort             int    `description:"段落排序"`
}

// GetSaCompareLabelByCond 获取比对标签
func GetSaCompareLabelByCond(condition string, pars []interface{}) (items []*SaCompareLabelItem, err error) {
	o := orm.NewOrm()
	sql := `SELECT
				a.sa_compare_label_id,
				a.label_id,
				a.label_name,
				a.compare_id,
				a.doc_id,
				a.content AS compare_content,
				b.title,
				c.sa_doc_section_id AS section_id,
				c.content AS section_content,
				c.sort
			FROM
				sa_compare_label AS a
			JOIN sa_doc AS b ON a.doc_id = b.sa_doc_id
			JOIN sa_doc_section AS c ON a.section_id = c.sa_doc_section_id
			WHERE
				1 = 1 %s
			ORDER BY
				a.doc_id,
				c.sort ASC`
	sql = fmt.Sprintf(sql, condition)
	_, err = o.Raw(sql, pars).QueryRows(&items)
	return
}