From 715e133be287ec3490e81529aa4fe963c12fde53 Mon Sep 17 00:00:00 2001
From: Jonas-Ian Kuche <jonas-ian.kuche@mni.thm.de>
Date: Wed, 14 Aug 2024 15:50:41 +0200
Subject: [PATCH] feat(full): add full image

---
 .github/workflows/ci.yml |  2 +-
 Earthfile                | 21 +++++++++++++++++++--
 2 files changed, 20 insertions(+), 3 deletions(-)

diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index 51a26e7..6045242 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -12,7 +12,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        image: [all-python-cpu, all-python-cuda, all-r, all-rust]
+        image: [all-python-cpu, all-python-cuda, all-r, all-rust, full-jupyter]
     env:
       FORCE_COLOR: 1
     steps:
diff --git a/Earthfile b/Earthfile
index 6bcbb03..1bea9bb 100644
--- a/Earthfile
+++ b/Earthfile
@@ -12,9 +12,13 @@ ARG --required --global VERSION
 common:
   FROM mambaorg/micromamba:1.5.8
   ENV SHELL=/bin/bash
-  RUN micromamba install -n base -c conda-forge -y htop nvtop curl wget unzip
+  RUN micromamba install -n base -c conda-forge -y htop curl wget unzip
   WORKDIR /workspace
 
+common-cuda:
+  FROM +common
+  RUN micromamba install -n base -c conda-forge -y nvtop
+
 build:
   ARG language
   ARG frontend
@@ -42,6 +46,19 @@ all-python-cpu:
 all-python-cuda:
   BUILD +build-all-frontends --language=python-cuda
 
+full:
+  FROM +common
+  DO python+SETUP_CPU
+  DO r+SETUP_CPU
+  DO rust+SETUP_CPU
+
+full-jupyter:
+  FROM +full
+  DO jupyter+SETUP
+  DO r+JUPYTER_POST_INSTALL
+  DO rust+JUPYTER_POST_INSTALL
+  SAVE IMAGE --push $REGISTRY/full-jupyter:$VERSION
+
 python-cpu:
   FROM +common
   DO python+SETUP_CPU
@@ -57,7 +74,7 @@ python-cpu-code:
   SAVE IMAGE --push $REGISTRY/python-code:$VERSION
 
 python-cuda:
-  FROM +common
+  FROM +common-cuda
   DO python+SETUP_CUDA
 
 python-cuda-jupyter: