@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 (clang-cl / cl) directly on target unity files
:: ------------------------------------------------------------------------------

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

set FBUILD=misc\fbuild.exe

set TARGET_JULIET=0
set TARGET_ROMEO=0
set TARGET_SHADER=0
set TARGET_COUNT=0

set CONFIG_DEBUG=0
set CONFIG_RELEASE=0
set CONFIG_PROFILE=0
set CONFIG_COUNT=0

:parse_args
if "%~1"=="" goto check_defaults

if /i "%~1"=="debug" (
    set CONFIG_DEBUG=1
    set /a CONFIG_COUNT+=1
    shift
    goto parse_args
)

if /i "%~1"=="release" (
    set CONFIG_RELEASE=1
    set /a CONFIG_COUNT+=1
    shift
    goto parse_args
)

if /i "%~1"=="profile" (
    set CONFIG_PROFILE=1
    set /a CONFIG_COUNT+=1
    shift
    goto parse_args
)

if /i "%~1"=="juliet" (
    set TARGET_JULIET=1
    set /a TARGET_COUNT+=1
    shift
    goto parse_args
)

if /i "%~1"=="romeo" (
    set TARGET_ROMEO=1
    set /a TARGET_COUNT+=1
    shift
    goto parse_args
)

if /i "%~1"=="shadercompiler" (
    set TARGET_SHADER=1
    set /a TARGET_COUNT+=1
    shift
    goto parse_args
)

shift
goto parse_args

:check_defaults
:: If no arguments passed ("build"), build ALL targets across ALL configurations (Debug, Profile, Release)
if %CONFIG_COUNT% equ 0 if %TARGET_COUNT% equ 0 (
    set CONFIG_DEBUG=1
    set CONFIG_RELEASE=1
    set CONFIG_PROFILE=1
    set TARGET_JULIET=1
    set TARGET_ROMEO=1
    set TARGET_SHADER=1
    goto start_build
)

:: If targets specified but no configs specified, default to Debug
if %CONFIG_COUNT% equ 0 (
    set CONFIG_DEBUG=1
)

:: If configs specified but no targets specified, build all targets for those configs
if %TARGET_COUNT% equ 0 (
    set TARGET_JULIET=1
    set TARGET_ROMEO=1
    set TARGET_SHADER=1
)

:start_build
if not exist "%FBUILD%" (
    echo.
    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
)

if %CONFIG_DEBUG% equ 1 call :build_config Debug
if %ERRORLEVEL% neq 0 exit /b %ERRORLEVEL%

if %CONFIG_PROFILE% equ 1 call :build_config Profile
if %ERRORLEVEL% neq 0 exit /b %ERRORLEVEL%

if %CONFIG_RELEASE% equ 1 call :build_config Release
if %ERRORLEVEL% neq 0 exit /b %ERRORLEVEL%

echo.
echo ==============================================================================
echo [ALL BUILDS SUCCEEDED] All requested executables built successfully.
echo ==============================================================================
exit /b 0

:: ------------------------------------------------------------------------------
:: Subroutine to build a specific configuration (Debug / Profile / Release)
:: ------------------------------------------------------------------------------
:build_config
set "CFG=%~1"

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

set FASTBUILD_TARGETS=Juliet-Unity
if %TARGET_JULIET% equ 1 set FASTBUILD_TARGETS=!FASTBUILD_TARGETS! JulietApp-Unity 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

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

:: Strip '#pragma message' from generated Unity files
powershell -Command "Get-ChildItem -Path Intermediate -Recurse -Filter '*_Unity*.cpp' | ForEach-Object { (Get-Content $_.FullName) | Where-Object { $_ -notmatch '#pragma message' } | Set-Content $_.FullName }"

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

if not exist "bin\x64-!CFG!" mkdir "bin\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

if /i "!CFG!"=="Debug" (
    set COMPILER_FLAGS=!COMMON_FLAGS! /MDd /Od /Zi /DDEBUG /DPROFILING_ENABLED /DJULIET_ENABLE_IMGUI /DJULIET_DEBUG
) else if /i "!CFG!"=="Profile" (
    set COMPILER_FLAGS=!COMMON_FLAGS! /MD /O2 /Zi /DRELEASE /DPROFILING_ENABLED
) else (
    set COMPILER_FLAGS=!COMMON_FLAGS! /MD /O2 /DRELEASE
)

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
set IMGUI_SRCS=External\imgui\imgui.cpp External\imgui\imgui_demo.cpp External\imgui\imgui_draw.cpp External\imgui\imgui_tables.cpp External\imgui\imgui_widgets.cpp External\imgui\backends\imgui_impl_win32.cpp External\imgui\backends\imgui_impl_dx12.cpp

:: Compile JulietApp Executable
if %TARGET_JULIET% equ 1 (
    echo.
    echo --- Building JulietApp.exe [%CFG%] ---
    %COMPILER% %COMPILER_FLAGS% /Fe"bin\x64-!CFG!\JulietApp.exe" ^
        Intermediate\Juliet\*.cpp ^
        Intermediate\JulietApp\*.cpp ^
        Intermediate\Game\*.cpp ^
        !IMGUI_SRCS! ^
        /link %COMMON_LIBS% /SUBSYSTEM:CONSOLE
    if %ERRORLEVEL% neq 0 exit /b %ERRORLEVEL%
)

:: Compile Romeo Executable
if %TARGET_ROMEO% equ 1 (
    echo.
    echo --- Building Romeo.exe [%CFG%] ---
    %COMPILER% %COMPILER_FLAGS% /Fe"bin\x64-!CFG!\Romeo.exe" ^
        Intermediate\Juliet\*.cpp ^
        Intermediate\Romeo\*.cpp ^
        !IMGUI_SRCS! ^
        /link %COMMON_LIBS% /SUBSYSTEM:CONSOLE
    if %ERRORLEVEL% neq 0 exit /b %ERRORLEVEL%
)

:: Compile JulietShaderCompiler Executable
if %TARGET_SHADER% equ 1 (
    echo.
    echo --- Building JulietShaderCompiler.exe [%CFG%] ---
    if exist "Intermediate\JulietShaderCompiler\*.cpp" (
        %COMPILER% %COMPILER_FLAGS% /I"JulietShaderCompiler" /Fe"bin\x64-!CFG!\JulietShaderCompiler.exe" ^
            Intermediate\Juliet\*.cpp ^
            Intermediate\JulietShaderCompiler\*.cpp ^
            JulietShaderCompiler\DXShaderCompiler\dxcompiler.lib ^
            /link %COMMON_LIBS% /SUBSYSTEM:CONSOLE
        if %ERRORLEVEL% neq 0 exit /b %ERRORLEVEL%
    )
)

exit /b 0