@echo off
setlocal enabledelayedexpansion

:: ------------------------------------------------------------------------------
:: Unity Build Script (misc/build.bat)
:: Step 1: Use FASTBuild to generate unity .cpp files directly in Intermediate/
:: Step 2: Invoke compiler directly on target unity files
:: ------------------------------------------------------------------------------

set SCRIPT_DIR=%~dp0
cd /d "%SCRIPT_DIR%.."

set FBUILD=misc\fbuild.exe
set CHECK_TOOL=misc\build_check.exe
set CHECK_SRC=tools\build_check.cpp

set TARGET_JULIET=0
set TARGET_ROMEO=0
set TARGET_SHADER=0
set TARGET_GAME=0
set TARGET_SPECIFIED=0

set CONFIG_DEBUG=0
set CONFIG_RELEASE=0
set CONFIG_PROFILE=0
set CONFIG_SPECIFIED=0

:parse_args
if "%~1"=="" goto check_defaults
if /i "%~1"=="debug" set CONFIG_DEBUG=1& set CONFIG_SPECIFIED=1
if /i "%~1"=="release" set CONFIG_RELEASE=1& set CONFIG_SPECIFIED=1
if /i "%~1"=="profile" set CONFIG_PROFILE=1& set CONFIG_SPECIFIED=1
if /i "%~1"=="juliet" set TARGET_JULIET=1& set TARGET_SPECIFIED=1
if /i "%~1"=="game" set TARGET_GAME=1& set TARGET_SPECIFIED=1
if /i "%~1"=="romeo" set TARGET_ROMEO=1& set TARGET_SPECIFIED=1
if /i "%~1"=="shadercompiler" set TARGET_SHADER=1& set TARGET_SPECIFIED=1
shift
goto parse_args

:check_defaults
if !CONFIG_SPECIFIED! equ 0 set CONFIG_DEBUG=1& if !TARGET_SPECIFIED! equ 0 set CONFIG_RELEASE=1& set CONFIG_PROFILE=1
if !TARGET_SPECIFIED! equ 0 set TARGET_JULIET=1& set TARGET_GAME=1& set TARGET_ROMEO=1& set TARGET_SHADER=1

:start_build
if exist "Intermediate\built_outputs.tmp" del /f /q "Intermediate\built_outputs.tmp" >nul 2>&1

if not exist "%FBUILD%" (
    echo [BUILD FAILED] Could not find %FBUILD%.
    exit /b 1
)

:: Find Clang / MSVC compiler
set COMPILER=clang-cl.exe
where %COMPILER% >nul 2>nul
if !ERRORLEVEL! neq 0 (
    set COMPILER=cl.exe
    where %COMPILER% >nul 2>nul
    if !ERRORLEVEL! neq 0 (
        echo [BUILD FAILED] Neither clang-cl.exe nor cl.exe was found in PATH.
        exit /b 1
    )
)

:: Check if build_check.exe needs to be compiled (missing, version changed, or source changed)
set CHECK_TOOL_VER=3
set NEED_RECOMPILE_CHECK_TOOL=0
if not exist "%CHECK_TOOL%" (
    set NEED_RECOMPILE_CHECK_TOOL=1
) else (
    set "CURR_VER="
    if exist "Intermediate\build_check.ver" (
        for /f "usebackq delims=" %%I in ("Intermediate\build_check.ver") do set "CURR_VER=%%I"
    )
    if not "!CURR_VER!"=="!CHECK_TOOL_VER!" set NEED_RECOMPILE_CHECK_TOOL=1
    set "SRC_TIME="
    for %%I in ("%CHECK_SRC%") do set "SRC_TIME=%%~tI"
    set "STAMP_TIME="
    if exist "Intermediate\build_check.stamp" (
        for /f "usebackq delims=" %%I in ("Intermediate\build_check.stamp") do set "STAMP_TIME=%%I"
    )
    if not "!SRC_TIME!"=="!STAMP_TIME!" set NEED_RECOMPILE_CHECK_TOOL=1
)

