Implement restart-policy handling. Closes #1734

Applications without a restart-policy will have their policies set to `on-failure:10`. Users can completely unset the restart-policy using the `docker-options` plugin, though this will be equivalent to setting it explicitly to `no`.

Restart policies must be explicitly set, and the following are all valid:

- no
- unless-stopped
- always
- on-failure
- on-failure:NUMBER

plugins/ps/commands

@@ -16,6 +16,8 @@ case "$1" in
     ps:restart <app>, Restart app container(s)
     ps:restartall, Restart all deployed app containers
     ps:restore, Start previously running apps e.g. after reboot
+    ps:restart-policy <app>, Shows the restart-policy for an app
+    ps:set-restart-policy <app> <policy>, Sets app restart-policy
 help_content
     }
 

plugins/ps/functions

@@ -165,3 +165,15 @@ ps_scale() {
     release_and_deploy "$APP" "$IMAGE_TAG"
   fi
 }
+
+get_restart_policies() {
+  declare desc="strips docker options and prints restart policies"
+  local -r phase_file_path=$1
+  [[ -r "$phase_file_path" ]] && sed -e '/^--restart=/!d' -e 's/^--restart=/    /' < "$phase_file_path"
+}
+
+get_raw_restart_policies() {
+  declare desc="strips docker options and prints raw restart policies"
+  local -r phase_file_path=$1
+  [[ -r "$phase_file_path" ]] && sed -e '/^--restart=/!d' < "$phase_file_path"
+}

plugins/ps/install

@@ -0,0 +1,21 @@
+#!/usr/bin/env bash
+set -eo pipefail; [[ $DOKKU_TRACE ]] && set -x
+source "$PLUGIN_CORE_AVAILABLE_PATH/common/functions"
+source "$PLUGIN_AVAILABLE_PATH/docker-options/functions"
+source "$PLUGIN_AVAILABLE_PATH/ps/functions"
+
+set_default_restart_policies() {
+  declare desc="set the default restart policy for all applications if there is not one already set"
+  local APPS="$(dokku_apps)"
+  local APP
+
+  for APP in $APPS; do
+    local RESTART_POLICIES=$(get_restart_policies "$(get_phase_file_path "deploy")")
+    if [[ -z "$RESTART_POLICIES" ]]; then
+      local passed_phases=(deploy)
+      add_passed_docker_option passed_phases[@] "--restart=on-failure:10"
+    fi
+  done
+}
+
+set_default_restart_policies "$@"

plugins/ps/post-create

@@ -0,0 +1,14 @@
+#!/usr/bin/env bash
+set -eo pipefail; [[ $DOKKU_TRACE ]] && set -x
+source "$PLUGIN_CORE_AVAILABLE_PATH/common/functions"
+source "$PLUGIN_AVAILABLE_PATH/docker-options/functions"
+source "$PLUGIN_AVAILABLE_PATH/ps/functions"
+
+ps_post_create() {
+  declare desc="ps post-stop plugin trigger"
+  local trigger="ps_post_stop"
+  local APP="$1"
+
+  local passed_phases=(deploy)
+  add_passed_docker_option passed_phases[@] "--restart=on-failure:10"
+}

plugins/ps/subcommands/restart-policy

@@ -0,0 +1,18 @@
+#!/usr/bin/env bash
+set -eo pipefail; [[ $DOKKU_TRACE ]] && set -x
+source "$PLUGIN_CORE_AVAILABLE_PATH/common/functions"
+source "$PLUGIN_AVAILABLE_PATH/docker-options/functions"
+source "$PLUGIN_AVAILABLE_PATH/ps/functions"
+
+ps_restart_policy_cmd() {
+  declare desc="shows the restart-policy for an app"
+  local cmd="ps:restart-policy"
+  local passed_phases="deploy"
+  [[ -z $2 ]] && dokku_log_fail "Please specify an app to run the command on"
+  verify_app_name "$2" && local APP="$2"
+
+  echo "$APP restart-policy:"
+  get_restart_policies "$(get_phase_file_path "$passed_phases")"
+}
+
+ps_restart_policy_cmd "$@"

plugins/ps/subcommands/set-restart-policy

@@ -0,0 +1,31 @@
+#!/usr/bin/env bash
+set -eo pipefail; [[ $DOKKU_TRACE ]] && set -x
+source "$PLUGIN_CORE_AVAILABLE_PATH/common/functions"
+source "$PLUGIN_AVAILABLE_PATH/docker-options/functions"
+source "$PLUGIN_AVAILABLE_PATH/ps/functions"
+
+ps_set_restart_policy_cmd() {
+  declare desc="sets app restart-policy"
+  local cmd="ps:set-restart-policy"
+  [[ -z $2 ]] && dokku_log_fail "Please specify an app to run the command on"
+  verify_app_name "$2" && local APP="$2"
+  local RESTART_POLICY="$3"
+  local RE_RESTART_POLICY="^(no|unless-stopped|always|on-failure(:[1-9][0-9]*)?)$"
+
+  if [[ ! "$RESTART_POLICY" =~ $RE_RESTART_POLICY ]]; then
+    dokku_log_fail "Please specify an valid restart policy matching the following regex: ${RE_RESTART_POLICY}"
+  fi
+
+  local RESTART_POLICIES=$(get_raw_restart_policies "$(get_phase_file_path "deploy")")
+  local passed_phases=(deploy)
+
+  for policy in $RESTART_POLICIES; do
+    remove_passed_docker_option passed_phases[@] "$policy"
+  done
+
+  dokku_log_info1_quiet "Setting restart policy: ${RESTART_POLICY}"
+  local passed_phases=(deploy)
+  add_passed_docker_option passed_phases[@] "--restart=${RESTART_POLICY}"
+}
+
+ps_set_restart_policy_cmd "$@"