# 使用 3.13 而非 3.14，因为 openvino 尚未提供 3.14 的 wheels
# 等待 https://github.com/openvinotoolkit/openvino/issues/34802 解决后再升级
FROM python:3.13.13-slim-bookworm

# ======================
# Build args (architecture)
# ======================
ARG TARGETARCH
ARG DEBIAN_FRONTEND=noninteractive
ENV DEBIAN_FRONTEND=${DEBIAN_FRONTEND}

# ======================
# Python optimization
# ======================
ENV PYTHONDONTWRITEBYTECODE=1
ENV PYTHONUNBUFFERED=1

# ======================
# Virtualenv
# ======================
ENV VIRTUAL_ENV="/opt/venv"
ENV UV_PROJECT_ENVIRONMENT=$VIRTUAL_ENV
ENV PATH="$VIRTUAL_ENV/bin:$PATH"

# ======================
# Timezone
# ======================
ENV TZ=Asia/Shanghai

# ======================
# App environment variables
# ======================
ENV MARCH7TH_CLOUD_GAME_ENABLE=true
ENV MARCH7TH_BROWSER_HEADLESS_ENABLE=true
ENV MARCH7TH_BROWSER_HEADLESS_RESTART_ON_NOT_LOGGED_IN=false
# 默认使用官方源下载浏览器；如果下载缓慢，可以改为 true 启用镜像下载
ENV MARCH7TH_BROWSER_DOWNLOAD_USE_MIRROR=false
# 任务完成后循环执行
ENV MARCH7TH_AFTER_FINISH=Loop
# 标记从 Docker 启动，避免控制台阻塞
ENV MARCH7TH_DOCKER_STARTED=true

WORKDIR /m7a

COPY pyproject.toml uv.lock ./

# ======================
# System dependencies
# ======================
RUN \
    # 如果需要使用国内源，可以取消下面一行的注释
    # sed -i 's/deb.debian.org/mirrors.cloud.tencent.com/g' /etc/apt/sources.list.d/debian.sources && \
    apt-get update && apt-get install -yq --no-install-recommends \
    # Dependencies for OpenCV
    libgl1 \
    # Dependencies for headless Chrome
    libglib2.0-0 \
    libnss3 \
    libfontconfig1 \
    libx11-6 \
    libx11-xcb1 \
    libxcb1 \
    libxcomposite1 \
    libxdamage1 \
    libxrandr2 \
    libxkbcommon0 \
    libatk1.0-0 \
    libatk-bridge2.0-0 \
    libcups2 \
    libdrm2 \
    libgbm1 \
    libasound2 \
    libpangocairo-1.0-0 \
    libpango-1.0-0 \
    libgtk-3-0 \
    ca-certificates \
    fonts-liberation \
    \
    # -------- arm64 only --------
    && if [ "$TARGETARCH" = "arm64" ]; then \
        apt-get install -yq --no-install-recommends \
            chromium \
            chromium-driver ; \
    fi \
    \
    && rm -rf /var/lib/apt/lists/*

# ======================
# Python deps
# ======================
COPY --from=ghcr.io/astral-sh/uv:0.11.15 /uv /uvx /bin/
RUN uv sync --only-group docker
    # 如果需要使用国内源，可以取消下面一行的注释
    # RUN uv sync --only-group docker --index-url https://mirrors.cloud.tencent.com/pypi/simple/

COPY build.py ./

RUN if [ "$TARGETARCH" != "arm64" ]; then \
        python build.py --task browser ; \
    else \
        echo "Skipping browser build on arm64"; \
    fi

COPY . .

RUN python build.py --task ocr \
    && rm -rf ./logs \
    && rm -f config.yaml

# ======================
# arm64 browser env
# ======================
COPY entrypoint.sh /entrypoint.sh
RUN chmod +x /entrypoint.sh

ENTRYPOINT ["/entrypoint.sh"]
CMD ["python", "main.py"]