#!/usr/bin/env bash
# restore-display-ram.sh — restore nvidia_drm to the host default (no
# modeset). Counterpart to unlock-display-ram.sh (the temporary 2 GB
# display-reserve unlock): run after the EXL3 2x containers are stopped.
#
# Temporary by design: no /etc/modprobe.d entry is ever written; the
# default modprobe state matches a fresh boot (firmware display reserve
# stays at its 0.5 GB default until the next unlock).
set -euo pipefail

if [ ! -d /sys/module/nvidia_drm ]; then
    echo "nvidia_drm not loaded — nothing to restore"
    exit 0
fi

modeset=$(cat /sys/module/nvidia_drm/parameters/modeset)
if [ "$modeset" != "Y" ]; then
    echo "nvidia_drm already at default (modeset=$modeset) — nothing to do"
    exit 0
fi

if ! sudo rmmod nvidia_drm 2>/dev/null; then
    echo "nvidia_drm busy — waiting 3 s and retrying once"
    sleep 3
    sudo rmmod nvidia_drm
fi
sudo modprobe nvidia_drm
echo "nvidia_drm restored to default (modeset=$(cat /sys/module/nvidia_drm/parameters/modeset))"