@echo off
setlocal enabledelayedexpansion

:: upload.bat -- Compile and upload the sketch to the board
::
:: Reads all settings from .anvil.toml. No Anvil binary required.
::
:: Usage:
::   upload.bat                     Auto-detect port, compile + upload
::   upload.bat -p COM3             Specify port
::   upload.bat --board mega        Use a named board
::   upload.bat --monitor           Open serial monitor after upload
::   upload.bat --clean             Clean build cache first
::   upload.bat --verbose           Full compiler + avrdude output

set "SCRIPT_DIR=%~dp0"
set "SCRIPT_DIR=%SCRIPT_DIR:~0,-1%"
set "CONFIG=%SCRIPT_DIR%\.anvil.toml"
set "LOCAL_CONFIG=%SCRIPT_DIR%\.anvil.local"

if not exist "%CONFIG%" (
    echo FAIL: No .anvil.toml found in %SCRIPT_DIR%
    exit /b 1
)

:: -- Parse .anvil.toml (flat keys) ----------------------------------------
for /f "usebackq tokens=1,* delims==" %%a in ("%CONFIG%") do (
    set "_K=%%a"
    if not "!_K:~0,1!"=="#" if not "!_K:~0,1!"=="[" (
        set "_K=!_K: =!"
        set "_V=%%b"
        if defined _V (
            set "_V=!_V: =!"
            set "_V=!_V:"=!"
        )
        if "!_K!"=="name" set "SKETCH_NAME=!_V!"
        if "!_K!"=="default" set "DEFAULT_BOARD=!_V!"
        if "!_K!"=="warnings" set "WARNINGS=!_V!"
        if "!_K!"=="baud" set "BAUD=!_V!"
    )
)

:: -- Parse .anvil.local ---------------------------------------------------
set "LOCAL_PORT="
set "LOCAL_VID_PID="
if exist "%LOCAL_CONFIG%" (
    for /f "usebackq tokens=1,* delims==" %%a in ("%LOCAL_CONFIG%") do (
        set "_K=%%a"
        if not "!_K:~0,1!"=="#" (
            set "_K=!_K: =!"
            set "_V=%%b"
            if defined _V (
                set "_V=!_V: =!"
                set "_V=!_V:"=!"
            )
            if "!_K!"=="port" set "LOCAL_PORT=!_V!"
            if "!_K!"=="vid_pid" set "LOCAL_VID_PID=!_V!"
        )
    )
)

if "%SKETCH_NAME%"=="" (
    echo FAIL: Could not read project name from .anvil.toml
    exit /b 1
)
if "%BAUD%"=="" set "BAUD=115200"

set "SKETCH_DIR=%SCRIPT_DIR%\%SKETCH_NAME%"
set "BUILD_DIR=%SCRIPT_DIR%\.build"

:: -- Parse arguments ------------------------------------------------------
set "PORT="
set "DO_MONITOR=0"
set "DO_CLEAN=0"
set "VERBOSE="
set "BOARD_NAME="

:parse_args
if "%~1"=="" goto done_args
if "%~1"=="-p" set "PORT=%~2" & shift & shift & goto parse_args
if "%~1"=="--port" set "PORT=%~2" & shift & shift & goto parse_args
if "%~1"=="--board" set "BOARD_NAME=%~2" & shift & shift & goto parse_args
if "%~1"=="--monitor" set "DO_MONITOR=1" & shift & goto parse_args
if "%~1"=="--clean" set "DO_CLEAN=1" & shift & goto parse_args
if "%~1"=="--verbose" set "VERBOSE=--verbose" & shift & goto parse_args
if "%~1"=="--help" goto show_help
if "%~1"=="-h" goto show_help
echo FAIL: Unknown option: %~1
exit /b 1

:show_help
echo Usage: upload.bat [-p PORT] [--board NAME] [--monitor] [--clean] [--verbose]
echo   Compiles and uploads the sketch. Settings from .anvil.toml.
echo   --board NAME selects a board from [boards.NAME].
exit /b 0

:done_args

:: -- Resolve board --------------------------------------------------------
if "%BOARD_NAME%"=="" set "BOARD_NAME=%DEFAULT_BOARD%"

set "BOARD_SECTION=[boards.%BOARD_NAME%]"
set "IN_SECTION=0"
set "FQBN="
set "BOARD_BAUD="
for /f "usebackq tokens=*" %%L in ("%CONFIG%") do (
    set "_LINE=%%L"
    if "!_LINE!"=="!BOARD_SECTION!" (
        set "IN_SECTION=1"
    ) else if "!IN_SECTION!"=="1" (
        if "!_LINE:~0,1!"=="[" (
            set "IN_SECTION=0"
        ) else if not "!_LINE:~0,1!"=="#" (
            for /f "tokens=1,* delims==" %%a in ("!_LINE!") do (
                set "_BK=%%a"
                set "_BK=!_BK: =!"
                set "_BV=%%b"
                if defined _BV (
                    set "_BV=!_BV: =!"
                    set "_BV=!_BV:"=!"
                )
                if "!_BK!"=="fqbn" set "FQBN=!_BV!"
                if "!_BK!"=="baud" set "BOARD_BAUD=!_BV!"
            )
        )
    )
)

if "!FQBN!"=="" (
    echo FAIL: No board '%BOARD_NAME%' in .anvil.toml.
    echo   Add it: anvil board --add %BOARD_NAME%
    exit /b 1
)
if not "!BOARD_BAUD!"=="" set "BAUD=!BOARD_BAUD!"

if not "%BOARD_NAME%"=="%DEFAULT_BOARD%" (
    echo ok   Using board: %BOARD_NAME% -- !FQBN!
)

:: -- Preflight ------------------------------------------------------------
where arduino-cli >nul 2>nul
if errorlevel 1 (
    echo FAIL: arduino-cli not found in PATH.
    exit /b 1
)

:: -- Resolve port ---------------------------------------------------------
:: Priority: -p flag > VID:PID resolve > saved port > auto-detect
if "%PORT%"=="" (
    for /f "delims=" %%p in ('powershell -NoProfile -ExecutionPolicy Bypass -File "%SCRIPT_DIR%\_detect_port.ps1" -VidPid "%LOCAL_VID_PID%" -SavedPort "%LOCAL_PORT%"') do (
        if "!PORT!"=="" set "PORT=%%p"
    )
    if "!PORT!"=="" (
        echo FAIL: No serial port detected. Is the board plugged in?
        echo   Specify manually: upload.bat -p COM3
        echo   Or save a default: anvil devices --set
        exit /b 1
    )
    if not "%LOCAL_VID_PID%"=="" (
        if not "!PORT!"=="%LOCAL_PORT%" (
            echo info  Device %LOCAL_VID_PID% found on !PORT! ^(moved from %LOCAL_PORT%^)
        ) else (
            echo info  Using port !PORT! ^(from .anvil.local^)
        )
    ) else if not "%LOCAL_PORT%"=="" (
        echo info  Using port !PORT! ^(from .anvil.local^)
    ) else (
        echo warn Auto-detected port: !PORT! ^(use -p to override, or: anvil devices --set^)
    )
)

:: -- Clean ----------------------------------------------------------------
if "%DO_CLEAN%"=="1" (
    if exist "%BUILD_DIR%" rmdir /s /q "%BUILD_DIR%"
)

:: -- Build include flags --------------------------------------------------
set "BUILD_FLAGS="
for %%d in (lib\hal lib\app) do (
    if exist "%SCRIPT_DIR%\%%d" (
        set "BUILD_FLAGS=!BUILD_FLAGS! -I%SCRIPT_DIR%\%%d"
    )
)
set "BUILD_FLAGS=!BUILD_FLAGS! -Werror"

:: -- Compile --------------------------------------------------------------
echo Compiling %SKETCH_NAME%...
if not exist "%BUILD_DIR%" mkdir "%BUILD_DIR%"

arduino-cli compile --fqbn %FQBN% --build-path "%BUILD_DIR%" --warnings %WARNINGS% --build-property "compiler.cpp.extra_flags=%BUILD_FLAGS%" --build-property "compiler.c.extra_flags=%BUILD_FLAGS%" %VERBOSE% "%SKETCH_DIR%"
if errorlevel 1 (
    echo FAIL: Compilation failed.
    exit /b 1
)
echo ok   Compile succeeded.

:: -- Upload ---------------------------------------------------------------
echo.
echo Uploading to %PORT%...

arduino-cli upload --fqbn %FQBN% --port %PORT% --input-dir "%BUILD_DIR%" %VERBOSE%
if errorlevel 1 (
    echo FAIL: Upload failed.
    exit /b 1
)
echo ok   Upload complete!

:: -- Monitor --------------------------------------------------------------
if "%DO_MONITOR%"=="1" (
    echo.
    echo Opening serial monitor on %PORT% at %BAUD% baud...
    echo Press Ctrl+C to exit.
    echo.
    arduino-cli monitor -p %PORT% -c "baudrate=%BAUD%"
)