selkies-gstreamer-entrypoint.sh 1.6 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152
  1. #!/bin/bash -e
  2. # Update env for gstreamer
  3. source /opt/gstreamer/gst-env
  4. export LD_LIBRARY_PATH="/usr/local/cuda/lib64:${LD_LIBRARY_PATH}"
  5. export DISPLAY=":0"
  6. export GST_DEBUG="${GST_DEBUG:-*:2}"
  7. if [ "${ENABLE_BASIC_AUTH,,}" = "true" ] && [ -z "$BASIC_AUTH_PASSWORD" ]; then export BASIC_AUTH_PASSWORD="$PASSWD"; fi
  8. # Wait for X11 to start
  9. echo "Waiting for X socket"
  10. until [ -S "/tmp/.X11-unix/X${DISPLAY/:/}" ]; do sleep 1; done
  11. echo "X socket is ready"
  12. # Write Progressive Web App (PWA) config.
  13. export PWA_APP_NAME="Selkies WebRTC"
  14. export PWA_APP_SHORT_NAME="selkies"
  15. export PWA_START_URL="/index.html"
  16. sed -i \
  17. -e "s|PWA_APP_NAME|${PWA_APP_NAME}|g" \
  18. -e "s|PWA_APP_SHORT_NAME|${PWA_APP_SHORT_NAME}|g" \
  19. -e "s|PWA_START_URL|${PWA_START_URL}|g" \
  20. /opt/gst-web/manifest.json && \
  21. sed -i \
  22. -e "s|PWA_CACHE|${PWA_APP_SHORT_NAME}-webrtc-pwa|g" \
  23. /opt/gst-web/sw.js
  24. # Write default user config.
  25. export SELKIES_USER_CONFIG_FILE="${HOME}/.config/selkies/selkies-gstreamer-conf.json"
  26. mkdir -p $(dirname "$SELKIES_USER_CONFIG_FILE")
  27. if [ ! -f "${SELKIES_USER_CONFIG_FILE}" ]; then
  28. cat - > "${SELKIES_USER_CONFIG_FILE}" <<EOF
  29. {
  30. "framerate": 60,
  31. "video_bitrate": 4000,
  32. "audio_bitrate": 64000,
  33. "enable_audio": "${ENABLE_AUDIO:-true}",
  34. "enable_resize": "${WEBRTC_ENABLE_RESIZE:-true}",
  35. "encoder": "${WEBRTC_ENCODER:-nvh264enc}"
  36. }
  37. EOF
  38. fi
  39. # Clear the cache registry to force the cuda elements to refresh
  40. rm -f "${HOME}/.cache/gstreamer-1.0/registry.x86_64.bin"
  41. # Start the selkies webrtc gstreamer app
  42. selkies-gstreamer \
  43. --json_config="${SELKIES_USER_CONFIG_FILE}" \
  44. --addr="0.0.0.0" \
  45. --port="8080" \
  46. $@