Clean postgres driver variables and add postgres-9.1 support

[hcoop/domtool2.git] / src / plugins / domtool-postgres

diff --git a/src/plugins/domtool-postgres b/src/plugins/domtool-postgres
dissimilarity index 60%
index 1d945d4..03320d0 100755 (executable)
--- a/src/plugins/domtool-postgres
+++ b/src/plugins/domtool-postgres
@@ -1,20 +1,37 @@
-#!/bin/sh -e
-
-case $1 in
-       adduser)
-               USERNAME=$2
-
-               sudo -u postgres psql -c "CREATE USER $USERNAME" template1
-               sudo -u postgres psql -c "ALTER TABLESPACE user_$USERNAME OWNER TO $USERNAME"
-       ;;
-       createdb)
-               USERNAME=$2
-               DBNAME_BASE=$3
-               DBNAME="${USERNAME}_${DBNAME_BASE}"
-
-               sudo -u postgres createdb -O $USERNAME -D user_$USERNAME $DBNAME
-       ;;
-       *)
-               echo "Usage: domtool-postgres [adduser <user> | createdb <user> <table>]"
-       ;;
-esac
+#!/bin/sh -e
+
+export PGPORT="$2"
+
+case $1 in
+       adduser)
+               USERNAME=$3
+
+               sudo -u postgres psql -h postgres -c "CREATE USER $USERNAME" template1
+               sudo -u postgres psql -h postgres -c "ALTER TABLESPACE user_$USERNAME OWNER TO $USERNAME"
+       ;;
+
+       createdb)
+               USERNAME=$3
+               DBNAME_BASE=$4
+               ENCODING=$5
+               DBNAME="${USERNAME}_${DBNAME_BASE}"
+
+               if [ -n "$ENCODING" ]; then
+                   ENCODING="-E $ENCODING"
+               fi
+
+               sudo -u postgres createdb -h postgres -O $USERNAME -D user_$USERNAME $ENCODING $DBNAME
+       ;;
+
+       dropdb)
+               USERNAME=$3
+               DBNAME_BASE=$4
+               DBNAME="${USERNAME}_${DBNAME_BASE}"
+
+               sudo -u postgres dropdb -h postgres $DBNAME
+       ;;
+
+       *)
+               echo "Usage: domtool-postgres [adduser <user> | createdb <user> <db> | dropdb <user> <db>]"
+       ;;
+esac