@echo off
setlocal EnableExtensions DisableDelayedExpansion

:: --- SELF-ELEVATION CODE ---
net session >nul 2>&1
if %errorLevel% == 0 (
    goto :MainMenu
) else (
    echo Requesting Administrator privileges...
    powershell -Command "Start-Process '%~dpnx0' -Verb RunAs"
    exit
)

:MainMenu
    title TurboTax and TurboImpot Fixer
    echo ===================================================
    echo      TURBOTAX (TTD) and TURBOIMPOT (TID) MANAGER
    echo ===================================================
    echo.
    echo [1] Enable Contingency Mode 
    echo [2] Revert to Original Mode 
    echo.
    set /p choice="Enter your choice (1 or 2): "

    if "%choice%"=="1" goto :EnableContingency
    if "%choice%"=="2" goto :RevertContingency
    
    echo.
    echo Invalid choice. Please type 1 or 2.
    pause
    goto :MainMenu

:: =========================================================
:: OPTION 1: ENABLE CONTINGENCY (RENAME)
:: =========================================================
:EnableContingency
    echo.
    echo ---------------------------------------
    echo STARTING CONTINGENCY MODE...
    echo ---------------------------------------
    echo.

    :: --- TURBOTAX (TTD) ---
    call :SafeRename "TurboTax 2020" "C:\Program Files (x86)\TurboTax 2020"
    call :SafeRename "TurboTax 2021" "C:\Program Files (x86)\TurboTax 2021"
    call :SafeRename "TurboTax 2022" "C:\Program Files (x86)\TurboTax 2022"

    :: --- TURBOIMPOT (TID) ---
    call :SafeRename "TurboImpot 2020" "C:\Program Files (x86)\TurboImpot 2020"
    call :SafeRename "TurboImpot 2021" "C:\Program Files (x86)\TurboImpot 2021"
    call :SafeRename "TurboImpot 2022" "C:\Program Files (x86)\TurboImpot 2022"

    goto :EndScript

:: =========================================================
:: OPTION 2: REVERT (RESTORE)
:: =========================================================
:RevertContingency
    echo.
    echo ---------------------------------------
    echo REVERTING TO ORIGINAL MODE...
    echo ---------------------------------------
    echo.

    :: --- TURBOTAX (TTD) ---
    call :SafeRestore "TurboTax 2020" "C:\Program Files (x86)\TurboTax 2020"
    call :SafeRestore "TurboTax 2021" "C:\Program Files (x86)\TurboTax 2021"
    call :SafeRestore "TurboTax 2022" "C:\Program Files (x86)\TurboTax 2022"

    :: --- TURBOIMPOT (TID) ---
    call :SafeRestore "TurboImpot 2020" "C:\Program Files (x86)\TurboImpot 2020"
    call :SafeRestore "TurboImpot 2021" "C:\Program Files (x86)\TurboImpot 2021"
    call :SafeRestore "TurboImpot 2022" "C:\Program Files (x86)\TurboImpot 2022"

    goto :EndScript

:: =========================================================
:: HELPER FUNCTIONS (DO NOT TOUCH BELOW)
:: =========================================================

:SafeRename
    echo Switching %~1...
    if exist "%~2\" (
        pushd "%~2"
        if exist "CefSharp.BrowserSubprocess.exe" (
            ren "CefSharp.BrowserSubprocess.exe" "_CefSharp.BrowserSubprocess.exe"
            echo [SUCCESS]  %~1 in Contingency Mode.
        ) else (
            if exist "_CefSharp.BrowserSubprocess.exe" (
                echo [INFO] %~1 is already in contingency Mode.
            ) else (
                echo [ERROR] File not found in %~1.
            )
        )
        popd
    ) else (
        echo [SKIPPED] %~1 folder not found.
    )
    echo.
    exit /b

:SafeRestore
    echo Switching %~1...
    if exist "%~2\" (
        pushd "%~2"
        if exist "_CefSharp.BrowserSubprocess.exe" (
            ren "_CefSharp.BrowserSubprocess.exe" "CefSharp.BrowserSubprocess.exe"
            echo [SUCCESS] Back to Original mode for %~1.
        ) else (
            if exist "CefSharp.BrowserSubprocess.exe" (
                echo [INFO] %~1 is already in original Mode.
            ) else (
                echo [ERROR] File not found in %~1.
            )
        )
        popd
    ) else (
        echo [SKIPPED] %~1 folder not found.
    )
    echo.
    exit /b

:: =========================================================
:: END OF SCRIPT
:: =========================================================
:EndScript
    echo.
    echo ---------------------------------------------------
    echo Task Complete.
    echo ---------------------------------------------------
    pause