initial commit

pg_backup_restore.sh

@@ -0,0 +1,76 @@
+#!/bin/env bash
+
+# Variables
+database_name=""
+backup_file=""
+
+# Function to display help text
+display_help() {
+    echo "Usage: $0 [-b|--backup] [-r|--restore] [-d|--database] [-h|--help] database_name"
+    echo "  -b or --backup: backup the specified database"
+    echo "  -r or --restore: restore the specified database"
+    echo "  -d or --database: specify the name of the database"
+    echo "  -h or --help: display this help text"
+}
+
+# Parse command line arguments
+while [[ $# -gt 0 ]]; do
+    key="$1"
+    case $key in
+        -b|--backup)
+            option="backup"
+            ;;
+        -r|--restore)
+            option="restore"
+            ;;
+        -d|--database)
+            database_name="$2"
+            shift # past argument
+            ;;
+        -h|--help)
+            display_help
+            exit 0
+            ;;
+        *)
+            echo "Invalid option: $1"
+            display_help
+            exit 1
+            ;;
+    esac
+    shift # past argument or value
+done
+
+# Validate input
+if [[ -z $database_name ]]; then
+    echo "Error: missing database name"
+    display_help
+    exit 1
+fi
+if [[ -z $option ]]; then
+    echo "Error: missing option"
+    display_help
+    exit 1
+fi
+
+# Backup or restore the database
+if [[ $option == "backup" ]]; then
+    timestamp=$(date +%Y-%m-%d-%H-%M-%S)
+    backup_file="$database_name-$timestamp.sql"
+    echo "Backing up $database_name to $backup_file..."
+    pg_dump $database_name > $backup_file
+    echo "Backup complete."
+elif [[ $option == "restore" ]]; then
+    backup_file="$database_name.sql"
+    echo "Deleting $database_name..."
+    dropdb $database_name
+    echo "Creating $database_name..."
+    createdb $database_name
+    echo "Restoring $database_name from $backup_file..."
+    psql $database_name < $backup_file
+    echo "Restore complete."
+else
+    echo "Error: invalid option"
+    display_help
+    exit 1
+fi
+