ARG UV_VERSION="latest"

FROM ghcr.io/astral-sh/uv:${UV_VERSION} AS uv

# https://gitlab.com/nvidia/container-images/cuda/blob/master/doc/supported-tags.md
FROM nvidia/cuda:13.0.3-cudnn-runtime-ubuntu24.04

ENV COMFYUI_HOME="/opt/comfyui"

# hadolint ignore=DL3008
RUN --mount=type=cache,target=/var/cache/apt \
  --mount=type=cache,target=/var/lib/apt/lists \
  set -ex \
  && apt-get update \
  && apt-get install --no-install-recommends -y \
  ca-certificates \
  clang \
  curl \
  ffmpeg \
  git \
  gosu \
  libcupti12 \
  libcusparselt0 \
  libgl1 \
  libglib2.0-0

RUN set -ex \
  && groupmod -n comfyui ubuntu \
  && usermod -l comfyui -m -d ${COMFYUI_HOME} ubuntu \
  && chown -R comfyui:comfyui ${COMFYUI_HOME}

COPY --from=uv /uv /uvx /usr/local/bin/

WORKDIR ${COMFYUI_HOME}

COPY ./pylock.toml ./.python-version ./

ARG COMFYUI_VERSION

# hadolint ignore=DL3003
RUN set -ex \
  && git clone --depth 1 --branch "v${COMFYUI_VERSION}" https://github.com/Comfy-Org/ComfyUI.git app/ \
  && install -d \
  "python" \
  "app/custom_nodes" \
  "app/input" \
  "app/models" \
  "app/output" \
  "app/user" \
  "patches" \
  && chown -R comfyui:comfyui .

COPY --chown=comfyui --chmod=0755 ./entrypoint.sh .
COPY --chown=comfyui --chmod=0644 ./extensions.sh .
COPY --chown=comfyui --chmod=0755 ./patches ./patches

EXPOSE 8188
ENTRYPOINT ["/opt/comfyui/entrypoint.sh"]
CMD ["--verbose"]