You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
93 lines
2.7 KiB
Bash
93 lines
2.7 KiB
Bash
#!/usr/bin/env bash
|
|
# vim:ts=4:sts=4:sw=4:et
|
|
#
|
|
# Author: Hari Sekhon
|
|
# Date: 2020-03-12 16:15:38 +0000 (Thu, 12 Mar 2020)
|
|
#
|
|
# https://github.com/HariSekhon/DevOps-Bash-tools
|
|
#
|
|
# License: see accompanying Hari Sekhon LICENSE file
|
|
#
|
|
# If you're using my code you're welcome to connect with me on LinkedIn and optionally send me feedback to help steer this or other code I publish
|
|
#
|
|
# https://www.linkedin.com/in/HariSekhon
|
|
#
|
|
|
|
set -euo pipefail
|
|
[ -n "${DEBUG:-}" ] && set -x
|
|
srcdir="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
|
|
|
# shellcheck disable=SC1090
|
|
. "$srcdir/lib/utils.sh"
|
|
|
|
# shellcheck disable=SC2034,SC2154
|
|
usage_description="
|
|
Script to more easily connect to PostgreSQL without having to repeatedly specify options like host, username and password
|
|
|
|
Leverages standard PostgresSQL options as well as others likely to be found in the environment
|
|
|
|
https://www.postgresql.org/docs/9.0/libpq-envars.html
|
|
|
|
See also - GNU sql
|
|
|
|
Tested on PostgreSQL 8.4, 9.x, 10.x, 11.x, 12.x, 13.0
|
|
AWS RDS PostgreSQL 9.5.15
|
|
"
|
|
|
|
# used by usage() in lib/utils.sh
|
|
# shellcheck disable=SC2034
|
|
usage_args="[<psql_options>]"
|
|
|
|
# would catch -h but this is a legit option we should let through
|
|
#help_usage "$@"
|
|
|
|
#min_args 1 "$@"
|
|
|
|
for arg; do
|
|
case "$arg" in
|
|
--help) usage
|
|
;;
|
|
esac
|
|
done
|
|
|
|
opts="${POSTGRES_OPTS:-}"
|
|
|
|
POSTGRES_HOST="${PGHOST:-${POSTGRESQL_HOST:-${POSTGRES_HOST:-${HOST:-}}}}"
|
|
if [ -n "${POSTGRES_HOST:-}" ]; then
|
|
# more intuitive to see in the process list what is going on
|
|
#export PGHOST="$POSTGRES_HOST"
|
|
opts="$opts -h $POSTGRES_HOST"
|
|
fi
|
|
|
|
POSTGRES_PORT="${PGPORT:-${POSTGRESQL_PORT:-${POSTGRES_PORT:-${PORT:-}}}}"
|
|
if [ -n "${POSTGRES_PORT:-}" ]; then
|
|
# more intuitive to see in the process list what is going on
|
|
#export PGPORT="$POSTGRES_PORT"
|
|
opts="$opts -p $POSTGRES_PORT"
|
|
fi
|
|
|
|
POSTGRES_USER="${PGUSER:-${POSTGRESQL_USER:-${POSTGRES_USER:-${USER:-}}}}"
|
|
if [ -n "${POSTGRES_USER:-}" ]; then
|
|
# more intuitive to see in the process list what is going on
|
|
#export PGUSER="$POSTGRES_USER"
|
|
opts="$opts -U $POSTGRES_USER"
|
|
fi
|
|
|
|
POSTGRES_PASSWORD="${PGPASSWORD:-${POSTGRESQL_PASSWORD:-${POSTGRES_PASSWORD:-${PASSWORD:-}}}}"
|
|
if [ -n "${POSTGRES_PASSWORD:-}" ]; then
|
|
# can't do this and wouldn't want to as it'd expose it in the password list
|
|
#opts="$opts -U $POSTGRES_PASSWORD"
|
|
export PGPASSWORD="$POSTGRES_PASSWORD"
|
|
fi
|
|
|
|
POSTGRES_DATABASE="${PGDATABASE:-${POSTGRESQL_DATABASE:-${POSTGRES_DATABASE:-${DATABASE:-}}}}"
|
|
if [ -n "${POSTGRES_DATABASE:-}" ]; then
|
|
# more intuitive to see in the process list what is going on
|
|
#export PGDATABASE="$POSTGRES_DATABASE"
|
|
opts="$opts -d $POSTGRES_DATABASE"
|
|
fi
|
|
|
|
# split opts
|
|
# shellcheck disable=SC2086
|
|
exec psql $opts "$@"
|