package main

import (
	"ankao/internal/database"
	"log"
	"os"
)

func main() {
	// 直接设置数据库配置
	os.Setenv("DB_HOST", "localhost")
	os.Setenv("DB_PORT", "5432")
	os.Setenv("DB_USER", "postgres")
	os.Setenv("DB_PASSWORD", "root")
	os.Setenv("DB_NAME", "ankao")

	// 连接数据库
	if err := database.InitDB(); err != nil {
		log.Fatalf("数据库连接失败: %v", err)
	}

	db := database.GetDB()

	log.Println("开始迁移：删除 wrong_questions 表的旧字段...")

	// 删除 wrong_answer 和 correct_answer 字段
	// 这些字段在新版本中已移至 wrong_question_history 表
	migrations := []string{
		"ALTER TABLE wrong_questions DROP COLUMN IF EXISTS wrong_answer",
		"ALTER TABLE wrong_questions DROP COLUMN IF EXISTS correct_answer",
		"ALTER TABLE wrong_questions DROP COLUMN IF EXISTS wrong_count", // 也删除旧的 wrong_count 字段（如果存在）
	}

	for _, migration := range migrations {
		log.Printf("执行: %s", migration)
		if err := db.Exec(migration).Error; err != nil {
			log.Printf("警告: 执行失败 - %v (字段可能已不存在)", err)
		} else {
			log.Println("✓ 执行成功")
		}
	}

	log.Println("迁移完成！")
}