if !NEED_RECOMPILE_CHECK_TOOL! equ 1 (
    if exist "Intermediate\build_check.stamp" del /f /q "Intermediate\build_check.stamp"
    if exist "Intermediate\build_check.ver" del /f /q "Intermediate\build_check.ver"
    if exist "%CHECK_TOOL%" (
        del /f /q "%CHECK_TOOL%" >nul 2>&1
        if exist "%CHECK_TOOL%" (
            echo [BUILD FAILED] Could not overwrite %CHECK_TOOL%. It may be in use by another process.
            exit /b 1
        )
    )
    echo [BUILD] Compiling fast native helper tool [tools\build_check.cpp -^> misc\build_check.exe]...
    %COMPILER% /nologo /std:c++20 /MT /O2 /EHa- /W4 "%CHECK_SRC%" /Fe"%CHECK_TOOL%"
    if !ERRORLEVEL! neq 0 (
        echo [BUILD FAILED] Could not compile build helper tool.
        exit /b 1
    )
    if not exist "%CHECK_TOOL%" (
        echo [BUILD FAILED] Compilation failed to produce %CHECK_TOOL%.
        exit /b 1
    )
    "%CHECK_TOOL%" check_alive >nul 2>&1
    if !ERRORLEVEL! neq 0 (
        echo [BUILD FAILED] %CHECK_TOOL% was built but failed to execute ^(missing DLL or AV block^).
        exit /b 1
    )
    if not exist "Intermediate" mkdir "Intermediate"
    (echo !CHECK_TOOL_VER!)>"Intermediate\build_check.ver"
    for %%I in ("%CHECK_SRC%") do (echo %%~tI)>"Intermediate\build_check.stamp"
)

"%CHECK_TOOL%" start_build
if !ERRORLEVEL! neq 0 (
    echo [BUILD FAILED] %CHECK_TOOL% start_build failed with exit code !ERRORLEVEL!.
    exit /b 1
)

if !CONFIG_DEBUG! equ 1 call :build_config Debug || exit /b !ERRORLEVEL!
if !CONFIG_PROFILE! equ 1 call :build_config Profile || exit /b !ERRORLEVEL!
if !CONFIG_RELEASE! equ 1 call :build_config Release || exit /b !ERRORLEVEL!

echo.
echo ==============================================================================
echo [BUILD SUMMARY]
echo ==============================================================================
echo Compiler Used : %COMPILER%
"%CHECK_TOOL%" finish_build
echo ==============================================================================
echo [ALL BUILDS SUCCEEDED] All requested targets processed successfully.
echo ==============================================================================
exit /b 0

:: ------------------------------------------------------------------------------
:: Build Configuration Subroutine
:: ------------------------------------------------------------------------------
:build_config
set "CFG=%~1"

echo.
echo ==============================================================================
echo Generating Unity Files into Intermediate/ via FASTBuild [%CFG%]...
echo ==============================================================================

set FASTBUILD_TARGETS=Juliet-Unity
if /i not "!CFG!"=="Release" set FASTBUILD_TARGETS=!FASTBUILD_TARGETS! ImGui-Unity
if !TARGET_JULIET! equ 1 set FASTBUILD_TARGETS=!FASTBUILD_TARGETS! JulietApp-Unity Game-Unity
if !TARGET_GAME! equ 1 set FASTBUILD_TARGETS=!FASTBUILD_TARGETS! Game-Unity
if !TARGET_ROMEO! equ 1 set FASTBUILD_TARGETS=!FASTBUILD_TARGETS! Romeo-Unity
if !TARGET_SHADER! equ 1 set FASTBUILD_TARGETS=!FASTBUILD_TARGETS! JulietShaderCompiler-Unity

