selkies-gstreamer-entrypoint.sh 1.7 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455
  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 [ -n "$ENABLE_BASIC_AUTH" ] && [ -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. # Fix keyboard mapping for "<" and ">"
  13. sudo xmodmap -e "keycode 94 shift = less less" &
  14. # Write Progressive Web App (PWA) config.
  15. export PWA_APP_NAME="Selkies WebRTC"
  16. export PWA_APP_SHORT_NAME="selkies"
  17. export PWA_START_URL="/index.html"
  18. sed -i \
  19. -e "s|PWA_APP_NAME|${PWA_APP_NAME}|g" \
  20. -e "s|PWA_APP_SHORT_NAME|${PWA_APP_SHORT_NAME}|g" \
  21. -e "s|PWA_START_URL|${PWA_START_URL}|g" \
  22. /opt/gst-web/manifest.json && \
  23. sed -i \
  24. -e "s|PWA_CACHE|${PWA_APP_SHORT_NAME}-webrtc-pwa|g" \
  25. /opt/gst-web/sw.js
  26. # Write default user config.
  27. export SELKIES_USER_CONFIG_FILE="${HOME}/.config/selkies/selkies-gstreamer-conf.json"
  28. mkdir -p $(dirname "$SELKIES_USER_CONFIG_FILE")
  29. if [ ! -f "${SELKIES_USER_CONFIG_FILE}" ]; then
  30. cat - > "${SELKIES_USER_CONFIG_FILE}" <<EOF
  31. {
  32. "framerate": 60,
  33. "video_bitrate": 4000,
  34. "audio_bitrate": 64000,
  35. "enable_audio": "${ENABLE_AUDIO:-true}",
  36. "enable_resize": "${WEBRTC_ENABLE_RESIZE:-true}",
  37. "encoder": "nvh264enc"
  38. }
  39. EOF
  40. fi
  41. # Clear the cache registry to force the cuda elements to refresh
  42. rm -f "${HOME}/.cache/gstreamer-1.0/registry.x86_64.bin"
  43. # Start the selkies webrtc gstreamer app
  44. selkies-gstreamer \
  45. --json_config="${SELKIES_USER_CONFIG_FILE}" \
  46. --addr="0.0.0.0" \
  47. --port="8080" \
  48. $@