@echo off
setlocal enabledelayedexpansion
title Sigmod WSL installer
rem Escape character for color codes
for /F %%a in ('echo prompt $E ^| cmd') do set "ESC=%%a"

set "params=%*"
cd /d "%~dp0" && ( if exist "%temp%\getadmin.vbs" del "%temp%\getadmin.vbs" ) && fsutil dirty query %systemdrive% 1>nul 2>nul || (  echo Set UAC = CreateObject("Shell.Application"^) : UAC.ShellExecute "cmd.exe", "/k cd ""%~sdp0"" && %~s0 %params%", "", "runas", 1 >> "%temp%\getadmin.vbs" && "%temp%\getadmin.vbs" && exit /B )
powershell -Command Get-ComputerInfo -property "HyperVisorPresent" | find /i "True"
cls

if not errorlevel 1 (
    echo !ESC![93m
    echo ----------------------------------------
    echo Checking for hardware virtualization...
    echo ----------------------------------------
    echo !ESC![0m
) else (
	powershell -Command Get-ComputerInfo -property "HyperVRequirementVirtualizationFirmwareEnabled" | find /i "True"
	if errorlevel 1 (
        cls
        echo !ESC![91m
        echo ----------------------------------------
        echo HARDWARE VIRTUALIZATION IS DISABLED!
        echo Enable it in BIOS settings and try again
        echo ----------------------------------------
        echo !ESC![0m
		pause
		exit
	)
)
rem apparently you must install wsl to the default windows drive???
rem there are some powershell hacks to manually get the distro
rem :setubuntudrive
rem choice /C YN /M "Install linux to your current drive? No will install to the default location"
rem if errorlevel 1 (
rem     echo "[wsl2]" > %USERPROFILE%\.wslconfig
rem     echo memory=2GB >> %USERPROFILE%\.wslconfig
rem     echo "[automount]" >> %USERPROFILE%\.wslconfig
rem     echo root = "%CD:~0,3%\\" >> %USERPROFILE%\.wslconfig
rem )
rem exit /b

set "current_path=%~dp0"
echo %current_path% | findstr /i "onedrive" > nul
if not errorlevel 1 (
    cls
    echo !ESC![91m
    echo -----------------------------------------------------
    echo WARNING: RUNNING FROM ONEDRIVE!
    echo.
    echo Please move the files to a local drive and try again.
    echo -----------------------------------------------------
    echo !ESC![0m
    
    choice /C YN /M "Continue anyway? This may cause issues with the installation"
    if errorlevel 2 exit
    echo.
)
if exist TF2Server (

    wsl --status | find /i "running"
    if errorlevel 1 (

        wsl test -e /var/tf2server/tf/gameinfo.txt
        if errorlevel 0 (

            echo !ESC![96m
            choice /C YN /M "Server files already exist. Do you want to fully reinstall the server?"
            if errorlevel 2 (
                rem continue
            ) else (
                echo !ESC![92m
                echo Reinstalling server...
                wsl rm -rf /var/tf2server
                wsl rm -rf /var/steamcmd
                rmdir /q /s TF2Server
                echo !ESC![0m
            )
        )
    )
)

mkdir TF2Server > nul 2>&1
cd TF2Server

rem create batch scripts
(
    echo @echo off
    echo cd /d "%%~dp0"
    echo wsl --set-default-version 2

    echo echo !ESC![92m
    echo echo -----------------
    echo echo Installing ubuntu
    echo echo -----------------
    echo echo !ESC![0m
    echo ubuntu install --root

    echo echo !ESC![92m
    echo echo -------------------
    echo echo Installing packages
    echo echo -------------------
    echo echo !ESC![0m
    rem WSL likes to set its own DNS servers that don't always work
    echo wsl -d ubuntu -u root bash -c "echo nameserver 8.8.8.8 > /etc/resolv.conf"
    echo wsl -d ubuntu -u root bash -c "echo generateResolveConf=false >> /etc/wsl.conf"
    echo wsl -d ubuntu -u root dpkg --add-architecture i386
    rem for libncurses5
    echo wsl -d ubuntu -u root bash -c "mkdir -p /etc/apt/sources.list.d"
    echo wsl -d ubuntu -u root bash -c "echo 'deb http://security.ubuntu.com/ubuntu focal-security main universe' > /etc/apt/sources.list.d/ubuntu-focal-sources.list"
    echo wsl -d ubuntu -u root apt update
    echo wsl -d ubuntu -u root apt install -y python3-venv gcc g++-multilib lbzip2 unzip wget cpufrequtils lib32z1 libncurses5 lib32tinfo6 libbz2-1.0 libcurl4-gnutls-dev lib32stdc++6 libstdc++6:i386 libncurses5:i386 libbz2-1.0:i386 lib32gcc-s1 libc6-i386 libstdc++6 
    rem do we need 32-bit libcurl for anything? this conflicts with the 64-bit version
    echo wsl -d ubuntu -u root apt install -y libcurl4-gnutls-dev:i386
    echo wsl -d ubuntu -u root ufw disable

    echo echo !ESC![92m
    echo echo ----------------------
    echo echo Installing game server
    echo echo ----------------------
    echo echo !ESC![0m
    echo wsl -d ubuntu -u root mkdir -p /var/tf2server
    echo wsl -d ubuntu -u root mkdir -p /var/steamcmd
    echo wsl -d ubuntu -u root useradd -m gameserver
    echo wsl -d ubuntu -u root chown -R gameserver:gameserver /var/tf2server
    echo wsl -d ubuntu -u root chown -R gameserver:gameserver /var/steamcmd
    echo echo !ESC![35m
    echo wsl -d ubuntu -u gameserver cd /var/steamcmd; wget "https://steamcdn-a.akamaihd.net/client/installer/steamcmd_linux.tar.gz"; tar zxf steamcmd_linux.tar.gz; ./steamcmd.sh +force_install_dir ../tf2server +login anonymous +app_update 232250 +quit

    echo echo !ESC![36m
    echo echo ----------------------------
    echo echo Installing Metamod/Sourcemod
    echo echo ----------------------------
    echo wsl -d ubuntu -u gameserver cd /var/tf2server; curl https://mms.alliedmods.net/mmsdrop/1.12/`curl https://mms.alliedmods.net/mmsdrop/1.12/mmsource-latest-linux` -o metamod.tar.gz; tar -xf metamod.tar.gz -C ../tf2server/tf;
    echo wsl -d ubuntu -u gameserver cd /var/tf2server; curl https://sm.alliedmods.net/smdrop/1.12/`curl https://sm.alliedmods.net/smdrop/1.12/sourcemod-latest-linux` -o sourcemod.tar.gz; tar -xf sourcemod.tar.gz -C ../tf2server/tf;

    echo echo !ESC![93m
    echo echo --------------------------------
    echo echo Installing Sigsegv-MvM extension
    echo echo --------------------------------
    echo wsl -d ubuntu -u gameserver cd /var/tf2server; curl -LJ https://github.com/rafradek/sigsegv-mvm/releases/latest/download/package-linux.zip -o package-linux.zip; unzip -d ../tf2server/tf package-linux.zip;

    @REM echo echo.
    @REM echo echo ----------------------
    @REM echo echo Compiling Accelerator
    @REM echo echo ----------------------
    @REM echo echo.
    @REM echo wsl -d ubuntu -u root rm -rf /var/accelerator; mkdir -p /var/accelerator; cd /var/accelerator; git clone https://github.com/asherkin/accelerator.git .; chmod -R 755 /var/accelerator;
    @REM echo wsl -d ubuntu -u root cd /var/accelerator; git clone https://github.com/alliedmodders/ambuild.git --depth 1;git clone https://github.com/alliedmodders/metamod-source.git --depth 1 -b 1.12-dev; git clone --recursive https://github.com/alliedmodders/sourcemod.git --depth 1 -b 1.12-dev; chmod -R 755 /var/accelerator
    @REM echo wsl -d ubuntu -u root mkdir -p ~/.venvs; python3 -m venv ~/.venvs/ambuild; chmod -R 755 ~/.venvs; cd /var/accelerator; ~/.venvs/ambuild/bin/pip install /var/accelerator/ambuild
    @REM echo wsl -d ubuntu -u root chmod +x /var/accelerator/configure.py
    @REM @REM echo wsl -d ubuntu -u root mkdir -p /var/accelerator/build/release; cd /var/accelerator/build/release; CC=gcc CXX=g++ ~/.venvs/ambuild/bin/python /var/accelerator/configure.py --targets=x86 --mms-path=/var/ambuild/metamod-source --sm-path=/var/ambuild/sourcemod
    @REM echo wsl -d ubuntu -u root mkdir -p /var/accelerator/build/release; cd /var/accelerator/build/release; CC=gcc CXX=g++ ~/.venvs/ambuild/bin/python /var/accelerator/configure.py
    echo echo !ESC![92m
    echo echo ----------------------
    echo echo INSTALLATION COMPLETE!
    echo echo ----------------------
    echo echo !ESC![96m
    echo echo To run the game server, enter the newly created TF2Server directory and run the !ESC![97mRun Server.bat!ESC![96m file

    echo echo !ESC![93m 
    echo echo ---------------------------------
    echo echo If your server crashes on launch:
    echo echo ---------------------------------
    echo echo !ESC![96m
    echo echo 1. Grab the latest rafmod build from the latest GitHub action.
    echo echo.
    echo echo   a. !ESC![97mhttps://github.com/rafradek/sigsegv-mvm/actions!ESC![96m
    echo echo   b. Click on the most recent run and click !ESC![97mbuild-and-package!ESC![96m
    echo echo   c. Click on !ESC![97mUpload package-linux!ESC![96m and click the !ESC![97mArtifact download URL:!ESC![96m link
    echo echo   d. Unzip the !ESC![97mpackage-linux.zip!ESC![96m file, unzip the other one inside too
    echo echo !ESC![96m
    echo echo 2. Click !ESC![97mBrowse Files.bat!ESC![96m in TF2Server and manually install rafmod
    echo echo 3. Restart the server
    echo echo !ESC![0m

    echo pause
    echo start /b "" cmd /c del "%%~f0"^&exit /b
    echo exit
) > "SetupSigsegvAfterBoot.bat"

(
    echo @echo off
    echo wsl -d ubuntu -u root bash -c "echo nameserver 8.8.8.8 > /etc/resolv.conf"
    echo wsl -d ubuntu -u root bash -c "echo generateResolveConf=false >> /etc/wsl.conf"
    echo wsl -d ubuntu -u gameserver cd /var/steamcmd; ./steamcmd.sh +force_install_dir ../tf2server +login anonymous +app_update 232250 +quit
    echo wsl -d ubuntu -u gameserver /var/tf2server/srcds_run +maxplayers 32 +map mvm_bigrock -enablefakeip
) > "Run Server.bat"

(
    echo @echo off
    echo wsl -d ubuntu -u root bash -c "echo nameserver 8.8.8.8 > /etc/resolv.conf"
    echo wsl -d ubuntu -u root bash -c "echo generateResolveConf=false >> /etc/wsl.conf"
    echo echo Installing Metamod
    echo wsl -d ubuntu -u gameserver cd /var/tf2server; curl https://mms.alliedmods.net/mmsdrop/1.12/`curl https://mms.alliedmods.net/mmsdrop/1.12/mmsource-latest-linux` -o metamod.tar.gz; tar -xf metamod.tar.gz -C ../tf2server/tf;
    echo echo Installing Sourcemod
    echo wsl -d ubuntu -u gameserver cd /var/tf2server; curl https://sm.alliedmods.net/smdrop/1.12/`curl https://sm.alliedmods.net/smdrop/1.12/sourcemod-latest-linux` -o sourcemod.tar.gz; tar -xf sourcemod.tar.gz -C ../tf2server/tf --exclude "addons/sourcemod/configs" --exclude "cfg";
    echo echo Installing Sigsegv-MvM extension
    echo wsl -d ubuntu -u gameserver cd /var/tf2server; curl -LJ https://github.com/rafradek/sigsegv-mvm/releases/latest/download/package-linux.zip -o package-linux.zip; unzip -o -x "cfg/*" -d ../tf2server/tf package-linux.zip; 
    echo echo Update complete
) > "Update Server.bat"

(
    echo @echo off
    echo wsl -d ubuntu -u gameserver cd /var/tf2server/tf; explorer.exe .
) > "Browse Files.bat"

(
    echo @echo off
    echo choice /C YN /M "Do you want to delete the linux VM and all server files?"
    echo if errorlevel 2 ^(
        echo echo Server files are not deleted
    echo ^) else ^(
        echo wsl --unregister ubuntu
        echo cd ..
        echo pause
        echo rmdir /q /s TF2Server
    echo ^)
) > "Delete Everything.bat"

(
    echo @echo off
    echo choise /C YN /M "Do you want to delete server files and keep the linux VM?"
    echo if errorlevel 2 ^(
        echo echo Server files are not deleted
    echo ^) else ^(
        echo wsl rm -rf /var/tf2server
        echo wsl rm -rf /var/steamcmd
        echo pause
    echo ^)
) > "Delete Server Files.bat"

if not exist "%windir%\System32\bash.exe" (
    echo !ESC![92m
    echo -----------------
    echo Installing WSL...
    echo -----------------
    echo !ESC![0m
    wsl --install -d Ubuntu-22.04 --root
    echo [wsl2] > %USERPROFILE%/.wslconfig
    echo memory=2GB >> %USERPROFILE%/.wslconfig

    reg add HKEY_CURRENT_USER\SOFTWARE\Microsoft\Windows\CurrentVersion\RunOnce /f /v "WslInstallContinue" /t REG_SZ /d "%COMSPEC% /c """%cd%\SetupSigsegvAfterBoot.bat"""
    echo !ESC![96m
    echo Reboot system to continue installation
    choice /C YN /M "Do you want to reboot now?"

    if errorlevel 2 (
        echo Reboot the system later to continue
    ) else (
        shutdown -r -t 0
    )
    pause
) else (
    cls
    echo !ESC![92mUpdating WSL...!ESC![0m
    wsl --update > nul 2>&1
    SetupSigsegvAfterBoot.bat
)
exit