From fd9d558154eac3585310626f697b943d90e1bf7f Mon Sep 17 00:00:00 2001
From: Bobby <30593201+luciferreeves@users.noreply.github.com>
Date: Tue, 10 Mar 2026 23:50:58 +0530
Subject: feat: add support for LibSQL database driver and enhance seeding
 process

---
 scripts/entrypoint.sh | 21 +++++++++++++++++++
 scripts/seed.sh       | 56 +++++++++++++++++++++++++++++++++++++++++++--------
 2 files changed, 69 insertions(+), 8 deletions(-)
 create mode 100644 scripts/entrypoint.sh

(limited to 'scripts')

diff --git a/scripts/entrypoint.sh b/scripts/entrypoint.sh
new file mode 100644
index 0000000..3ff3886
--- /dev/null
+++ b/scripts/entrypoint.sh
@@ -0,0 +1,21 @@
+#!/bin/bash
+set -e
+
+./shrine &
+SHRINE_PID=$!
+
+if [ "$SEED" = "true" ]; then
+  RETRIES=0
+  MAX_RETRIES=30
+  until bash -c "echo > /dev/tcp/localhost/${PORT:-3000}" 2>/dev/null; do
+    RETRIES=$((RETRIES + 1))
+    if [ "$RETRIES" -ge "$MAX_RETRIES" ]; then
+      echo "Server failed to start after ${MAX_RETRIES}s"
+      exit 1
+    fi
+    sleep 1
+  done
+  bash scripts/seed.sh
+fi
+
+wait $SHRINE_PID
\ No newline at end of file
diff --git a/scripts/seed.sh b/scripts/seed.sh
index 4cded48..c6812d2 100755
--- a/scripts/seed.sh
+++ b/scripts/seed.sh
@@ -1,16 +1,56 @@
 #!/bin/bash
 set -euo pipefail
 
-DB_PATH="shrine/pagoda.db"
 SEED_DIR="seed"
+DB_DRIVER="${DB_DRIVER:-sqlite}"
+DSN="${DSN:-shrine/pagoda.db}"
 
-if [ ! -f "$DB_PATH" ]; then
-  echo "Database not found at $DB_PATH"
+if [ "$DB_DRIVER" = "libsql" ]; then
+  TURSO_URL=$(echo "$DSN" | sed 's|^libsql://|https://|; s|?.*||')
+  TURSO_TOKEN=$(echo "$DSN" | sed -n 's/.*authToken=\(.*\)/\1/p')
+fi
+
+exec_sql_file() {
+  if [ "$DB_DRIVER" = "sqlite" ]; then
+    sqlite3 "$DSN" < "$1"
+  elif [ "$DB_DRIVER" = "libsql" ]; then
+    local STMTS=""
+    while IFS= read -r line; do
+      line=$(echo "$line" | sed 's/^[[:space:]]*//;s/[[:space:]]*$//')
+      [ -z "$line" ] && continue
+      [ "$line" = "BEGIN TRANSACTION;" ] && continue
+      [ "$line" = "COMMIT;" ] && continue
+      local ESCAPED
+      ESCAPED=$(printf '%s' "$line" | sed 's/\\/\\\\/g; s/"/\\"/g')
+      STMTS="${STMTS}{\"type\":\"execute\",\"stmt\":{\"sql\":\"${ESCAPED}\"}},"
+    done < "$1"
+    STMTS="${STMTS%,}"
+    curl -sf "${TURSO_URL}/v2/pipeline" \
+      -H "Authorization: Bearer ${TURSO_TOKEN}" \
+      -H "Content-Type: application/json" \
+      -d "{\"requests\":[${STMTS},{\"type\":\"close\"}]}" > /dev/null
+  fi
+}
+
+query_sql() {
+  if [ "$DB_DRIVER" = "sqlite" ]; then
+    sqlite3 "$DSN" "$1"
+  elif [ "$DB_DRIVER" = "libsql" ]; then
+    curl -sf "${TURSO_URL}/v2/pipeline" \
+      -H "Authorization: Bearer ${TURSO_TOKEN}" \
+      -H "Content-Type: application/json" \
+      -d "{\"requests\":[{\"type\":\"execute\",\"stmt\":{\"sql\":\"$1\"}},{\"type\":\"close\"}]}" \
+      | sed -n 's/.*"value":"\([^"]*\)".*/\1/p' | head -1
+  fi
+}
+
+if [ "$DB_DRIVER" = "sqlite" ] && [ ! -f "$DSN" ]; then
+  echo "Database not found at $DSN"
   exit 1
 fi
 
-EXISTING=$(sqlite3 "$DB_PATH" "SELECT COUNT(*) FROM users;")
-if [ "$EXISTING" -gt 0 ]; then
+EXISTING=$(query_sql "SELECT COUNT(*) FROM users;")
+if [ -n "$EXISTING" ] && [ "$EXISTING" -gt 0 ] 2>/dev/null; then
   echo "Database already has $EXISTING users, skipping seed"
   exit 0
 fi
@@ -273,8 +313,8 @@ done
 
 echo "COMMIT;" >> "$SQL_FILE"
 
-echo "Inserting into database..."
-sqlite3 "$DB_PATH" < "$SQL_FILE"
+echo "Inserting into database ($DB_DRIVER)..."
+exec_sql_file "$SQL_FILE"
 
-TOTAL=$(sqlite3 "$DB_PATH" "SELECT COUNT(*) FROM users;")
+TOTAL=$(query_sql "SELECT COUNT(*) FROM users;")
 echo "Done. Total users: $TOTAL (admins: $ADMIN_COUNT, mods: $MOD_COUNT, banned: $BANNED_COUNT, disabled: $DISABLED_COUNT, unverified: $UNVERIFIED_COUNT)"
\ No newline at end of file
-- 
cgit v1.2.3