FROM ghcr.io/jhhgiyv/highlight2api:latest

# 设置可写缓存位置，防止根目录写入权限问题
ENV XDG_CACHE_HOME=/app/.cache

# 确保 /app 可写，避免权限问题
RUN mkdir -p /app/.cache && chmod -R 777 /app/.cache || true

# 安装 socat 和 nc (用于端口转发和监听检测)
RUN if [ -f /etc/alpine-release ]; then \
      apk add --no-cache socat netcat-openbsd; \
    elif [ -f /etc/debian_version ]; then \
      apt-get update && apt-get install -y socat netcat-openbsd && rm -rf /var/lib/apt/lists/*; \
    else \
      echo "Unknown base distro, please ensure socat and nc are available"; \
    fi

# Hugging Face Spaces 会路由到 7860
EXPOSE 7860

# 启动脚本：先启动原应用监听 8080，待其就绪后再转发 7860 -> 8080
COPY <<'EOF' /usr/local/bin/start-with-proxy.sh
#!/bin/sh
set -e

# 启动原应用
start_app() {
  if [ "$#" -ne 0 ]; then
    "$@" &
    APP_PID=$!
  else
    if command -v uv >/dev/null 2>&1 && [ -f ./main.py ]; then
      uv run main.py --host 0.0.0.0 --port 8080 &
      APP_PID=$!
    elif command -v uvicorn >/dev/null 2>&1 && [ -f ./app/main.py ]; then
      uvicorn app.main:app --host 0.0.0.0 --port 8080 &
      APP_PID=$!
    else
      echo "No known start command found, keeping container alive."
      tail -f /dev/null &
      APP_PID=$!
    fi
  fi
}

start_app "$@"

# 等待应用启动监听 8080
WAIT_SECONDS=30
SLEEP_INTERVAL=0.5
elapsed=0
echo "Waiting for application to listen on 127.0.0.1:8080 ..."
while ! nc -z 127.0.0.1 8080 >/dev/null 2>&1; do
  sleep $SLEEP_INTERVAL
  elapsed=$(awk "BEGIN {print $elapsed+$SLEEP_INTERVAL}")
  if [ "$(echo "$elapsed > $WAIT_SECONDS" | bc)" -eq 1 ]; then
    echo "Timeout waiting for 127.0.0.1:8080 after ${WAIT_SECONDS}s"
    break
  fi
done

echo "Starting socat to forward 7860 -> 127.0.0.1:8080"
socat TCP-LISTEN:7860,fork,reuseaddr TCP:127.0.0.1:8080 &

# 等待两个进程
wait
EOF

RUN chmod +x /usr/local/bin/start-with-proxy.sh

ENTRYPOINT ["/usr/local/bin/start-with-proxy.sh"]
CMD []