"%CHECK_TOOL%" start_step "FASTBuild Unity Gen [!CFG!]"
"%FBUILD%" !FASTBUILD_TARGETS!
set "FBUILD_ERR=!ERRORLEVEL!"
"%CHECK_TOOL%" end_step
if !FBUILD_ERR! neq 0 (
    echo [BUILD FAILED] FASTBuild failed to generate unity files for %CFG%.
    exit /b !FBUILD_ERR!
)

:: Evaluate all target needs in a single fast native pass (<5ms)
"%CHECK_TOOL%" start_step "Config Evaluation [!CFG!]"
"%CHECK_TOOL%" evaluate "!CFG!" "!TARGET_JULIET!" "!TARGET_GAME!" "!TARGET_ROMEO!" "!TARGET_SHADER!"
if !ERRORLEVEL! neq 0 (
    echo [BUILD FAILED] %CHECK_TOOL% evaluate failed with exit code !ERRORLEVEL!.
    exit /b 1
)
"%CHECK_TOOL%" end_step

set FASTBUILD_UNITY_CHANGED=0
set JULIET_NEED_COMPILE=0
set IMGUI_NEED_COMPILE=0
set GAME_NEED_COMPILE=0
set JULIETAPP_NEED_COMPILE=0
set ROMEO_NEED_COMPILE=0
set SHADER_NEED_COMPILE=0

if exist "Intermediate\config_status.tmp" (
    for /f "usebackq tokens=1,2 delims==" %%A in ("Intermediate\config_status.tmp") do set %%A=%%B
    del /f /q "Intermediate\config_status.tmp"
)

if !FASTBUILD_UNITY_CHANGED! equ 1 (
    echo [FASTBuild] Unity structure or source file list changed.
) else (
    echo [FASTBuild] No Unity file structure changes detected [up-to-date].
)

set ANY_NEED_BUILD=0
for %%T in (JULIET IMGUI GAME JULIETAPP ROMEO SHADER) do (
    if !%%T_NEED_COMPILE! equ 1 set ANY_NEED_BUILD=1
)

if !ANY_NEED_BUILD! equ 0 (
    echo [SKIP / UP-TO-DATE] All requested targets in [%CFG%] are up-to-date.
    exit /b 0
)

echo.
echo ==============================================================================
echo Compiling Unity Files directly via %COMPILER% [%CFG%]...
echo ==============================================================================

if not exist "bin\x64-!CFG!" mkdir "bin\x64-!CFG!"
if not exist "Intermediate\x64-!CFG!" mkdir "Intermediate\x64-!CFG!"

set COMMON_FLAGS=/nologo /std:c++20 /W4 /EHa- /utf-8 /DUNICODE /D_UNICODE /DWIN32_LEAN_AND_MEAN /D_CRT_SECURE_NO_WARNINGS /I"Juliet/include" /I"Juliet/src" /I"Game" /I"External/imgui" /I"External/imgui/backends" /DJULIET_WIN32 /wd5267 /wd4061 /wd4505 /wd4514 /wd4577 /wd4625 /wd4710 /wd4711 /wd4746 /wd4820 /wd5045 /wd5220 /wd5245 /wd4626 /wd5026 /wd5027 /wd4530 /Fd"Intermediate\x64-!CFG!\vc140.pdb"

if /i "!CFG!"=="Debug" (
    set COMPILER_FLAGS=!COMMON_FLAGS! /MDd /Od /Zi /Gw /Gy /Zc:inline /DDEBUG /DPROFILING_ENABLED /DJULIET_ENABLE_IMGUI /DJULIET_DEBUG
    set IMGUI_UNITY_SRCS=Intermediate\External\imgui\*.cpp
) else if /i "!CFG!"=="Profile" (
    set COMPILER_FLAGS=!COMMON_FLAGS! /MD /O2 /Zi /Gw /Gy /Zc:inline /DRELEASE /DPROFILING_ENABLED /DJULIET_ENABLE_IMGUI
    set IMGUI_UNITY_SRCS=Intermediate\External\imgui\*.cpp
) else (
    set COMPILER_FLAGS=!COMMON_FLAGS! /MD /O2 /Gw /Gy /Zc:inline /DRELEASE
    set IMGUI_UNITY_SRCS=
)

