package service

import (
	"apsClient/constvar"
	"apsClient/model"
	"apsClient/model/common"
	"apsClient/model/request"
	"apsClient/model/response"
	"apsClient/pkg/ecode"
	"apsClient/pkg/logx"
	"fmt"
	"github.com/jinzhu/gorm"
	"time"
)

type TaskService struct {
}

func NewTaskService() *TaskService {
	return &TaskService{}
}

// GetTask 获取任务，未完成的开始时间小于等于当前时间，结束时间大于当前时间的任务
func (slf TaskService) GetTask(deviceID string, page, pageSize int, mode constvar.TaskMode, channels []int32) (taskResp *response.TaskResponse, code int) {
	var taskList []*response.TaskData
	var count int64
	var workers []*common.ProcedureWorker

	taskResp = &response.TaskResponse{
		Tasks:     taskList,
		TaskCount: count,
		Workers:   workers,
	}

	var (
		err          error
		orders       []*model.Order
		procedures   []*model.Procedures
		workOrderIds []string
	)
	search := model.NewProceduresSearch(nil).
		SetDeviceId(deviceID).
		SetPage(page, pageSize)

	if mode == constvar.TaskModeUnStarted {
		search.SetStatus(model.ProcedureStatusWaitProcess).
			SetOrder("start_time asc")
	} else if mode == constvar.TaskModeCurrent {
		search.SetStatus(model.ProcedureStatusProcessing).
			SetOrder("start_time asc")
	} else if mode == constvar.TaskModeLastFinished {
		search.SetStatus(model.ProcedureStatusFinished).SetOrder("updated_at desc")
		if len(channels) > 0 {
			search.SetChannels(channels)
		}
	}

	procedures, err = search.FindNotTotal()
	if err != nil {
		return taskResp, ecode.DBErr
	}

	if err != nil {
		return taskResp, ecode.DBErr
	}
	if len(procedures) == 0 {
		return taskResp, ecode.OK
	}

	for _, procedure := range procedures {
		workOrderIds = append(workOrderIds, procedure.WorkOrderID)
	}
	orders, err = model.NewOrderSearch(nil).SetWorkOrderIds(workOrderIds).FindNotTotal() //由工序反推工单
	if err != nil {
		return taskResp, ecode.DBErr
	}

	orderMap := make(map[string]*model.Order, len(workOrderIds))
	for _, order := range orders {
		orderMap[order.WorkOrderID] = order
	}
	for _, procedure := range procedures {
		taskData := response.TaskData{
			Order:                 orderMap[procedure.WorkOrderID],
			Procedure:             procedure,
			AllProcedures:         procedure.ProceduresInfo.AllProcedureNames,
			CurrentProcedureIndex: 0,
			Channel:               procedure.Channel,
		}
		index := 0
		for _, name := range procedure.ProceduresInfo.AllProcedureNames {
			if name == procedure.ProceduresInfo.ProcedureName {
				break
			}
			index++
		}
		taskData.CurrentProcedureIndex = index
		taskList = append(taskList, &taskData)
		workers = append(workers, procedure.ProceduresInfo.Workers...)
	}
	taskResp = &response.TaskResponse{
		Tasks:     taskList,
		TaskCount: count,
		Workers:   slf.WorkerDistinct(workers),
	}
	return taskResp, ecode.OK
}
func (slf TaskService) NewTaskCount(deviceId string) (count int64) {
	count, _ = model.NewProceduresSearch(nil).SetDeviceId(deviceId).SetStatus(model.ProcedureStatusWaitProcess).Count()
	return count
}

func (slf TaskService) WorkerDistinct(workers []*common.ProcedureWorker) (NoRepeatedWorkers []*common.ProcedureWorker) {
	exists := make(map[string]bool, 0)
	for _, worker := range workers {
		key := fmt.Sprintf("%v%v", worker.WorkerID, worker.StartTime)
		if !exists[key] {
			NoRepeatedWorkers = append(NoRepeatedWorkers, worker)
			exists[key] = true
		}
	}
	return NoRepeatedWorkers
}

func (slf TaskService) GetProcedureById(id uint) (*model.Procedures, int) {
	procedure, err := model.NewProceduresSearch(nil).SetId(id).First()
	if err != nil {
		return nil, ecode.DBErr
	}
	return procedure, ecode.OK
}

func (slf TaskService) UpdateProcedureStatusAndChannel(db *gorm.DB, id uint, status model.ProcedureStatus, channel int32, processModelNumber string) error {
	if status == model.ProcedureStatusFinished || status == model.ProcedureStatusWaitProcess {
		ProgressCacheUnset(channel)
	}

	upMap := map[string]interface{}{"status": status}
	if status == model.ProcedureStatusProcessing {
		upMap["process_model_number"] = processModelNumber
		upMap["real_start_time"] = time.Now().Unix()
	} else if status == model.ProcedureStatusWaitProcess {
		upMap["process_model_number"] = ""
	} else if status == model.ProcedureStatusFinished {
		upMap["real_end_time"] = time.Now().Unix()
	}

	return model.NewProceduresSearch(db).SetId(id).UpdateByMap(upMap)
}

