From e278eb16254b962360f4ad44023d59615840f659 Mon Sep 17 00:00:00 2001
From: Cian Hughes <chughes000@gmail.com>
Date: Tue, 21 Jan 2025 13:42:14 +0000
Subject: [PATCH] Added justfile utility functions

---
 .gitignore |  2 ++
 justfile   | 67 ++++++++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 69 insertions(+)

diff --git a/.gitignore b/.gitignore
index d0b127d..2910b4e 100644
--- a/.gitignore
+++ b/.gitignore
@@ -42,3 +42,5 @@ yarn-debug.log*
 yarn-error.log*
 
 .svelte-kit
+
+versions
diff --git a/justfile b/justfile
index 2ae1075..d67e9f9 100644
--- a/justfile
+++ b/justfile
@@ -1,6 +1,8 @@
 update:
+    just tag-version
     docker compose down
     git pull
+    docker compose pull
     docker compose build --no-cache
     docker compose up -d
 
@@ -9,3 +11,68 @@ merge_and_push_prod:
     git merge main
     git switch main
     git push --all
+
+remote-update:
+    #!/usr/bin/env bash
+    if ssh server "[ ! -f /tmp/update.lock ]"; then
+        ssh server "touch /tmp/update.lock"
+        just merge_and_push_prod
+        ssh server "cd am-d-model.eu && just update"
+        ssh server "rm /tmp/update.lock"
+        ssh server "just healthcheck || just rollback"
+    else
+        echo "Update already in progress"
+    fi
+
+tag-version:
+    git tag backup-$(date +%Y%m%d-%H%M%S)
+    mkdir -p versions
+    docker compose images | grep -v "REPOSITORY" > versions/$(date +%Y%m%d-%H%M%S).txt
+
+healthcheck:
+    #!/usr/bin/env bash
+    for i in 1 2 3; do
+        curl -f http://localhost/health && exit 0
+        sleep 5
+    done
+    exit 1
+
+rollback version="":
+    #!/usr/bin/env bash
+    current_branch=$(git branch --show-current)
+    if [ -z "{{version}}" ]; then # Get last backup tag if no version specified
+        version=$(git tag | grep backup | sort -r | head -n1)
+    else # Or get the specified version
+        version="backup-{{version}}"
+    fi
+
+    # Find corresponding version file
+    version_date=${version#backup-}  # Remove 'backup-' prefix
+    version_file="versions/$version_date.txt"
+
+    if [ ! -f "$version_file" ]; then
+        echo "No version file found for $version"
+        exit 1
+    fi
+
+    # Read the old image versions and pull them specifically
+    while read -r repo tag image_id _; do
+        if [ ! -z "$repo" ]; then
+            docker pull "$repo:$tag"
+        fi
+    done < "$version_file"
+
+    # Then check out the version and update
+    git checkout $version
+    docker compose down
+    docker compose build --no-cache
+    docker compose up -d
+    git switch $current_branch  # Return to original branch
+
+cleanup-versions:
+    #!/usr/bin/env bash
+    # Keep last 5 backup tags
+    for tag in $(git tag | grep backup | sort -r | tail -n +6); do
+        git tag -d $tag
+        rm -f "versions/${tag#backup-}.txt"
+    done