package speech_recognition

import (
	"eta_gn/eta_api/global"
	"eta_gn/eta_api/utils"
	"fmt"
	"strings"
	"time"
)

const (
	SpeechRecognitionMenuNodeTypeDefault = 0
	SpeechRecognitionMenuNodeTypeSpeech  = 1
)

type SpeechRecognitionMenu struct {
	SpeechRecognitionMenuId int       `gorm:"primaryKey;column:speech_recognition_menu_id;type:int(10) unsigned;not null"` // 语音识别目录Id
	UniqueCode              string    `gorm:"column:unique_code;type:varchar(64);not null;default:''"`                     // 唯一编码
	MenuName                string    `gorm:"column:menu_name;type:varchar(255);not null;default:''"`                      // 目录名称
	ParentId                int       `gorm:"column:parent_id;type:int(10) unsigned;not null;default:0"`                   // 父级Id
	Level                   int       `gorm:"column:level;type:tinyint(4) unsigned;not null;default:0"`                    // 目录层级
	Sort                    int       `gorm:"column:sort;type:int(10) unsigned;not null;default:0"`                        // 排序
	RootId                  int       `gorm:"column:root_id;type:int(10) unsigned;not null;default:0"`                     // 顶级Id
	CreateTime              time.Time `gorm:"column:create_time;type:datetime"`                                            // 创建时间
	ModifyTime              time.Time `gorm:"column:modify_time;type:datetime"`                                            // 更新时间
}

var SpeechRecognitionMenuCols = struct {
	SpeechRecognitionMenuId string
	UniqueCode              string
	MenuName                string
	ParentId                string
	Level                   string
	Sort                    string
	RootId                  string
	CreateTime              string
	ModifyTime              string
}{
	SpeechRecognitionMenuId: "speech_recognition_menu_id",
	UniqueCode:              "unique_code",
	MenuName:                "menu_name",
	ParentId:                "parent_id",
	Level:                   "level",
	Sort:                    "sort",
	RootId:                  "root_id",
	CreateTime:              "create_time",
	ModifyTime:              "modify_time",
}

func (m *SpeechRecognitionMenu) TableName() string {
	return "speech_recognition_menu"
}

func (m *SpeechRecognitionMenu) PrimaryId() string {
	return SpeechRecognitionMenuCols.SpeechRecognitionMenuId
}

func (m *SpeechRecognitionMenu) Create() (err error) {
	err = global.DEFAULT_DmSQL.Create(m).Error
	return
}

func (m *SpeechRecognitionMenu) CreateMulti(items []*SpeechRecognitionMenu) (err error) {
	if len(items) == 0 {
		return
	}
	err = global.DEFAULT_DmSQL.CreateInBatches(items, utils.MultiAddNum).Error
	return
}

func (m *SpeechRecognitionMenu) Update(cols []string) (err error) {
	err = global.DEFAULT_DmSQL.Select(cols).Updates(m).Error
	return
}

func (m *SpeechRecognitionMenu) Del() (err error) {
	sql := fmt.Sprintf(`DELETE FROM %s WHERE %s = ? LIMIT 1`, m.TableName(), m.PrimaryId())
	err = global.DEFAULT_DmSQL.Exec(sql, m.SpeechRecognitionMenuId).Error
	return
}

func (m *SpeechRecognitionMenu) MultiDel(menuIds []int) (err error) {
	if len(menuIds) == 0 {
		return
	}
	sql := fmt.Sprintf(`DELETE FROM %s WHERE %s IN (%s)`, m.TableName(), m.PrimaryId(), utils.GetOrmInReplace(len(menuIds)))
	err = global.DEFAULT_DmSQL.Exec(sql, menuIds).Error
	return
}

func (m *SpeechRecognitionMenu) GetItemById(id int) (item *SpeechRecognitionMenu, err error) {
	sql := fmt.Sprintf(`SELECT * FROM %s WHERE %s = ? LIMIT 1`, m.TableName(), m.PrimaryId())
	err = global.DEFAULT_DmSQL.Raw(sql, id).First(&item).Error
	return
}

func (m *SpeechRecognitionMenu) GetItemByCondition(condition string, pars []interface{}, orderRule string) (item *SpeechRecognitionMenu, err error) {
	order := ``
	if orderRule != "" {
		order = ` ORDER BY ` + orderRule
	}
	sql := fmt.Sprintf(`SELECT * FROM %s WHERE 1=1 %s %s LIMIT 1`, m.TableName(), condition, order)
	err = global.DEFAULT_DmSQL.Raw(sql, pars...).First(&item).Error
	return
}

func (m *SpeechRecognitionMenu) GetCountByCondition(condition string, pars []interface{}) (count int, err error) {
	sql := fmt.Sprintf(`SELECT COUNT(1) FROM %s WHERE 1=1 %s`, m.TableName(), condition)
	err = global.DEFAULT_DmSQL.Raw(sql, pars...).Scan(&count).Error
	return
}

func (m *SpeechRecognitionMenu) GetItemsByCondition(condition string, pars []interface{}, fieldArr []string, orderRule string) (items []*SpeechRecognitionMenu, err error) {
	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 = global.DEFAULT_DmSQL.Raw(sql, pars...).Find(&items).Error
	return
}

