1 files changed, 13 insertions, 1 deletions

diff --git a/roles/contexts/fretlink/files/workdir/start b/roles/contexts/fretlink/files/workdir/start
index 840943a..8f440bf 100755
--- a/roles/contexts/fretlink/files/workdir/start
+++ b/roles/contexts/fretlink/files/workdir/start
@@ -1,6 +1,7 @@
 #!/bin/bash
 
 APP="$1"
+shift
 
 if [ -z "$APP" ]; then
   if [ $(dirname $(pwd)) = "$HOME/workdir" ]; then
@@ -11,9 +12,12 @@ if [ -z "$APP" ]; then
   fi
 fi
 
-if [ "$APP" != "psql" ]; then
+if [ "$APP" != "psql" -a "$APP" != "pg_restore" ]; then
   cd $HOME/workdir/$APP
   source ../environment
+elif [ "$APP" = "pg_restore" ]; then
+  source ../environment
+  APP=pg_restore
 else
   source ../environment
   APP=psql
@@ -41,14 +45,22 @@ function start_psql() {
   psql -h $POSTGRESQL_ADDON_HOST -p $POSTGRESQL_ADDON_PORT -U $POSTGRESQL_ADDON_USER -d $POSTGRESQL_ADDON_DB
 }
 
+function start_pg_restore() {
+  export PGPASSWORD=$POSTGRESQL_ADDON_PASSWORD
+  pg_restore -h $POSTGRESQL_ADDON_HOST -p $POSTGRESQL_ADDON_PORT -U $POSTGRESQL_ADDON_USER -d $POSTGRESQL_ADDON_DB
+}
+
 case "$APP${FLAVOR:+-}$FLAVOR" in
   "psql") start_psql;;
+  "pg_restore") start_pg_restore;;
   "app") start_app;;
   "admin-root") start_stack server;;
+  "admin-user") start_stack server;;
   "carrier-directory") start_stack server;;
   "freight") start_stack freight-server;;
   "notifier") start_stack notifier-exe;;
   "notifier-amqp") start_stack notifier-exe-amqp;;
   "pricer") start_stack pricer;;
+  "third-parties") start_stack server;;
   "toolbox") start_stack toolbox-api;;
 esac