Add an entrypoint script for Nsight streamer that sets the TURN host to the correct IP. Pass the BREV_ENV_ID variable into the containers.

Author: brycelelbach

build/docker-compose.yml

@@ -20,6 +20,8 @@ services:
       - "0.0.0.0:8888:8888" # JupyterLab
     volumes:
       - /home/ubuntu/pyhpc-tutorial:/pyhpc-tutorial
+    environment:
+      - BREV_ENV_ID
     user: root
     working_dir: /pyhpc-tutorial/notebooks
     restart: unless-stopped
@@ -37,18 +39,14 @@ services:
             - driver: nvidia
               count: all
               capabilities: [gpu]
-    environment:
-      - NVIDIA_DRIVER_CAPABILITIES=all
-      - HOST_IP=0.0.0.0
-      - WEB_USERNAME=""
-      - WEB_PASSWORD=""
-      - SELKIES_TURN_HOST=nsight-turn0-${BREV_ENV_ID}.brevlab.com
-      - TURN_PORT=80
     ports:
       - "0.0.0.0:8080:8080" # HTTP
       - "0.0.0.0:3478:3478" # TURN
     volumes:
       - /home/ubuntu/pyhpc-tutorial:/pyhpc-tutorial
+    environment:
+      - BREV_ENV_ID
     user: root
     working_dir: /pyhpc-tutorial/notebooks
+    entrypoint: ["/pyhpc-tutorial/build/nsight-start.bash"]
     restart: unless-stopped

build/nsight-start.bash

@@ -0,0 +1,21 @@
+#!/bin/bash
+
+apt-get update
+apt-get install -y curl
+
+EXTERNAL_IP=$(curl -sSL ifconfig.me)
+
+export NVIDIA_DRIVER_CAPABILITIES=all
+export HOST_IP=0.0.0.0
+export WEB_USERNAME=""
+export WEB_PASSWORD=""
+export SELKIES_TURN_HOST=$(curl -sSL ifconfig.me)
+
+VARS=("NVIDIA_DRIVER_CAPABILITIES" "HOST_IP" "WEB_USERNAME" "WEB_PASSWORD" "SELKIES_TURN_HOST")
+for VAR in "${VARS[@]}"; do
+  if [[ -n "${!VAR+x}" ]]; then
+    echo "export $VAR=${!VAR}" >> ~/.bashrc
+  fi
+done
+
+source /setup/entrypoint.sh "$@"