package model

import (
	"apsClient/constvar"
	"apsClient/pkg/sqlitex"
	"fmt"
	"github.com/jinzhu/gorm"
)

type (
	// TaskStatusSync 任务状态同步
	TaskStatusSync struct {
		CommonModel
		WorkOrderId        string              `json:"workOrderId"`                                  //工单编号
		ProductProcedureID string              `gorm:"type:varchar(191);" json:"productProcedureID"` //产品工序id
		ProcedureID        string              `json:"procedureId"`                                  // 工序编号
		DeviceId           string              `json:"deviceId"`                                     //设备编号
		IsProcessing       bool                //是否处理中
		IsFinish           bool                //是否完成
		StartTs            int64               `json:"start_ts"`  //开始时间
		FinishTs           int64               `json:"finish_ts"` //结束时间
		FinishedQuantity   int                 `json:"finishedQuantity"`
		SyncStatus         constvar.SyncStatus `json:"-"`
	}

	TaskStatusSyncSearch struct {
		TaskStatusSync
		Order    string
		PageNum  int
		PageSize int
		Orm      *gorm.DB
		IDs      []uint
		SyncWait bool
	}
)

func (slf *TaskStatusSync) TableName() string {
	return "task_status_sync"
}

func NewTaskStatusSyncSearch(db *gorm.DB) *TaskStatusSyncSearch {
	if db == nil {
		db = sqlitex.GetDB()
	}
	return &TaskStatusSyncSearch{Orm: db}
}

func (slf *TaskStatusSyncSearch) SetOrm(tx *gorm.DB) *TaskStatusSyncSearch {
	slf.Orm = tx
	return slf
}

func (slf *TaskStatusSyncSearch) SetPage(page, size int) *TaskStatusSyncSearch {
	slf.PageNum, slf.PageSize = page, size
	return slf
}

func (slf *TaskStatusSyncSearch) SetOrder(order string) *TaskStatusSyncSearch {
	slf.Order = order
	return slf
}

func (slf *TaskStatusSyncSearch) SetID(id uint) *TaskStatusSyncSearch {
	slf.ID = id
	return slf
}

func (slf *TaskStatusSyncSearch) SetProductProcedureId(productProcedureId string) *TaskStatusSyncSearch {
	slf.ProductProcedureID = productProcedureId
	return slf
}

func (slf *TaskStatusSyncSearch) SetWorkOrderId(workOrderId string) *TaskStatusSyncSearch {
	slf.WorkOrderId = workOrderId
	return slf
}

func (slf *TaskStatusSyncSearch) SetIDs(ids []uint) *TaskStatusSyncSearch {
	slf.IDs = ids
	return slf
}

func (slf *TaskStatusSyncSearch) SetSyncWait() *TaskStatusSyncSearch {
	slf.SyncWait = true
	return slf
}

func (slf *TaskStatusSyncSearch) build() *gorm.DB {
	var db = slf.Orm.Table(slf.TableName())

	if slf.ID != 0 {
		db = db.Where("id = ?", slf.ID)
	}

	if slf.ProductProcedureID != "" {
		db = db.Where("product_procedure_id = ?", slf.ProductProcedureID)
	}

	if len(slf.IDs) != 0 {
		db = db.Where("id in (?)", slf.IDs)
	}

	if slf.Order != "" {
		db = db.Order(slf.Order)
	}

	if slf.WorkOrderId != "" {
		db = db.Where("work_order_id = ?", slf.WorkOrderId)
	}

	if slf.SyncWait {
		db = db.Where("sync_status != ?", constvar.SyncStatusOk)
	}

	return db
}

// Create 单条插入
func (slf *TaskStatusSyncSearch) Create(record *TaskStatusSync) error {
	var db = slf.build()

	if err := db.Create(record).Error; err != nil {
		return fmt.Errorf("create err: %v, record: %+v", err, record)
	}

	return nil
}

func (slf *TaskStatusSyncSearch) Save(record *TaskStatusSync) error {
	var db = slf.build()

	if err := db.Updates(record).Error; err != nil {
		return fmt.Errorf("create err: %v, record: %+v", err, record)
	}
	return nil
}

