From 27c74bff6e90ce91d26ad7570153332de2b7852d Mon Sep 17 00:00:00 2001
From: Richard Bowman <richard.bowman@cantab.net>
Date: Thu, 2 Nov 2023 21:18:59 +0000
Subject: [PATCH] Add shutdown/restart actions

---
 src/openflexure_microscope_server/server.py   |  2 +
 .../things/system_control.py                  | 59 +++++++++++++++++++
 2 files changed, 61 insertions(+)
 create mode 100644 src/openflexure_microscope_server/things/system_control.py

diff --git a/src/openflexure_microscope_server/server.py b/src/openflexure_microscope_server/server.py
index 699e7c28..ab84d4cf 100644
--- a/src/openflexure_microscope_server/server.py
+++ b/src/openflexure_microscope_server/server.py
@@ -10,6 +10,7 @@ from labthings_picamera2.thing import StreamingPiCamera2
 
 from .things.autofocus import AutofocusThing
 from .things.camera_stage_mapping import CameraStageMapper
+from .things.system_control import SystemControlThing
 from .serve_static_files import add_static_files
 import openflexure_microscope_server
 
@@ -20,6 +21,7 @@ thing_server.add_thing(StreamingPiCamera2(), "/camera/")
 thing_server.add_thing(SangaboardThing(), "/stage/")
 thing_server.add_thing(AutofocusThing(), "/autofocus/")
 thing_server.add_thing(CameraStageMapper(), "/camera_stage_mapping/")
+thing_server.add_thing(SystemControlThing(), "/system_control/")
 try:
     add_static_files(thing_server.app)
 except RuntimeError:
diff --git a/src/openflexure_microscope_server/things/system_control.py b/src/openflexure_microscope_server/things/system_control.py
new file mode 100644
index 00000000..74a0be44
--- /dev/null
+++ b/src/openflexure_microscope_server/things/system_control.py
@@ -0,0 +1,59 @@
+"""
+OpenFlexure Microscope API extension for stage calibration
+
+This file contains the HTTP API for camera/stage calibration. It
+includes calibration functions that measure the relationship between
+stage coordinates and camera coordinates, as well as functions that
+move by a specified displacement in pixels, perform closed-loop moves,
+and return the calibration data.
+
+This module is only intended to be called from the OpenFlexure Microscope
+server, and depends on that server and its underlying LabThings library.
+"""
+import subprocess
+import os
+from labthings_fastapi.thing import Thing
+from labthings_fastapi.decorators import thing_action, thing_property
+from pydantic import BaseModel
+
+class CommandOutput(BaseModel):
+    output: str
+    error: str
+
+
+class SystemControlThing(Thing):
+    """
+    Attempt to shutdown the device 
+    """
+
+    @thing_action
+    def shutdown(self) -> CommandOutput:
+        """
+        Attempt to shutdown the device
+        """
+        p = subprocess.Popen(
+            ["sudo", "shutdown", "-h", "now"],
+            stderr=subprocess.PIPE,
+            stdout=subprocess.PIPE,
+        )
+
+        out, err = p.communicate()
+        return CommandOutput(output=out, error=err)
+
+    @thing_property
+    def is_raspberrypi() -> bool:
+        """
+        Checks if we are running on a Raspberry Pi.
+        """
+        return os.path.exists("/usr/bin/raspi-config")
+    
+    @thing_action
+    def reboot(self) -> CommandOutput:
+        """Attempt to reboot the device"""
+        p = subprocess.Popen(
+            ["sudo", "shutdown", "-r", "now"],
+            stderr=subprocess.PIPE,
+            stdout=subprocess.PIPE,
+        )
+        out, err = p.communicate()
+        return CommandOutput(output=out, error=err)