From 88a8bdf7b3d7c882733fa20b30b393f1d7492240 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Gina=20H=C3=A4u=C3=9Fge?= <gina@octoprint.org>
Date: Fri, 20 Jan 2017 14:27:06 +0100
Subject: [PATCH] Add "Restart in safe mode" system menu entry

---
 src/octoprint/server/api/system.py | 33 +++++++++++++++++++++++-------
 1 file changed, 26 insertions(+), 7 deletions(-)

diff --git a/src/octoprint/server/api/system.py b/src/octoprint/server/api/system.py
index 97ec21fd..058476d3 100644
--- a/src/octoprint/server/api/system.py
+++ b/src/octoprint/server/api/system.py
@@ -72,9 +72,19 @@ def executeSystemCommand(source, command):
 	if not "command" in command_spec:
 		return make_response("Command {}:{} does not define a command to execute, can't proceed".format(source, command), 500)
 
-	async = command_spec["async"] if "async" in command_spec else False
-	ignore = command_spec["ignore"] if "ignore" in command_spec else False
+	do_async = command_spec.get("async", False)
+	do_ignore = command_spec.get("ignore", False)
 	logger.info("Performing command for {}:{}: {}".format(source, command, command_spec["command"]))
+
+	try:
+		if "before" in command_spec and callable(command_spec["before"]):
+			command_spec["before"]()
+	except Exception as e:
+		if not do_ignore:
+			error = "Command \"before\" failed: {}".format(str(e))
+			logger.warn(error)
+			return make_response(error, 500)
+
 	try:
 		# we run this with shell=True since we have to trust whatever
 		# our admin configured as command and since we want to allow
@@ -83,9 +93,9 @@ def executeSystemCommand(source, command):
 		              stdout=sarge.Capture(),
 		              stderr=sarge.Capture(),
 		              shell=True,
-		              async=async)
-		if not async:
-			if not ignore and p.returncode != 0:
+		              async=do_async)
+		if not do_async:
+			if not do_ignore and p.returncode != 0:
 				returncode = p.returncode
 				stdout_text = p.stdout.text
 				stderr_text = p.stderr.text
@@ -94,7 +104,7 @@ def executeSystemCommand(source, command):
 				logger.warn(error)
 				return make_response(error, 500)
 	except Exception as e:
-		if not ignore:
+		if not do_ignore:
 			error = "Command failed: {}".format(str(e))
 			logger.warn(error)
 			return make_response(error, 500)
@@ -126,6 +136,10 @@ def _get_command_spec(source, action):
 
 
 def _get_core_command_specs():
+	def enable_safe_mode():
+		s().set(["server", "startOnceInSafeMode"], True)
+		s().save()
+
 	commands = collections.OrderedDict(
 		shutdown=dict(
 			command=s().get(["server", "commands", "systemShutdownCommand"]),
@@ -138,7 +152,12 @@ def _get_core_command_specs():
 		restart=dict(
 			command=s().get(["server", "commands", "serverRestartCommand"]),
 			name=gettext("Restart OctoPrint"),
-			confirm=gettext("You are about to restart the OctoPrint server."))
+			confirm=gettext("You are about to restart the OctoPrint server.")),
+		restart_safe=dict(
+			command=s().get(["server", "commands", "serverRestartCommand"]),
+			name=gettext("Restart OctoPrint in safe mode"),
+			confirm=gettext("You are about to restart the OctoPrint server in safe mode."),
+			before=enable_safe_mode)
 	)
 
 	available_commands = collections.OrderedDict()