Add GitHub Actions workflow for building and pushing Docker images on tag

Author: MatInGit

.github/workflows/docker-build.yml

@@ -0,0 +1,63 @@
+name: Build & Push Docker Images on Tag
+on:
+  push:
+    tags:
+      - 'v*'  # Triggers only on tags like v0.1.2, v0.1.2.dev1, etc.
+  workflow_dispatch:
+
+jobs:
+  build-and-push:
+    runs-on: ubuntu-latest
+    env:
+      IMAGE_BASE: isisacceleratorcontrols/poly-lithic
+    steps:
+      - name: Checkout repository
+        uses: actions/checkout@v4
+
+      - name: Extract version from tag
+        id: vars
+        run: |
+          TAG_NAME=${GITHUB_REF#refs/tags/}
+          echo "VERSION=${TAG_NAME}" >> $GITHUB_ENV
+          echo "version=${TAG_NAME}" >> $GITHUB_OUTPUT
+
+      - name: Log in to Docker Hub
+        uses: docker/login-action@v3
+        with:
+          username: ${{ secrets.DOCKERHUB_USERNAME }}
+          password: ${{ secrets.DOCKERHUB_TOKEN }}
+
+      - name: Build Docker images with version injection
+        run: |
+          docker build --pull --rm --target vanilla \
+            --build-arg VERSION=${VERSION} \
+            -f Dockerfile -t $IMAGE_BASE:base-${VERSION} .
+
+          docker build --pull --rm --target torch \
+            --build-arg VERSION=${VERSION} \
+            -f Dockerfile -t $IMAGE_BASE:torch-${VERSION} .
+
+          docker build --pull --rm --target tensorflow \
+            --build-arg VERSION=${VERSION} \
+            -f Dockerfile -t $IMAGE_BASE:tensorflow-${VERSION} .
+
+      - name: Push versioned images
+        run: |
+          docker push $IMAGE_BASE:base-${VERSION}
+          docker push $IMAGE_BASE:torch-${VERSION}
+          docker push $IMAGE_BASE:tensorflow-${VERSION}
+
+      - name: Tag and push latest if version is pure semver
+        if: ${{ steps.vars.outputs.version =~ '^v[0-9]+\\.[0-9]+\\.[0-9]+$' }}
+        run: |
+          echo "Tag is a pure semver release — tagging latest"
+
+          docker tag $IMAGE_BASE:base-${VERSION} $IMAGE_BASE:base-latest
+          docker tag $IMAGE_BASE:torch-${VERSION} $IMAGE_BASE:torch-latest
+          docker tag $IMAGE_BASE:tensorflow-${VERSION} $IMAGE_BASE:tensorflow-latest
+          docker tag $IMAGE_BASE:base-latest $IMAGE_BASE:latest
+
+          docker push $IMAGE_BASE:base-latest
+          docker push $IMAGE_BASE:torch-latest
+          docker push $IMAGE_BASE:tensorflow-latest
+          docker push $IMAGE_BASE:latest

.github/workflows/main.yml

@@ -1,4 +1,4 @@
-name: CI Pipeline
+name: Main build and test workflow
 
 # only on tagged commits or manual trigger or on change to main.yml
 on:

Dockerfile

@@ -15,48 +15,70 @@
 # # CMD tail -f /dev/null
 # Stage 1: Base image with common dependencies
 
+# ===== Base stage =====
 FROM python:3.11.8-slim-bullseye AS base
+
+# Accept version as build argument
+ARG VERSION=""
+ENV POLY_LITHIC_VERSION=$VERSION
+
+LABEL org.opencontainers.image.version="${POLY_LITHIC_VERSION}"
+
+# System dependencies
 RUN apt-get update && apt-get install -y --no-install-recommends \
     git \
     iputils-ping \
     curl \
     && rm -rf /var/lib/apt/lists/*
 
+# Python requirements
 COPY requirements.txt /app/requirements.txt
 RUN python -m pip install -r /app/requirements.txt --no-cache-dir
 
+# Embed version into the container (optional)
+RUN echo "${POLY_LITHIC_VERSION}" > /opt/deployment/version.txt
 
+# ===== Torch stage =====
 FROM base AS torch
-# Install additional ML libraries
+
+# Additional ML dependencies
 RUN python -m pip install botorch pydantic torch==2.6.0 --no-cache-dir
-# need to copy poly_lithic, pyptoject.toml and tests
+
+# Copy code
 COPY poly_lithic /opt/poly_lithic
 COPY pyproject.toml /opt/deployment/pyproject.toml
 COPY tests /opt/deployment/tests
+
 WORKDIR /opt/deployment
+
 RUN python -m pip install poly-lithic --no-cache-dir
 
 CMD pl -c config.yaml -r -e env.json && pl --publish -c config.yaml -e env.json
 
+# ===== TensorFlow stage =====
 FROM base AS tensorflow
-# Install additional ML libraries
+
 RUN python -m pip install pydantic tensorflow --no-cache-dir
 
 COPY poly_lithic /opt/poly_lithic
 COPY pyproject.toml /opt/deployment/pyproject.toml
 COPY tests /opt/deployment/tests
+
 WORKDIR /opt/deployment
+
 RUN python -m pip install poly-lithic --no-cache-dir
 
 CMD pl -c config.yaml -r -e env.json && pl --publish -c config.yaml -e env.json
 
-
+# ===== Vanilla stage =====
 FROM base AS vanilla
 
 COPY poly_lithic /opt/poly_lithic
 COPY pyproject.toml /opt/deployment/pyproject.toml
 COPY tests /opt/deployment/tests
+
 WORKDIR /opt/deployment
+
 RUN python -m pip install poly-lithic --no-cache-dir
 
 CMD pl -c config.yaml -r -e env.json && pl --publish -c config.yaml -e env.json