diff --git a/pwnagotchi/__init__.py b/pwnagotchi/__init__.py
index c0cac857..827f8018 100644
--- a/pwnagotchi/__init__.py
+++ b/pwnagotchi/__init__.py
@@ -112,3 +112,14 @@ def reboot():
logging.warning("rebooting ...")
os.system("sync")
os.system("shutdown -r now")
+
+
+def reboot_into_auto():
+ logging.warning("rebooting into AUTO mode ...")
+ if view.ROOT:
+ view.ROOT.on_reboot()
+ # give it some time to refresh the ui
+ time.sleep(10)
+ os.system("touch /root/.pwnagotchi-auto")
+ os.system("sync")
+ os.system("shutdown -r now")
diff --git a/pwnagotchi/ui/web.py b/pwnagotchi/ui/web.py
index 8376ce4f..b47601b9 100644
--- a/pwnagotchi/ui/web.py
+++ b/pwnagotchi/ui/web.py
@@ -52,8 +52,11 @@ INDEX = """
-
+
@@ -73,6 +76,17 @@ SHUTDOWN = """
"""
+REBOOT_INTO_AUTO = """
+
+ %s
+
+
+
+
+ Rebooting into AUTO mode ...
+
+
+"""
class Handler(BaseHTTPRequestHandler):
AllowedOrigin = None # CORS headers are not sent
@@ -116,6 +130,11 @@ class Handler(BaseHTTPRequestHandler):
self._html(SHUTDOWN % pwnagotchi.name())
pwnagotchi.shutdown()
+ # serve a message and reboot the unit into auto mode
+ def _reboot(self):
+ self._html(REBOOT % pwnagotchi.name())
+ pwnagotchi.reboot_into_auto()
+
# serve the PNG file with the display image
def _image(self):
global frame_lock, frame_path, frame_ctype