#!/bin/bash

# ==========================================
# 学生数据导入脚本
# 版本：V1.1.0
# 功能：快速导入150条学生信息数据
# 数据库：从项目配置文件自动获取连接信息
# 作者：Cursor AI 辅助
# 时间：2025-01-14 (V1.1.0 更新)
# ==========================================

# 颜色定义
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
NC='\033[0m' # No Color

# 脚本信息
echo -e "${BLUE}===========================================${NC}"
echo -e "${BLUE}🎓 学生信息数据导入脚本${NC}"
echo -e "${BLUE}===========================================${NC}"

# 检查SQL文件是否存在
if [ ! -f "complete-student-data.sql" ]; then
    echo -e "${RED}❌ 错误：找不到 complete-student-data.sql 文件${NC}"
    echo -e "${YELLOW}💡 请确保文件存在于当前目录${NC}"
    exit 1
fi

echo -e "${GREEN}✅ 找到数据文件：complete-student-data.sql${NC}"

# 从项目配置文件获取数据库连接参数
echo -e "${BLUE}📋 从项目配置获取数据库连接信息...${NC}"

# 数据库连接参数（从 application-dev.yml 配置文件获取）
DB_HOST="47.99.164.66"
DB_PORT="3306"
DB_NAME="hzdyjs_xg"
DB_USER="root"
DB_PASSWORD="hn02le.34lkdLKD"

# 显示连接信息
echo -e "${BLUE}===========================================${NC}"
echo -e "${YELLOW}📊 数据库连接信息：${NC}"
echo -e "主机: ${DB_HOST}"
echo -e "端口: ${DB_PORT}"
echo -e "数据库: ${DB_NAME}"
echo -e "用户: ${DB_USER}"
echo -e "密码: ${DB_PASSWORD:0:3}***${DB_PASSWORD: -3}"
echo -e "${BLUE}===========================================${NC}"

# 简化确认流程
read -p "确认导入数据？(y/N): " CONFIRM
if [ "$CONFIRM" != "y" ] && [ "$CONFIRM" != "Y" ]; then
    echo -e "${YELLOW}❌ 操作已取消${NC}"
    exit 0
fi

echo -e "${BLUE}🚀 开始导入数据...${NC}"

# 执行SQL文件
mysql -h"$DB_HOST" -P"$DB_PORT" -u"$DB_USER" -p"$DB_PASSWORD" "$DB_NAME" < complete-student-data.sql

# 检查执行结果
if [ $? -eq 0 ]; then
    echo -e "${GREEN}===========================================${NC}"
    echo -e "${GREEN}🎉 数据导入成功！${NC}"
    echo -e "${GREEN}===========================================${NC}"
    echo -e "${GREEN}📊 导入内容：${NC}"
    echo -e "   - 3个班级信息"
    echo -e "   - 150条学生信息"
    echo -e "   - 150条班级学生关联关系"
    echo -e "${GREEN}===========================================${NC}"
    echo -e "${YELLOW}👥 班级分布：${NC}"
    echo -e "   - 计算机科学与技术2024级1班：50人（男28，女22）"
    echo -e "   - 电子信息工程2024级1班：50人（男32，女18）"
    echo -e "   - 工商管理2024级1班：50人（男20，女30）"
    echo -e "${GREEN}===========================================${NC}"
else
    echo -e "${RED}===========================================${NC}"
    echo -e "${RED}❌ 数据导入失败！${NC}"
    echo -e "${RED}===========================================${NC}"
    echo -e "${YELLOW}💡 可能的原因：${NC}"
    echo -e "   - 数据库连接失败"
    echo -e "   - 数据表不存在"
    echo -e "   - 权限不足"
    echo -e "   - 数据冲突"
    echo -e "${YELLOW}🔧 建议检查：${NC}"
    echo -e "   - 确认数据库连接信息"
    echo -e "   - 确认数据表已创建"
    echo -e "   - 确认用户权限"
    exit 1
fi

echo -e "${BLUE}🎓 学生信息数据导入完成！${NC}"