# syntax=docker/dockerfile:1

# Maintained by Matthias Wild <mauwii@outlook.de>

ARG PYTHON_VERSION=3.9
##################
##  base image  ##
##################
FROM python:${PYTHON_VERSION}-slim AS python-base

# prepare for buildkit cache
RUN rm -f /etc/apt/apt.conf.d/docker-clean

# Install necesarry packages
RUN \
  --mount=type=cache,target=/var/cache/apt,sharing=locked \
  apt-get update \
  && apt-get install -y \
    --no-install-recommends \
    libgl1-mesa-glx=20.3.* \
    libglib2.0-0=2.66.* \
    libopencv-dev=4.5.* \
  && rm -rf /var/lib/apt/lists/*

# set working directory and path
ARG APPDIR=/usr/src
ARG APPNAME=InvokeAI
WORKDIR ${APPDIR}
ENV PATH=${APPDIR}/${APPNAME}/bin:$PATH

######################
##  build frontend  ##
######################
FROM node:lts as frontend-builder

# Copy Sources
ARG APPDIR=/usr/src
ARG FRONTEND_DIR=invokeai/frontend
WORKDIR ${APPDIR}/${FRONTEND_DIR}
COPY ${FRONTEND_DIR} .

# install dependencies
RUN \
  --mount=type=cache,target=/usr/local/share/.cache/yarn \
  yarn install \
  --frozen-lockfile \
  --non-interactive

# build frontend
RUN yarn build

###################################
##  install python dependencies  ##
###################################
FROM python-base AS pyproject-builder
ENV PIP_USE_PEP517=1

# Install dependencies
RUN \
  --mount=type=cache,target=/var/cache/apt,sharing=locked \
  apt-get update \
  && apt-get install -y \
    --no-install-recommends \
    build-essential=12.9 \
    gcc=4:10.2.* \
    python3-dev=3.9.* \
  && rm -rf /var/lib/apt/lists/*

# create virtual environment
RUN python3 -m venv "${APPNAME}" \
  --upgrade-deps

# copy sources
COPY --link . .
ARG FRONTEND_DIR=invokeai/frontend
COPY --from=frontend-builder ${APPDIR}/${FRONTEND_DIR}/dist ${FRONTEND_DIR}/dist

# install pyproject.toml
ARG PIP_EXTRA_INDEX_URL
ENV PIP_EXTRA_INDEX_URL ${PIP_EXTRA_INDEX_URL}
ARG PIP_PACKAGE=.
RUN --mount=type=cache,target=/root/.cache/pip,sharing=locked \
  "${APPDIR}/${APPNAME}/bin/pip" install ${PIP_PACKAGE}

# build patchmatch
RUN python3 -c "from patchmatch import patch_match"

#####################
##  runtime image  ##
#####################
FROM python-base AS runtime

# setup environment
COPY --from=pyproject-builder ${APPDIR}/${APPNAME} ${APPDIR}/${APPNAME}
ENV INVOKEAI_ROOT=/data
ENV INVOKE_MODEL_RECONFIGURE="--yes --default_only"

# set Entrypoint and default CMD
ENTRYPOINT [ "invokeai" ]
CMD [ "--web", "--host=0.0.0.0" ]
VOLUME [ "/data" ]