60 lines
1.7 KiB
Bash
60 lines
1.7 KiB
Bash
#!/usr/bin/env bash
|
||
set -e
|
||
|
||
# 1) Wait for MySQL
|
||
DB_HOST="${DB_HOST:-${MYSQL_HOST:-db}}"
|
||
DB_PORT="${DB_PORT:-${MYSQL_PORT:-3306}}"
|
||
echo "[⏳] Waiting for database at $DB_HOST:$DB_PORT..."
|
||
until nc -z "$DB_HOST" "$DB_PORT"; do sleep 1; done
|
||
echo "[✔] Database is up"
|
||
|
||
# 2) Ensure uploads dir exists/writable
|
||
UPLOAD_DIR="/app/${UPLOAD_FOLDER:-data/uploads}"
|
||
mkdir -p "$UPLOAD_DIR"
|
||
chown -R appuser:appuser "$UPLOAD_DIR"
|
||
chmod -R u+rwX,g+rwX,o+rX "$UPLOAD_DIR"
|
||
echo "⏺️ Upload directory ready at $UPLOAD_DIR"
|
||
|
||
# 3) If launching Flask, handle schema + migrations + seeding
|
||
if [ "$1" = "flask" ]; then
|
||
# A) Create any truly missing tables (so FKs won’t break)
|
||
echo "[🛠️] Ensuring tables via create_all()"
|
||
python <<EOF
|
||
from app import create_app, db
|
||
app = create_app()
|
||
with app.app_context():
|
||
db.create_all()
|
||
EOF
|
||
|
||
# B) If this is the very first run (no revisions), stamp head
|
||
VERSIONS_DIR="/app/migrations/versions"
|
||
if [ ! -d "$VERSIONS_DIR" ] || [ -z "$(ls -A "$VERSIONS_DIR")" ]; then
|
||
mkdir -p "$VERSIONS_DIR"
|
||
echo "[🔖] No Alembic revisions found—stamping head"
|
||
flask db stamp head
|
||
fi
|
||
|
||
# C) Always try to auto-generate a new migration (if models changed)
|
||
echo "[🛠️] Auto-generating a new migration (if needed)"
|
||
flask db migrate --message "Auto migration on $(date +%Y-%m-%d_%H-%M-%S)" || true
|
||
|
||
# D) Apply all pending migrations
|
||
echo "[🛠️] Applying migrations"
|
||
flask db upgrade
|
||
|
||
# E) Seed if configured
|
||
if [ "${ENABLE_DB_SEEDING,,}" = "true" ] || [ "${ENABLE_DB_SEEDING}" = "1" ]; then
|
||
echo "[🌱] Seeding demo data"
|
||
flask preload-data
|
||
fi
|
||
|
||
echo "[🚀] Ready to start Flask"
|
||
fi
|
||
|
||
# 4) Drop privileges if root; then exec
|
||
if [ "$(id -u)" = "0" ]; then
|
||
exec gosu appuser "$@"
|
||
else
|
||
exec "$@"
|
||
fi
|