Add support for skip databases on save

Author: macropin

commands/common.sh

@@ -36,6 +36,7 @@ global_options() {
     echoerr "    -d|--dbname    database to connect to"
     echoerr "    -W|--password    password to connection to (not recommended. Use password-file)"
     echoerr "    --password-file    password file to read password from"
+    echoerr "    --skip-database    databases to skip"
 }
 
 parse_options() {
@@ -73,6 +74,16 @@ parse_options() {
         fi
     done
 
+    # Pull in environment variables prefixed with TOOLBOX_
+    # TOOLBOX_SKIP_DATABASES should be comma separated
+    for item in skip_databases; do
+        local varname
+        varname="TOOLBOX_${item^^}"
+        if [[ -n "${!varname:-}" ]]; then
+          eval ${item}="${!varname}"
+        fi
+    done
+
     # Options and long options
     local options="h:p:U:d:W:"
     local longopts="host:,port:,username:,dbname:,password:,password-file:,no-create-database,no-revoke-public-create,drop-database,format:,compression:,skip-globals,skip-database:,skip-analyze,full-count"
@@ -190,6 +201,11 @@ parse_options() {
         umask "${old_umask}"
         echoerr ">>> Password written to ${PGPASSFILE}"
     fi
+
+    # If skip_databases was set, split into an array
+    if [[ -n "${skip_databases:-}" ]]; then
+        IFS=',' read -r -a skip_databases <<< "${skip_databases}"
+    fi
 }
 
 echoerr() { echo "$@" 1>&2; }