package model

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

type (
	ReportWork struct {
		CommonModel
		ProceduresID       uint   `gorm:"index;type:varchar(191)" json:"proceduresId"`        //procedures表的id
		ProductProcedureID string `gorm:"index;type:varchar(191);" json:"productProcedureID"` //产品工序id
		WorkOrderID        string `gorm:"index;type:varchar(191);" json:"workOrderId"`
		DeviceID           string `gorm:"index;type:varchar(191)" json:"deviceId"`
		DeviceName         string `gorm:"index;type:varchar(191)" json:"deviceName"`
		ProcedureID        string `gorm:"index;type:varchar(191)" json:"procedureId"`
		StartTime          int64  `json:"startTime"`
		EndTime            int64  `json:"endTime"`
		ReportAmount       int    `json:"reportAmount"` //报工数量
		FinishAmount       int    `json:"finishAmount"` //本次报工 - 上次报工
		WorkerID           string `json:"workerID"`     //报工人id
		WorkerName         string `json:"workerName"`   //报工人姓名
		WorkerTime         int64  `json:"workerTime"`   //工时，单位秒
		BarCode            string `json:"barCode"`      //条形码
	}

	ReportWorkSearch struct {
		ReportWork
		Order        string
		PageNum      int
		PageSize     int
		Orm          *gorm.DB
		Preload      bool
		ProcedureIds []string
		Channels     []int32
		Offset       int
		Limit        int
	}
)

func (slf *ReportWork) TableName() string {
	return "report_work"
}

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

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

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

func (slf *ReportWorkSearch) SetOffset(offset, limit int) *ReportWorkSearch {
	slf.Offset, slf.Limit = offset, limit
	return slf
}

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

func (slf *ReportWorkSearch) SetWorkOrderId(orderId string) *ReportWorkSearch {
	slf.WorkOrderID = orderId
	return slf
}

func (slf *ReportWorkSearch) SetProceduresId(id uint) *ReportWorkSearch {
	slf.ProceduresID = id
	return slf
}

func (slf *ReportWorkSearch) SetId(id uint) *ReportWorkSearch {
	slf.ID = id
	return slf
}

func (slf *ReportWorkSearch) SetProcedureIds(procedureIds []string) *ReportWorkSearch {
	slf.ProcedureIds = procedureIds
	return slf
}

func (slf *ReportWorkSearch) SetDeviceId(id string) *ReportWorkSearch {
	slf.DeviceID = id
	return slf
}

func (slf *ReportWorkSearch) SetPreload(preload bool) *ReportWorkSearch {
	slf.Preload = preload
	return slf
}

func (slf *ReportWorkSearch) SetChannels(channels []int32) *ReportWorkSearch {
	slf.Channels = channels
	return slf
}

func (slf *ReportWorkSearch) build() *gorm.DB {
	var db = slf.Orm.Model(&ReportWork{})

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

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

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

	if slf.DeviceID != "" {
		db = db.Where("device_id = ?", slf.DeviceID)
	}

	if slf.Preload {
		db = db.Preload("InputMaterials").Preload("OutputMaterials")
	}

	if len(slf.ProcedureIds) > 0 {
		db = db.Where("procedure_id IN (?)", slf.ProcedureIds)
	}

	if slf.ProcedureID != "" {
		db = db.Where("procedure_id = ?", slf.ProcedureID)
	}

	if slf.ProceduresID != 0 {
		db = db.Where("procedures_id = ?", slf.ProceduresID)
	}

	if len(slf.Channels) > 0 {
		db = db.Where("channel IN (?)", slf.Channels)
	}

	return db
}

// Create 单条插入
func (slf *ReportWorkSearch) Create(record *ReportWork) 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
}

// CreateBatch 批量插入
func (slf *ReportWorkSearch) CreateBatch(records []*ReportWork) error {
	var db = slf.build()
	for _, record := range records {
		if err := db.Create(record).Error; err != nil {
			return fmt.Errorf("create batch err: %v, records: %+v", err, records)
		}
	}
	return nil
}

func (slf *ReportWorkSearch) Upsert(record *ReportWork) error {
	var db = slf.build()
	old, err := slf.First()
	if err != gorm.ErrRecordNotFound && old.ID != 0 {
		record.ID = old.ID
		err = db.Save(&record).Error
	} else {
		err = db.Create(&record).Error
	}
	if err != nil {
		return fmt.Errorf("save err: %v, record: %+v", err, record)
	}

	return nil
}

func (slf *ReportWorkSearch) Updates(record *ReportWork) error {
	var db = slf.build()

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

	return nil
}

func (slf *ReportWorkSearch) 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 *ReportWorkSearch) 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 *ReportWorkSearch) Delete() error {
	var db = slf.build()

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

	return nil
}

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

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

	return record, nil
}

func (slf *ReportWorkSearch) Find() ([]*ReportWork, int64, error) {
	var (
		records = make([]*ReportWork, 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 slf.Limit > 0 {
		db = db.Offset(slf.Offset).Limit(slf.Limit)
	}

	if err := db.Find(&records).Error; err != nil {
		return records, total, fmt.Errorf("find records err: %v", err)
	}

	return records, total, nil
}

func (slf *ReportWorkSearch) Count() (int64, error) {
	var (
		total int64
		db    = slf.build()
	)
	if err := db.Count(&total).Error; err != nil {
		return total, fmt.Errorf("find count err: %v", err)
	}
	return total, nil
}

func (slf *ReportWorkSearch) FindNotTotal() ([]*ReportWork, error) {
	var (
		records = make([]*ReportWork, 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 *ReportWorkSearch) FindByQuery(query string, args []interface{}) ([]*ReportWork, int64, error) {
	var (
		records = make([]*ReportWork, 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 *ReportWorkSearch) FindByQueryNotTotal(query string, args []interface{}) ([]*ReportWork, error) {
	var (
		records = make([]*ReportWork, 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
}