Port the saml mapping providers to new module interface

1 files changed, 8 insertions, 0 deletions

diff --git a/synapse/module_api/__init__.py b/synapse/module_api/__init__.py

index b11fa6393b..f337a0f65b 100644
--- a/synapse/module_api/__init__.py
+++ b/synapse/module_api/__init__.py
@@ -117,6 +117,7 @@ class ModuleApi:
         self._account_validity_handler = hs.get_account_validity_handler()
         self._third_party_event_rules = hs.get_third_party_event_rules()
         self._presence_router = hs.get_presence_router()
+        self._saml2_user_mapping_provider = hs.get_saml2_user_mapping_provider()
 
     #################################################################################
     # The following methods should only be called during the module's initialisation.
@@ -141,6 +142,13 @@ class ModuleApi:
         """Registers callbacks for presence router capabilities."""
         return self._presence_router.register_presence_router_callbacks
 
+    @property
+    def register_saml2_user_mapping_provider_callbacks(self):
+        """Registers callbacks for presence router capabilities."""
+        return (
+            self._saml2_user_mapping_provider.register_saml2_user_mapping_provider_callbacks
+        )
+
     def register_web_resource(self, path: str, resource: IResource):
         """Registers a web resource to be served at the given path.