38 lines
1 KiB
Bash
Executable file
38 lines
1 KiB
Bash
Executable file
#!/bin/bash
|
|
set -e
|
|
|
|
# Check if database name is provided
|
|
if [ $# -eq 0 ]; then
|
|
echo "Usage: $0 <database-name>"
|
|
doctl databases list
|
|
exit 1
|
|
fi
|
|
|
|
DATABASE_NAME="$1"
|
|
DATABASE_ID=$(doctl databases list --format ID,Name --no-header | grep "$DATABASE_NAME" | awk '{print $1}')
|
|
|
|
if [ -z "$DATABASE_ID" ]; then
|
|
echo "Error: Database '$DATABASE_NAME' not found"
|
|
exit 1
|
|
fi
|
|
CURRENT_IP=$(curl -s https://api.ipify.org)
|
|
if [ -z "$CURRENT_IP" ]; then
|
|
echo "Error: Failed to get current IP address"
|
|
exit 1
|
|
fi
|
|
|
|
EXISTING_RULE=$(doctl databases firewalls list "$DATABASE_ID" | grep "ip_addr" | grep "$CURRENT_IP")
|
|
|
|
if [ -z "$EXISTING_RULE" ]; then
|
|
echo "IP not found in whitelist. Adding $CURRENT_IP to database firewall..."
|
|
doctl databases firewalls append "$DATABASE_ID" --rule ip_addr:"$CURRENT_IP"
|
|
fi
|
|
|
|
CONNECTION_URL=$(doctl databases connection "$DATABASE_ID" --format URI --no-header)
|
|
|
|
if [ -z "$CONNECTION_URL" ]; then
|
|
echo "Error: Failed to get database connection details"
|
|
exit 1
|
|
fi
|
|
|
|
psql "$CONNECTION_URL"
|