@echo off
title FastRDP-NG v2.0 - Installer
cd /d "%~dp0"
echo.
echo  ╔══════════════════════════════════════════╗
echo  ║     FastRDP-NG v2.0 Installer            ║
echo  ╚══════════════════════════════════════════╝
echo.

:: Check Python
python --version >nul 2>&1
if %errorlevel% neq 0 (
    echo  [FAIL] Python is not installed or not in PATH!
    echo.
    echo  Download Python 3.8+ from: https://python.org/downloads
    echo  Make sure to check "Add Python to PATH" during installation.
    echo.
    pause
    exit /b 1
)

for /f "tokens=2" %%i in ('python --version 2^>^&1') do set pyver=%%i
echo  [OK] Python %pyver% found

:: Create directories
if not exist "wordlists" mkdir wordlists
if not exist "results" mkdir results
echo  [OK] Directories created

:: Verify core files exist
set FILES=main.py gui.py scanner.py bruteforce.py ip_utils.py
for %%f in (%FILES%) do (
    if exist "%%f" (
        echo  [OK] %%f
    ) else (
        echo  [WARN] %%f not found
    )
)

:: Create initial good.txt
if not exist "results\good.txt" (
    echo. > "results\good.txt"
    echo  [OK] Created results\good.txt
)

:: Test import
echo.
echo  Testing imports...
python -c "from scanner import scan_ips, RDP_PORTS; from ip_utils import parse_ranges_file; from bruteforce import spray_all_hosts, TOP50_PASSWORDS; print('  [OK] All modules loaded successfully')" 2>&1
if %errorlevel% neq 0 (
    echo  [FAIL] Import test failed. There may be a syntax error.
    pause
    exit /b 1
)

echo.
echo  ╔══════════════════════════════════════════╗
echo  ║     Installation Complete!               ║
echo  ║                                          ║
echo  ║  Run:  run.bat  or  python main.py       ║
echo  ╚══════════════════════════════════════════╝
echo.
pause