From 137cde5fc98ec39497e85f335e2f636cf9a5e5ea Mon Sep 17 00:00:00 2001
From: Maxime Desroches <desroches.maxime@gmail.com>
Date: Wed, 4 Dec 2024 21:55:54 -0800
Subject: [PATCH] check

---
 selfdrive/test/docker_build.sh  | 9 +++++++++
 selfdrive/test/docker_common.sh | 1 +
 2 files changed, 10 insertions(+)

diff --git a/selfdrive/test/docker_build.sh b/selfdrive/test/docker_build.sh
index 4d58a1507c..4a7d901d0a 100755
--- a/selfdrive/test/docker_build.sh
+++ b/selfdrive/test/docker_build.sh
@@ -19,6 +19,15 @@ source $SCRIPT_DIR/docker_common.sh $1 "$TAG_SUFFIX"
 
 DOCKER_BUILDKIT=1 docker buildx build --provenance false --pull --platform $PLATFORM --load --cache-to type=inline --cache-from type=registry,ref=$REMOTE_TAG -t $DOCKER_IMAGE:latest -t $REMOTE_TAG -t $LOCAL_TAG -f $OPENPILOT_DIR/$DOCKER_FILE $OPENPILOT_DIR
 
+if [ -n "$DOCKER_MAX_SIZE" ]; then
+  SIZE="$(docker inspect $DOCKER_IMAGE:latest --format='{{.Size}}')"
+  echo "$DOCKER_IMAGE size: $SIZE bytes"
+  if [ $(echo "$SIZE > $DOCKER_MAX_SIZE" | bc) -eq 1 ]; then
+    echo "Image too large, max is $DOCKER_MAX_SIZE bytes"
+    exit 1
+  fi
+fi
+
 if [ -n "$PUSH_IMAGE" ]; then
   docker push $REMOTE_TAG
   docker tag $REMOTE_TAG $REMOTE_SHA_TAG
diff --git a/selfdrive/test/docker_common.sh b/selfdrive/test/docker_common.sh
index 2887fff74b..cef507b391 100644
--- a/selfdrive/test/docker_common.sh
+++ b/selfdrive/test/docker_common.sh
@@ -1,6 +1,7 @@
 if [ "$1" = "base" ]; then
   export DOCKER_IMAGE=openpilot-base
   export DOCKER_FILE=Dockerfile.openpilot_base
+  export DOCKER_MAX_SIZE="7.1*10^9"
 elif [ "$1" = "prebuilt" ]; then
   export DOCKER_IMAGE=openpilot-prebuilt
   export DOCKER_FILE=Dockerfile.openpilot