set COMMON_LIBS=kernel32.lib user32.lib gdi32.lib dxguid.lib Ws2_32.lib dxgi.lib imm32.lib dwmapi.lib d3dcompiler.lib shell32.lib winhttp.lib delayimp.lib

:: --- Juliet Shared Objs ---
echo.
echo --- Compiling Shared Object Files (Juliet / ImGui) [%CFG%] ---
if !JULIET_NEED_COMPILE! equ 0 (
    echo [SKIP / UP-TO-DATE] Intermediate\x64-!CFG!\Juliet_Unity*.obj [Unity files unchanged]
) else (
    echo [REBUILDING] Compiling Intermediate\x64-!CFG!\Juliet_Unity*.obj...
    "%CHECK_TOOL%" start_step "Compiling Juliet Shared Objs [!CFG!]"
    %COMPILER% %COMPILER_FLAGS% /DJULIET_EXPORT /c Intermediate\Juliet\*.cpp /Fo"Intermediate\x64-!CFG!\\"
    set "COMP_ERR=!ERRORLEVEL!"
    "%CHECK_TOOL%" end_step
    if !COMP_ERR! neq 0 exit /b !COMP_ERR!
)

:: --- ImGui Shared Objs ---
if defined IMGUI_UNITY_SRCS (
    if !IMGUI_NEED_COMPILE! equ 0 (
        echo [SKIP / UP-TO-DATE] Intermediate\x64-!CFG!\ImGui_Unity*.obj [Unity files unchanged]
    ) else (
        echo [REBUILDING] Compiling Intermediate\x64-!CFG!\ImGui_Unity*.obj...
        "%CHECK_TOOL%" start_step "Compiling ImGui Shared Objs [!CFG!]"
        %COMPILER% %COMPILER_FLAGS% /c !IMGUI_UNITY_SRCS! /Fo"Intermediate\x64-!CFG!\\"
        set "COMP_ERR=!ERRORLEVEL!"
        "%CHECK_TOOL%" end_step
        if !COMP_ERR! neq 0 exit /b !COMP_ERR!
    )
)

set JULIET_OBJS=Intermediate\x64-!CFG!\Juliet_Unity*.obj
set IMGUI_OBJS=
if defined IMGUI_UNITY_SRCS set IMGUI_OBJS=Intermediate\x64-!CFG!\ImGui_Unity*.obj

set BUILD_GAME_DLL=0
if !TARGET_JULIET! equ 1 set BUILD_GAME_DLL=1
if !TARGET_GAME! equ 1 set BUILD_GAME_DLL=1

:: --- Juliet.dll ---
if !TARGET_JULIET! equ 1 (
    call :link_target "Juliet.dll" "Building Juliet.dll [!CFG!]" "!JULIET_NEED_COMPILE!" "%COMPILER_FLAGS% /DJULIET_EXPORT /LD" "%JULIET_OBJS% %IMGUI_OBJS%" "/link %COMMON_LIBS% /INCREMENTAL /ILK:Intermediate\x64-!CFG!\Juliet.ilk /PDB:Intermediate\x64-!CFG!\Juliet.pdb"
    if !ERRORLEVEL! neq 0 exit /b !ERRORLEVEL!
)

:: --- Game.dll ---
if !BUILD_GAME_DLL! equ 1 (
    call :link_target "Game.dll" "Building Game.dll [!CFG!]" "!GAME_NEED_COMPILE!" "%COMPILER_FLAGS% /DJULIET_EXPORT /LD" "Intermediate\Game\*.cpp %JULIET_OBJS% %IMGUI_OBJS%" "/link %COMMON_LIBS% bin\x64-!CFG!\Juliet.lib /INCREMENTAL /ILK:Intermediate\x64-!CFG!\Game.ilk /PDB:Intermediate\x64-!CFG!\Game.pdb"
    if !ERRORLEVEL! neq 0 exit /b !ERRORLEVEL!
)

