Upload 3 files

README.md

@@ -26,6 +26,10 @@ app_port: 7860
 - `DATABASE_URL=postgresql://USER:PASSWORD@HOST:PORT/DBNAME?sslmode=require&channel_binding=require`
 - `REDIS_URL=rediss://:PASSWORD@HOST:PORT/0`
 
+## 脚本行为
+- 启动脚本会把 `DATABASE_URL` / `REDIS_URL` 解析为应用实际需要的 `DATABASE_*` / `REDIS_*` 变量。
+- 不要在 HF 变量里加引号（否则解析会失败）。
+
 ## 建议配置（无默认值）
 - JWT_SECRET
 - TOTP_ENCRYPTION_KEY
@@ -34,11 +38,15 @@ app_port: 7860
 
 ## 可选环境变量（含默认值）
 - PORT=7860
-- GIN_MODE=release
+- SERVER_HOST=0.0.0.0
+- SERVER_PORT=7860
+- SERVER_MODE=release
+- RUN_MODE=standard
+- AUTO_SETUP=true
 - TZ=Asia/Shanghai
 - DATA_DIR=/tmp/sub2api
 
 ## 说明
 - Hugging Face Docker Space 默认端口是 7860，可通过 `app_port` 覆盖。([huggingface.co](https://huggingface.co/docs/hub/spaces-sdks-docker?utm_source=openai))
 - 空间里通常需要把可写目录放在 `/tmp`；如启用持久化存储，可使用 `/data`。([huggingface.co](https://huggingface.co/docs/hub/spaces-storage?utm_source=openai))([discuss.huggingface.co](https://discuss.huggingface.co/t/permission-denied-for-writing-files-within-spaces/29799?utm_source=openai))
-- 本脚本默认 `DATA_DIR=/tmp/sub2api`。
+- 应用侧实际使用 `DATABASE_*` / `REDIS_*` 变量（见官方 compose 示例）。([github.com](https://github.com/Wei-Shaw/sub2api/blob/main/docker-compose.yml?utm_source=openai))

start.sh

@@ -14,6 +14,30 @@ set -eu
 mkdir -p "$DATA_DIR" "$BIN_DIR" || true
 chmod 777 "$DATA_DIR" "$BIN_DIR" 2>/dev/null || true
 
+strip_quotes() {
+  val="$1"
+  case "$val" in
+    "\""*"\"") val="${val#\"}"; val="${val%\"}" ;;
+  esac
+  case "$val" in
+    "'"*"'") val="${val#'}"; val="${val%'}" ;;
+  esac
+  printf '%s' "$val"
+}
+
+url_decode() {
+  printf '%b' "$(printf '%s' "$1" | sed 's/+/ /g; s/%/\\x/g')"
+}
+
+get_query_param() {
+  query="$1"
+  key="$2"
+  if [ -z "$query" ]; then
+    return 0
+  fi
+  printf '%s' "$query" | tr '&' '\n' | awk -F= -v k="$key" '$1==k {print $2; exit}'
+}
+
 require_env() {
   name="$1"
   eval "val=\${$name:-}"
@@ -23,9 +47,161 @@ require_env() {
   fi
 }
 
+parse_database_url() {
+  url="$1"
+  rest="${url#*://}"
+  base="$rest"
+  query=""
+
+  if [ "$rest" != "${rest%%\?*}" ]; then
+    base="${rest%%\?*}"
+    query="${rest#*\?}"
+  fi
+
+  userinfo="${base%%@*}"
+  hostport_db="${base#*@}"
+  if [ "$hostport_db" = "$base" ]; then
+    userinfo=""
+    hostport_db="$base"
+  fi
+
+  hostport="${hostport_db%%/*}"
+  dbname="${hostport_db#*/}"
+  if [ "$dbname" = "$hostport_db" ]; then
+    dbname=""
+  fi
+
+  user=""
+  pass=""
+  if [ -n "$userinfo" ]; then
+    case "$userinfo" in
+      *:*)
+        user="${userinfo%%:*}"
+        pass="${userinfo#*:}"
+        ;;
+      *)
+        user="$userinfo"
+        pass=""
+        ;;
+    esac
+  fi
+
+  host="${hostport%%:*}"
+  port="${hostport#*:}"
+  if [ "$port" = "$hostport" ]; then
+    port=""
+  fi
+
+  [ -n "$user" ] && export DATABASE_USER="$(url_decode "$user")"
+  [ -n "$pass" ] && export DATABASE_PASSWORD="$(url_decode "$pass")"
+  [ -n "$host" ] && export DATABASE_HOST="$(url_decode "$host")"
+  [ -n "$dbname" ] && export DATABASE_DBNAME="$(url_decode "$dbname")"
+  [ -n "$port" ] && export DATABASE_PORT="$port"
+
+  sslmode="$(get_query_param "$query" "sslmode")"
+  if [ -n "$sslmode" ]; then
+    export DATABASE_SSLMODE="$(url_decode "$sslmode")"
+  else
+    : "${DATABASE_SSLMODE:=require}"
+  fi
+}
+
+parse_redis_url() {
+  url="$1"
+  scheme="${url%%://*}"
+  rest="${url#*://}"
+  base="$rest"
+  query=""
+
+  if [ "$rest" != "${rest%%\?*}" ]; then
+    base="${rest%%\?*}"
+    query="${rest#*\?}"
+  fi
+
+  userinfo="${base%%@*}"
+  hostport_db="${base#*@}"
+  if [ "$hostport_db" = "$base" ]; then
+    userinfo=""
+    hostport_db="$base"
+  fi
+
+  hostport="${hostport_db%%/*}"
+  dbname="${hostport_db#*/}"
+  if [ "$dbname" = "$hostport_db" ] || [ -z "$dbname" ]; then
+    dbname=""
+  fi
+
+  pass=""
+  if [ -n "$userinfo" ]; then
+    case "$userinfo" in
+      *:*)
+        pass="${userinfo#*:}"
+        ;;
+      *)
+        pass=""
+        ;;
+    esac
+  fi
+
+  host="${hostport%%:*}"
+  port="${hostport#*:}"
+  if [ "$port" = "$hostport" ]; then
+    port=""
+  fi
+
+  [ -n "$host" ] && export REDIS_HOST="$(url_decode "$host")"
+  [ -n "$port" ] && export REDIS_PORT="$port"
+  [ -n "$pass" ] && export REDIS_PASSWORD="$(url_decode "$pass")"
+
+  db_query="$(get_query_param "$query" "db")"
+  if [ -n "$db_query" ]; then
+    dbname="$db_query"
+  fi
+  [ -n "$dbname" ] && export REDIS_DB="$(url_decode "$dbname")"
+
+  tls="false"
+  tls_query="$(get_query_param "$query" "ssl")"
+  if [ -z "$tls_query" ]; then
+    tls_query="$(get_query_param "$query" "tls")"
+  fi
+  if [ -n "$tls_query" ]; then
+    case "$(url_decode "$tls_query")" in
+      1|true|TRUE|True|yes|YES|on|ON)
+        tls="true"
+        ;;
+      *)
+        tls="false"
+        ;;
+    esac
+  else
+    if [ "$scheme" = "rediss" ]; then
+      tls="true"
+    fi
+  fi
+  export REDIS_ENABLE_TLS="$tls"
+}
+
 require_env DATABASE_URL
 require_env REDIS_URL
 
+DATABASE_URL="$(strip_quotes "$DATABASE_URL")"
+REDIS_URL="$(strip_quotes "$REDIS_URL")"
+export DATABASE_URL REDIS_URL
+
+parse_database_url "$DATABASE_URL"
+parse_redis_url "$REDIS_URL"
+
+: "${DATABASE_DBNAME:=sub2api}"
+: "${DATABASE_PORT:=5432}"
+: "${DATABASE_SSLMODE:=require}"
+: "${REDIS_PORT:=6379}"
+: "${REDIS_DB:=0}"
+
+require_env DATABASE_HOST
+require_env DATABASE_USER
+require_env DATABASE_PASSWORD
+require_env REDIS_HOST
+
 export PORT SERVER_HOST SERVER_PORT SERVER_MODE RUN_MODE AUTO_SETUP TZ DATA_DIR
 
 copy_and_exec() {