func (slf *TaskStatusSyncSearch) UpdateByMap(upMap map[string]interface{}) error {
	var (
		db = slf.build()
	)

	if err := db.Updates(upMap).Error; err != nil {
		return fmt.Errorf("update by map err: %v, upMap: %+v", err, upMap)
	}

	return nil
}

func (slf *TaskStatusSyncSearch) UpdateByQuery(query string, args []interface{}, upMap map[string]interface{}) error {
	var (
		db = slf.Orm.Table(slf.TableName()).Where(query, args...)
	)

	if err := db.Updates(upMap).Error; err != nil {
		return fmt.Errorf("update by query err: %v, query: %s, args: %+v, upMap: %+v", err, query, args, upMap)
	}

	return nil
}

func (slf *TaskStatusSyncSearch) Delete() error {
	var db = slf.build()

	if err := db.Unscoped().Delete(&TaskStatusSync{}).Error; err != nil {
		return err
	}

	return nil
}

func (slf *TaskStatusSyncSearch) First() (*TaskStatusSync, error) {
	var (
		record = new(TaskStatusSync)
		db     = slf.build()
	)

	if err := db.First(record).Error; err != nil {
		return record, err
	}

	return record, nil
}

func (slf *TaskStatusSyncSearch) Find() ([]*TaskStatusSync, int64, error) {
	var (
		records = make([]*TaskStatusSync, 0)
		total   int64
		db      = slf.build()
	)

	if err := db.Count(&total).Error; err != nil {
		return records, total, fmt.Errorf("find count err: %v", err)
	}
	if slf.PageNum*slf.PageSize > 0 {
		db = db.Offset((slf.PageNum - 1) * slf.PageSize).Limit(slf.PageSize)
	}
	if err := db.Find(&records).Error; err != nil {
		return records, total, fmt.Errorf("find records err: %v", err)
	}

	return records, total, nil
}

func (slf *TaskStatusSyncSearch) FindNotTotal() ([]*TaskStatusSync, error) {
	var (
		records = make([]*TaskStatusSync, 0)
		db      = slf.build()
	)

	if slf.PageNum*slf.PageSize > 0 {
		db = db.Offset((slf.PageNum - 1) * slf.PageSize).Limit(slf.PageSize)
	}
	if err := db.Find(&records).Error; err != nil {
		return records, fmt.Errorf("find records err: %v", err)
	}

	return records, nil
}

// FindByQuery 指定条件查询.
func (slf *TaskStatusSyncSearch) FindByQuery(query string, args []interface{}) ([]*TaskStatusSync, int64, error) {
	var (
		records = make([]*TaskStatusSync, 0)
		total   int64
		db      = slf.Orm.Table(slf.TableName()).Where(query, args...)
	)

	if err := db.Count(&total).Error; err != nil {
		return records, total, fmt.Errorf("find by query count err: %v", err)
	}
	if slf.PageNum*slf.PageSize > 0 {
		db = db.Offset((slf.PageNum - 1) * slf.PageSize).Limit(slf.PageSize)
	}
	if err := db.Find(&records).Error; err != nil {
		return records, total, fmt.Errorf("find by query records err: %v, query: %s, args: %+v", err, query, args)
	}

	return records, total, nil
}

// FindByQueryNotTotal 指定条件查询&不查询总条数.
func (slf *TaskStatusSyncSearch) FindByQueryNotTotal(query string, args []interface{}) ([]*TaskStatusSync, error) {
	var (
		records = make([]*TaskStatusSync, 0)
		db      = slf.Orm.Table(slf.TableName()).Where(query, args...)
	)

	if slf.PageNum*slf.PageSize > 0 {
		db = db.Offset((slf.PageNum - 1) * slf.PageSize).Limit(slf.PageSize)
	}
	if err := db.Find(&records).Error; err != nil {
		return records, fmt.Errorf("find by query records err: %v, query: %s, args: %+v", err, query, args)
	}

	return records, nil
}