:: --- JulietApp.exe ---
if !TARGET_JULIET! equ 1 (
    call :link_target "JulietApp.exe" "Building JulietApp.exe [!CFG!]" "!JULIETAPP_NEED_COMPILE!" "%COMPILER_FLAGS%" "Intermediate\JulietApp\*.cpp %JULIET_OBJS% %IMGUI_OBJS%" "/link %COMMON_LIBS% bin\x64-!CFG!\Juliet.lib /SUBSYSTEM:CONSOLE /INCREMENTAL /ILK:Intermediate\x64-!CFG!\JulietApp.ilk /PDB:Intermediate\x64-!CFG!\JulietApp.pdb"
    if !ERRORLEVEL! neq 0 exit /b !ERRORLEVEL!
)

:: --- Romeo.exe ---
if !TARGET_ROMEO! equ 1 (
    call :link_target "Romeo.exe" "Building Romeo.exe [!CFG!]" "!ROMEO_NEED_COMPILE!" "%COMPILER_FLAGS%" "%JULIET_OBJS% Intermediate\Romeo\*.cpp" "/link %COMMON_LIBS% /SUBSYSTEM:CONSOLE /INCREMENTAL /ILK:\"Intermediate\x64-!CFG!\Romeo.ilk\" /PDB:\"Intermediate\x64-!CFG!\Romeo.pdb\""
    if !ERRORLEVEL! neq 0 exit /b !ERRORLEVEL!
)

:: --- JulietShaderCompiler.exe ---
if !TARGET_SHADER! equ 1 (
    if exist "Intermediate\JulietShaderCompiler\*.cpp" (
        call :link_target "JulietShaderCompiler.exe" "Building JulietShaderCompiler.exe [!CFG!]" "!SHADER_NEED_COMPILE!" "%COMPILER_FLAGS% /I\"JulietShaderCompiler\"" "%JULIET_OBJS% Intermediate\JulietShaderCompiler\*.cpp" "/link %COMMON_LIBS% dxcompiler.lib /DELAYLOAD:dxcompiler.dll /SUBSYSTEM:CONSOLE /INCREMENTAL /ILK:\"Intermediate\x64-!CFG!\JulietShaderCompiler.ilk\" /PDB:\"Intermediate\x64-!CFG!\JulietShaderCompiler.pdb\""
        if !ERRORLEVEL! neq 0 exit /b !ERRORLEVEL!
    )
)

exit /b 0

:: ------------------------------------------------------------------------------
:: Helper Subroutine: Link / Build Target Binary
:: ------------------------------------------------------------------------------
:link_target
set "TARGET_NAME=%~1"
set "STEP_TITLE=%~2"
set "NEED_COMPILE=%~3"
set "EXTRA_FLAGS=%~4"
set "SOURCES_AND_OBJS=%~5"
set "LINK_FLAGS=%~6"

echo.
echo --- Building %TARGET_NAME% [%CFG%] ---
if "%NEED_COMPILE%"=="0" (
    echo [SKIP / UP-TO-DATE] bin\x64-!CFG!\%TARGET_NAME% [Binary up to date, skipping compilation]
    exit /b 0
)

echo [REBUILDING] Compiling bin\x64-!CFG!\%TARGET_NAME%...
"%CHECK_TOOL%" start_step "%STEP_TITLE%"
%COMPILER% %EXTRA_FLAGS% /Fe"bin\x64-!CFG!\%TARGET_NAME%" %SOURCES_AND_OBJS% %LINK_FLAGS%
set "COMP_ERR=!ERRORLEVEL!"
"%CHECK_TOOL%" end_step
if !COMP_ERR! neq 0 exit /b !COMP_ERR!

echo bin\x64-!CFG!\%TARGET_NAME%>>Intermediate\built_outputs.tmp
exit /b 0