From 1bfea517cdcf3531947936830424d08e01e0fd00 Mon Sep 17 00:00:00 2001
From: James Swineson <github@public.swineson.me>
Date: Sat, 13 Apr 2019 11:05:42 +0800
Subject: [PATCH] Update menhera.sh

---
 menhera.sh | 30 ++++++++++++++++++++++++++++--
 1 file changed, 28 insertions(+), 2 deletions(-)

diff --git a/menhera.sh b/menhera.sh
index 7942f3a..95f950e 100755
--- a/menhera.sh
+++ b/menhera.sh
@@ -9,6 +9,30 @@ ROOTFS=""
 OLDROOT="/"
 NEWROOT=""
 
+# environment compatibility
+__compat_restart_ssh() {
+    if [ -x "$(command -v systemctl)" ]; then
+        systemctl restart ssh
+    elif [ -x "$(command -v service)" ]; then
+        service ssh restart
+    else
+        echo "ERROR: Cannot restart SSH server" >&2
+        exit 1
+    fi
+}
+
+__compat_reload_init() {
+    if [ -x "$(command -v systemctl)" ]; then
+        systemctl daemon-reexec
+    elif [ -x "$(command -v telinit)" ]; then
+        telinit u
+    else
+        echo "ERROR: Cannot re-exec init" >&2
+        exit 1
+    fi
+}
+
+# helper functions
 # https://stackoverflow.com/a/3232082/2646069
 confirm() {
     # call with a prompt string or use a default
@@ -23,6 +47,7 @@ confirm() {
     esac
 }
 
+# jobs
 get_rootfs() {
     if [ -n ${ROOTFS} ]; then 
         echo "Getting rootfs URL..."
@@ -128,7 +153,7 @@ swap_root() {
     mount --move "${OLDROOT}/${WORKDIR}" "${WORKDIR}"
 
     echo "Restarting SSH daemon..."
-    systemctl restart ssh
+    __compat_restart_ssh
 }
 
 clear_processes() {
@@ -136,7 +161,8 @@ clear_processes() {
     swapoff -a
 
     echo "Restarting systemd..."
-    telinit u
+    __compat_reload_init
+    # hope 15s is enough
     sleep 15
 
     echo "Killing all programs still using the old root..."