84 lines
1.8 KiB
Bash
84 lines
1.8 KiB
Bash
#!/usr/bin/env bash
|
|
set -euo pipefail
|
|
|
|
ROOT="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
|
|
cd "$ROOT"
|
|
|
|
MODE="${1:-auto}"
|
|
|
|
if [[ -f ".env" ]]; then
|
|
set -a
|
|
# shellcheck disable=SC1091
|
|
source ./.env
|
|
set +a
|
|
fi
|
|
|
|
normalize_host() {
|
|
local host="${1:-0.0.0.0}"
|
|
case "$host" in
|
|
127.0.0.1|localhost|::1)
|
|
echo "0.0.0.0"
|
|
;;
|
|
*)
|
|
echo "$host"
|
|
;;
|
|
esac
|
|
}
|
|
|
|
pick_python() {
|
|
if [[ -n "${VISUAL_CHAT_PYTHON:-}" && -x "${VISUAL_CHAT_PYTHON}" ]]; then
|
|
echo "$VISUAL_CHAT_PYTHON"
|
|
return 0
|
|
fi
|
|
if [[ -n "${VIRTUAL_ENV:-}" && -x "${VIRTUAL_ENV}/bin/python" ]]; then
|
|
echo "${VIRTUAL_ENV}/bin/python"
|
|
return 0
|
|
fi
|
|
if [[ -x "/home/xsl/miniconda3/envs/MuseTalk/bin/python" ]]; then
|
|
echo "/home/xsl/miniconda3/envs/MuseTalk/bin/python"
|
|
return 0
|
|
fi
|
|
if command -v python3 >/dev/null 2>&1; then
|
|
command -v python3
|
|
return 0
|
|
fi
|
|
command -v python
|
|
}
|
|
|
|
PY="$(pick_python)"
|
|
HOST="$(normalize_host "${WEBRTC_HOST:-0.0.0.0}")"
|
|
PORT="${WEBRTC_PORT:-8080}"
|
|
SCHEME="http"
|
|
ARGS=(main:app --host "$HOST" --port "$PORT")
|
|
|
|
case "$MODE" in
|
|
auto)
|
|
if [[ -n "${SSL_CERTFILE:-}" && -n "${SSL_KEYFILE:-}" ]]; then
|
|
SCHEME="https"
|
|
ARGS+=(--ssl-certfile "$SSL_CERTFILE" --ssl-keyfile "$SSL_KEYFILE")
|
|
fi
|
|
;;
|
|
https)
|
|
if [[ -z "${SSL_CERTFILE:-}" || -z "${SSL_KEYFILE:-}" ]]; then
|
|
echo "SSL_CERTFILE / SSL_KEYFILE 未配置，无法以 https 模式启动" >&2
|
|
exit 1
|
|
fi
|
|
SCHEME="https"
|
|
ARGS+=(--ssl-certfile "$SSL_CERTFILE" --ssl-keyfile "$SSL_KEYFILE")
|
|
;;
|
|
http)
|
|
;;
|
|
*)
|
|
echo "usage: bash scripts/start-test.sh [auto|http|https]" >&2
|
|
exit 2
|
|
;;
|
|
esac
|
|
|
|
echo "repo: $ROOT"
|
|
echo "python: $PY"
|
|
echo "mode: $MODE"
|
|
echo "bind: ${HOST}:${PORT}"
|
|
echo "listen: ${SCHEME}://${HOST}:${PORT}"
|
|
echo "browser test page: ${SCHEME}://<server-ip>:${PORT}/"
|
|
|
|
exec "$PY" -m uvicorn "${ARGS[@]}" |