func (slf TaskService) UpdateProcessModelNumber(db *gorm.DB, id uint, processModelNumber string) error {
	upMap := map[string]interface{}{"process_model_number": processModelNumber}
	return model.NewProceduresSearch(db).SetId(id).UpdateByMap(upMap)
}

func (slf TaskService) UpdateOrderStatus(db *gorm.DB, id uint, status model.OrderStatus) error {
	return model.NewOrderSearch(db).SetId(id).UpdateByMap(map[string]interface{}{
		"status": status,
	})
}

func (slf TaskService) GetOrderByWorkOrderId(workOrderId string) (order *model.Order, err error) {
	return model.NewOrderSearch(nil).SetWorkOrderId(workOrderId).First()
}

func (slf TaskService) GetProcessParams(procedure *model.Procedures, order *model.Order) (processModel *model.ProcessModel, err error) {
	data, err := model.NewProcessModelSearch().
		SetProcedure(procedure.ProceduresInfo.ProcedureName).
		SetProduct(order.ProductName).
		SetOrder("id desc").First()
	if err != nil && err != gorm.ErrRecordNotFound {
		logx.Errorf("TaskStart GetProcessModel  err:%v", err)
		return nil, err
	}
	if err == nil {
		return data, nil
	}
	return
}

// GetNextTask 获取未开始的任务
func (slf TaskService) GetNextTask() (workOrder *model.Order, err error) {
	nowTs := time.Now().Unix()
	orderSearch := model.NewOrderSearch(nil)
	return orderSearch.SetOrder("start_time asc").
		SetStartTimeMin(nowTs).
		SetStatus(model.OrderStatusWaitProcess).First()
}

// SaveTaskStatusSync 保存任务同步信息
func (slf TaskService) SaveTaskStatusSync(db *gorm.DB, record *model.TaskStatusSync) (err error) {
	return model.NewTaskStatusSyncSearch(db).Create(record)
}

// GetTaskStatusSync 获取待同步任务信息
func (slf TaskService) GetTaskStatusSync(limit int) ([]*model.TaskStatusSync, error) {
	return model.NewTaskStatusSyncSearch(nil).SetOrder("id desc").SetPage(1, limit).FindNotTotal()
}

// RemoveTaskStatusSync 删除同步完的记录
func (slf TaskService) RemoveTaskStatusSync(ids []uint) (err error) {
	return model.NewTaskStatusSyncSearch(nil).SetIDs(ids).Delete()
}

// GetTask2 获取任务列表2
func (slf TaskService) GetTask2(deviceID string, offset, limit int, channels []int32, queryType request.QueryType) (taskResp *response.TaskResponse, err error) {
	var taskList []*response.TaskData
	var count int64
	var workers []*common.ProcedureWorker

	taskResp = &response.TaskResponse{
		Tasks:     taskList,
		TaskCount: count,
		Workers:   workers,
	}

	var (
		orders       []*model.Order
		procedures   []*model.Procedures
		workOrderIds []string
	)
	search := model.NewProceduresSearch(nil).
		SetDeviceId(deviceID).
		SetOffset(offset, limit).SetChannels(channels)
	nowTs := time.Now().Unix()
	switch queryType {
	case request.QueryTypeUnFinish:
		search.SetStatusNot(model.ProcedureStatusFinished).SetOrder("status desc, start_time asc")
	case request.QueryTypeToday:
		search.SetStatusNot(model.ProcedureStatusFinished).SetStartTimeMin(nowTs).SetEndTimeMax(nowTs + 86400).SetOrder("status desc, start_time asc")
	case request.QueryTypeFinished:
		search.SetStatus(model.ProcedureStatusFinished).SetOrder("start_time desc")
	}

	procedures, total, err := search.Find()
	if err != nil {
		return taskResp, err
	}

	if err != nil {
		return taskResp, err
	}
	if len(procedures) == 0 {
		return taskResp, nil
	}

	for _, procedure := range procedures {
		workOrderIds = append(workOrderIds, procedure.WorkOrderID)
	}
	orders, err = model.NewOrderSearch(nil).SetWorkOrderIds(workOrderIds).FindNotTotal() //由工序反推工单
	if err != nil {
		return taskResp, err
	}

	orderMap := make(map[string]*model.Order, len(workOrderIds))
	for _, order := range orders {
		orderMap[order.WorkOrderID] = order
	}
	for _, procedure := range procedures {
		taskData := response.TaskData{
			Order:                 orderMap[procedure.WorkOrderID],
			Procedure:             procedure,
			AllProcedures:         procedure.ProceduresInfo.AllProcedureNames,
			CurrentProcedureIndex: 0,
			Channel:               procedure.Channel,
		}
		index := 0
		for _, name := range procedure.ProceduresInfo.AllProcedureNames {
			if name == procedure.ProceduresInfo.ProcedureName {
				break
			}
			index++
		}
		taskData.CurrentProcedureIndex = index
		taskList = append(taskList, &taskData)
		workers = append(workers, procedure.ProceduresInfo.Workers...)
	}
	taskResp = &response.TaskResponse{
		Tasks:     taskList,
		TaskCount: total,
		Workers:   slf.WorkerDistinct(workers),
	}
	return taskResp, err
}