package logic

import (
	"context"
	"errors"
	"github.com/zeromicro/go-zero/core/logx"
	"gorm.io/gorm"
	plantModel "sundynix-micro-go/app/plant/model"
	"sundynix-micro-go/app/plant/rpc/internal/svc"
	"sundynix-micro-go/app/plant/rpc/plant"
)

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

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

func (l *CreateWikiLogic) CreateWiki(in *plant.CreateWikiReq) (*plant.CommonResp, error) {
	if !errors.Is(l.svcCtx.DB.Where("name LIKE ?", "%"+in.Name+"%").First(&plantModel.Wiki{}).Error, gorm.ErrRecordNotFound) {
		return nil, errors.New("植物已经存在")
	}
	classID := in.ClassId
	if classID == "" && len(in.ClassIds) > 0 {
		classID = in.ClassIds[0]
	}
	w := plantModel.Wiki{
		Name: in.Name, LatinName: in.LatinName, Aliases: in.Aliases,
		Genus: in.Genus, Difficulty: int(in.Difficulty), IsHot: int(in.IsHot),
		GrowthHabit: in.GrowthHabit, LightIntensity: in.LightIntensity,
		OptimalTempPeriod: in.OptimalTempPeriod, ClassID: classID,
		DistributionArea: in.DistributionArea, LifeCycle: in.LifeCycle,
		ReproductionMethod: in.ReproductionMethod, PestsDiseases: in.PestsDiseases,
		LightType: in.LightType, Stem: in.Stem, Fruit: in.Fruit,
		FoliageType: in.FoliageType, FoliageColor: in.FoliageColor,
		FoliageShape: in.FoliageShape, Height: int(in.Height),
		FloweringPeriod: in.FloweringPeriod, FloweringColor: in.FloweringColor,
		FloweringShape: in.FloweringShape, FlowerDiameter: int(in.FloweringDiameter),
	}
	if err := l.svcCtx.DB.Transaction(func(tx *gorm.DB) error {
		if err := tx.Create(&w).Error; err != nil {
			return err
		}
		return replaceWikiRelations(tx, w.ID, in.ClassIds, in.OssIds, in.RelatedWikiIds)
	}); err != nil {
		return nil, err
	}
	go func() {
		_, _ = NewSyncWikiVectorLogic(context.Background(), l.svcCtx).SyncWikiVector(&plant.SyncWikiVectorReq{WikiId: w.ID})
	}()
	return &plant.CommonResp{Code: 0, Msg: w.ID}, nil
}
