From 67a1148e858e51973ac4d6abfe6c723b9f1b892b Mon Sep 17 00:00:00 2001
From: MayaTheShy <MayaTheShy@gmail.com>
Date: Sun, 22 Mar 2026 03:15:58 -0400
Subject: [PATCH] Add reboot listener to handle remote reboot commands

---
 startup/bridge.lua | 26 +++++++++++++++++++++++++-
 1 file changed, 25 insertions(+), 1 deletion(-)

diff --git a/startup/bridge.lua b/startup/bridge.lua
index 4ede619..2e940e3 100644
--- a/startup/bridge.lua
+++ b/startup/bridge.lua
@@ -55,4 +55,28 @@ print("")
 print("Starting inventoryWebBridge...")
 sleep(1)
 
-shell.run("inventoryWebBridge.lua")
+-- Reboot listener: reboots this computer on remote command
+local SYSTEM_CHANNEL = 4205
+local ROLE = "bridge"
+
+local function rebootListener()
+    local m = peripheral.find("modem")
+    if not m then return end
+    m.open(SYSTEM_CHANNEL)
+    while true do
+        local _, _, channel, _, message = os.pullEvent("modem_message")
+        if channel == SYSTEM_CHANNEL and type(message) == "table" and message.type == "reboot" then
+            local target = message.target or "all"
+            if target == "all" or target == ROLE or target == tostring(os.getComputerID()) then
+                print("[SYSTEM] Reboot command received. Rebooting...")
+                sleep(0.5)
+                os.reboot()
+            end
+        end
+    end
+end
+
+parallel.waitForAny(
+    function() shell.run("inventoryWebBridge.lua") end,
+    rebootListener
+)