254 lines
7.0 KiB
Go
254 lines
7.0 KiB
Go
package cmd
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"strings"
|
|
|
|
"github.com/gogf/gf/v2/frame/g"
|
|
"github.com/gogf/gf/v2/os/gfile"
|
|
"github.com/gogf/gf/v2/os/gtime"
|
|
"golang.org/x/crypto/bcrypt"
|
|
)
|
|
|
|
// InitDatabase 初始化数据库
|
|
func InitDatabase(ctx context.Context) error {
|
|
db := g.DB()
|
|
|
|
// 检查数据库连接
|
|
if err := db.PingMaster(); err != nil {
|
|
return fmt.Errorf("数据库连接失败: %v", err)
|
|
}
|
|
|
|
g.Log().Info(ctx, "数据库连接成功，开始初始化数据库...")
|
|
|
|
// 读取SQL文件
|
|
sqlFile := "nl_video_database.sql"
|
|
if !gfile.Exists(sqlFile) {
|
|
return fmt.Errorf("SQL文件不存在: %s", sqlFile)
|
|
}
|
|
|
|
sqlContent, err := ioutil.ReadFile(sqlFile)
|
|
if err != nil {
|
|
return fmt.Errorf("读取SQL文件失败: %v", err)
|
|
}
|
|
|
|
// 分割SQL语句
|
|
sqlStatements := strings.Split(string(sqlContent), ";")
|
|
|
|
// 执行SQL语句
|
|
for i, statement := range sqlStatements {
|
|
statement = strings.TrimSpace(statement)
|
|
if statement == "" || strings.HasPrefix(statement, "--") || strings.HasPrefix(statement, "/*") {
|
|
continue
|
|
}
|
|
|
|
g.Log().Infof(ctx, "执行SQL语句 %d/%d", i+1, len(sqlStatements))
|
|
|
|
if _, err := db.Exec(ctx, statement); err != nil {
|
|
// 如果是表已存在的错误，跳过
|
|
if strings.Contains(err.Error(), "already exists") {
|
|
g.Log().Warning(ctx, "表已存在，跳过创建:", statement[:50])
|
|
continue
|
|
}
|
|
return fmt.Errorf("执行SQL失败: %v, SQL: %s", err, statement[:100])
|
|
}
|
|
}
|
|
|
|
// 初始化基础数据
|
|
if err := initBaseData(ctx); err != nil {
|
|
return fmt.Errorf("初始化基础数据失败: %v", err)
|
|
}
|
|
|
|
g.Log().Info(ctx, "数据库初始化完成！")
|
|
return nil
|
|
}
|
|
|
|
// initBaseData 初始化基础数据
|
|
func initBaseData(ctx context.Context) error {
|
|
db := g.DB()
|
|
|
|
// 检查是否已有管理员数据
|
|
count, err := db.Model("nl_admin").Count()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if count > 0 {
|
|
g.Log().Info(ctx, "管理员数据已存在，跳过初始化")
|
|
return nil
|
|
}
|
|
|
|
// 创建默认超级管理员
|
|
hashedPassword, err := bcrypt.GenerateFromPassword([]byte("admin123"), bcrypt.DefaultCost)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// 插入超级管理员角色
|
|
roleResult, err := db.Model("nl_role").Data(g.Map{
|
|
"name": "超级管理员",
|
|
"code": "super_admin",
|
|
"level": 1,
|
|
"is_system": 1,
|
|
"description": "系统超级管理员，拥有所有权限",
|
|
"status": 1,
|
|
"created_at": gtime.Now(),
|
|
"updated_at": gtime.Now(),
|
|
}).Insert()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
roleId, err := roleResult.LastInsertId()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// 插入超级管理员账号
|
|
_, err = db.Model("nl_admin").Data(g.Map{
|
|
"username": "admin",
|
|
"password": string(hashedPassword),
|
|
"email": "admin@nlvideo.com",
|
|
"real_name": "系统管理员",
|
|
"nickname": "超级管理员",
|
|
"role_id": roleId,
|
|
"status": 1,
|
|
"created_at": gtime.Now(),
|
|
"updated_at": gtime.Now(),
|
|
}).Insert()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// 初始化基础权限
|
|
if err := initPermissions(ctx, roleId); err != nil {
|
|
return err
|
|
}
|
|
|
|
// 初始化影片分类
|
|
if err := initCategories(ctx); err != nil {
|
|
return err
|
|
}
|
|
|
|
// 初始化VIP等级
|
|
if err := initVipLevels(ctx); err != nil {
|
|
return err
|
|
}
|
|
|
|
g.Log().Info(ctx, "基础数据初始化完成")
|
|
g.Log().Info(ctx, "默认管理员账号: admin")
|
|
g.Log().Info(ctx, "默认管理员密码: admin123")
|
|
|
|
return nil
|
|
}
|
|
|
|
// initPermissions 初始化权限数据
|
|
func initPermissions(ctx context.Context, roleId int64) error {
|
|
db := g.DB()
|
|
|
|
permissions := []g.Map{
|
|
{
|
|
"name": "系统管理", "code": "system", "type": "menu", "parent_id": 0,
|
|
"path": "/system", "component": "Layout", "icon": "system",
|
|
"sort": 1, "status": 1, "created_at": gtime.Now(), "updated_at": gtime.Now(),
|
|
},
|
|
{
|
|
"name": "用户管理", "code": "user", "type": "menu", "parent_id": 0,
|
|
"path": "/user", "component": "Layout", "icon": "user",
|
|
"sort": 2, "status": 1, "created_at": gtime.Now(), "updated_at": gtime.Now(),
|
|
},
|
|
{
|
|
"name": "影片管理", "code": "movie", "type": "menu", "parent_id": 0,
|
|
"path": "/movie", "component": "Layout", "icon": "movie",
|
|
"sort": 3, "status": 1, "created_at": gtime.Now(), "updated_at": gtime.Now(),
|
|
},
|
|
{
|
|
"name": "权限管理", "code": "permission", "type": "menu", "parent_id": 0,
|
|
"path": "/permission", "component": "Layout", "icon": "permission",
|
|
"sort": 4, "status": 1, "created_at": gtime.Now(), "updated_at": gtime.Now(),
|
|
},
|
|
}
|
|
|
|
for _, perm := range permissions {
|
|
result, err := db.Model("nl_permission").Data(perm).Insert()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
permId, err := result.LastInsertId()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// 给超级管理员分配权限
|
|
_, err = db.Model("nl_role_permission").Data(g.Map{
|
|
"role_id": roleId,
|
|
"permission_id": permId,
|
|
"created_at": gtime.Now(),
|
|
}).Insert()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// initCategories 初始化影片分类
|
|
func initCategories(ctx context.Context) error {
|
|
db := g.DB()
|
|
|
|
categories := []g.Map{
|
|
{"name": "电影", "code": "movie", "parent_id": 0, "sort": 1, "status": 1, "created_at": gtime.Now(), "updated_at": gtime.Now()},
|
|
{"name": "电视剧", "code": "tv", "parent_id": 0, "sort": 2, "status": 1, "created_at": gtime.Now(), "updated_at": gtime.Now()},
|
|
{"name": "综艺", "code": "variety", "parent_id": 0, "sort": 3, "status": 1, "created_at": gtime.Now(), "updated_at": gtime.Now()},
|
|
{"name": "动漫", "code": "anime", "parent_id": 0, "sort": 4, "status": 1, "created_at": gtime.Now(), "updated_at": gtime.Now()},
|
|
{"name": "纪录片", "code": "documentary", "parent_id": 0, "sort": 5, "status": 1, "created_at": gtime.Now(), "updated_at": gtime.Now()},
|
|
}
|
|
|
|
for _, category := range categories {
|
|
_, err := db.Model("nl_category").Data(category).Insert()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// initVipLevels 初始化VIP等级
|
|
func initVipLevels(ctx context.Context) error {
|
|
db := g.DB()
|
|
|
|
vipLevels := []g.Map{
|
|
{
|
|
"name": "普通用户", "level": 0, "price": 0, "duration": 0,
|
|
"description": "免费用户，可观看部分免费内容",
|
|
"privileges": `{"free_content": true, "hd_quality": false, "download": false}`,
|
|
"status": 1, "created_at": gtime.Now(), "updated_at": gtime.Now(),
|
|
},
|
|
{
|
|
"name": "月度VIP", "level": 1, "price": 1500, "duration": 30,
|
|
"description": "月度会员，享受高清观看和下载权限",
|
|
"privileges": `{"free_content": true, "hd_quality": true, "download": true, "ad_free": true}`,
|
|
"status": 1, "created_at": gtime.Now(), "updated_at": gtime.Now(),
|
|
},
|
|
{
|
|
"name": "年度VIP", "level": 2, "price": 15000, "duration": 365,
|
|
"description": "年度会员，享受所有内容和特权",
|
|
"privileges": `{"free_content": true, "hd_quality": true, "download": true, "ad_free": true, "exclusive_content": true}`,
|
|
"status": 1, "created_at": gtime.Now(), "updated_at": gtime.Now(),
|
|
},
|
|
}
|
|
|
|
for _, vip := range vipLevels {
|
|
_, err := db.Model("nl_vip_level").Data(vip).Insert()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
} |