#!/bin/bash
# Installation Script for Subnautica + Nitrox

# Install dependencies
dpkg --add-architecture i386
apt update
apt -y --no-install-recommends install curl jq unzip libstdc++6 ca-certificates libsdl2-2.0-0:i386

# Resolve Nitrox download URL (latest or pinned NITROX_VERSION)
GITHUB_PACKAGE="SubnauticaNitrox/Nitrox"
case "$(uname -m)" in
  aarch64|arm64) ARCH_TAG="linux_arm64" ;;
  *) ARCH_TAG="linux_x64" ;;
esac

if [ -z "${NITROX_VERSION}" ] || [ "${NITROX_VERSION}" = "latest" ]; then
    RELEASE_JSON=$(curl -fsSL "https://api.github.com/repos/${GITHUB_PACKAGE}/releases/latest")
else
    ALL_RELEASES=$(curl -fsSL "https://api.github.com/repos/${GITHUB_PACKAGE}/releases")
    RELEASE_JSON=$(printf '%s' "${ALL_RELEASES}" | jq -c --arg v "${NITROX_VERSION}" '.[] | select(.tag_name == $v)')
    if [ -z "${RELEASE_JSON}" ]; then
        echo "Nitrox version ${NITROX_VERSION} not found; falling back to latest"
        RELEASE_JSON=$(curl -fsSL "https://api.github.com/repos/${GITHUB_PACKAGE}/releases/latest")
    fi
fi

DOWNLOAD_URL=$(printf '%s' "${RELEASE_JSON}" | jq -r --arg arch "${ARCH_TAG}" '.assets[] | select(.name | test($arch; "i")) | .browser_download_url' | head -n1)

if [ -z "${DOWNLOAD_URL}" ] || [ "${DOWNLOAD_URL}" = "null" ]; then
    echo "Could not determine Nitrox download URL for ${ARCH_TAG}"
    printf '%s' "${RELEASE_JSON}" | jq '.assets[] | {name, browser_download_url}'
    exit 1
fi

# Install SteamCMD
mkdir -p /mnt/server/steamcmd
cd /tmp
curl -fsSL -o steamcmd.tar.gz https://steamcdn-a.akamaihd.net/client/installer/steamcmd_linux.tar.gz
tar -xzf steamcmd.tar.gz -C /mnt/server/steamcmd
cd /mnt/server/steamcmd

chown -R root:root /mnt
export HOME=/mnt/server

if [ -z "${STEAM_GUARDCODE}" ]; then
    echo "### No Steam Guard code provided."
    echo "### One should be emailed to you shortly; set STEAM_GUARDCODE and reinstall."
    sleep 5
    timeout 60 ./steamcmd.sh +force_install_dir "${HOME}/subnautica" +login "${STEAM_USERNAME}" "${STEAM_PASSWORD}" +quit
    exit 1
fi

mkdir -p "${HOME}/subnautica"
./steamcmd.sh +force_install_dir "${HOME}/subnautica" +@sSteamCmdForcePlatformType windows +set_steam_guard_code "${STEAM_GUARDCODE}" +login "${STEAM_USERNAME}" "${STEAM_PASSWORD}" +app_update "${APPID}" ${EXTRA_FLAGS} validate +quit
status=$?

if [ $status -ne 0 ]; then
    echo "### Subnautica download failed (likely a bad/expired Steam Guard code)."
    echo "### A new code should arrive shortly; update STEAM_GUARDCODE and reinstall."
    sleep 5
    timeout 30 ./steamcmd.sh +force_install_dir "${HOME}/subnautica" +login "${STEAM_USERNAME}" "${STEAM_PASSWORD}" +quit
    exit 1
fi

mkdir -p "${HOME}/.steam/sdk32" "${HOME}/.steam/sdk64"
cp -v linux32/steamclient.so "${HOME}/.steam/sdk32/steamclient.so"
cp -v linux64/steamclient.so "${HOME}/.steam/sdk64/steamclient.so"

rm -rf "${HOME}/nitrox"
mkdir -p "${HOME}/nitrox"
cd "${HOME}/nitrox"
curl -fsSL "${DOWNLOAD_URL}" -o Nitrox.zip
unzip -q Nitrox.zip
rm Nitrox.zip
if [ "$(find . -mindepth 1 -maxdepth 1 | wc -l)" = "1" ]; then
    WRAPPER=$(find . -mindepth 1 -maxdepth 1 -type d)
    if [ -n "${WRAPPER}" ]; then
        mv "${WRAPPER}"/* "${WRAPPER}"/.[!.]* . 2>/dev/null || true
        rmdir "${WRAPPER}"
    fi
fi
if [ -f "${HOME}/nitrox/Nitrox.Server.Subnautica" ]; then
    ln -sf Nitrox.Server.Subnautica "${HOME}/nitrox/server"
else
    ln -sf NitroxServer-Subnautica "${HOME}/nitrox/server"
fi
chmod +x "${HOME}/nitrox/server" 2>/dev/null || true

mkdir -p "${HOME}/.config"

mkdir -p "${HOME}/NitroxData/saves/nitrox_world"
CFG="${HOME}/NitroxData/saves/nitrox_world/server.cfg"
if [ ! -e "${CFG}" ]; then
    {
        echo "CreateFullEntityCache=False"
        echo "Seed="
        echo "ServerPort=11000"
        echo "SaveInterval=120000"
        echo "PostSaveCommandPath="
        echo "MaxConnections=${MAX_CONNECTIONS}"
        echo "InitialSyncTimeout=300000"
        echo "DisableConsole=${DISABLE_CONSOLE}"
        echo "DisableAutoSave=False"
        echo "ServerPassword="
        echo "AdminPassword=PleaseChangeMe"
        echo "GameMode=SURVIVAL"
        echo "SerializerMode=JSON"
        echo "DefaultPlayerPerm=${DEFAULT_PERMISSION}"
        echo "DefaultOxygenValue=45"
        echo "DefaultMaxOxygenValue=45"
        echo "DefaultHealthValue=80"
        echo "DefaultHungerValue=50.5"
        echo "DefaultThirstValue=90.5"
        echo "DefaultInfectionValue=0.1"
        echo "AutoPortForward=False"
    } > "${CFG}"
fi