upgrade_postgis_debian.sh

#!/bin/bash

# Copyright (c) 2016, Philippe Ivaldi <www.piprime.fr>

# This program is free software ; you can redistribute it and/or modify
# it under the terms of the GNU Lesser General Public License as published by
# the Free Software Foundation ; either version 3 of the License, or
# (at your option) any later version.

# This program is distributed in the hope that it will be useful, but
# WITHOUT ANY WARRANTY ; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
# Lesser General Public License for more details.

# You should have received a copy of the GNU Lesser General Public License
# along with this program ; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA

CURRENT_DIR=$(dirname "$0")
. ${CURRENT_DIR}/functions.rc

usage() {
    echo "This scrit restore the custom pg dump file DUMP_PATH in the database DB_NAME when Postgist extension need a hard upgraded.
Based on the process described here http://www.postgis.org/docs/postgis_installation.html#hard_upgrade

Usage: $0 -p PORT_NUM -d DB_NAME -f DUMP_PATH [OPTIONS...]

Available options:
   -p N   The port number of the new PostgreSQL server version.
   -d S   The database name, it must already exists.
   -f S   The custom binary pg_dump file to upgrade.
   -c     Do NOT create postgis extension in the database.
   -v S   Force the target version of postgresql (9.5 or 10.2 etc)
   -g S   Force the target version of postgis (2.1 or 3.0 etc.)
   -h     This help.
"
    echo -e "$1"

    exit 1;
}

## Init config ######################

PORT=0
DB_NAME=''
FILE_PATH=''
CREATE_POSTGIS=true
PG_TARGET_VERSION=0
PGIS_TARGET_VERSION=0

while getopts 'p:d:f:g:hc' OPT; do
    case "$OPT" in
        p) # Port
            PORT="$OPTARG"
            ;;
        d) # db name
            DB_NAME="$OPTARG"
            ;;
        f) # file path
            FILE_PATH="$OPTARG"
            ;;
        g)
            PGIS_TARGET_VERSION="$OPTARG"
            ;;
        c) # Create postgis extension
            CREATE_POSTGIS=false
            ;;
        h) # short help
            usage
            ;;
        \?) # error
            usage
            ;;
    esac
done

## Check config ######################

ERROR=false
MSG='\n'

[ $OPTIND -gt $# ] || ERROR=true  # no non-option args

[ $PORT -eq 0 ] && {
    ERROR=true
    MSG='Missing port number\n'
}

[ "$DB_NAME" = "" ] && {
    ERROR=true
    MSG='Missing database name\n'
}

[ "$FILE_PATH" = "" ] && {
    ERROR=true
    MSG='Missing pg dump file path\n'
}


$ERROR && {
    usage "$MSG"
}

[ -f "$FILE_PATH" ] || {
    ERROR "'${FILE_PATH}' does not exits or is not readable"
    exit 1
}

### Starting commands ######################
[ "$USER" = "postgres" ] || {
    ERROR "$0"
    ERROR 'This script must be executed by the user postgres.'
    ERROR 'Process aborted...'
    exit 1
}

CONFIRM=false
[ $PG_TARGET_VERSION -eq 0 ] && {
    INFO "Detecting target Postgresql version..."
    PG_TARGET_VERSION=$(pg_lsclusters -h | awk '{print $1}' | sort -n -t. -k 1,2n | tail -1)
    CONFIRM=true
}

[ $PGIS_TARGET_VERSION -eq 0 ] && {
    INFO "Detecting target Postgis version..."
    PGIS_TARGET_VERSION=$(locate postgis_restore.pl | sed -E 's/.*postgis-([0-9.]+).*/\1/' | sort -n -t. -k 1,2n | tail -1)
    CONFIRM=true
}

PGIS_CURRENT_VERSION=$(psql --tuples-only -d ${DB_NAME} -U postgres -c 'SELECT PostGIS_version()' | head -n1 | awk '{print $1}')

[ "$PGIS_CURRENT_VERSION" = "$PGIS_TARGET_VERSION" ] && {
    INFO "The current version of Postgis is the same as the target version : ${PGIS_TARGET_VERSION}"
    confirm "Do you still wish to continue ?" || {
        INFO "You can restore your dump in the new cluster, executing this command as postgresql"
        INFO_EXEC "/usr/lib/postgresql/${PG_TARGET_VERSION}/bin/pg_restore -p ${PORT} -e -F c -d ${DB_NAME} -v '${FILE_PATH}'"
        ERROR 'Aborted...'
        exit 1
    }
}

$CONFIRM && {
    INFO "The folowing TARGET versions will be used :"
    echo "Postgresql : ${PG_TARGET_VERSION}"
    echo "Postgis    : ${PGIS_TARGET_VERSION}"
    echo
    confirm "Are you agree ?" || {
        echo 'Aborted...'
        exit 1
    }
}

PSQL_CMD="/usr/lib/postgresql/${PG_TARGET_VERSION}/bin/psql -p ${PORT}"
PGIS_CMD="perl /usr/share/postgresql/${PG_TARGET_VERSION}/contrib/postgis-${PGIS_TARGET_VERSION}/postgis_restore.pl"

$CREATE_POSTGIS && {
$PSQL_CMD -d ${DB_NAME} -c "CREATE EXTENSION postgis;"
$PSQL_CMD -d ${DB_NAME} -c "CREATE EXTENSION postgis_topology;"
}

ERROR_FILE="/tmp/${DB_NAME}_err.txt"

$PGIS_CMD $FILE_PATH | $PSQL_CMD ${DB_NAME} 2> $ERROR_FILE && INFO "Operation done !"

cat "${ERROR_FILE}"

# Local variables:
# coding: utf-8
# End: