Add initial create-database command

Author: kane-c

commands

@@ -43,6 +43,10 @@ case "$1" in
     "$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)/subcommands/create" "$@"
     ;;
 
+  $PLUGIN_COMMAND_PREFIX:create-database)
+    "$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)/subcommands/create-database" "$@"
+    ;;
+
   $PLUGIN_COMMAND_PREFIX:destroy)
     "$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)/subcommands/destroy" "$@"
     ;;
@@ -112,6 +116,7 @@ case "$1" in
     $PLUGIN_COMMAND_PREFIX:clone <name> <new-name>, Create container <new-name> then copy data from <name> into <new-name>
     $PLUGIN_COMMAND_PREFIX:connect <name>, Connect via psql to a $PLUGIN_SERVICE service
     $PLUGIN_COMMAND_PREFIX:create <name>, Create a $PLUGIN_SERVICE service
+    $PLUGIN_COMMAND_PREFIX:create-database <name> <db>, Create a $PLUGIN_SERVICE database in the specified service
     $PLUGIN_COMMAND_PREFIX:destroy <name>, Delete the $PLUGIN_SERVICE service and stop its container if there are no links left
     $PLUGIN_COMMAND_PREFIX:export <name>, Export a dump of the $PLUGIN_SERVICE service database
     $PLUGIN_COMMAND_PREFIX:expose <name> [port], Expose a $PLUGIN_SERVICE service on custom port if provided (random port otherwise)

functions

@@ -22,9 +22,12 @@ service_create() {
   mkdir -p "$SERVICE_ROOT" || dokku_log_fail "Unable to create service directory"
   mkdir -p "$SERVICE_ROOT/data" || dokku_log_fail "Unable to create service data directory"
   mkdir -p "$SERVICE_ROOT/config" || dokku_log_fail "Unable to create service config directory"
+  mkdir -p "$SERVICE_ROOT/databases" || dokku_log_fail "Unable to create service database directory"
+  mkdir -p "$SERVICE_ROOT/auth" || dokku_log_fail "Unable to create service auth directory"
+  chmod 750 "$SERVICE_ROOT/auth"
   touch "$LINKS_FILE"
   password=$(openssl rand -hex 16)
-  echo "$password" > "$SERVICE_ROOT/PASSWORD"
+  echo "$password" > "$SERVICE_ROOT/auth/PASSWORD"
   chmod 640 "$SERVICE_ROOT/PASSWORD"
 
   if [[ -n $POSTGRES_CUSTOM_ENV ]]; then
@@ -35,6 +38,33 @@ service_create() {
   service_create_container "$SERVICE"
 }
 
+database_create() {
+  local SERVICE="$1"
+  local NAME="$2"
+  [[ -z "$SERVICE" ]] && dokku_log_fail "Please specify a name for the service"
+  [[ -z "$NAME" ]] && dokku_log_fail "Please specify a name for the database"
+
+  local SERVICE_ROOT="$PLUGIN_DATA_ROOT/$SERVICE"
+  local SERVICE_NAME="$(get_service_name "$SERVICE")"
+
+  dokku_log_verbose_quiet "Creating user"
+  password=$(openssl rand -hex 16)
+  if docker exec "$SERVICE_NAME" su - postgres -c "psql -c \"CREATE USER $NAME WITH PASSWORD '$password';\"" 2> /dev/null; then
+    echo "$password" > "$SERVICE_ROOT/auth/$NAME"
+    chmod 640 "$SERVICE_ROOT/auth/$NAME"
+  else
+    echo 'Already exists'
+  fi
+
+  dokku_log_verbose_quiet "Creating database"
+  if docker exec "$SERVICE_NAME" su - postgres -c "createdb -E utf8 $NAME" 2> /dev/null; then
+    touch "$SERVICE_ROOT/databases/$NAME"
+  else
+    echo 'Already exists'
+  fi
+  docker exec "$SERVICE_NAME" su - postgres -c "psql -c \"GRANT ALL PRIVILEGES ON DATABASE $NAME TO $NAME;\"" > /dev/null
+}
+
 service_create_container() {
   local SERVICE="$1"
   local SERVICE_ROOT="$PLUGIN_DATA_ROOT/$SERVICE"

subcommands/create-database

@@ -0,0 +1,15 @@
+#!/usr/bin/env bash
+source "$(dirname "$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)")/config"
+set -eo pipefail; [[ $DOKKU_TRACE ]] && set -x
+source "$PLUGIN_BASE_PATH/common/functions"
+source "$(dirname "$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)")/functions"
+
+postgres-create-database-cmd() {
+  declare desc="create a $PLUGIN_SERVICE database"
+  local cmd="$PLUGIN_COMMAND_PREFIX:create-database" argv=("$@"); [[ ${argv[0]} == "$cmd" ]] && shift 1
+  declare SERVICE="$1" NAME="$2"
+
+  database_create "$SERVICE" "$NAME"
+}
+
+postgres-create-database-cmd "$@"