package logic

import (
	"context"
	"lone-services/pkg/modelbase"
	"lone-services/pkg/utils"
	"lone-services/rpc/express/pb"
	"lone-services/services/express/internal/dao"
	"lone-services/services/express/internal/model"
	"lone-services/services/express/internal/svc"
	"lone-services/services/express/validator"
	"strconv"

	"github.com/zeromicro/go-zero/core/logx"
)

type PrintExpressLogic struct {
	ctx    context.Context
	svcCtx *svc.ServiceContext
	logx.Logger
	BaseLogic
}

func NewPrintExpressLogic(ctx context.Context, svcCtx *svc.ServiceContext) *PrintExpressLogic {
	return &PrintExpressLogic{
		ctx:    ctx,
		svcCtx: svcCtx,
		Logger: logx.WithContext(ctx),
	}
}

func (l *PrintExpressLogic) PrintExpress(in *express.PrintExpressRequest) (*express.Response, error) {
	var v validator.PrintOrderValidator
	if fail := l.checkParams(in, &v); fail != nil {
		return fail, nil
	}
	adminInfo := utils.GetUserFromCtx(l.ctx)
	if adminInfo.ID < utils.NumberOne {
		return l.fail(utils.ErrorNoLoginInfo)
	}

	var info []dao.DeliveryOrderPrint
	w := modelbase.Params{
		In: map[string][]string{"status": {strconv.Itoa(dao.SendStatusOk), strconv.Itoa(dao.SendStatusIn), strconv.Itoa(dao.SendStatusWait)}},
	}
	if v.Status == utils.NumberOne {
		w.Page = utils.NumberOne
		w.Size = utils.SIZE
		w.Order = "id desc"
	}

	if v.Status == utils.NumberZero {
		w.Page = utils.NumberOne
		w.Size = utils.SIZE
		w.Order = "print_status desc, id desc"
	}

	w.Eq = make(map[string]string)
	if v.Status > utils.NumberZero {
		w.Eq["print_status"] = strconv.Itoa(int(v.Status))
	}

	if len(v.OrderSn) > utils.NumberZero {
		w.Eq["obj_sn"] = v.OrderSn
	}

	startTime, endTime := utils.TimeToStartAndEnd(v.Time, utils.NumberOne)
	if len(startTime) > utils.NumberZero {
		betweenTimeKey := "create_time"
		if v.Status == utils.NumberOne {
			betweenTimeKey = "update_time"
		}
		w.Between = map[string][2]string{betweenTimeKey: {startTime, endTime}}
	}

	items, err := model.DeliveryOrderModel{}.Init().Page(w, &info)

	if err != nil {
		l.Logger.Error(err)
		return l.fail(utils.Fail)
	}

	obj := svc.Print{}
	obj.GetOrderInfo(l.svcCtx.ProductSvcName, &info)
	items.Items = info

	return l.ok(items)
}