psql.sh

#!/bin/bash -e
# @installable
# queries the local database
X=$(dirname `readlink -f ${BASH_SOURCE[0]}`)
source $X/env
[[ -f $LOCAL_ENV ]] && source $LOCAL_ENV 
source $X/log.sh
source $X/prop.sh

MYSELF() { readlink -f "${BASH_SOURCE[0]}"; }
MYDIR() { echo "$(dirname $(MYSELF))"; }
MYNAME() { echo "$(basename $(MYSELF))"; }
CALLER=$(basename `readlink -f $0`)

separator="|"

# TODO support for different ports and hosts
connection="psql -U $DB_USER $DB_NAME"
ops='qAtX'

if [[ $# -lt 1 ]]; then
    $connection
fi

query="$1"; shift
if [[ ! -n "$query" ]]; then
    err "arg 1 must be the query"
    exit 1
fi

if [[ "$query" == --create-db ]]; then
    info "starting db $DB_NAME ..."
    psql -U $DB_USER -c "create database $DB_NAME"
    exit 0
fi

if [[ "$query" == --connection ]]; then
    echo "$connection"
    exit 0
fi

while test $# -gt 0
do
    case "$1" in
    --separator|-s)
        shift
        separator="$1"
    ;;
    --ops)
        shift
        ops="$1"
    ;;
    --full)
        ops="c"
    ;;
    --raw)
        noop=noop
    ;;
    -*)
        echo "bad option '$1'"
        exit 1
    ;;
    esac
    shift
done

if [[ -f "$query" ]]; then
    $connection -$ops --field-separator="$separator" < "$query"
else
    $connection -$ops --field-separator="$separator" -c "$query"
fi