1 files changed, 103 insertions, 0 deletions
diff --git a/src/pybind/mgr/mirroring/module.py b/src/pybind/mgr/mirroring/module.py
new file mode 100644
index 000000000..4b4354ab2
--- /dev/null
+++ b/src/pybind/mgr/mirroring/module.py
@@ -0,0 +1,103 @@
+from typing import List, Optional
+
+from mgr_module import MgrModule, CLIReadCommand, CLIWriteCommand, Option, NotifyType
+
+from .fs.snapshot_mirror import FSSnapshotMirror
+
+class Module(MgrModule):
+    MODULE_OPTIONS: List[Option] = []
+    NOTIFY_TYPES = [NotifyType.fs_map]
+
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.fs_snapshot_mirror = FSSnapshotMirror(self)
+
+    def notify(self, notify_type: NotifyType, notify_id):
+        self.fs_snapshot_mirror.notify(notify_type)
+
+    @CLIWriteCommand('fs snapshot mirror enable')
+    def snapshot_mirror_enable(self,
+                               fs_name: str):
+        """Enable snapshot mirroring for a filesystem"""
+        return self.fs_snapshot_mirror.enable_mirror(fs_name)
+
+    @CLIWriteCommand('fs snapshot mirror disable')
+    def snapshot_mirror_disable(self,
+                                fs_name: str):
+        """Disable snapshot mirroring for a filesystem"""
+        return self.fs_snapshot_mirror.disable_mirror(fs_name)
+
+    @CLIWriteCommand('fs snapshot mirror peer_add')
+    def snapshot_mirorr_peer_add(self,
+                                 fs_name: str,
+                                 remote_cluster_spec: str,
+                                 remote_fs_name: Optional[str] = None,
+                                 remote_mon_host: Optional[str] = None,
+                                 cephx_key: Optional[str] = None):
+        """Add a remote filesystem peer"""
+        conf = {}
+        if remote_mon_host and cephx_key:
+            conf['mon_host'] = remote_mon_host
+            conf['key'] = cephx_key
+        return self.fs_snapshot_mirror.peer_add(fs_name, remote_cluster_spec,
+                                                remote_fs_name, remote_conf=conf)
+
+    @CLIReadCommand('fs snapshot mirror peer_list')
+    def snapshot_mirror_peer_list(self,
+                                  fs_name: str):
+        """List configured peers for a file system"""
+        return self.fs_snapshot_mirror.peer_list(fs_name)
+
+    @CLIWriteCommand('fs snapshot mirror peer_remove')
+    def snapshot_mirror_peer_remove(self,
+                                    fs_name: str,
+                                    peer_uuid: str):
+        """Remove a filesystem peer"""
+        return self.fs_snapshot_mirror.peer_remove(fs_name, peer_uuid)
+
+    @CLIWriteCommand('fs snapshot mirror peer_bootstrap create')
+    def snapshot_mirror_peer_bootstrap_create(self,
+                                              fs_name: str,
+                                              client_name: str,
+                                              site_name: str):
+        """Bootstrap a filesystem peer"""
+        return self.fs_snapshot_mirror.peer_bootstrap_create(fs_name, client_name, site_name)
+
+    @CLIWriteCommand('fs snapshot mirror peer_bootstrap import')
+    def snapshot_mirror_peer_bootstrap_import(self,
+                                              fs_name: str,
+                                              token: str):
+        """Import a bootstrap token"""
+        return self.fs_snapshot_mirror.peer_bootstrap_import(fs_name, token)
+
+    @CLIWriteCommand('fs snapshot mirror add')
+    def snapshot_mirror_add_dir(self,
+                                fs_name: str,
+                                path: str):
+        """Add a directory for snapshot mirroring"""
+        return self.fs_snapshot_mirror.add_dir(fs_name, path)
+
+    @CLIWriteCommand('fs snapshot mirror remove')
+    def snapshot_mirror_remove_dir(self,
+                                   fs_name: str,
+                                   path: str):
+        """Remove a snapshot mirrored directory"""
+        return self.fs_snapshot_mirror.remove_dir(fs_name, path)
+
+    @CLIReadCommand('fs snapshot mirror dirmap')
+    def snapshot_mirror_dirmap(self,
+                               fs_name: str,
+                               path: str):
+        """Get current mirror instance map for a directory"""
+        return self.fs_snapshot_mirror.status(fs_name, path)
+
+    @CLIReadCommand('fs snapshot mirror show distribution')
+    def snapshot_mirror_distribution(self,
+                                     fs_name: str):
+        """Get current instance to directory map for a filesystem"""
+        return self.fs_snapshot_mirror.show_distribution(fs_name)
+
+    @CLIReadCommand('fs snapshot mirror daemon status')
+    def snapshot_mirror_daemon_status(self):
+        """Get mirror daemon status"""
+        return self.fs_snapshot_mirror.daemon_status()