FROM docker.io/python:3.10.14-slim-bullseye AS base

RUN pip install --upgrade --no-cache-dir pip

# install pytest

RUN pip install --no-cache-dir pytest

RUN apt -y update \

&& apt install -y clamav-daemon \

&& apt-get clean \

&& rm -rf /var/lib/apt/lists/* \

&& useradd -ms /bin/bash dpk

USER dpk

WORKDIR /home/dpk

FROM docker.io/clamav/clamav:latest AS clamav

FROM base AS clamav-local

USER root

COPY --chown=dpk:root --from=clamav /var/lib/clamav/ /var/lib/clamav/

COPY --chown=dpk:root clamd.conf /etc/clamav/clamd.conf

RUN freshclam \

&& mkdir -p /var/run/clamav \

&& chown -R dpk:root /var/run/clamav /var/log/clamav /var/lib/clamav

CMD ["/bin/bash", "-c", "clamd --debug --foreground"]

USER dpk

FROM base AS malware

COPY --chown=dpk:root --from=clamav-local /var/lib/clamav/ /var/lib/clamav/

COPY --chown=dpk:root --from=clamav-local /etc/clamav/clamd.conf /etc/clamav/clamd.conf

COPY --chown=dpk:root --from=clamav-local /var/log/clamav/clamav.log /var/log/clamav/clamav.log

COPY --chown=dpk:root --from=clamav-local /var/run/clamav /var/run/clamav

# Copy and install data processing libraries

# These are expected to be placed in the docker context before this is run (see the make image).

COPY --chown=dpk:root data-processing-lib-python/ data-processing-lib-python/

RUN cd data-processing-lib-python && pip install --no-cache-dir -e .

COPY --chown=dpk:root src/ src/

COPY --chown=dpk:root pyproject.toml pyproject.toml

RUN pip install --no-cache-dir -e .

# copy source data

COPY --chown=dpk:root ./src/malware_transform.py .

COPY --chown=dpk:root ./src/malware_transform_python.py .

COPY --chown=dpk:root ./src/malware_local.py local/

COPY test/ test/

COPY test-data/ test-data/

ENV PYTHONPATH /home/dpk

# Put these at the end since they seem to upset the docker cache.

ARG BUILD_DATE

ARG GIT_COMMIT

LABEL build-date=$BUILD_DATE

LABEL git-commit=$GIT_COMMIT