Update Dockerfile

Dockerfile

@@ -2,102 +2,115 @@
 FROM lbjlaq/antigravity-manager:latest
 
 USER root
-
-# curl is the only thing we need for Supabase REST API
-RUN apt-get update && apt-get install -y --no-install-recommends curl && rm -rf /var/lib/apt/lists/*
+RUN apt-get update && apt-get install -y --no-install-recommends curl jq && rm -rf /var/lib/apt/lists/*
 
 ENV DATA_DIR=/root/.antigravity_tools
 ENV SYNC_INTERVAL=30
-
 EXPOSE 8045
 
-# Inline entrypoint — pull on start, push on change, push on shutdown
 COPY --chmod=755 <<'SCRIPT' /entrypoint.sh
 #!/bin/bash
 set -e
 
 DATA_DIR="${DATA_DIR:-/root/.antigravity_tools}"
 SYNC_INTERVAL="${SYNC_INTERVAL:-30}"
-SYNC_FILE="/tmp/antigravity_data.tar.gz"
 LAST_SYNC="/tmp/.last_sync"
 
-supabase_download() {
-  local code
-  code=$(curl -s -o "$SYNC_FILE" -w "%{http_code}" \
-    "${SUPABASE_URL}/storage/v1/object/${SUPABASE_BUCKET}/antigravity_data.tar.gz" \
+supabase_list() {
+  local prefix="$1" response files folders
+  response=$(curl -s -X POST \
+    "${SUPABASE_URL}/storage/v1/object/list/${SUPABASE_BUCKET}" \
     -H "Authorization: Bearer ${SUPABASE_KEY}" \
-    -H "Cache-Control: no-cache")
-  [ "$code" = "200" ]
+    -H "Content-Type: application/json" \
+    -d "{\"prefix\":\"${prefix}\",\"limit\":1000}" 2>/dev/null)
+  files=$(echo "$response" | jq -r '.[]? | select(.id != null) | .name' 2>/dev/null)
+  folders=$(echo "$response" | jq -r '.[]? | select(.id == null) | .name' 2>/dev/null)
+  for f in $files; do [ -n "$f" ] && echo "${prefix}${f}"; done
+  for d in $folders; do [ -n "$d" ] && supabase_list "${prefix}${d}/"; done
 }
 
-supabase_upload() {
-  tar czf "$SYNC_FILE" -C "$(dirname "$DATA_DIR")" "$(basename "$DATA_DIR")" 2>/dev/null || return 1
+supabase_pull() {
+  echo "[Sync] Pulling files from Supabase..."
+  supabase_list "" | while read -r fp; do
+    mkdir -p "$(dirname "$DATA_DIR/$fp")"
+    curl -s -o "$DATA_DIR/$fp" \
+      "${SUPABASE_URL}/storage/v1/object/${SUPABASE_BUCKET}/${fp}" \
+      -H "Authorization: Bearer ${SUPABASE_KEY}"
+    echo "[Sync]   <- $fp"
+  done
+  touch "$LAST_SYNC"
+  echo "[Sync] Pull complete!"
+}
+
+supabase_push_file() {
+  local rel="${1#$DATA_DIR/}"
   curl -s -X POST \
-    "${SUPABASE_URL}/storage/v1/object/${SUPABASE_BUCKET}/antigravity_data.tar.gz" \
+    "${SUPABASE_URL}/storage/v1/object/${SUPABASE_BUCKET}/${rel}" \
     -H "Authorization: Bearer ${SUPABASE_KEY}" \
-    -H "Content-Type: application/gzip" \
+    -H "Content-Type: application/octet-stream" \
     -H "x-upsert: true" \
-    --data-binary @"$SYNC_FILE" > /dev/null 2>&1
+    --data-binary "@${1}" > /dev/null 2>&1
+}
+
+supabase_push_all() {
+  echo "[Sync] Pushing all files..."
+  find "$DATA_DIR" -type f | while read -r f; do
+    supabase_push_file "$f"
+    echo "[Sync]   -> ${f#$DATA_DIR/}"
+  done
   touch "$LAST_SYNC"
-  echo "[Sync] Pushed to Supabase at $(date '+%H:%M:%S')"
+  echo "[Sync] Push complete!"
 }
 
-has_changes() {
-  [ ! -f "$LAST_SYNC" ] && return 0
-  [ -n "$(find "$DATA_DIR" -newer "$LAST_SYNC" -type f 2>/dev/null)" ]
+supabase_push_changed() {
+  local changed
+  changed=$(find "$DATA_DIR" -type f -newer "$LAST_SYNC" 2>/dev/null)
+  [ -z "$changed" ] && return 1
+  echo "$changed" | while read -r f; do
+    supabase_push_file "$f"
+    echo "[Sync]   -> ${f#$DATA_DIR/}"
+  done
+  touch "$LAST_SYNC"
+}
+
+supabase_has_data() {
+  local r
+  r=$(curl -s -X POST \
+    "${SUPABASE_URL}/storage/v1/object/list/${SUPABASE_BUCKET}" \
+    -H "Authorization: Bearer ${SUPABASE_KEY}" \
+    -H "Content-Type: application/json" -d '{"prefix":"","limit":1}' 2>/dev/null)
+  [ "$(echo "$r" | jq 'length' 2>/dev/null)" != "0" ]
 }
 
-# ── Step 1: Pull from Supabase (if data exists) ──
-if [ -n "$SUPABASE_URL" ] && [ -n "$SUPABASE_KEY" ] && [ -n "$SUPABASE_BUCKET" ]; then
-  echo "[Sync] Checking Supabase for existing data..."
-  if supabase_download; then
-    echo "[Sync] Found backup — restoring..."
-    mkdir -p "$DATA_DIR"
-    tar xzf "$SYNC_FILE" -C "$(dirname "$DATA_DIR")" 2>/dev/null
-    touch "$LAST_SYNC"
-    echo "[Sync] Restore complete!"
-  else
-    echo "[Sync] No backup found — first run."
-  fi
-  rm -f "$SYNC_FILE"
-else
-  echo "[Sync] No Supabase config — running without sync."
+SYNC=""
+[ -n "$SUPABASE_URL" ] && [ -n "$SUPABASE_KEY" ] && [ -n "$SUPABASE_BUCKET" ] && SYNC="1"
+
+if [ -n "$SYNC" ]; then
+  echo "[Sync] Checking Supabase..."
+  if supabase_has_data; then supabase_pull; else echo "[Sync] Empty bucket — first run."; fi
 fi
 
-# ── Step 2: Start the app ──
 /app/antigravity-tools --headless &
 APP_PID=$!
 
-# ── Step 3: First run — wait for the app to create the data dir, then push ──
-if [ -n "$SUPABASE_URL" ] && [ ! -f "$LAST_SYNC" ]; then
-  echo "[Sync] Waiting for data directory..."
+if [ -n "$SYNC" ] && [ ! -f "$LAST_SYNC" ]; then
+  echo "[Sync] Waiting for data dir..."
   while [ ! -d "$DATA_DIR" ]; do sleep 2; done
-  sleep 5
-  supabase_upload
+  sleep 5 && supabase_push_all
 fi
 
-# ── Step 4: Background sync — push on change every N seconds ──
-if [ -n "$SUPABASE_URL" ]; then
-  (
-    while kill -0 $APP_PID 2>/dev/null; do
-      sleep "$SYNC_INTERVAL"
-      has_changes && supabase_upload
-    done
-  ) &
+if [ -n "$SYNC" ]; then
+  ( while kill -0 $APP_PID 2>/dev/null; do sleep "$SYNC_INTERVAL"; supabase_push_changed || true; done ) &
   SYNC_PID=$!
 fi
 
-# ── Step 5: Final sync on shutdown ──
 cleanup() {
-  echo "[Sync] Shutting down — final push..."
-  [ -n "$SUPABASE_URL" ] && supabase_upload
-  kill $APP_PID 2>/dev/null
-  [ -n "$SYNC_PID" ] && kill $SYNC_PID 2>/dev/null
-  wait $APP_PID 2>/dev/null
-  exit 0
+  echo "[Sync] Shutting down..."
+  [ -n "$SYNC" ] && [ -d "$DATA_DIR" ] && supabase_push_all
+  kill $APP_PID 2>/dev/null; [ -n "$SYNC_PID" ] && kill $SYNC_PID 2>/dev/null
+  wait $APP_PID 2>/dev/null; exit 0
 }
 trap cleanup SIGTERM SIGINT
-
 wait $APP_PID
 SCRIPT