setup-env.sh

#!/usr/bin/env bash
# Resolve host-specific values and write them into .env.
# Run once on the target Linux host before `docker compose up -d`.
set -euo pipefail

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
ENV_FILE="$SCRIPT_DIR/.env"

# --------------------------------------------------------------------------- #
# Validation
# --------------------------------------------------------------------------- #
[[ -f "$ENV_FILE" ]] || { echo "ERROR: $ENV_FILE not found." >&2; exit 1; }
command -v getent &>/dev/null || { echo "ERROR: getent not found (non-glibc system?)." >&2; exit 1; }
command -v sed   &>/dev/null || { echo "ERROR: sed not found." >&2; exit 1; }

RENDER_GID=$(getent group render 2>/dev/null | cut -d: -f3)
VIDEO_GID=$(getent group video  2>/dev/null | cut -d: -f3)

[[ -n "$RENDER_GID" ]] || { echo "ERROR: 'render' group not found on this host." >&2; exit 1; }
[[ -n "$VIDEO_GID"  ]] || { echo "ERROR: 'video' group not found on this host."  >&2; exit 1; }

# Sanity-check: GIDs must be numeric
[[ "$RENDER_GID" =~ ^[0-9]+$ ]] || { echo "ERROR: unexpected RENDER_GID value: $RENDER_GID" >&2; exit 1; }
[[ "$VIDEO_GID"  =~ ^[0-9]+$ ]] || { echo "ERROR: unexpected VIDEO_GID value: $VIDEO_GID"   >&2; exit 1; }

echo "RENDER_GID=$RENDER_GID"
echo "VIDEO_GID=$VIDEO_GID"

# --------------------------------------------------------------------------- #
# 2. HF cache dir — reuse the current user's llama.cpp download cache
# --------------------------------------------------------------------------- #
HF_CACHE_DIR="${HOME}/.cache/llama.cpp"
mkdir -p "$HF_CACHE_DIR"
echo "HF_CACHE_DIR=$HF_CACHE_DIR"

# --------------------------------------------------------------------------- #
# Apply with sed (in-place)
# --------------------------------------------------------------------------- #
sed -i \
  -e "s|^RENDER_GID=.*|RENDER_GID=$RENDER_GID|" \
  -e "s|^VIDEO_GID=.*|VIDEO_GID=$VIDEO_GID|" \
  -e "s|^HF_CACHE_DIR=.*|HF_CACHE_DIR=$HF_CACHE_DIR|" \
  "$ENV_FILE"

echo "Done. Updated $ENV_FILE:"
grep -E '^(RENDER_GID|VIDEO_GID|HF_CACHE_DIR)=' "$ENV_FILE"