Add a script to connect to the database. (#32023)

Conrad Irwin created 6 months ago

This avoids needing passwords in plaintext on the command line....

Release Notes:

- N/A

Change summary

script/digital-ocean-db.sh | 38 ++++++++++++++++++++++++++++++++++++++
1 file changed, 38 insertions(+)

Detailed changes

script/digital-ocean-db.sh

@@ -0,0 +1,38 @@
+#!/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"