From 2d426ed39e9603535eda255f605ee6f4284e4961 Mon Sep 17 00:00:00 2001
From: Yann MAHE <yannmh@users.noreply.github.com>
Date: Thu, 13 Nov 2025 15:35:40 -0500
Subject: [PATCH] USE_PREBAKED_SETUP for niteshift-setup

---
 niteshift-setup.sh | 71 +++++++++++++++++++++++++++++++++-------------
 1 file changed, 51 insertions(+), 20 deletions(-)

diff --git a/niteshift-setup.sh b/niteshift-setup.sh
index 3dbf05b9..6c072afc 100755
--- a/niteshift-setup.sh
+++ b/niteshift-setup.sh
@@ -20,6 +20,21 @@ log_error() {
 # Pre-flight checks
 log "Starting niteshift setup for umami..."
 
+USE_PREBAKED_SETUP=0
+if [[ "${UMAMI_PREBAKED:-0}" == "1" ]]; then
+  if command -v umami-prebake-sync >/dev/null 2>&1; then
+    log "Prebaked Modal image detected, syncing cached artifacts..."
+    if umami-prebake-sync "$(pwd)"; then
+      USE_PREBAKED_SETUP=1
+      log "✓ Prebaked dependencies and artifacts restored"
+    else
+      log_error "Prebaked sync failed, falling back to full setup"
+    fi
+  else
+    log_error "UMAMI_PREBAKED=1 but umami-prebake-sync is missing; falling back to full setup"
+  fi
+fi
+
 # 1. Check DATABASE_URL is set
 if [ -z "${DATABASE_URL:-}" ]; then
   log_error "DATABASE_URL environment variable is not set"
@@ -54,12 +69,16 @@ fi
 log "✓ pnpm is available"
 
 # 4. Install dependencies
-log "Installing dependencies with pnpm..."
-if ! pnpm install --frozen-lockfile; then
-  log_error "Failed to install dependencies"
-  exit 1
+if [[ "$USE_PREBAKED_SETUP" -eq 0 ]]; then
+  log "Installing dependencies with pnpm..."
+  if ! pnpm install --frozen-lockfile; then
+    log_error "Failed to install dependencies"
+    exit 1
+  fi
+  log "✓ Dependencies installed"
+else
+  log "Skipping pnpm install (prebaked dependencies in place)"
 fi
-log "✓ Dependencies installed"
 
 # 5. Build only what's needed for dev (skip production Next.js build)
 # For dev mode we need:
@@ -76,12 +95,16 @@ if ! pnpm run check-env; then
 fi
 log "✓ Environment validated"
 
-log "Building database client..."
-if ! pnpm run build-db; then
-  log_error "Database client build failed"
-  exit 1
+if [[ "$USE_PREBAKED_SETUP" -eq 0 ]]; then
+  log "Building database client..."
+  if ! pnpm run build-db; then
+    log_error "Database client build failed"
+    exit 1
+  fi
+  log "✓ Database client built"
+else
+  log "Skipping build-db (prebaked Prisma client detected)"
 fi
-log "✓ Database client built"
 
 log "Checking database and applying migrations..."
 if ! pnpm run check-db; then
@@ -90,19 +113,27 @@ if ! pnpm run check-db; then
 fi
 log "✓ Database migrations applied"
 
-log "Building tracker script..."
-if ! pnpm run build-tracker; then
-  log_error "Tracker build failed"
-  exit 1
+if [[ "$USE_PREBAKED_SETUP" -eq 0 ]]; then
+  log "Building tracker script..."
+  if ! pnpm run build-tracker; then
+    log_error "Tracker build failed"
+    exit 1
+  fi
+  log "✓ Tracker script built"
+else
+  log "Skipping tracker build (prebaked bundle detected)"
 fi
-log "✓ Tracker script built"
 
-log "Building geo database..."
-if ! pnpm run build-geo; then
-  log_error "Geo build failed"
-  exit 1
+if [[ "$USE_PREBAKED_SETUP" -eq 0 ]]; then
+  log "Building geo database..."
+  if ! pnpm run build-geo; then
+    log_error "Geo build failed"
+    exit 1
+  fi
+  log "✓ Geo database built"
+else
+  log "Skipping geo build (prebaked GeoLite database detected)"
 fi
-log "✓ Geo database built"
 
 # 6. Start the dev server in the background
 log "Starting development server on port 3001 (with hot reload)..."