package service

import (
	"github.com/spf13/cast"
	"gorm.io/gorm"
	"strconv"
	"wms/constvar"
	"wms/models"
	"wms/pkg/blevex"
	"wms/pkg/logx"
	"wms/pkg/structx"
)

type HistoryReport struct {
	ID                int
	BaseOperationType constvar.BaseOperationType
	Number            string //操作单号
	ProductId         string //产品编号
	ProductName       string //产品名称
	OperationTypeName string //业务名称
	FromLocation      string //源位置名称
	ToLocation        string //目标位置名称
}

const (
	HistoryReportIndexName = "HistoryReport.bleve"
)

func InitHistoryReportData() {
	docCount, err := blevex.DocCount(HistoryReportIndexName)
	if err != nil {
		logx.Errorf("InitHistoryReportData get doc count err:%v", err)
		return
	}
	if docCount > 0 {
		return
	}
	records := make([]*models.MoveHistory, 0, 100)
	reports := make([]*HistoryReport, 0, 100)
	err = models.NewMoveHistorySearch().Orm.FindInBatches(&records, 100, func(tx *gorm.DB, batch int) error {
		err = structx.AssignTo(records, &reports)
		if err != nil {
			logx.Errorf("AddNewHistoryReportRecord AssignTo err:%v", err)
			return err
		}
		for _, report := range reports {
			err = blevex.Add(HistoryReportIndexName, strconv.Itoa(report.ID), report)
			if err != nil {
				logx.Errorf("InitHistoryReportData add failed, err:%v, index:%v, data:%v", err, HistoryReportIndexName, report)
				return err
			}
		}
		return nil
	}).Error

	if err != nil {
		logx.Errorf("InitHistoryReportData scan err:%v", err)
	}
	return
}

func SearchHistoryReport(keyword string, operationType constvar.BaseOperationType, page, pageSize int) (recordIds []int, total uint64, err error) {
	var ids []string
	from := (page - 1) * pageSize
	ids, total, err = blevex.ComplexSearch(HistoryReportIndexName, keyword, map[string]interface{}{"BaseOperationType": operationType}, from, pageSize)
	if err != nil {
		return
	}
	if len(ids) == 0 {
		return
	}
	recordIds = make([]int, 0, len(ids))
	for _, id := range ids {
		recordIds = append(recordIds, cast.ToInt(id))
	}
	return
}

func AddNewHistoryReportRecord(moveHistoryId int) {
	var report HistoryReport
	record, err := models.NewMoveHistorySearch().SetID(uint(moveHistoryId)).First()
	if err != nil {
		logx.Errorf("AddNewHistoryReportRecord Find err:%v", err)
		return
	}
	err = structx.AssignTo(record, &report)
	if err != nil {
		logx.Errorf("AddNewHistoryReportRecord AssignTo err:%v", err)
		return
	}
	err = blevex.Add(HistoryReportIndexName, strconv.Itoa(moveHistoryId), report)
	if err != nil {
		logx.Errorf("AddNewHistoryReportRecord bleve add err:%v", err)
		return
	}
}