package system

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

type SysDepartmentAddReq struct {
	DepartmentId   int    `description:"部门Id"`
	DepartmentName string `description:"部门名称"`
	Sort           int    `description:"排序"`
	CreateTime     string `description:"创建时间"`
}

type SysDepartment struct {
	DepartmentId   int    `orm:"column(department_id);pk" description:"部门Id"`
	DepartmentName string `description:"部门名称"`
	Sort           int    `description:"排序"`
	CreateTime     string `description:"创建时间"`
}

func GetSysDepartmentCount(departmentName string) (count int, err error) {
	sql := `SELECT COUNT(1) AS count FROM sys_department WHERE department_name=? `
	o := orm.NewOrm()
	err = o.Raw(sql, departmentName).QueryRow(&count)
	return
}

func AddSysDepartment(item *SysDepartment) (lastId int64, err error) {
	o := orm.NewOrm()
	lastId, err = o.Insert(item)
	return
}

type SysDepartmentEditReq struct {
	DepartmentId   int    `description:"部门Id"`
	DepartmentName string `description:"部门名称"`
}

func GetSysDepartmentById(departmentId int) (item *SysDepartment, err error) {
	sql := `SELECT * FROM sys_department WHERE department_id=? `
	o := orm.NewOrm()
	err = o.Raw(sql, departmentId).QueryRow(&item)
	return
}

func GetSysDepartmentByName(departmentName string) (item *SysDepartment, err error) {
	sql := `SELECT * FROM sys_department WHERE department_name=? `
	o := orm.NewOrm()
	err = o.Raw(sql, departmentName).QueryRow(&item)
	return
}

func ModifySysDepartment(departmentName string, departmentId int) (err error) {
	o := orm.NewOrm()
	to, err := o.Begin()
	if err != nil {
		return
	}
	defer func() {
		if err != nil {
			_ = to.Rollback()
		} else {
			_ = to.Commit()
		}
	}()
	sql := `UPDATE sys_department SET department_name=? WHERE department_id=? `
	_, err = to.Raw(sql, departmentName, departmentId).Exec()
	if err != nil {
		return
	}
	sql = `UPDATE admin SET department_name=? WHERE department_id=? `
	_, err = to.Raw(sql, departmentName, departmentId).Exec()
	return
}

type SysDepartmentDeleteReq struct {
	DepartmentId int `description:"部门Id"`
}

func DeleteSysDepartment(departmentId int) (err error) {
	sql := `DELETE FROM sys_department WHERE department_id=? `
	o := orm.NewOrm()
	_, err = o.Raw(sql, departmentId).Exec()
	return
}

func GetSysDepartmentAll() (item []*SysDepartment, err error) {
	sql := `SELECT * FROM sys_department ORDER BY sort ASC, department_id ASC `
	o := orm.NewOrm()
	_, err = o.Raw(sql).QueryRows(&item)
	return
}

type DepartmentSort struct {
	DepartmentId int
	Sort         int
}

// Update 更新用户基础信息
func (item *SysDepartment) Update(cols []string) (err error) {
	o := orm.NewOrm()
	_, err = o.Update(item, cols...)
	return
}

type SysDepartmentSyncReq struct {
	List []*SysDepartment
}

// 获取所有部门信息
func GetAllSysDepartment() (items []*SysDepartment, err error) {
	sql := `SELECT * FROM sys_department `
	o := orm.NewOrm()
	_, err = o.Raw(sql).QueryRows(&items)
	return
}

// Update 更新用户基础信息
func (item *SysDepartment) AddOrUpdate() (err error) {
	o := orm.NewOrm()
	_, err = o.InsertOrUpdate(item)
	return
}