libervia-backend: sat/plugins/plugin_misc_email

comparison sat/plugins/plugin_misc_email_invitation.py @ 3709:09f5ac48ffe3

merge bookmark @

author	Goffi <goffi@goffi.org>
date	Fri, 12 Nov 2021 17:21:24 +0100
parents	cfc06915de15
children	524856bd7b19

comparison

equal deleted inserted replaced

-:8353cc3b8db9
+:09f5ac48ffe3
 # You should have received a copy of the GNU Affero General Public License
 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
 import shortuuid
+from typing import Optional
 from twisted.internet import defer
 from twisted.words.protocols.jabber import jid
 from twisted.words.protocols.jabber import error
 from twisted.words.protocols.jabber import sasl
 from sat.core.i18n import _, D_
 "url_template", "message_subject", "message_body", "profile"):
 value = locals()[key]
 if value:
 kwargs[key] = str(value)
 return defer.ensureDeferred(self.create(**kwargs))
+async def getExistingInvitation(self, email: Optional[str]) -> Optional[dict]:
+"""Retrieve existing invitation with given email
+@param email: check if any invitation exist with this email
+@return: first found invitation, or None if nothing found
+"""
+# FIXME: This method is highly inefficient, it get all invitations and check them
+# one by one, this is just a temporary way to avoid creating creating new accounts
+# for an existing email. A better way will be available with Libervia 0.9.
+# TODO: use a better way to check existing invitations
+if email is None:
+return None
+all_invitations = await self.invitations.all()
+for id_, invitation in all_invitations.items():
+if invitation.get("email") == email:
+invitation[KEY_ID] = id_
+return invitation
+async def _createAccountAndProfile(
+self,
+id_: str,
+kwargs: dict,
+extra: dict
+) -> None:
+"""Create XMPP account and Libervia profile for guest"""
+## XMPP account creation
+password = kwargs.pop('password', None)
+if password is None:
+password = utils.generatePassword()
+assert password
+# XXX: password is here saved in clear in database
+#      it is needed for invitation as the same password is used for profile
+#      and SàT need to be able to automatically open the profile with the uuid
+# FIXME: we could add an extra encryption key which would be used with the
+#        uuid when the invitee is connecting (e.g. with URL). This key would
+#        not be saved and could be used to encrypt profile password.
+extra[KEY_PASSWORD] = password
+jid_ = kwargs.pop('jid_', None)
+if not jid_:
+domain = self.host.memory.getConfig(None, 'xmpp_domain')
+if not domain:
+# TODO: fallback to profile's domain
+raise ValueError(_("You need to specify xmpp_domain in sat.conf"))
+jid_ = "invitation-{uuid}@{domain}".format(uuid=shortuuid.uuid(),
+domain=domain)
+jid_ = jid.JID(jid_)
+extra[KEY_JID] = jid_.full()
+if jid_.user:
+# we don't register account if there is no user as anonymous login is then
+# used
+try:
+await self.host.plugins['XEP-0077'].registerNewAccount(jid_, password)
+except error.StanzaError as e:
+prefix = jid_.user
+idx = 0
+while e.condition == 'conflict':
+if idx >= SUFFIX_MAX:
+raise exceptions.ConflictError(_("Can't create XMPP account"))
+jid_.user = prefix + '_' + str(idx)
+log.info(_("requested jid already exists, trying with {}".format(
+jid_.full())))
+try:
+await self.host.plugins['XEP-0077'].registerNewAccount(
+jid_,
+password
+)
+except error.StanzaError:
+idx += 1
+else:
+break
+if e.condition != 'conflict':
+raise e
+log.info(_("account {jid_} created").format(jid_=jid_.full()))
+## profile creation
+extra[KEY_GUEST_PROFILE] = guest_profile = INVITEE_PROFILE_TPL.format(
+uuid=id_
+)
+# profile creation should not fail as we generate unique name ourselves
+await self.host.memory.createProfile(guest_profile, password)
+await self.host.memory.startSession(password, guest_profile)
+await self.host.memory.setParam("JabberID", jid_.full(), "Connection",
+profile_key=guest_profile)
+await self.host.memory.setParam("Password", password, "Connection",
+profile_key=guest_profile)
 async def create(self, **kwargs):
 r"""Create an invitation
 This will create an XMPP account and a profile, and use a UUID to retrieve them.
 ', '.join(set(kwargs).intersection(extra))))
 self.checkExtra(extra)
 email = kwargs.pop('email', None)
+existing = await self.getExistingInvitation(email)
+if existing is not None:
+log.info(f"There is already an invitation for {email!r}")
+extra.update(existing)
+del extra[KEY_ID]
 emails_extra = kwargs.pop('emails_extra', [])
 if not email and emails_extra:
 raise ValueError(
 _('You need to provide a main email address before using emails_extra'))
 raise ValueError(
 _("You need to provide url_template if you use default message body"))
 ## uuid
 log.info(_("creating an invitation"))
-id_ = str(shortuuid.uuid())
+id_ = existing[KEY_ID] if existing else str(shortuuid.uuid())
-## XMPP account creation
+if existing is None:
-password = kwargs.pop('password', None)
+await self._createAccountAndProfile(id_, kwargs, extra)
-if password is None:
-password = utils.generatePassword()
+profile = kwargs.pop('profile', None)
-assert password
+guest_profile = extra[KEY_GUEST_PROFILE]
-# XXX: password is here saved in clear in database
+jid_ = jid.JID(extra[KEY_JID])
-#      it is needed for invitation as the same password is used for profile
-#      and SàT need to be able to automatically open the profile with the uuid
+## identity
-# FIXME: we could add an extra encryption key which would be used with the uuid
-#        when the invitee is connecting (e.g. with URL). This key would not be
-#        saved and could be used to encrypt profile password.
-extra[KEY_PASSWORD] = password
-jid_ = kwargs.pop('jid_', None)
-if not jid_:
-domain = self.host.memory.getConfig(None, 'xmpp_domain')
-if not domain:
-# TODO: fallback to profile's domain
-raise ValueError(_("You need to specify xmpp_domain in sat.conf"))
-jid_ = "invitation-{uuid}@{domain}".format(uuid=shortuuid.uuid(),
-domain=domain)
-jid_ = jid.JID(jid_)
-if jid_.user:
-# we don't register account if there is no user as anonymous login is then
-# used
-try:
-await self.host.plugins['XEP-0077'].registerNewAccount(jid_, password)
-except error.StanzaError as e:
-prefix = jid_.user
-idx = 0
-while e.condition == 'conflict':
-if idx >= SUFFIX_MAX:
-raise exceptions.ConflictError(_("Can't create XMPP account"))
-jid_.user = prefix + '_' + str(idx)
-log.info(_("requested jid already exists, trying with {}".format(
-jid_.full())))
-try:
-await self.host.plugins['XEP-0077'].registerNewAccount(jid_,
-password)
-except error.StanzaError:
-idx += 1
-else:
-break
-if e.condition != 'conflict':
-raise e
-log.info(_("account {jid_} created").format(jid_=jid_.full()))
-## profile creation
-extra[KEY_GUEST_PROFILE] = guest_profile = INVITEE_PROFILE_TPL.format(uuid=id_)
-# profile creation should not fail as we generate unique name ourselves
-await self.host.memory.createProfile(guest_profile, password)
-await self.host.memory.startSession(password, guest_profile)
-await self.host.memory.setParam("JabberID", jid_.full(), "Connection",
-profile_key=guest_profile)
-await self.host.memory.setParam("Password", password, "Connection",
-profile_key=guest_profile)
 name = kwargs.pop('name', None)
+password = extra[KEY_PASSWORD]
 if name is not None:
 extra['name'] = name
 try:
 id_plugin = self.host.plugins['IDENTITY']
 except KeyError:
 if name is None:
 format_args['name'] = email
 else:
 format_args['name'] = name
-profile = kwargs.pop('profile', None)
 if profile is None:
 format_args['profile'] = ''
 else:
 format_args['profile'] = extra['profile'] = profile
 else:
 await self.host.updateContact(client, jid_, name, ['guests'])
 if kwargs:
 log.warning(_("Not all arguments have been consumed: {}").format(kwargs))
-extra[KEY_JID] = jid_.full()
 ## extra data saving
 self.invitations[id_] = extra
 extra[KEY_ID] = id_

Mercurial > libervia-backend

comparison sat/plugins/plugin_misc_email_invitation.py @ 3709:09f5ac48ffe3