package dao

import (
	"github.com/druidcaesa/gotool"
	"ulink-admin/frame"
	"ulink-admin/modules/car/models/model"
	"ulink-admin/modules/car/models/req"
	"ulink-admin/pkg/base"
	"ulink-admin/pkg/page"
)

type CompanyDao struct {
	base.BaseDao
}

// Page 查询公司管理分页数据
func (this CompanyDao) Page(query *req.CompanyQuery, list interface{}) int64 {
	session := this.GetSession().Table(model.Company{}.TableName())

	if query.Id > 0 {
		session.And("id = ?", query.Id)
	}
	if !gotool.StrUtils.HasEmpty(query.Label) {
		session.And("label = ?", query.Label)
	}
	if !gotool.StrUtils.HasEmpty(query.Head) {
		session.And("head = ?", query.Head)
	}
	if !gotool.StrUtils.HasEmpty(query.TradeNo) {
		session.And("trade_no = ?", query.TradeNo)
	}
	if !gotool.StrUtils.HasEmpty(query.UnitNo) {
		session.And("unit_no = ?", query.UnitNo)
	}
	if !gotool.StrUtils.HasEmpty(query.Phone) {
		session.And("phone = ?", query.Phone)
	}
	if !gotool.StrUtils.HasEmpty(query.Address) {
		session.And("address = ?", query.Address)
	}
	if !gotool.StrUtils.HasEmpty(query.PostalCode) {
		session.And("postal_code = ?", query.PostalCode)
	}
	if query.UserId > 0 {
		session.And("user_id = ?", query.UserId)
	}
	if query.Status > 0 {
		session.And("status = ?", query.Status)
	}
	if !gotool.StrUtils.HasEmpty(query.TradeDate) {
		session.And("trade_date = ?", query.TradeDate)
	}
	if !gotool.StrUtils.HasEmpty(query.Examine) {
		session.And("examine = ?", query.Examine)
	}
	if !gotool.StrUtils.HasEmpty(query.Remark) {
		session.And("remark = ?", query.Remark)
	}
	if !gotool.StrUtils.HasEmpty(query.CreateBy) {
		session.And("create_by = ?", query.CreateBy)
	}
	if !gotool.StrUtils.HasEmpty(query.UpdateBy) {
		session.And("update_by = ?", query.UpdateBy)
	}
	if !gotool.StrUtils.HasEmpty(query.BeginTime) {
		session.And("date_format(u.create_time,'%y%m%d') >= date_format(?,'%y%m%d')", query.BeginTime)
	}
	if !gotool.StrUtils.HasEmpty(query.EndTime) {
		session.And("date_format(u.create_time,'%y%m%d') <= date_format(?,'%y%m%d')", query.EndTime)
	}
	session.Desc("id")
	total, err := session.Limit(query.PageSize, page.StartSize(query.PageNum, query.PageSize)).FindAndCount(list)
	if err != nil {
		frame.Throw(frame.SQL_CODE, "数据查询错误"+err.Error())
	}
	return total
}

// List 查询公司管理分页数据
func (this CompanyDao) List(query *req.CompanyQuery, list interface{}) {
	session := this.GetSession().Table(model.Company{}.TableName())

	if query.Id > 0 {
		session.And("id = ?", query.Id)
	}
	if !gotool.StrUtils.HasEmpty(query.Label) {
		session.And("label = ?", query.Label)
	}
	if !gotool.StrUtils.HasEmpty(query.Head) {
		session.And("head = ?", query.Head)
	}
	if !gotool.StrUtils.HasEmpty(query.TradeNo) {
		session.And("trade_no = ?", query.TradeNo)
	}
	if !gotool.StrUtils.HasEmpty(query.UnitNo) {
		session.And("unit_no = ?", query.UnitNo)
	}
	if !gotool.StrUtils.HasEmpty(query.Phone) {
		session.And("phone = ?", query.Phone)
	}
	if !gotool.StrUtils.HasEmpty(query.Address) {
		session.And("address = ?", query.Address)
	}
	if !gotool.StrUtils.HasEmpty(query.PostalCode) {
		session.And("postal_code = ?", query.PostalCode)
	}
	if query.UserId > 0 {
		session.And("user_id = ?", query.UserId)
	}
	if query.Status > 0 {
		session.And("status = ?", query.Status)
	}
	if !gotool.StrUtils.HasEmpty(query.TradeDate) {
		session.And("trade_date = ?", query.TradeDate)
	}
	if !gotool.StrUtils.HasEmpty(query.Examine) {
		session.And("examine = ?", query.Examine)
	}
	if !gotool.StrUtils.HasEmpty(query.Remark) {
		session.And("remark = ?", query.Remark)
	}
	if !gotool.StrUtils.HasEmpty(query.CreateBy) {
		session.And("create_by = ?", query.CreateBy)
	}
	if !gotool.StrUtils.HasEmpty(query.UpdateBy) {
		session.And("update_by = ?", query.UpdateBy)
	}
	if !gotool.StrUtils.HasEmpty(query.BeginTime) {
		session.And("date_format(u.create_time,'%y%m%d') >= date_format(?,'%y%m%d')", query.BeginTime)
	}
	if !gotool.StrUtils.HasEmpty(query.EndTime) {
		session.And("date_format(u.create_time,'%y%m%d') <= date_format(?,'%y%m%d')", query.EndTime)
	}
	err := session.Find(list)
	if err != nil {
		frame.Throw(frame.SQL_CODE, "数据查询错误"+err.Error())
	}
}