package preload

import (
	"context"

	"bj_power_mes/ent"

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

// SeedProcessDict 首次启动时预置 12 道工序字典（Q9 种子数据），后续可在页面维护
func SeedProcessDict(entClient *ent.Client) {
	ctx := context.Background()
	n, err := entClient.ProcessDict.Query().Count(ctx)
	if err != nil || n > 0 {
		return
	}
	names := [12]string{
		"1序 上料", "2序 粗铣", "3序 精铣", "4序 钻孔", "5序 攻丝", "6序 去毛刺",
		"7序 清洗", "8序 检验", "9序 打标", "10序 装配", "11序 拧紧", "12序 下线",
	}
	bulk := make([]*ent.ProcessDictCreate, 0, 12)
	for i := 1; i <= 12; i++ {
		bulk = append(bulk, entClient.ProcessDict.Create().
			SetCode(i).
			SetName(names[i-1]).
			SetDescription("").
			SetSortNum(i).
			SetEnabled(true))
	}
	if _, err = entClient.ProcessDict.CreateBulk(bulk...).Save(ctx); err != nil {
		logx.Errorf("seed process dict failed: %v", err)
	} else {
		logx.Info("seeded default 12 processes")
	}
}