package main

import (
	"ankao/pkg/config"
	"fmt"
	"log"

	"gorm.io/driver/postgres"
	"gorm.io/gorm"
	"gorm.io/gorm/logger"
)

func main() {
	// 直接连接数据库，不使用 InitDB
	cfg := config.GetDatabaseConfig()
	dsn := cfg.GetDSN()

	db, err := gorm.Open(postgres.Open(dsn), &gorm.Config{
		Logger: logger.Default.LogMode(logger.Info),
	})
	if err != nil {
		log.Fatal("数据库连接失败:", err)
	}

	log.Println("开始清理 exam_shares 表...")

	// 删除 exam_shares 表（如果存在）
	if err := db.Exec("DROP TABLE IF EXISTS exam_shares CASCADE").Error; err != nil {
		log.Fatal("删除 exam_shares 表失败:", err)
	}

	log.Println("✓ 已删除 exam_shares 表")
	log.Println("\n清理完成！现在可以重新运行主程序。")
	fmt.Println("\n执行步骤:")
	fmt.Println("1. go run main.go  # 这会自动创建正确的表结构")
	fmt.Println("2. 如果有旧的分享数据需要迁移，运行:")
	fmt.Println("   go run cmd/migrate_exam_shares.go")
}