@echo off
chcp 65001 >nul
cd /d "%~dp0"

:menu
cls
echo ================================================================================
echo    万花筒API性能监控系统
echo ================================================================================
echo.
echo    [1] 启动API服务器 (后台运行)
echo    [2] 停止API服务器
echo    [3] 查看性能统计
echo    [4] 测试登录功能
echo    [5] 重置密码
echo    [6] 测试文章接口
echo    [0] 退出
echo.
echo ================================================================================
set /p choice=请选择操作 (0-6): 

if "%choice%"=="1" goto start_server
if "%choice%"=="2" goto stop_server
if "%choice%"=="3" goto show_stats
if "%choice%"=="4" goto test_login
if "%choice%"=="5" goto reset_pwd
if "%choice%"=="6" goto test_article
if "%choice%"=="0" goto end
goto menu

:start_server
echo.
echo [步骤1/2] 停止现有服务器...
taskkill /F /FI "IMAGENAME eq python.exe" /FI "WINDOWTITLE eq Flask API Server*" 2>nul
timeout /t 2 /nobreak >nul

echo [步骤2/2] 启动新服务器...
start "Flask API Server" /MIN python flask_wht_server_api.py
timeout /t 3 /nobreak >nul

echo.
echo ✓ 服务器已启动在后台
echo   访问地址: http://127.0.0.1:8216
echo.
pause
goto menu

:stop_server
echo.
echo 正在停止服务器...
taskkill /F /FI "IMAGENAME eq python.exe" /FI "WINDOWTITLE eq Flask API Server*" 2>nul
echo.
echo ✓ 服务器已停止
echo.
pause
goto menu

:show_stats
echo.
python 性能统计查询.py
echo.
pause
goto menu

:test_login
echo.
echo 正在测试登录...
python test_login.py
echo.
pause
goto menu

:reset_pwd
echo.
python reset_password.py
echo.
pause
goto menu

:test_article
echo.
echo 正在测试文章接口...
python 测试文章接口性能.py
echo.
pause
goto menu

:end
echo.
echo 再见!
timeout /t 1 /nobreak >nul
exit