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 分支。
	full := func(code, label, icon, group string, sort int) model.RecordType {
		return model.RecordType{Code: code, Label: label, Icon: icon, GroupKey: group,
			Sort: sort, Enabled: true, Form: model.RecordFormFull, Locked: true}
	}
	simple := func(code, label, icon, group string, sort int) model.RecordType {
		return model.RecordType{Code: code, Label: label, Icon: icon, GroupKey: group,
			Sort: sort, Enabled: true, Form: model.RecordFormSimple, Locked: false}
	}

	d, h, c, cl := model.RecordGroupDaily, model.RecordGroupHealth, model.RecordGroupCare, model.RecordGroupClean
	types := []model.RecordType{
		// 日常
		full("weight", "体重", "weight", d, 10),
		full("poop", "排便", "poop", d, 20),
		full("food", "饮食", "food", d, 30),
		simple("water", "喝水", "water", d, 40),
		full("cost", "记账", "cost", d, 50),
		full("photo", "照片", "photo", d, 60),
		// 健康
		full("symptom", "异常", "symptom", h, 10),
		full("vaccine", "疫苗", "vaccine", h, 20),
		full("deworm", "驱虫", "deworm", h, 30),
		simple("checkup", "体检", "checkup", h, 40),
		simple("vet", "看病", "vet", h, 50),
		full("medicine", "给药", "medicine", h, 60),
		simple("supplement", "保健品", "supplement", h, 70),
		// 洗护
		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),
		// 清洁
		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
}