#!/bin/bash -e

# Update env for gstreamer
source /opt/gstreamer/gst-env
export LD_LIBRARY_PATH="/usr/local/cuda/lib64:${LD_LIBRARY_PATH}"
export DISPLAY=":0"
export GST_DEBUG="${GST_DEBUG:-*:2}"
if [ -n "$ENABLE_BASIC_AUTH" ] && [ -z "$BASIC_AUTH_PASSWORD" ]; then export BASIC_AUTH_PASSWORD="$PASSWD"; fi

# Wait for X11 to start
echo "Waiting for X socket"
until [ -S "/tmp/.X11-unix/X${DISPLAY/:/}" ]; do sleep 1; done
echo "X socket is ready"

# Fix keyboard mapping for "<" and ">"
sudo xmodmap -e "keycode 94 shift = less less" &

# Write Progressive Web App (PWA) config.
export PWA_APP_NAME="Selkies WebRTC"
export PWA_APP_SHORT_NAME="selkies"
export PWA_START_URL="/index.html"
sed -i \
    -e "s|PWA_APP_NAME|${PWA_APP_NAME}|g" \
    -e "s|PWA_APP_SHORT_NAME|${PWA_APP_SHORT_NAME}|g" \
    -e "s|PWA_START_URL|${PWA_START_URL}|g" \
/opt/gst-web/manifest.json && \
sed -i \
    -e "s|PWA_CACHE|${PWA_APP_SHORT_NAME}-webrtc-pwa|g" \
/opt/gst-web/sw.js

# Write default user config.
export SELKIES_USER_CONFIG_FILE="${HOME}/.config/selkies/selkies-gstreamer-conf.json"
mkdir -p $(dirname "$SELKIES_USER_CONFIG_FILE")
if [ ! -f "${SELKIES_USER_CONFIG_FILE}" ]; then
    cat - > "${SELKIES_USER_CONFIG_FILE}" <<EOF
{
    "framerate": 60,
    "video_bitrate": 4000,
    "audio_bitrate": 64000,
    "enable_audio": "${ENABLE_AUDIO:-true}",
    "enable_resize": "${WEBRTC_ENABLE_RESIZE:-true}",
    "encoder": "nvh264enc"
}
EOF
fi

# Clear the cache registry to force the cuda elements to refresh
rm -f "${HOME}/.cache/gstreamer-1.0/registry.x86_64.bin"

# Start the selkies webrtc gstreamer app
selkies-gstreamer \
    --json_config="${SELKIES_USER_CONFIG_FILE}" \
    --addr="0.0.0.0" \
    --port="8080" \
    $@