From c6c09606132b0a612ffba7e0fe2f4bae70b120fa Mon Sep 17 00:00:00 2001
From: Roland <roland@learningselfreliance.com>
Date: Sat, 14 Feb 2026 17:30:07 -0700
Subject: [PATCH] No longer restricting the Controller from unmounting the
 wrolpi drive.

---
 controller/lib/disks.py       | 25 +------------------------
 controller/test/test_disks.py |  8 --------
 2 files changed, 1 insertion(+), 32 deletions(-)

diff --git a/controller/lib/disks.py b/controller/lib/disks.py
index 1c00551f..9580d470 100644
--- a/controller/lib/disks.py
+++ b/controller/lib/disks.py
@@ -13,7 +13,7 @@ from dataclasses import dataclass
 from pathlib import Path
 from typing import Optional
 
-from controller.lib.config import get_config, is_docker_mode
+from controller.lib.config import is_docker_mode
 
 SUPPORTED_FILESYSTEMS = {"ext4", "btrfs", "vfat", "exfat"}
 
@@ -204,29 +204,6 @@ def unmount_drive(mount_point: str, lazy: bool = False) -> dict:
     if is_docker_mode():
         return {"success": False, "error": "Not available in Docker mode"}
 
-    # Safety rule: Don't unmount /media/wrolpi if no drives configured
-    if mount_point == "/media/wrolpi":
-        config = get_config()
-        configured_mounts = config.get("drives", {}).get("mounts", [])
-
-        if not configured_mounts:
-            return {
-                "success": False,
-                "error": "Cannot unmount /media/wrolpi: No drives are configured. "
-                         "This prevents accidental data loss.",
-            }
-
-        # Check if this is the primary drive
-        primary_mount = next(
-            (m for m in configured_mounts if m.get("is_primary")),
-            None
-        )
-        if primary_mount:
-            return {
-                "success": False,
-                "error": "Cannot unmount /media/wrolpi: This is the primary WROLPi drive.",
-            }
-
     # Check if mount is busy
     if is_mount_busy(mount_point) and not lazy:
         return {
diff --git a/controller/test/test_disks.py b/controller/test/test_disks.py
index 4ce49c87..c1a6418c 100644
--- a/controller/test/test_disks.py
+++ b/controller/test/test_disks.py
@@ -293,14 +293,6 @@ class TestUnmountDrive:
             assert result["success"] is False
             assert "Docker" in result["error"]
 
-    def test_protects_primary_wrolpi_mount(self):
-        """Should protect /media/wrolpi from accidental unmount."""
-        with mock.patch("controller.lib.disks.is_docker_mode", return_value=False):
-            with mock.patch("controller.lib.disks.get_config", return_value={"drives": {}}):
-                result = unmount_drive("/media/wrolpi")
-                assert result["success"] is False
-                assert "Cannot unmount /media/wrolpi" in result["error"]
-
     def test_unmounts_successfully(self):
         """Should unmount successfully."""
         with mock.patch("controller.lib.disks.is_docker_mode", return_value=False):