#!/bin/bashset -e# Check if database name is providedif [ $# -eq 0 ]; then echo "Usage: $0 <database-name>" doctl databases list exit 1fiDATABASE_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 1fiCURRENT_IP=$(curl -s https://api.ipify.org)if [ -z "$CURRENT_IP" ]; then echo "Error: Failed to get current IP address" exit 1fiEXISTING_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"fiCONNECTION_URL=$(doctl databases connection "$DATABASE_ID" --format URI --no-header)if [ -z "$CONNECTION_URL" ]; then echo "Error: Failed to get database connection details" exit 1fipsql "$CONNECTION_URL"
#!/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"