package mysql

import (
	"gorm.io/gorm"

	"vul-backend/internal/model"
	vulnmodel "vul-backend/internal/modules/vuln/model"
)

func AutoMigrate(db *gorm.DB) error {
	return db.AutoMigrate(
		&model.User{},
		&vulnmodel.Vulnerability{},
		&vulnmodel.Tag{},
		&vulnmodel.VulnerabilityTagRelation{},
		&vulnmodel.DiscoveryChannel{},
	)
}
