X-Git-Url: https://git.hcoop.net/hcoop/domtool2.git/blobdiff_plain/53cb7cc1a416d38a26a5287478e00920f3d67b7c..ebf47697644deafd0bde2dfb207b0048a8f2753c:/src/plugins/domtool-postgres

diff --git a/src/plugins/domtool-postgres b/src/plugins/domtool-postgres
dissimilarity index 60%
index 1d945d4..45d7735 100755
--- 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 -p $PGPORT -c "CREATE USER $USERNAME" template1
+		sudo -u postgres psql -h postgres -p $PGPORT -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 -p $PGPORT -O $USERNAME -D user_$USERNAME $ENCODING $DBNAME
+	;;
+
+	dropdb)
+		USERNAME=$3
+		DBNAME_BASE=$4
+		DBNAME="${USERNAME}_${DBNAME_BASE}"
+
+		sudo -u postgres dropdb -h postgres -p $PGPORT $DBNAME
+	;;
+
+	*)
+		echo "Usage: domtool-postgres [adduser <user> | createdb <user> <db> | dropdb <user> <db>]"
+	;;
+esac