diff --git a/functions b/functions index bb8fd6a..e66e6c9 100755 --- a/functions +++ b/functions @@ -38,6 +38,20 @@ service_alias() { echo "$SERVICE_NAME" | tr ._ - } +service_export() { + local SERVICE="$1" + local SERVICE_ROOT="$PLUGIN_DATA_ROOT/$SERVICE" + local SERVICE_NAME="$(get_service_name "$SERVICE")" + local DATABASE_NAME="$(get_database_name "$SERVICE")" + local PASSWORD=$(cat "$SERVICE_ROOT/PASSWORD") + + [[ -n $SSH_TTY ]] && stty -opost + docker exec "$SERVICE_NAME" env PGPASSWORD="$PASSWORD" pg_dump -Fc --no-acl --no-owner -h localhost -U postgres -w "$DATABASE_NAME" + status=$? + [[ -n $SSH_TTY ]] && stty opost + exit $status +} + service_info() { local SERVICE="$1" local SERVICE_URL=$(service_url "$SERVICE") @@ -45,6 +59,19 @@ service_info() { echo " DSN: $SERVICE_URL" } +service_import() { + local SERVICE="$1" + SERVICE_ROOT="$PLUGIN_DATA_ROOT/$SERVICE" + SERVICE_NAME="$(get_service_name "$SERVICE")" + DATABASE_NAME="$(get_database_name "$SERVICE")" + PASSWORD=$(cat "$SERVICE_ROOT/PASSWORD") + + if [[ -t 0 ]]; then + dokku_log_fail "No data provided on stdin." + fi + docker exec -i "$SERVICE_NAME" env PGPASSWORD="$PASSWORD" pg_restore -h localhost -cO -d "$DATABASE_NAME" -U postgres -w +} + service_list() { local SERVICES=$(ls "$PLUGIN_DATA_ROOT" 2> /dev/null) if [[ -z $SERVICES ]]; then @@ -252,6 +279,30 @@ service_start() { fi } +service_create() { + local SERVICE="$1" + [[ -z "$SERVICE" ]] && dokku_log_fail "Please specify a name for the service" + [[ ! -d "$PLUGIN_DATA_ROOT/$SERVICE" ]] || dokku_log_fail "$PLUGIN_SERVICE service $SERVICE already exists" + SERVICE_ROOT="$PLUGIN_DATA_ROOT/$SERVICE"; LINKS_FILE="$SERVICE_ROOT/LINKS" + + if ! docker images | grep -e "^$PLUGIN_IMAGE " | grep -q " $PLUGIN_IMAGE_VERSION " ; then + docker pull "$PLUGIN_IMAGE:$PLUGIN_IMAGE_VERSION" || dokku_log_fail "$PLUGIN_SERVICE image $PLUGIN_IMAGE:$PLUGIN_IMAGE_VERSION pull failed" + fi + + mkdir -p "$SERVICE_ROOT" || dokku_log_fail "Unable to create service directory" + password=$(openssl rand -hex 16) + echo "$password" > "$SERVICE_ROOT/PASSWORD" + chmod 640 "$SERVICE_ROOT/PASSWORD" + touch "$LINKS_FILE" + + if [[ -n $POSTGRES_CUSTOM_ENV ]]; then + echo "$POSTGRES_CUSTOM_ENV" | tr ';' "\n" > "$SERVICE_ROOT/ENV" + else + echo "" > "$SERVICE_ROOT/ENV" + fi + service_create_container "$SERVICE" +} + service_create_container() { local SERVICE="$1" local SERVICE_ROOT="$PLUGIN_DATA_ROOT/$SERVICE" diff --git a/subcommands/clone b/subcommands/clone index 45e945e..8b6fe93 100755 --- a/subcommands/clone +++ b/subcommands/clone @@ -13,9 +13,9 @@ postgres-clone-cmd() { [[ -z "$NEW_SERVICE" ]] && dokku_log_fail "Please specify a name for the new service" verify_service_name "$SERVICE" - dokku "$PLUGIN_COMMAND_PREFIX:create" "$NEW_SERVICE" + service_create "$NEW_SERVICE" dokku_log_info1 "Copying data from $SERVICE to $NEW_SERVICE" - dokku "$PLUGIN_COMMAND_PREFIX:export" "$SERVICE" | dokku "$PLUGIN_COMMAND_PREFIX:import" "$NEW_SERVICE" > /dev/null 2>&1 || true + service_export "$SERVICE" | service_import "$NEW_SERVICE" > /dev/null 2>&1 || true dokku_log_info1 "Done" } diff --git a/subcommands/create b/subcommands/create index cebeef4..56ef58a 100755 --- a/subcommands/create +++ b/subcommands/create @@ -9,26 +9,7 @@ postgres-create-cmd() { local cmd="$PLUGIN_COMMAND_PREFIX:create" argv=("$@"); [[ ${argv[0]} == "$cmd" ]] && shift 1 declare SERVICE="$1" - [[ -z "$SERVICE" ]] && dokku_log_fail "Please specify a name for the service" - [[ ! -d "$PLUGIN_DATA_ROOT/$SERVICE" ]] || dokku_log_fail "$PLUGIN_SERVICE service $2 already exists" - SERVICE_ROOT="$PLUGIN_DATA_ROOT/$SERVICE"; LINKS_FILE="$SERVICE_ROOT/LINKS" - - if ! docker images | grep -e "^$PLUGIN_IMAGE " | grep -q " $PLUGIN_IMAGE_VERSION " ; then - docker pull "$PLUGIN_IMAGE:$PLUGIN_IMAGE_VERSION" || dokku_log_fail "$PLUGIN_SERVICE image $PLUGIN_IMAGE:$PLUGIN_IMAGE_VERSION pull failed" - fi - - mkdir -p "$SERVICE_ROOT" || dokku_log_fail "Unable to create service directory" - password=$(openssl rand -hex 16) - echo "$password" > "$SERVICE_ROOT/PASSWORD" - chmod 640 "$SERVICE_ROOT/PASSWORD" - touch "$LINKS_FILE" - - if [[ -n $POSTGRES_CUSTOM_ENV ]]; then - echo "$POSTGRES_CUSTOM_ENV" | tr ';' "\n" > "$SERVICE_ROOT/ENV" - else - echo "" > "$SERVICE_ROOT/ENV" - fi - service_create_container "$SERVICE" + service_create "$SERVICE" } postgres-create-cmd "$@" diff --git a/subcommands/export b/subcommands/export index 3a2ea06..e4394ca 100755 --- a/subcommands/export +++ b/subcommands/export @@ -11,16 +11,7 @@ postgres-export-cmd() { [[ -z "$SERVICE" ]] && dokku_log_fail "Please specify a name for the service" verify_service_name "$SERVICE" - SERVICE_ROOT="$PLUGIN_DATA_ROOT/$SERVICE" - SERVICE_NAME="$(get_service_name "$SERVICE")" - DATABASE_NAME="$(get_database_name "$SERVICE")" - PASSWORD=$(cat "$SERVICE_ROOT/PASSWORD") - - [[ -n $SSH_TTY ]] && stty -opost - docker exec "$SERVICE_NAME" env PGPASSWORD="$PASSWORD" pg_dump -Fc --no-acl --no-owner -h localhost -U postgres -w "$DATABASE_NAME" - status=$? - [[ -n $SSH_TTY ]] && stty opost - exit $status + service_export "$SERVICE" } postgres-export-cmd "$@" diff --git a/subcommands/import b/subcommands/import index 1e08f76..667b8c5 100755 --- a/subcommands/import +++ b/subcommands/import @@ -11,15 +11,7 @@ postgres-import-cmd() { [[ -z "$SERVICE" ]] && dokku_log_fail "Please specify a name for the service" verify_service_name "$SERVICE" - SERVICE_ROOT="$PLUGIN_DATA_ROOT/$SERVICE" - SERVICE_NAME="$(get_service_name "$SERVICE")" - DATABASE_NAME="$(get_database_name "$SERVICE")" - PASSWORD=$(cat "$SERVICE_ROOT/PASSWORD") - - if [[ -t 0 ]]; then - dokku_log_fail "No data provided on stdin." - fi - docker exec -i "$SERVICE_NAME" env PGPASSWORD="$PASSWORD" pg_restore -h localhost -cO -d "$DATABASE_NAME" -U postgres -w + service_import "$SERVICE" } postgres-import-cmd "$@"