facts+operations/server: add mount support for FreeBSD

Author: DtxdF

pyinfra/facts/server.py

@@ -1,6 +1,8 @@
 from __future__ import annotations
 
+import json
 import os
+import platform
 import re
 import shutil
 from datetime import datetime
@@ -206,7 +208,10 @@ class Mounts(FactBase[Dict[str, MountsDict]]):
 
     @override
     def command(self):
-        return "cat /proc/self/mountinfo"
+        if platform.system() == "FreeBSD":
+            return "mount -p --libxo json"
+        else:
+            return "cat /proc/self/mountinfo"
 
     @override
     def process(self, output) -> dict[str, MountsDict]:
@@ -222,6 +227,21 @@ def replace_octal(s: str) -> str:
             """
             return re.sub(r"\\[0-7]{3}", unescape_octal, s)
 
+        if platform.system() == "FreeBSD":
+            full_output = "\n".join(output)
+            json_output = json.loads(full_output)
+            mount_fstab = json_output["mount"]["fstab"]
+
+            for entry in mount_fstab:
+                path = entry["mntpoint"]
+                type_ = entry["fstype"]
+                device = entry["device"]
+                options = [option.strip() for option in entry["opts"].split(",")]
+
+                devices[path] = {"device": device, "type": type_, "options": options}
+
+            return devices
+
         for line in output:
             # ignore mount ID, parent ID, major:minor, root
             _, _, _, _, mount_point, mount_options, line = line.split(sep=" ", maxsplit=6)

pyinfra/operations/server.py

@@ -5,6 +5,7 @@
 
 from __future__ import annotations
 
+import platform
 from io import StringIO
 from itertools import filterfalse, tee
 from os import path
@@ -335,7 +336,15 @@ def mount(
         mounted_options = mounts[path]["options"]
         needed_options = set(options) - set(mounted_options)
         if needed_options:
-            yield "mount -o remount,{0} {1}".format(options_string, path)
+            if platform.system() == "FreeBSD":
+                fs_type = mounts[path]["type"]
+                device = mounts[path]["device"]
+
+                yield "mount -o update,{options} -t {fs_type} {device} {path}".format(
+                    options=options_string, fs_type=fs_type, device=device, path=path
+                )
+            else:
+                yield "mount -o remount,{0} {1}".format(options_string, path)
 
     else:
         host.noop(