DevOps-Bash-tools/psql_colorized.sh

#!/usr/bin/env bash
#  vim:ts=4:sts=4:sw=4:et
#
#  Author: Hari Sekhon
#  Date: 2020-10-20 12:21:38 +0100 (Tue, 20 Oct 2020)
#
#  https://github.com/HariSekhon/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="
Runs 'psql' with colorized output

Uses adjacent psql.sh, see there for more details

Tested on PostgreSQL 8.4, 9.x, 10.x, 11.x, 12.x, 13.0
"

# 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

GREEN="$(echo -e '\033[0;32m')"
# RED="$(echo -e '\033[1;31m')"
NOCOLOR="$(echo -e '\033[0m')"

#export LESS="-iMSx4 -FXR"
export LESS="-RFXig --tabs=4"

if type -P less &>/dev/null; then
    pager="less"
else
    pager="more"
fi

sed="sed"
if is_mac; then
    # doesn't work inside PAGER evaluation
    #sed(){
    #    command gsed "$@"
    #}
    sed="command gsed"
fi

                # can't colourize errors because they go to stderr and 2>&1 at start of pager breaks everything
                #s/\(ERROR:\)/$RED\1$NOCOLOR/g;
                #s/^\(\.psqlrc loaded\)/$GREEN\1$NOCOLOR/;
export PAGER="$sed '
                s/^\(([0-9]\+ rows*)\)/$GREEN\1$NOCOLOR/;
                s/^\(-\[\ RECORD\ [0-9]\+\ \][-+]\+\)/$GREEN\1$NOCOLOR/;
                s/|/$GREEN|$NOCOLOR/g;
                s/^\([-+]\+\)/$GREEN\1$NOCOLOR/
                ' 2>&1 |
              $pager"

"$srcdir/psql.sh" --pset pager=always "$@"