@echo off
chcp 65001 >nul
cd /d "%~dp0"
set "ROOT=%~dp0.."
set "TOOLS=%ROOT%\tools"
set "NAME=CS55-Q05-Installer"
set "SRC=CS55-Q05_Installer.py"
set "ICON=cs55-q05.ico"
title %NAME% - Cython Build

echo ============================================================
echo   %NAME% - Cython Build
echo ============================================================
echo.

where python >nul 2>&1
if errorlevel 1 (
    echo [ERROR] Python not found
    pause
    exit /b 1
)
for /f "delims=" %%i in ('where python') do set "PY=%%i"
echo Python: %PY%

if not exist "%SRC%" (
    echo [ERROR] Source not found: %SRC%
    pause
    exit /b 1
)
if not exist "%ICON%" (
    echo [ERROR] Icon not found: %ICON%
    pause
    exit /b 1
)
if not exist "%TOOLS%\adb.exe" (
    echo [ERROR] adb.exe not found in %TOOLS%
    pause
    exit /b 1
)
if not exist "%TOOLS%\7za.exe" (
    echo [ERROR] 7za.exe not found in %TOOLS%
    pause
    exit /b 1
)

echo [1/6] Installing deps...
%PY% -m pip install pyinstaller cython pyzipper -q
if errorlevel 1 (
    %PY% -m pip install pyinstaller cython pyzipper -q -i https://pypi.tuna.tsinghua.edu.cn/simple
)
if errorlevel 1 (
    echo [ERROR] Dependency install failed
    pause
    exit /b 1
)

echo [2/6] Clean...
if exist "dist_cy" rmdir /s /q dist_cy 2>nul
if exist "build" rmdir /s /q build 2>nul
if exist "dist" rmdir /s /q dist 2>nul
if exist "%NAME%.spec" del /q "%NAME%.spec" 2>nul

echo [3/6] Cython compile...
mkdir dist_cy 2>nul
copy "%SRC%" dist_cy\_core.py >nul

%PY% -c "open('dist_cy/setup_cython.py','w',encoding='utf-8').write('from setuptools import setup\nfrom Cython.Build import cythonize\nsetup(ext_modules=cythonize(\"_core.py\", compiler_directives={\"language_level\":\"3\"}))\n')"

cd dist_cy
%PY% setup_cython.py build_ext --inplace
if errorlevel 1 (
    cd ..
    echo [ERROR] Cython failed. Build stopped.
    pause
    exit /b 1
)

set "PYD="
for %%f in (_core*.pyd) do set "PYD=%%f"
if "%PYD%"=="" (
    cd ..
    echo [ERROR] No Cython PYD generated. Build stopped.
    pause
    exit /b 1
)
echo PYD: %PYD%
copy "%PYD%" _core.pyd >nul
if errorlevel 1 (
    cd ..
    echo [ERROR] Failed to copy Cython PYD
    pause
    exit /b 1
)

%PY% -c "open('launcher.py','w',encoding='utf-8').write('# -*- coding: utf-8 -*-\nfrom _core import main\nmain()\n')"

echo [4/6] Copy resources...
copy "%TOOLS%\adb.exe" . >nul
copy "%TOOLS%\AdbWinApi.dll" . >nul
copy "%TOOLS%\AdbWinUsbApi.dll" . >nul
copy "%TOOLS%\7za.exe" . >nul
copy "..\%ICON%" . >nul
if errorlevel 1 (
    cd ..
    echo [ERROR] Copy resources failed
    pause
    exit /b 1
)

echo [5/6] PyInstaller...
%PY% -m PyInstaller --onefile --windowed --name="%NAME%" --icon="%ICON%" --add-data "adb.exe;." --add-data "AdbWinApi.dll;." --add-data "AdbWinUsbApi.dll;." --add-data "7za.exe;." --add-data "%ICON%;." --add-binary "_core.pyd;." --hidden-import=queue --hidden-import=threading --hidden-import=tkinter --hidden-import=tkinter.simpledialog --hidden-import=zipfile --hidden-import=json --hidden-import=urllib --hidden-import=urllib.parse --collect-all tkinter --uac-admin launcher.py
if errorlevel 1 (
    cd ..
    echo [ERROR] PyInstaller failed
    pause
    exit /b 1
)

echo [6/6] Cleanup...
del /q _core.py _core.c _core.pyd %PYD% launcher.py setup_cython.py "%ICON%" 2>nul
rmdir /s /q build 2>nul
cd ..

echo.
echo Done.
if exist "dist_cy\dist\%NAME%.exe" (
    echo Output: dist_cy\dist\%NAME%.exe
) else (
    echo [ERROR] Output exe was not generated
    pause
    exit /b 1
)
pause