Dockerfile

# =============================================================================
# Test environment.
#
# Example:
#   > docker build -t curryer_test . --target test
#   > docker run -a stdout -a stderr --rm \
#       -e SPACETRACK_USER=$SPACETRACK_USER -e SPACETRACK_PSWD=$SPACETRACK_PSWD \
#       curryer_test
# -----------------------------------------------------------------------------
# Python version with which to test (must be supported and available on dockerhub)
ARG BASE_IMAGE_PYTHON_VERSION
FROM python:${BASE_IMAGE_PYTHON_VERSION:-3.11}-slim-bookworm AS test
# TARGETPLATFORM is passed automatically by docker buildx or manually via --platform
ARG TARGETPLATFORM

USER root

ENV BASE_DIR=/app/curryer
ENV DATA_DIR=$BASE_DIR/data
WORKDIR $BASE_DIR

# Create virtual environment and permanently activate it for this image
# This adds not only the venv python executable but also all installed entrypoints to the PATH
# Upgrade pip to the latest version because poetry uses pip in the background to install packages
ENV VIRTUAL_ENV=/opt/venv
RUN python -m venv $VIRTUAL_ENV
ENV PATH="$VIRTUAL_ENV/bin:$PATH"
RUN pip install --upgrade pip

# Update the OS tools and install dependencies.
RUN apt-get update
RUN apt-get install -y curl wget libgdal-dev build-essential gcc g++ libc6-dev

# Install SPICE toolkit via conda for AMD or ARM architectures
ENV CONDA_DIR=/root/.local/conda
RUN <<EOF
set -ex
if [ "$TARGETPLATFORM" = "linux/arm64" ] || [ "$TARGETPLATFORM" = "linux/aarch64" ]; then ARCH="aarch64"; else ARCH="x86_64"; fi
CONDA_INSTALL_SCRIPT=Miniconda3-latest-Linux-$ARCH.sh
wget -q https://repo.anaconda.com/miniconda/$CONDA_INSTALL_SCRIPT
bash $CONDA_INSTALL_SCRIPT -b -p $CONDA_DIR
rm $CONDA_INSTALL_SCRIPT
export PATH="$CONDA_DIR/bin:$PATH"
conda tos accept
conda config --add channels conda-forge
conda config --set channel_priority strict
conda install -y cspice
echo "export CONDA_DIR=$CONDA_DIR" >> /root/.bashrc
echo ". $CONDA_DIR/etc/profile.d/conda.sh" >> /root/.bashrc
EOF

# Add Conda to path *after* venv so that venv python is used
ENV PATH="$PATH:$CONDA_DIR/bin"

# Install poetry and add to path.
RUN curl -sSL https://install.python-poetry.org | python -
ENV PATH="$PATH:/root/.local/bin"

# Copy data files.
COPY data $DATA_DIR

# Copy library files.
# TODO: First copy depdencies and install python before the whole lib? Makes things faster locally.
COPY bin $BASE_DIR/bin
COPY curryer $BASE_DIR/curryer
COPY tests $BASE_DIR/tests
COPY README.md $BASE_DIR
COPY pyproject.toml $BASE_DIR

# Ensure pip is upgraded
RUN pip install --upgrade pip

# Install all dependencies (including dev deps) specified in pyproject.toml
RUN poetry install --all-extras

# Download third-party data.
ENV CURRYER_DATA_DIR=$DATA_DIR

RUN PROJ_DOWNLOAD_DIR=$(python -c "import pyproj; print(pyproj.datadir.get_user_data_dir())") \
    && mkdir -p ${PROJ_DOWNLOAD_DIR} \
    && curl https://cdn.proj.org/us_nga_egm96_15.tif --output ${PROJ_DOWNLOAD_DIR}/us_nga_egm96_15.tif

# Create entrypoint script to allow passing additional pytest arguments
RUN echo '#!/bin/bash\nexec pytest -v --junitxml=junit.xml --cov --cov-branch --cov-report=xml:coverage.xml --disable-warnings tests "$@"' > /entrypoint.sh \
    && chmod +x /entrypoint.sh

ENTRYPOINT ["/entrypoint.sh"]

# =============================================================================
# Debug environment.
#
# Example:
#   > docker build -t curryer_debug . --target debug
#   > docker run --security-opt seccomp=unconfined --privileged -it --rm curryer_debug
#
#   > docker run -v .:/workspace -p 8889:8889 --security-opt seccomp=unconfined --privileged -it --rm curryer_debug
#   > cd /workspace ; export JUPYTER_CONFIG_DIR=/workspace/notebooks
#   > jupyter lab . --allow-root --ip 0.0.0.0 --port 8889 --no-browser
# -----------------------------------------------------------------------------
FROM test AS debug

# Command line tools and debug dependencies.
RUN apt-get install -y man less vim which tree

RUN pip install matplotlib jupyterlab cartopy basemap basemap-data-hires

CMD []
ENTRYPOINT ["/bin/bash"]

# =============================================================================
# L1A Geolocation Integration Test Case
#
# Example:
#   > docker build -t curryer_clarreo_l1a . --target demo
#   > docker run -a stdout -a stderr --rm curryer_clarreo_l1a
#       or to override with a different input file:
#   > docker run -a stdout -a stderr --rm -v `pwd`:/app/curryer/tests/data/demo curryer_clarreo_l1a
# -----------------------------------------------------------------------------
FROM test AS demo

SHELL ["/usr/bin/bash"]
WORKDIR $BASE_DIR

CMD ["tests/sit4_l1a_geolocation.py", "--demo", "--output_dir", "tests/data/demo"]
ENTRYPOINT ["python3"]