mirror of
https://github.com/Skyvern-AI/skyvern.git
synced 2026-04-28 03:30:10 +00:00
83 lines
2.9 KiB
Bash
Executable file
83 lines
2.9 KiB
Bash
Executable file
#!/bin/bash
|
|
|
|
set -e
|
|
|
|
# Set ALLOWED_SKIP_DB_MIGRATION_VERSION env var to the DB version you want to allow (select * from alembic_version)
|
|
# If current DB matches this version, migrations will be skipped. Use at your own risk.
|
|
ALLOWED_SKIP_DB_MIGRATION_VERSION=${ALLOWED_SKIP_DB_MIGRATION_VERSION:-}
|
|
|
|
# Run migrations by default
|
|
run_migration=true
|
|
|
|
if [ -n "$ALLOWED_SKIP_DB_MIGRATION_VERSION" ]; then
|
|
current_version=$(alembic current 2>&1 | grep -Eo "[0-9a-f]{12,}" | tail -n 1 || echo "")
|
|
echo "Current DB version: $current_version"
|
|
|
|
if [ "$current_version" = "$ALLOWED_SKIP_DB_MIGRATION_VERSION" ]; then
|
|
echo "⚠️ WARNING: Skipping database migrations"
|
|
echo "⚠️ DB is at version $current_version which matches ALLOWED_SKIP_DB_MIGRATION_VERSION"
|
|
echo "⚠️ Running older code against newer database schema"
|
|
echo "⚠️ Beware of compatibility risks!"
|
|
run_migration=false
|
|
else
|
|
echo "Current DB version ($current_version) does not match ALLOWED_SKIP_DB_MIGRATION_VERSION ($ALLOWED_SKIP_DB_MIGRATION_VERSION)"
|
|
fi
|
|
fi
|
|
|
|
if [ "$run_migration" = true ]; then
|
|
echo "Running database migrations..."
|
|
alembic upgrade head
|
|
alembic check
|
|
fi
|
|
|
|
if [ ! -f ".streamlit/secrets.toml" ]; then
|
|
echo "Creating organization and API token..."
|
|
org_output=$(python scripts/create_organization.py Skyvern-Open-Source)
|
|
api_token=$(echo "$org_output" | awk '/token=/{gsub(/.*token='\''|'\''.*/, ""); print}')
|
|
# Update the secrets-open-source.toml file
|
|
echo -e "[skyvern]\nconfigs = [\n {\"env\" = \"local\", \"host\" = \"http://skyvern:8000/api/v1\", \"orgs\" = [{name=\"Skyvern\", cred=\"$api_token\"}]}\n]" > .streamlit/secrets.toml
|
|
echo ".streamlit/secrets.toml file updated with organization details."
|
|
fi
|
|
|
|
_kill_xvfb_on_term() {
|
|
kill -TERM $xvfb
|
|
}
|
|
|
|
# Setup a trap to catch SIGTERM and relay it to child processes
|
|
trap _kill_xvfb_on_term TERM
|
|
|
|
echo "Starting Xvfb..."
|
|
# delete the lock file if any
|
|
rm -f /tmp/.X99-lock
|
|
# Set display environment variable
|
|
export DISPLAY=:99
|
|
# Start Xvfb
|
|
Xvfb :99 -screen 0 1920x1080x16 &
|
|
xvfb=$!
|
|
|
|
DISPLAY=:99 xterm 2>/dev/null &
|
|
|
|
# Wait for Xvfb to be ready before starting x11vnc
|
|
for i in $(seq 1 10); do
|
|
xdpyinfo -display :99 >/dev/null 2>&1 && break
|
|
echo "Waiting for Xvfb to start (attempt $i/10)..."
|
|
sleep 1
|
|
done
|
|
if ! xdpyinfo -display :99 >/dev/null 2>&1; then
|
|
echo "ERROR: Xvfb failed to start on display :99 after 10 attempts"
|
|
exit 1
|
|
fi
|
|
|
|
echo "Starting x11vnc on display :99..."
|
|
# VNC runs without a password (-nopw) because port 5900 is not exposed outside
|
|
# the container. Browser streaming reaches users via websockify on port 6080.
|
|
mkdir -p /data/log
|
|
x11vnc -display :99 -forever -nopw -shared -rfbport 5900 -bg -o /dev/null 2>/data/log/x11vnc.err
|
|
|
|
echo "Starting websockify on port 6080 -> localhost:5900..."
|
|
websockify 6080 localhost:5900 --daemon
|
|
|
|
python run_streaming.py > /dev/null &
|
|
|
|
# Run the command and pass in all three arguments
|
|
python -m skyvern.forge
|