#!/bin/bash
## WARNING TO MAINTAINERS - DO NOT EDIT this file in debian/.  It will be
## overwritten by debian/rules.  Edit the .in template instead.

# Enable a procedural language for PostgreSQL in one or more
# existing databases

enable_database() {
	# the database name is the first parameter to this function
	x="select count(*) from pg_language where lanname='$lang'"
	if ! psql -d $1 -qtc "$x" >$TMPFIL2 2>&1
	then
		echo "Cannot connect to $1"
		exit 2
	fi
	if [ `cat $TMPFIL2` -eq 0 ]
	then
		if ! /usr/lib/postgresql/bin/createlang $lang $1
		then
			echo "Failed to add $lang to $1"
			exit 2
		fi
		echo "$lang added to $1"
	else
		echo "$lang is already enabled in $1"
	fi

}

# Execution starts here

if ! ps --User postgres | grep -v grep | grep -q postmaster
then
	echo The postgresql package must be installed before this script can be run.
	exit 1
fi

# Need unrestricted access
ACCESS=`grep '^local[  ][    ]*all[   ]' /etc/postgresql/pg_hba.conf | awk '{print $3}'`
if [ "$ACCESS" != trust -a "$ACCESS" != ident ]
then
	echo Sorry! I need unrestricted access in /etc/postgresql/pg_hba.conf to update
	echo the databases.
	exit
fi

TMPFILE=`mktemp /tmp/enable_lang.XXXXXX`
TMPFIL2=`mktemp /tmp/enable_lang.XXXXXX`
trap "rm $TMPFILE $TMPFIL2" EXIT

. /etc/postgresql/postgresql.env
installed=`cat ${PGDATA}/PG_VERSION`
current=7.2

if [ -n "$installed" -a "$installed" != $current ]
then
	echo "I cannot install the $lang language while the database version differs from the"
	echo "current version. Run enable_lang as the postgres user, after using postgresql-dump"
	echo "(from the postgresql package) to dump and reload the database."
	echo
	echo -n "Press <return> to continue: "
	read x
	exit 0
fi

# read command line
while [ -n "$1" ]
do
	if [ -z "$lang" ]
	then
		lang=$1
	else
		dblist="$dblist $1"
	fi
	shift
done
	
if [ -z "$lang" -o -z "$dblist" ]
then
	echo "Syntax: $0 language --all | database ..."
	exit 1
fi

if [ ! -x /usr/lib/postgresql/bin/createlang ]
then
	echo "No executable createlang"
	exit 2
fi

if [ $dblist = "--all" ]
then
	if ! psql -d template1 -t -c "select datname from pg_database where datname != 'template0' order by datname" >$TMPFILE
	then
		echo Cannot select databases
		exit 2
	fi
	for database in `cat $TMPFILE`
	do
		enable_database $database
	done
else
	for database in $dblist
	do
		enable_database $database
		shift
	done
fi
