QWQLwToo
100 lines
2.2 KiB
Go
100 lines
2.2 KiB
Go
package database
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"log"
|
|
|
|
"entgo.io/ent/dialect"
|
|
entsql "entgo.io/ent/dialect/sql"
|
|
"home-vue-go/internal/ent"
|
|
"home-vue-go/internal/ent/migrate"
|
|
|
|
_ "github.com/mattn/go-sqlite3"
|
|
"golang.org/x/crypto/bcrypt"
|
|
)
|
|
|
|
type Database struct {
|
|
Client *ent.Client
|
|
}
|
|
|
|
func Init(dbPath string) (*Database, error) {
|
|
db, err := sql.Open("sqlite3", dbPath+"?_fk=1")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
drv := entsql.OpenDB(dialect.SQLite, db)
|
|
client := ent.NewClient(ent.Driver(drv))
|
|
|
|
// 运行数据库迁移
|
|
ctx := context.Background()
|
|
if err := client.Schema.Create(ctx, migrate.WithForeignKeys(false)); err != nil {
|
|
log.Fatalf("数据库迁移失败: %v", err)
|
|
}
|
|
|
|
// 初始化默认数据
|
|
if err := initDefaultData(ctx, client); err != nil {
|
|
log.Printf("初始化默认数据失败: %v", err)
|
|
}
|
|
|
|
return &Database{Client: client}, nil
|
|
}
|
|
|
|
func (d *Database) Close() error {
|
|
return d.Client.Close()
|
|
}
|
|
|
|
func initDefaultData(ctx context.Context, client *ent.Client) error {
|
|
// 检查是否已有站点配置
|
|
_, err := client.SiteConfig.Get(ctx, 1)
|
|
if err == nil {
|
|
// 已存在配置，不初始化
|
|
return nil
|
|
}
|
|
|
|
// 创建默认站点配置
|
|
_, err = client.SiteConfig.Create().
|
|
SetSiteName("个人主页").
|
|
SetSiteURL("https://example.com").
|
|
SetSiteIcon("/favicon.ico").
|
|
SetSiteDescription("一个基于Vue3的个人主页").
|
|
SetSiteKeywords("个人主页,Vue3").
|
|
SetUserName("用户").
|
|
SetProfileImageURL("").
|
|
SetIcpNumber("").
|
|
SetPoliceNumber("").
|
|
SetPageTitle("个人主页").
|
|
SetFavicon("/favicon.ico").
|
|
SetUmamiScript("").
|
|
SetUmamiWebsiteID("").
|
|
SetIconLibrary("//lib.baomitu.com/font-awesome/6.5.0/css/all.min.css").
|
|
SetFontLibrary("").
|
|
Save(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// 检查是否已有用户，如果没有才创建默认管理员用户
|
|
userCount, err := client.User.Query().Count(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// 只有在没有任何用户时才创建默认管理员
|
|
if userCount == 0 {
|
|
hashedPassword, err := bcrypt.GenerateFromPassword([]byte("admin123"), bcrypt.DefaultCost)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
_, err = client.User.Create().
|
|
SetUsername("admin").
|
|
SetPassword(string(hashedPassword)).
|
|
Save(ctx)
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|