update mount kmod detection to look for mmc platform drivers

Signed-off-by: Zen <[email protected]>

Author: desultory

src/ugrd/fs/mounts.py

@@ -6,6 +6,7 @@
 from typing import Union
 
 from ugrd.exceptions import AutodetectError, ValidationError
+from ugrd.kmod.platform import _get_platform_mmc_drivers
 from zenlib.util import colorize as c_
 from zenlib.util import contains, pretty_print
 
@@ -1075,6 +1076,7 @@ def resolve_blkdev_kmod(self, device) -> list[str]:
             kmods.append("virtio_scsi")
     elif device_name.startswith("mmcblk"):
         kmods.append("mmc_block")
+        kmods.extend(_get_platform_mmc_drivers(self, device_name))
     elif device_name.startswith("sr"):
         kmods.append("sr_mod")
     elif device_name.startswith("md"):

src/ugrd/kmod/kmod.py

@@ -109,6 +109,14 @@ def _get_kmod_info(self, module: str) -> dict:
     """
     Runs modinfo on a kernel module, parses the output and stored the results in self['_kmod_modinfo'].
     !!! Should be run after metadata is processed so the kver is set properly !!!
+
+    Returns the module info as a dictionary with the following keys:
+    - filename: The path to the module file.
+    - depends: A list of module dependencies.
+    - softdep: A list of soft dependencies.
+    - firmware: A list of firmware files required by the module.
+    Raises:
+        DependencyResolutionError: If the modinfo command fails, returns no output, or the module name can't be resolved.
     """
     module = _normalize_kmod_name(self, module)
     if module in self["_kmod_modinfo"]:

src/ugrd/kmod/platform.py

@@ -101,3 +101,34 @@ def autodetect_platform_bus_drivers(self):
     else:
         self.logger.info("No platform bus drivers detected.")
 
+
+def _get_platform_mmc_drivers(self, mmc_dev):
+    """Helper function to get MMC drivers from a given device.
+    Strips the partition number from the device name if present.
+    """
+    mmc_name = mmc_dev.split("p")[0].replace('blk', '')  # Strip partition number if present, and 'blk' prefix
+    mmc_path = Path(f"/sys/class/mmc_host/{mmc_name}/device")
+    if not mmc_path.exists():
+        self.logger.warning(f"[{c_(mmc_path, 'yellow')}] MMC device path does not exist, skipping detection.")
+        return []
+
+    drivers = set()
+    if driver := (mmc_path / "driver").resolve().name:
+        self.logger.info(f"[{c_(mmc_dev, 'green', bright=True)}] Detected MMC driver: {c_(driver, 'magenta', bright=True)}")
+        drivers.add(driver)
+
+    # Check for supplier drivers
+    for supplier in mmc_path.iterdir():
+        if not supplier.name.startswith("supplier:"):
+            continue
+
+        supplier_driver = (supplier / "supplier" / "driver")
+        if not supplier_driver.exists():
+            self.logger.warning(f"[{c_(mmc_dev, 'yellow', bright=True)}] Supplier driver not found, skipping: {c_(supplier_driver, 'red', bright=True)}")
+            continue
+
+        supplier_driver = supplier_driver.resolve()
+        self.logger.debug(f"[{c_(mmc_dev, 'green', bright=True)}:{c_(supplier, 'blue')}] Detected MMC supplier driver: {c_(supplier_driver.name, 'magenta', bright=True)}")
+        drivers.add(supplier_driver.name)
+
+    return list(drivers)