merge {reboot,poweroff}.sh into get.py

4 files changed, 30 insertions, 12 deletions

diff --git a/cgi-bin/get.py b/cgi-bin/get.py
index 1453dc2..e852561 100755
--- a/cgi-bin/get.py
+++ b/cgi-bin/get.py
@@ -173,6 +173,28 @@ class TopTask(Task):
         return self.task.result()
 
 
+class RebootTask(Task):
+    def __init__(self):
+        self.cmd = Command('systemctl', 'reboot')
+
+    def run(self):
+        self.task = self.cmd.run()
+
+    def result(self):
+        return self.task.result()
+
+
+class PoweroffTask(Task):
+    def __init__(self):
+        self.cmd = Command('systemctl', 'poweroff')
+
+    def run(self):
+        self.task = self.cmd.run()
+
+    def result(self):
+        return self.task.result()
+
+
 class InstanceStatusTask(Task):
     def __init__(self, systemctl):
         self.overview_cmd = systemctl('status')
@@ -342,6 +364,8 @@ class Request(Enum):
     STATUS = 'status'
     TIMERS = 'timers'
     TOP = 'top'
+    REBOOT = 'reboot'
+    POWEROFF = 'poweroff'
 
     def __str__(self):
         return self.value
@@ -359,6 +383,10 @@ class Request(Enum):
             return TimersTask().complete()
         if self is Request.TOP:
             return TopTask().complete()
+        if self is Request.REBOOT:
+            return RebootTask().complete()
+        if self is Request.POWEROFF:
+            return PoweroffTask().complete()
         raise NotImplementedError(f'unknown request: {self}')
 
 
diff --git a/cgi-bin/poweroff.sh b/cgi-bin/poweroff.sh
deleted file mode 100755
index d0df856..0000000
--- a/cgi-bin/poweroff.sh
+++ /dev/null
@@ -1,5 +0,0 @@
-#!/usr/bin/env bash
-
-set -o errexit -o nounset -o pipefail
-
-systemctl poweroff
diff --git a/cgi-bin/reboot.sh b/cgi-bin/reboot.sh
deleted file mode 100755
index b8bb350..0000000
--- a/cgi-bin/reboot.sh
+++ /dev/null
@@ -1,5 +0,0 @@
-#!/usr/bin/env bash
-
-set -o errexit -o nounset -o pipefail
-
-systemctl reboot
diff --git a/index.html b/index.html
index 35d118e..7c8f487 100644
--- a/index.html
+++ b/index.html
@@ -59,11 +59,11 @@ h1, .h1 {
     <script src="js/bootstrap.bundle.min.js"></script>
     <script>
 function reboot() {
-    $.get('cgi-bin/reboot.sh');
+    $.get('reboot');
 }
 
 function shutdown() {
-    $.get('cgi-bin/poweroff.sh');
+    $.get('poweroff');
 }
 
 function set_hostname(data) {