package service

import (
	"apsClient/conf"
	"apsClient/constvar"
	"apsClient/model"
	"apsClient/model/common"
	"apsClient/model/response"
	"apsClient/nsq"
	"apsClient/pkg/ecode"
	"apsClient/pkg/logx"
	"apsClient/pkg/structx"
	"encoding/json"
	"fmt"
	"gorm.io/gorm"
	"time"
)

type TaskService struct {
}

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

// GetTask 获取任务，未完成的开始时间最早的任务
func (slf TaskService) GetTask() (taskData *response.TaskData, code int) {
	if v, ok := TaskCacheGet(); ok {
		return v, ecode.OK
	}

	nowTs := time.Now().Unix()
	var (
		err       error
		order     *model.Order
		procedure *model.Procedures
	)
	procedure, err = model.NewProceduresSearch(nil).SetOrder("start_time asc").
		SetStartTimeMax(nowTs).
		SetDeviceId(conf.Conf.System.DeviceId).
		SetStatusNot(model.ProcedureStatusFinished).
		First()
	if err == gorm.ErrRecordNotFound {
		return nil, ecode.OK
	}

	order, err = model.NewOrderSearch(nil).SetWorkOrderId(procedure.WorkOrderID).First() //由工序反推工单
	if err == gorm.ErrRecordNotFound {
		err = nil
		return nil, ecode.OK
	}
	if err != nil {
		return nil, ecode.DBErr
	}
	taskData = new(response.TaskData)
	taskData.Order = order
	taskData.Procedure = procedure
	TaskCacheSet(taskData)
	return taskData, ecode.OK
}

func (slf TaskService) GetProcedureById(id int) (*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) UpdateProcedureStatus(db *gorm.DB, id int, status model.ProcedureStatus) error {
	if status == model.ProcedureStatusFinished {
		TaskCacheUnset()
		ProgressCacheUnset()
	}
	return model.NewProceduresSearch(db).SetId(id).UpdateByMap(map[string]interface{}{
		"status": status,
	})
}

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().
		SetWorkOrder(procedure.WorkOrderID).
		SetDevice(procedure.ProceduresInfo.DeviceName).
		SetOrderId(procedure.OrderID).
		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 {
		err = json.Unmarshal([]byte(data.Params), &data.ParamsMap)
		if err != nil {
			logx.Errorf("process model json.Unmarshal:%v", err)
			return
		}
		return data, nil
	}

	if err == gorm.ErrRecordNotFound { //如果数据库没有从云端获取
		caller := nsq.NewCaller(fmt.Sprintf(constvar.NsqTopicProcessParamsRequest, conf.Conf.NsqConf.NodeId), fmt.Sprintf(constvar.NsqTopicProcessParamsResponse, conf.Conf.NsqConf.NodeId))
		var result common.ResponseProcessParams
		err = caller.Call(common.RequestProcessParams{
			WorkOrder: procedure.WorkOrderID,
			OrderId:   procedure.OrderID,
			Product:   order.ProductName,
			Procedure: procedure.ProceduresInfo.ProcedureName,
			Device:    procedure.ProceduresInfo.DeviceName,
			DeviceId:  conf.Conf.System.DeviceId,
		}, &result, time.Second*3)
		if err != nil {
			logx.Errorf("TaskStart GetProcessModel error:%v", err.Error())
			return
		}
		if result.ParamsMap == nil {
			logx.Errorf("TaskStart GetProcessModel response miss process params:%v", result)
			return
		}
		processModel = new(model.ProcessModel)
		err = structx.AssignTo(result, &processModel)
		return processModel, err
	}
	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()
}