123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081 |
- # -*- coding: utf-8 -*-
- # Copyright 2014-2016 OpenMarket Ltd
- #
- # Licensed under the Apache License, Version 2.0 (the "License");
- # you may not use this file except in compliance with the License.
- # You may obtain a copy of the License at
- #
- # http://www.apache.org/licenses/LICENSE-2.0
- #
- # Unless required by applicable law or agreed to in writing, software
- # distributed under the License is distributed on an "AS IS" BASIS,
- # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- # See the License for the specific language governing permissions and
- # limitations under the License.
- import logging
- from typing import TYPE_CHECKING, Callable, Dict, Optional
- from synapse.push import Pusher, PusherConfig
- from synapse.push.emailpusher import EmailPusher
- from synapse.push.httppusher import HttpPusher
- from synapse.push.mailer import Mailer
- if TYPE_CHECKING:
- from synapse.app.homeserver import HomeServer
- logger = logging.getLogger(__name__)
- class PusherFactory:
- def __init__(self, hs: "HomeServer"):
- self.hs = hs
- self.config = hs.config
- self.pusher_types = {
- "http": HttpPusher
- } # type: Dict[str, Callable[[HomeServer, PusherConfig], Pusher]]
- logger.info("email enable notifs: %r", hs.config.email_enable_notifs)
- if hs.config.email_enable_notifs:
- self.mailers = {} # type: Dict[str, Mailer]
- self._notif_template_html = hs.config.email_notif_template_html
- self._notif_template_text = hs.config.email_notif_template_text
- self.pusher_types["email"] = self._create_email_pusher
- logger.info("defined email pusher type")
- def create_pusher(self, pusher_config: PusherConfig) -> Optional[Pusher]:
- kind = pusher_config.kind
- f = self.pusher_types.get(kind, None)
- if not f:
- return None
- logger.debug("creating %s pusher for %r", kind, pusher_config)
- return f(self.hs, pusher_config)
- def _create_email_pusher(
- self, _hs: "HomeServer", pusher_config: PusherConfig
- ) -> EmailPusher:
- app_name = self._app_name_from_pusherdict(pusher_config)
- mailer = self.mailers.get(app_name)
- if not mailer:
- mailer = Mailer(
- hs=self.hs,
- app_name=app_name,
- template_html=self._notif_template_html,
- template_text=self._notif_template_text,
- )
- self.mailers[app_name] = mailer
- return EmailPusher(self.hs, pusher_config, mailer)
- def _app_name_from_pusherdict(self, pusher_config: PusherConfig) -> str:
- data = pusher_config.data
- if isinstance(data, dict):
- brand = data.get("brand")
- if isinstance(brand, str):
- return brand
- return self.config.email_app_name
|