From 2b3e4e856fd00aee36423c2b4dcdb5b43d3e213c Mon Sep 17 00:00:00 2001
From: Azrenbeth <7782548+Azrenbeth@users.noreply.github.com>
Date: Tue, 24 Aug 2021 11:33:02 +0100
Subject: Port the saml mapping providers to new module interface

---
 synapse/module_api/__init__.py | 8 ++++++++
 synapse/module_api/errors.py   | 1 +
 2 files changed, 9 insertions(+)

(limited to 'synapse/module_api')

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.
 
diff --git a/synapse/module_api/errors.py b/synapse/module_api/errors.py
index 98ea911a81..1560cd1c36 100644
--- a/synapse/module_api/errors.py
+++ b/synapse/module_api/errors.py
@@ -20,3 +20,4 @@ from synapse.api.errors import (  # noqa: F401
     SynapseError,
 )
 from synapse.config._base import ConfigError  # noqa: F401
+from synapse.handlers.sso import MappingException  # noqa: F401
-- 
cgit 1.5.1