sundynix-pets/pets-be/internal/database/seed.go

package database

import (
	"golang.org/x/crypto/bcrypt"
	"gorm.io/gorm"

	"github.com/sundynix/pets-be/internal/config"
	"github.com/sundynix/pets-be/internal/model"
)

// Seed 初始化种子数据：管理员账号 + 记录类型。
//
// 文章、养护模板这类**内容**数据不放代码里——正文几百字塞进 Go 源文件，既看不清
// 也没法在后台改完再同步回来。上线时直接同步数据库。
//
// 记录类型是另一码事，它是**结构**：表空着的话记录页一个可记事项都没有，
// App 直接不能用。所以和管理员账号一样，只在表完全为空时铺一次底，
// 之后一切以数据库为准，代码里这份不再参与。
func Seed(db *gorm.DB, cfg *config.Config) error {
	if err := seedAdmin(db, cfg); err != nil {
		return err
	}
	return seedRecordTypes(db)
}

// seedRecordTypes 只在表为空时跑。已经有数据（哪怕只有一条）就完全不碰，
// 否则后台删掉的类型会在每次重启后自己长回来
func seedRecordTypes(db *gorm.DB) error {
	var count int64
	if err := db.Model(&model.RecordType{}).Count(&count).Error; err != nil {
		return err
	}
	if count > 0 {
		return nil
	}

	// locked=true 的这 9 种在 record.go / insight.go / report.go / home.go 里有真实
	// 代码分支（体重回写档案、食便相关性、消费统计、疫苗完成度…），并且被
	// care_template_items.sheet_type 引用，后台不许删，只能停用。
	// form=full 表示 bottom-sheet 里有它专用的 wx:elif 分支。
	// fields 是表单额外字段的配置，一行一个（写法见 model.RecordType.Fields 的注释）。
	// 空 = 只要「时间 + 描述 + 照片」。
	//
	// 原有 9 种的配置必须和现在的统计口径严丝合缝：
	//   weight  number → num_value，CreateRecord 靠它回写宠物档案
	//   cost    金额→num_value + 类别→category，账单是按 category 分组聚合的
	//   poop    状态→category，食便相关性分析排除 category='正常'
	//   symptom 两个单选：症状不占 category，严重程度占——周报按 category='高' 数高风险
	mk := func(code, label, icon, group string, sort int, locked bool, fields string) model.RecordType {
		form := model.RecordFormSimple
		if locked {
			form = model.RecordFormFull
		}
		return model.RecordType{Code: code, Label: label, Icon: icon, GroupKey: group,
			Sort: sort, Enabled: true, Form: form, Locked: locked, Fields: fields}
	}
	full := func(code, label, icon, group string, sort int, fields string) model.RecordType {
		return mk(code, label, icon, group, sort, true, fields)
	}
	simple := func(code, label, icon, group string, sort int, fields string) model.RecordType {
		return mk(code, label, icon, group, sort, false, fields)
	}

	d, h, c, cl := model.RecordGroupDaily, model.RecordGroupHealth, model.RecordGroupCare, model.RecordGroupClean
	types := []model.RecordType{
		// 日常
		full("weight", "体重", "weight", d, 10, "number:体重:kg"),
		full("poop", "排便", "poop", d, 20, "options:状态:正常|软便|拉稀"),
		full("food", "饮食", "food", d, 30, "text:吃了什么\noptions:食欲:正常|偏少|不吃"),
		simple("water", "喝水", "water", d, 40, "number:饮水量:ml"),
		full("cost", "记账", "cost", d, 50, "number:金额:元\noptions:类别:医疗|食品|用品|美容|其他"),
		full("photo", "照片", "photo", d, 60, ""),
		// 健康
		full("symptom", "异常", "symptom", h, 10,
			"options:发生了什么:呕吐|拉稀|精神差|皮肤红|其他:-\noptions:严重程度:低|中|高"),
		full("vaccine", "疫苗", "vaccine", h, 20, "text:疫苗名称"),
		full("deworm", "驱虫", "deworm", h, 30, "options:类型:体内|体外|体内外同驱"),
		simple("checkup", "体检", "checkup", h, 40, "text:在哪做的"),
		simple("vet", "看病", "vet", h, 50, "text:诊断\nnumber:花费:元"),
		full("medicine", "给药", "medicine", h, 60, "text:药品名称\ntext:剂量"),
		simple("supplement", "保健品", "supplement", h, 70, "text:名称\ntext:剂量"),
		// 洗护
		simple("bath", "洗澡", "bath", c, 10, ""),
		simple("nail", "剪指甲", "nail", c, 20, ""),
		simple("ear", "洗耳朵", "ear", c, 30, ""),
		simple("tooth", "刷牙", "tooth", c, 40, ""),
		simple("brush", "梳毛", "brush", c, 50, ""),
		simple("groom", "美容", "groom", c, 60, "number:花费:元"),
		// 清洁
		simple("litter", "换猫砂", "litter", cl, 10, ""),
		simple("litterbox", "洗猫砂盆", "litterbox", cl, 20, ""),
		simple("bowl", "洗食盆", "bowl", cl, 30, ""),
		simple("waterbowl", "洗水盆", "waterbowl", cl, 40, ""),
		simple("clean", "消毒", "clean", cl, 50, ""),
	}
	return db.Create(&types).Error
}

func seedAdmin(db *gorm.DB, cfg *config.Config) error {
	var count int64
	if err := db.Model(&model.Admin{}).Where("username = ?", cfg.Admin.Username).Count(&count).Error; err != nil {
		return err
	}
	if count > 0 {
		return nil
	}
	hash, err := bcrypt.GenerateFromPassword([]byte(cfg.Admin.Password), bcrypt.DefaultCost)
	if err != nil {
		return err
	}
	return db.Create(&model.Admin{
		Username:     cfg.Admin.Username,
		PasswordHash: string(hash),
		Role:         "admin",
	}).Error
}