package logic

import (
	"encoding/json"
	"errors"
	"fmt"
	"fusenapi/constants"
	"fusenapi/utils/auth"
	"fusenapi/utils/basic"
	"fusenapi/utils/template_switch_info"
	"gorm.io/gorm"
	"strings"

	"context"

	"fusenapi/server/product/internal/svc"
	"fusenapi/server/product/internal/types"

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

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

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

func (l *GetTemplateByPidLogic) GetTemplateByPid(req *types.GetTemplateByPidReq, userinfo *auth.UserInfo) (resp *basic.Response) {
	req.Pid = strings.Trim(req.Pid, " ")
	if req.Pid == "" {
		return resp.SetStatusWithMessage(basic.CodeRequestParamsErr, "err param:pid is empty")
	}
	if req.TemplateTag == "" {
		return resp.SetStatusWithMessage(basic.CodeRequestParamsErr, "err param:template_tag")
	}
	//获取产品信息(只获取id)
	productInfo, err := l.svcCtx.AllModels.FsProduct.FindOneBySn(l.ctx, req.Pid, "id")
	if err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return resp.SetStatusWithMessage(basic.CodeDbRecordNotFoundErr, "the product is not exists")
		}
		logx.Error(err)
		return resp.SetStatusWithMessage(basic.CodeDbSqlErr, "failed to get product info")
	}
	//没有指定物料
	sizeIds := make([]int64, 0, 10)
	if req.ProductSizeId <= 0 {
		//获取产品所有物料
		sizeList, err := l.svcCtx.AllModels.FsProductSize.GetAllByProductIds(l.ctx, []int64{productInfo.Id}, "id")
		if err != nil {
			logx.Error(err)
			return resp.SetStatusWithMessage(basic.CodeDbRecordNotFoundErr, "failed to get product  size list")
		}
		if len(sizeList) == 0 {
			return resp.SetStatusWithMessage(basic.CodeDbRecordNotFoundErr, "product size list is empty")
		}
		for _, v := range sizeList {
			sizeIds = append(sizeIds, v.Id)
		}
	} else { //指定物料
		sizeIds = append(sizeIds, req.ProductSizeId)
	}
	//根据尺寸id获取模型
	modelList, err := l.svcCtx.AllModels.FsProductModel3d.GetAllBySizeIdsTag(l.ctx, sizeIds, constants.TAG_MODEL)
	if err != nil {
		logx.Error(err)
		return resp.SetStatusWithMessage(basic.CodeDbSqlErr, "failed to get model list")
	}
	if len(modelList) == 0 {
		return resp.SetStatusWithMessage(basic.CodeDbSqlErr, "model list is empty")
	}
	modelIds := make([]int64, 0, len(modelList))
	mapModel := make(map[int64]int)
	for k, v := range modelList {
		modelIds = append(modelIds, v.Id)
		mapModel[v.Id] = k
	}
	//查询模型ids下对应tag标签的模板
	templateList, err := l.svcCtx.AllModels.FsProductTemplateV2.FindAllByModelIdsTemplateTag(l.ctx, modelIds, req.TemplateTag, "")
	if err != nil {
		logx.Error(err)
		return resp.SetStatusWithMessage(basic.CodeDbSqlErr, "failed to get template list")
	}
	rsp := make(map[string]interface{})
	for _, templateInfo := range templateList {
		//没有设置模板据不要
		modelIndex, ok := mapModel[*templateInfo.ModelId]
		if !ok {
			continue
		}
		//基础模板信息
		var info interface{}
		if err = json.Unmarshal([]byte(*templateInfo.TemplateInfo), &info); err != nil {
			logx.Error(err)
			return resp.SetStatusWithMessage(basic.CodeJsonErr, fmt.Sprintf("failed to parse json product template info(may be old data):%d", templateInfo.Id))
		}
		modelInfo := modelList[modelIndex]
		mapKey := fmt.Sprintf("_%d", *modelInfo.SizeId)
		rsp[mapKey] = template_switch_info.GetTemplateSwitchInfo(templateInfo.Id, templateInfo.TemplateInfo, *templateInfo.MaterialImg)
	}
	return resp.SetStatusWithMessage(basic.CodeOK, "success", rsp)
}