func (m *SpeechRecognitionMenu) GetPageItemsByCondition(condition string, pars []interface{}, fieldArr []string, orderRule string, startSize, pageSize int) (items []*SpeechRecognitionMenu, err error) {
	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 LIMIT ?,?`, fields, m.TableName(), condition, order)
	err = global.DEFAULT_DmSQL.Raw(sql, pars...).Find(&items).Error
	return
}

// SpeechRecognitionMenuItem 目录
type SpeechRecognitionMenuItem struct {
	UniqueCode string                       `description:"唯一编码"`
	MenuId     int                          `description:"目录ID"`
	MenuName   string                       `description:"目录名称"`
	ParentId   int                          `description:"父级ID"`
	Level      int                          `description:"目录层级"`
	Sort       int                          `description:"排序"`
	CreateTime string                       `description:"创建时间"`
	Children   []*SpeechRecognitionMenuItem `gorm:"-" description:"子目录"`
}

// SpeechRecognitionMenuNodeItem 语音识别目录节点
type SpeechRecognitionMenuNodeItem struct {
	UniqueCode            string                           `description:"唯一编码"`
	NodeType              int                              `description:"节点类型：0-目录；1-语音识别"`
	MenuId                int                              `description:"目录ID"`
	MenuName              string                           `description:"目录名称"`
	SpeechRecognitionId   int                              `description:"语音识别ID"`
	SpeechRecognitionName string                           `description:"语音识别名称"`
	ParentId              int                              `description:"父级ID"`
	Level                 int                              `description:"目录层级"`
	Sort                  int                              `description:"排序"`
	CreateTime            string                           `description:"创建时间"`
	Children              []*SpeechRecognitionMenuNodeItem `gorm:"-" description:"子节点"`
}

// SpeechRecognitionMenuAddReq 新增语音识别目录
type SpeechRecognitionMenuAddReq struct {
	ParentId int    `description:"父级ID"`
	MenuName string `description:"目录名称"`
}

// SpeechRecognitionMenuEditReq 编辑语音识别目录
type SpeechRecognitionMenuEditReq struct {
	MenuId   int    `description:"目录ID"`
	MenuName string `description:"目录名称"`
}

// SpeechRecognitionMenuRemoveReq 删除语音识别目录
type SpeechRecognitionMenuRemoveReq struct {
	MenuId int `description:"目录ID"`
}

// SpeechRecognitionMenuRemoveCheckResp 删除语音识别目录-响应体
type SpeechRecognitionMenuRemoveCheckResp struct {
	CheckResult int    `description:"校验结果：0-可删除；1-关联内容；2-关联空目录"`
	Tips        string `description:"提示信息"`
}

// SpeechRecognitionMenuMoveReq 移动目录请求体
type SpeechRecognitionMenuMoveReq struct {
	MenuId       int `description:"目录ID"`
	ParentMenuId int `description:"父级目录ID"`
	PrevMenuId   int `description:"上一个兄弟节点目录ID"`
	NextMenuId   int `description:"下一个兄弟节点目录ID"`
	SpeechId     int `description:"语音识别ID, 大于0则表示移动语音识别"`
	PrevSpeechId int `description:"上一个语音识别ID"`
	NextSpeechId int `description:"下一个语音识别ID"`
}

// UpdateSortByParentId 根据父级ID更新排序
func (m *SpeechRecognitionMenu) UpdateSortByParentId(parentId, menuId, nowSort int, updateSort string) (err error) {
	sql := fmt.Sprintf(`UPDATE %s SET %s = %s WHERE %s = ? AND %s > ?`, m.TableName(), SpeechRecognitionMenuCols.Sort, updateSort, SpeechRecognitionMenuCols.ParentId, SpeechRecognitionMenuCols.Sort)
	if menuId > 0 {
		sql += fmt.Sprintf(` OR (%s > %d AND %s = %d)`, SpeechRecognitionMenuCols.SpeechRecognitionMenuId, menuId, SpeechRecognitionMenuCols.Sort, nowSort)
	}
	err = global.DEFAULT_DmSQL.Exec(sql, parentId, nowSort).Error
	return
}

// GetMaxSortByParentId 获取父级分类下最大Sort
func (m *SpeechRecognitionMenu) GetMaxSortByParentId(parentId int) (sort int, err error) {
	sql := fmt.Sprintf(`SELECT COALESCE(MAX(sort),0) AS sort FROM %s WHERE %s = ?`, m.TableName(), SpeechRecognitionMenuCols.ParentId)
	err = global.DEFAULT_DmSQL.Raw(sql, parentId).Scan(&sort).Error
	return
}

// GetFirstByParentId 获取父级目录下排序第一的目录
func (m *SpeechRecognitionMenu) GetFirstByParentId(parentId int) (item *SpeechRecognitionMenu, err error) {
	sql := fmt.Sprintf(`SELECT * FROM %s WHERE %s = ? ORDER BY %s ASC, %s ASC LIMIT 1`, m.TableName(), SpeechRecognitionMenuCols.ParentId, SpeechRecognitionMenuCols.Sort, SpeechRecognitionMenuCols.SpeechRecognitionMenuId)
	err = global.DEFAULT_DmSQL.Raw(sql, parentId).First(&item).Error
	return
}

// UpdateChildByParentMenuId 通过父级目录ID更新子目录
func (m *SpeechRecognitionMenu) UpdateChildByParentMenuId(menuIds []int, rootId int, levelStep int) (err error) {
	if len(menuIds) == 0 {
		return
	}
	var pars []interface{}
	pars = append(pars, rootId, levelStep)
	pars = append(pars, menuIds)
	sql := fmt.Sprintf(`UPDATE %s SET %s = ?, %s = %s + ? WHERE %s IN (%s)`, m.TableName(), SpeechRecognitionMenuCols.RootId, SpeechRecognitionMenuCols.Level, SpeechRecognitionMenuCols.Level, SpeechRecognitionMenuCols.SpeechRecognitionMenuId, utils.GetOrmInReplace(len(menuIds)))
	err = global.DEFAULT_DmSQL.Exec(sql, pars...).Error
	return
}