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
}