FROM boosapi-comfyui:multi-tenant

ARG COMFYUI_TAG=v0.22.2

WORKDIR /workspace/ComfyUI

COPY docker/build-comfyui-src /tmp/ComfyUI-updated

RUN find /workspace/ComfyUI -mindepth 1 \
        ! -path '/workspace/ComfyUI/venv' \
        ! -path '/workspace/ComfyUI/venv/*' \
        -exec rm -rf {} + \
    && cp -a /tmp/ComfyUI-updated/. /workspace/ComfyUI/ \
    && rm -rf /tmp/ComfyUI-updated \
    && UV_INDEX_URL=https://pypi.tuna.tsinghua.edu.cn/simple \
       UV_EXTRA_INDEX_URL=https://mirrors.tuna.tsinghua.edu.cn/pytorch-wheels/cpu/ \
       uv pip install --python venv/bin/python -r requirements.txt \
    && UV_INDEX_URL=https://pypi.tuna.tsinghua.edu.cn/simple \
       uv pip install --python venv/bin/python six \
    && mkdir -p /workspace/ComfyUI/custom_nodes/boosapi_nodes

COPY comfy/boosapi_comfy_nodes.py /workspace/ComfyUI/custom_nodes/boosapi_nodes/__init__.py
COPY scripts/comfyui/install-runninghub-template-nodes.sh /tmp/install-runninghub-template-nodes.sh

RUN bash /tmp/install-runninghub-template-nodes.sh \
    && rm -f /tmp/install-runninghub-template-nodes.sh

RUN python - <<'PY'
from pathlib import Path
path = Path("/workspace/ComfyUI/custom_nodes/boosapi_nodes/__init__.py")
if not path.exists():
    raise SystemExit("BoosAPI custom node missing after ComfyUI update")
PY