From 9558782532bb2377874e9e1d99ef605bee0b58d9 Mon Sep 17 00:00:00 2001
From: Davis Broda <15205060+DavisBroda@users.noreply.github.com>
Date: Mon, 12 Aug 2024 13:23:10 -0400
Subject: [PATCH] Adding push to latest to dockerize script

Signed-off-by: Davis Broda <15205060+DavisBroda@users.noreply.github.com>
---
 bin/dockerize.sh          | 23 ++++++++++++++++++++++-
 src/loader/output_step.py |  1 +
 2 files changed, 23 insertions(+), 1 deletion(-)

diff --git a/bin/dockerize.sh b/bin/dockerize.sh
index eb6a1ba..e83c138 100755
--- a/bin/dockerize.sh
+++ b/bin/dockerize.sh
@@ -32,6 +32,15 @@ while [[ $# -gt 0 ]]; do
       esac
       shift
       ;;
+    --latest)
+      LATEST="true"
+      shift
+      ;;
+    --version)
+      shift; #move past arg
+      VERSION="$1"
+      shift
+      ;;
     *) # ignore other args
       shift
       ;;
@@ -44,10 +53,11 @@ function showHelp {
     echo " "
     echo "Usage:"
     echo " "
-    echo "    docker.sh [--publish registry]"
+    echo "    docker.sh [--publish registry] [--latest] [--version ver]"
     echo " "
     echo "    --publish argument controls whether the docker image will be published to dockerhub."
     echo "        This argument defaults to false. Must be set to one of [false, dockerhub, custom]"
+    echo "        The DOCKER_REGISTRY environment variable "
     echo " "
     echo "    The DOCKER_USERNAME environment variable must exist to run this script, and "
     echo "        should be set to the user's docker username. "
@@ -159,6 +169,17 @@ publish() {
   else
     echo "Attempt to publish docker image failed!"
   fi
+
+  if [[ "$LATEST" == "true" ]]
+    echo "Pushing image to Docker as latest"
+    docker push "$IMAGE_NAME:latest"
+    DOCKER_RETURN=$?
+
+    if [[ $DOCKER_RETURN -eq 0 ]]; then
+      echo "Image successfully published as latest!"
+    else
+      echo "Attempt to publish docker image as latest failed!"
+    fi
 }
 
 echo "Changing directory to sandbox directory ($WORKING_DIR)"
diff --git a/src/loader/output_step.py b/src/loader/output_step.py
index 29167cc..0785614 100644
--- a/src/loader/output_step.py
+++ b/src/loader/output_step.py
@@ -64,6 +64,7 @@ class LocalDuckdbOutputStep(OutputStep):
 
     def __init__(self, conf_dict: Dict[str, str]):
         self.conf = LocalDuckdbOutputStepConf(**conf_dict)
+        self._vaidate_conf(self.conf)
 
     def _vaidate_conf(self, conf: LocalDuckdbOutputStepConf):
         if conf.database_dir is None: