#!/bin/bash

# 首次登录强制修改密码功能安装脚本

echo "=========================================="
echo "首次登录强制修改密码功能安装"
echo "=========================================="
echo ""

# 获取数据库配置
read -p "请输入数据库主机 (默认: localhost): " DB_HOST
DB_HOST=${DB_HOST:-localhost}

read -p "请输入数据库端口 (默认: 3306): " DB_PORT
DB_PORT=${DB_PORT:-3306}

read -p "请输入数据库名称: " DB_NAME
if [ -z "$DB_NAME" ]; then
    echo "错误: 数据库名称不能为空"
    exit 1
fi

read -p "请输入数据库用户名: " DB_USER
if [ -z "$DB_USER" ]; then
    echo "错误: 数据库用户名不能为空"
    exit 1
fi

read -sp "请输入数据库密码: " DB_PASS
echo ""

# 执行 SQL
echo ""
echo "正在添加 is_paw 字段..."

mysql -h"$DB_HOST" -P"$DB_PORT" -u"$DB_USER" -p"$DB_PASS" "$DB_NAME" < add_is_paw_field.sql

if [ $? -eq 0 ]; then
    echo "✓ 数据库字段添加成功"
    echo ""
    echo "=========================================="
    echo "安装完成！"
    echo "=========================================="
    echo ""
    echo "功能说明："
    echo "1. 当管理员的 is_paw 字段为 0 时，登录后会强制跳转到修改密码页面"
    echo "2. 修改密码成功后，is_paw 字段会自动更新为 1"
    echo "3. 如需设置某个管理员首次登录修改密码，请执行："
    echo "   UPDATE la_admin SET is_paw = 0 WHERE id = <管理员ID>;"
    echo ""
else
    echo "✗ 数据库字段添加失败"
    echo "请检查数据库连接信息是否正确"
    exit 1
fi