room.py 33 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934
  1. # -*- coding: utf-8 -*-
  2. # Copyright 2014 - 2016 OpenMarket Ltd
  3. # Copyright 2018 New Vector Ltd
  4. #
  5. # Licensed under the Apache License, Version 2.0 (the "License");
  6. # you may not use this file except in compliance with the License.
  7. # You may obtain a copy of the License at
  8. #
  9. # http://www.apache.org/licenses/LICENSE-2.0
  10. #
  11. # Unless required by applicable law or agreed to in writing, software
  12. # distributed under the License is distributed on an "AS IS" BASIS,
  13. # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14. # See the License for the specific language governing permissions and
  15. # limitations under the License.
  16. """Contains functions for performing events on rooms."""
  17. import itertools
  18. import logging
  19. import math
  20. import string
  21. from collections import OrderedDict
  22. from six import iteritems, string_types
  23. from twisted.internet import defer
  24. from synapse.api.constants import (
  25. DEFAULT_ROOM_VERSION,
  26. KNOWN_ROOM_VERSIONS,
  27. EventTypes,
  28. JoinRules,
  29. RoomCreationPreset,
  30. )
  31. from synapse.api.errors import AuthError, Codes, NotFoundError, StoreError, SynapseError
  32. from synapse.storage.state import StateFilter
  33. from synapse.types import RoomAlias, RoomID, RoomStreamToken, StreamToken, UserID
  34. from synapse.util import stringutils
  35. from synapse.util.async_helpers import Linearizer
  36. from synapse.visibility import filter_events_for_client
  37. from ._base import BaseHandler
  38. logger = logging.getLogger(__name__)
  39. id_server_scheme = "https://"
  40. class RoomCreationHandler(BaseHandler):
  41. PRESETS_DICT = {
  42. RoomCreationPreset.PRIVATE_CHAT: {
  43. "join_rules": JoinRules.INVITE,
  44. "history_visibility": "shared",
  45. "original_invitees_have_ops": False,
  46. "guest_can_join": True,
  47. },
  48. RoomCreationPreset.TRUSTED_PRIVATE_CHAT: {
  49. "join_rules": JoinRules.INVITE,
  50. "history_visibility": "shared",
  51. "original_invitees_have_ops": True,
  52. "guest_can_join": True,
  53. },
  54. RoomCreationPreset.PUBLIC_CHAT: {
  55. "join_rules": JoinRules.PUBLIC,
  56. "history_visibility": "shared",
  57. "original_invitees_have_ops": False,
  58. "guest_can_join": False,
  59. },
  60. }
  61. def __init__(self, hs):
  62. super(RoomCreationHandler, self).__init__(hs)
  63. self.spam_checker = hs.get_spam_checker()
  64. self.event_creation_handler = hs.get_event_creation_handler()
  65. self.room_member_handler = hs.get_room_member_handler()
  66. # linearizer to stop two upgrades happening at once
  67. self._upgrade_linearizer = Linearizer("room_upgrade_linearizer")
  68. @defer.inlineCallbacks
  69. def upgrade_room(self, requester, old_room_id, new_version):
  70. """Replace a room with a new room with a different version
  71. Args:
  72. requester (synapse.types.Requester): the user requesting the upgrade
  73. old_room_id (unicode): the id of the room to be replaced
  74. new_version (unicode): the new room version to use
  75. Returns:
  76. Deferred[unicode]: the new room id
  77. """
  78. yield self.ratelimit(requester)
  79. user_id = requester.user.to_string()
  80. with (yield self._upgrade_linearizer.queue(old_room_id)):
  81. # start by allocating a new room id
  82. r = yield self.store.get_room(old_room_id)
  83. if r is None:
  84. raise NotFoundError("Unknown room id %s" % (old_room_id,))
  85. new_room_id = yield self._generate_room_id(
  86. creator_id=user_id, is_public=r["is_public"],
  87. )
  88. logger.info("Creating new room %s to replace %s", new_room_id, old_room_id)
  89. # we create and auth the tombstone event before properly creating the new
  90. # room, to check our user has perms in the old room.
  91. tombstone_event, tombstone_context = (
  92. yield self.event_creation_handler.create_event(
  93. requester, {
  94. "type": EventTypes.Tombstone,
  95. "state_key": "",
  96. "room_id": old_room_id,
  97. "sender": user_id,
  98. "content": {
  99. "body": "This room has been replaced",
  100. "replacement_room": new_room_id,
  101. }
  102. },
  103. token_id=requester.access_token_id,
  104. )
  105. )
  106. yield self.auth.check_from_context(tombstone_event, tombstone_context)
  107. yield self.clone_exiting_room(
  108. requester,
  109. old_room_id=old_room_id,
  110. new_room_id=new_room_id,
  111. new_room_version=new_version,
  112. tombstone_event_id=tombstone_event.event_id,
  113. )
  114. # now send the tombstone
  115. yield self.event_creation_handler.send_nonmember_event(
  116. requester, tombstone_event, tombstone_context,
  117. )
  118. old_room_state = yield tombstone_context.get_current_state_ids(self.store)
  119. # update any aliases
  120. yield self._move_aliases_to_new_room(
  121. requester, old_room_id, new_room_id, old_room_state,
  122. )
  123. # and finally, shut down the PLs in the old room, and update them in the new
  124. # room.
  125. yield self._update_upgraded_room_pls(
  126. requester, old_room_id, new_room_id, old_room_state,
  127. )
  128. defer.returnValue(new_room_id)
  129. @defer.inlineCallbacks
  130. def _update_upgraded_room_pls(
  131. self, requester, old_room_id, new_room_id, old_room_state,
  132. ):
  133. """Send updated power levels in both rooms after an upgrade
  134. Args:
  135. requester (synapse.types.Requester): the user requesting the upgrade
  136. old_room_id (unicode): the id of the room to be replaced
  137. new_room_id (unicode): the id of the replacement room
  138. old_room_state (dict[tuple[str, str], str]): the state map for the old room
  139. Returns:
  140. Deferred
  141. """
  142. old_room_pl_event_id = old_room_state.get((EventTypes.PowerLevels, ""))
  143. if old_room_pl_event_id is None:
  144. logger.warning(
  145. "Not supported: upgrading a room with no PL event. Not setting PLs "
  146. "in old room.",
  147. )
  148. return
  149. old_room_pl_state = yield self.store.get_event(old_room_pl_event_id)
  150. # we try to stop regular users from speaking by setting the PL required
  151. # to send regular events and invites to 'Moderator' level. That's normally
  152. # 50, but if the default PL in a room is 50 or more, then we set the
  153. # required PL above that.
  154. pl_content = dict(old_room_pl_state.content)
  155. users_default = int(pl_content.get("users_default", 0))
  156. restricted_level = max(users_default + 1, 50)
  157. updated = False
  158. for v in ("invite", "events_default"):
  159. current = int(pl_content.get(v, 0))
  160. if current < restricted_level:
  161. logger.info(
  162. "Setting level for %s in %s to %i (was %i)",
  163. v, old_room_id, restricted_level, current,
  164. )
  165. pl_content[v] = restricted_level
  166. updated = True
  167. else:
  168. logger.info(
  169. "Not setting level for %s (already %i)",
  170. v, current,
  171. )
  172. if updated:
  173. try:
  174. yield self.event_creation_handler.create_and_send_nonmember_event(
  175. requester, {
  176. "type": EventTypes.PowerLevels,
  177. "state_key": '',
  178. "room_id": old_room_id,
  179. "sender": requester.user.to_string(),
  180. "content": pl_content,
  181. }, ratelimit=False,
  182. )
  183. except AuthError as e:
  184. logger.warning("Unable to update PLs in old room: %s", e)
  185. logger.info("Setting correct PLs in new room")
  186. yield self.event_creation_handler.create_and_send_nonmember_event(
  187. requester, {
  188. "type": EventTypes.PowerLevels,
  189. "state_key": '',
  190. "room_id": new_room_id,
  191. "sender": requester.user.to_string(),
  192. "content": old_room_pl_state.content,
  193. }, ratelimit=False,
  194. )
  195. @defer.inlineCallbacks
  196. def clone_exiting_room(
  197. self, requester, old_room_id, new_room_id, new_room_version,
  198. tombstone_event_id,
  199. ):
  200. """Populate a new room based on an old room
  201. Args:
  202. requester (synapse.types.Requester): the user requesting the upgrade
  203. old_room_id (unicode): the id of the room to be replaced
  204. new_room_id (unicode): the id to give the new room (should already have been
  205. created with _gemerate_room_id())
  206. new_room_version (unicode): the new room version to use
  207. tombstone_event_id (unicode|str): the ID of the tombstone event in the old
  208. room.
  209. Returns:
  210. Deferred[None]
  211. """
  212. user_id = requester.user.to_string()
  213. if not self.spam_checker.user_may_create_room(user_id):
  214. raise SynapseError(403, "You are not permitted to create rooms")
  215. creation_content = {
  216. "room_version": new_room_version,
  217. "predecessor": {
  218. "room_id": old_room_id,
  219. "event_id": tombstone_event_id,
  220. }
  221. }
  222. initial_state = dict()
  223. types_to_copy = (
  224. (EventTypes.JoinRules, ""),
  225. (EventTypes.Name, ""),
  226. (EventTypes.Topic, ""),
  227. (EventTypes.RoomHistoryVisibility, ""),
  228. (EventTypes.GuestAccess, ""),
  229. (EventTypes.RoomAvatar, ""),
  230. )
  231. old_room_state_ids = yield self.store.get_filtered_current_state_ids(
  232. old_room_id, StateFilter.from_types(types_to_copy),
  233. )
  234. # map from event_id to BaseEvent
  235. old_room_state_events = yield self.store.get_events(old_room_state_ids.values())
  236. for k, old_event_id in iteritems(old_room_state_ids):
  237. old_event = old_room_state_events.get(old_event_id)
  238. if old_event:
  239. initial_state[k] = old_event.content
  240. yield self._send_events_for_new_room(
  241. requester,
  242. new_room_id,
  243. # we expect to override all the presets with initial_state, so this is
  244. # somewhat arbitrary.
  245. preset_config=RoomCreationPreset.PRIVATE_CHAT,
  246. invite_list=[],
  247. initial_state=initial_state,
  248. creation_content=creation_content,
  249. )
  250. # XXX invites/joins
  251. # XXX 3pid invites
  252. @defer.inlineCallbacks
  253. def _move_aliases_to_new_room(
  254. self, requester, old_room_id, new_room_id, old_room_state,
  255. ):
  256. directory_handler = self.hs.get_handlers().directory_handler
  257. aliases = yield self.store.get_aliases_for_room(old_room_id)
  258. # check to see if we have a canonical alias.
  259. canonical_alias = None
  260. canonical_alias_event_id = old_room_state.get((EventTypes.CanonicalAlias, ""))
  261. if canonical_alias_event_id:
  262. canonical_alias_event = yield self.store.get_event(canonical_alias_event_id)
  263. if canonical_alias_event:
  264. canonical_alias = canonical_alias_event.content.get("alias", "")
  265. # first we try to remove the aliases from the old room (we suppress sending
  266. # the room_aliases event until the end).
  267. #
  268. # Note that we'll only be able to remove aliases that (a) aren't owned by an AS,
  269. # and (b) unless the user is a server admin, which the user created.
  270. #
  271. # This is probably correct - given we don't allow such aliases to be deleted
  272. # normally, it would be odd to allow it in the case of doing a room upgrade -
  273. # but it makes the upgrade less effective, and you have to wonder why a room
  274. # admin can't remove aliases that point to that room anyway.
  275. # (cf https://github.com/matrix-org/synapse/issues/2360)
  276. #
  277. removed_aliases = []
  278. for alias_str in aliases:
  279. alias = RoomAlias.from_string(alias_str)
  280. try:
  281. yield directory_handler.delete_association(
  282. requester, alias, send_event=False,
  283. )
  284. removed_aliases.append(alias_str)
  285. except SynapseError as e:
  286. logger.warning(
  287. "Unable to remove alias %s from old room: %s",
  288. alias, e,
  289. )
  290. # if we didn't find any aliases, or couldn't remove anyway, we can skip the rest
  291. # of this.
  292. if not removed_aliases:
  293. return
  294. try:
  295. # this can fail if, for some reason, our user doesn't have perms to send
  296. # m.room.aliases events in the old room (note that we've already checked that
  297. # they have perms to send a tombstone event, so that's not terribly likely).
  298. #
  299. # If that happens, it's regrettable, but we should carry on: it's the same
  300. # as when you remove an alias from the directory normally - it just means that
  301. # the aliases event gets out of sync with the directory
  302. # (cf https://github.com/vector-im/riot-web/issues/2369)
  303. yield directory_handler.send_room_alias_update_event(
  304. requester, old_room_id,
  305. )
  306. except AuthError as e:
  307. logger.warning(
  308. "Failed to send updated alias event on old room: %s", e,
  309. )
  310. # we can now add any aliases we successfully removed to the new room.
  311. for alias in removed_aliases:
  312. try:
  313. yield directory_handler.create_association(
  314. requester, RoomAlias.from_string(alias),
  315. new_room_id, servers=(self.hs.hostname, ),
  316. send_event=False,
  317. )
  318. logger.info("Moved alias %s to new room", alias)
  319. except SynapseError as e:
  320. # I'm not really expecting this to happen, but it could if the spam
  321. # checking module decides it shouldn't, or similar.
  322. logger.error(
  323. "Error adding alias %s to new room: %s",
  324. alias, e,
  325. )
  326. try:
  327. if canonical_alias and (canonical_alias in removed_aliases):
  328. yield self.event_creation_handler.create_and_send_nonmember_event(
  329. requester,
  330. {
  331. "type": EventTypes.CanonicalAlias,
  332. "state_key": "",
  333. "room_id": new_room_id,
  334. "sender": requester.user.to_string(),
  335. "content": {"alias": canonical_alias, },
  336. },
  337. ratelimit=False
  338. )
  339. yield directory_handler.send_room_alias_update_event(
  340. requester, new_room_id,
  341. )
  342. except SynapseError as e:
  343. # again I'm not really expecting this to fail, but if it does, I'd rather
  344. # we returned the new room to the client at this point.
  345. logger.error(
  346. "Unable to send updated alias events in new room: %s", e,
  347. )
  348. @defer.inlineCallbacks
  349. def create_room(self, requester, config, ratelimit=True,
  350. creator_join_profile=None):
  351. """ Creates a new room.
  352. Args:
  353. requester (synapse.types.Requester):
  354. The user who requested the room creation.
  355. config (dict) : A dict of configuration options.
  356. ratelimit (bool): set to False to disable the rate limiter
  357. creator_join_profile (dict|None):
  358. Set to override the displayname and avatar for the creating
  359. user in this room. If unset, displayname and avatar will be
  360. derived from the user's profile. If set, should contain the
  361. values to go in the body of the 'join' event (typically
  362. `avatar_url` and/or `displayname`.
  363. Returns:
  364. Deferred[dict]:
  365. a dict containing the keys `room_id` and, if an alias was
  366. requested, `room_alias`.
  367. Raises:
  368. SynapseError if the room ID couldn't be stored, or something went
  369. horribly wrong.
  370. ResourceLimitError if server is blocked to some resource being
  371. exceeded
  372. """
  373. user_id = requester.user.to_string()
  374. yield self.auth.check_auth_blocking(user_id)
  375. if not self.spam_checker.user_may_create_room(user_id):
  376. raise SynapseError(403, "You are not permitted to create rooms")
  377. if ratelimit:
  378. yield self.ratelimit(requester)
  379. room_version = config.get("room_version", DEFAULT_ROOM_VERSION)
  380. if not isinstance(room_version, string_types):
  381. raise SynapseError(
  382. 400,
  383. "room_version must be a string",
  384. Codes.BAD_JSON,
  385. )
  386. if room_version not in KNOWN_ROOM_VERSIONS:
  387. raise SynapseError(
  388. 400,
  389. "Your homeserver does not support this room version",
  390. Codes.UNSUPPORTED_ROOM_VERSION,
  391. )
  392. if "room_alias_name" in config:
  393. for wchar in string.whitespace:
  394. if wchar in config["room_alias_name"]:
  395. raise SynapseError(400, "Invalid characters in room alias")
  396. room_alias = RoomAlias(
  397. config["room_alias_name"],
  398. self.hs.hostname,
  399. )
  400. mapping = yield self.store.get_association_from_room_alias(
  401. room_alias
  402. )
  403. if mapping:
  404. raise SynapseError(
  405. 400,
  406. "Room alias already taken",
  407. Codes.ROOM_IN_USE
  408. )
  409. else:
  410. room_alias = None
  411. invite_list = config.get("invite", [])
  412. for i in invite_list:
  413. try:
  414. UserID.from_string(i)
  415. except Exception:
  416. raise SynapseError(400, "Invalid user_id: %s" % (i,))
  417. yield self.event_creation_handler.assert_accepted_privacy_policy(
  418. requester,
  419. )
  420. invite_3pid_list = config.get("invite_3pid", [])
  421. visibility = config.get("visibility", None)
  422. is_public = visibility == "public"
  423. room_id = yield self._generate_room_id(creator_id=user_id, is_public=is_public)
  424. if room_alias:
  425. directory_handler = self.hs.get_handlers().directory_handler
  426. yield directory_handler.create_association(
  427. requester=requester,
  428. room_id=room_id,
  429. room_alias=room_alias,
  430. servers=[self.hs.hostname],
  431. send_event=False,
  432. )
  433. preset_config = config.get(
  434. "preset",
  435. RoomCreationPreset.PRIVATE_CHAT
  436. if visibility == "private"
  437. else RoomCreationPreset.PUBLIC_CHAT
  438. )
  439. raw_initial_state = config.get("initial_state", [])
  440. initial_state = OrderedDict()
  441. for val in raw_initial_state:
  442. initial_state[(val["type"], val.get("state_key", ""))] = val["content"]
  443. creation_content = config.get("creation_content", {})
  444. # override any attempt to set room versions via the creation_content
  445. creation_content["room_version"] = room_version
  446. yield self._send_events_for_new_room(
  447. requester,
  448. room_id,
  449. preset_config=preset_config,
  450. invite_list=invite_list,
  451. initial_state=initial_state,
  452. creation_content=creation_content,
  453. room_alias=room_alias,
  454. power_level_content_override=config.get("power_level_content_override"),
  455. creator_join_profile=creator_join_profile,
  456. )
  457. if "name" in config:
  458. name = config["name"]
  459. yield self.event_creation_handler.create_and_send_nonmember_event(
  460. requester,
  461. {
  462. "type": EventTypes.Name,
  463. "room_id": room_id,
  464. "sender": user_id,
  465. "state_key": "",
  466. "content": {"name": name},
  467. },
  468. ratelimit=False)
  469. if "topic" in config:
  470. topic = config["topic"]
  471. yield self.event_creation_handler.create_and_send_nonmember_event(
  472. requester,
  473. {
  474. "type": EventTypes.Topic,
  475. "room_id": room_id,
  476. "sender": user_id,
  477. "state_key": "",
  478. "content": {"topic": topic},
  479. },
  480. ratelimit=False)
  481. for invitee in invite_list:
  482. content = {}
  483. is_direct = config.get("is_direct", None)
  484. if is_direct:
  485. content["is_direct"] = is_direct
  486. yield self.room_member_handler.update_membership(
  487. requester,
  488. UserID.from_string(invitee),
  489. room_id,
  490. "invite",
  491. ratelimit=False,
  492. content=content,
  493. )
  494. for invite_3pid in invite_3pid_list:
  495. id_server = invite_3pid["id_server"]
  496. address = invite_3pid["address"]
  497. medium = invite_3pid["medium"]
  498. yield self.hs.get_room_member_handler().do_3pid_invite(
  499. room_id,
  500. requester.user,
  501. medium,
  502. address,
  503. id_server,
  504. requester,
  505. txn_id=None,
  506. )
  507. result = {"room_id": room_id}
  508. if room_alias:
  509. result["room_alias"] = room_alias.to_string()
  510. yield directory_handler.send_room_alias_update_event(
  511. requester, room_id
  512. )
  513. defer.returnValue(result)
  514. @defer.inlineCallbacks
  515. def _send_events_for_new_room(
  516. self,
  517. creator, # A Requester object.
  518. room_id,
  519. preset_config,
  520. invite_list,
  521. initial_state,
  522. creation_content,
  523. room_alias=None,
  524. power_level_content_override=None,
  525. creator_join_profile=None,
  526. ):
  527. def create(etype, content, **kwargs):
  528. e = {
  529. "type": etype,
  530. "content": content,
  531. }
  532. e.update(event_keys)
  533. e.update(kwargs)
  534. return e
  535. @defer.inlineCallbacks
  536. def send(etype, content, **kwargs):
  537. event = create(etype, content, **kwargs)
  538. logger.info("Sending %s in new room", etype)
  539. yield self.event_creation_handler.create_and_send_nonmember_event(
  540. creator,
  541. event,
  542. ratelimit=False
  543. )
  544. config = RoomCreationHandler.PRESETS_DICT[preset_config]
  545. creator_id = creator.user.to_string()
  546. event_keys = {
  547. "room_id": room_id,
  548. "sender": creator_id,
  549. "state_key": "",
  550. }
  551. creation_content.update({"creator": creator_id})
  552. yield send(
  553. etype=EventTypes.Create,
  554. content=creation_content,
  555. )
  556. logger.info("Sending %s in new room", EventTypes.Member)
  557. yield self.room_member_handler.update_membership(
  558. creator,
  559. creator.user,
  560. room_id,
  561. "join",
  562. ratelimit=False,
  563. content=creator_join_profile,
  564. )
  565. # We treat the power levels override specially as this needs to be one
  566. # of the first events that get sent into a room.
  567. pl_content = initial_state.pop((EventTypes.PowerLevels, ''), None)
  568. if pl_content is not None:
  569. yield send(
  570. etype=EventTypes.PowerLevels,
  571. content=pl_content,
  572. )
  573. else:
  574. power_level_content = {
  575. "users": {
  576. creator_id: 100,
  577. },
  578. "users_default": 0,
  579. "events": {
  580. EventTypes.Name: 50,
  581. EventTypes.PowerLevels: 100,
  582. EventTypes.RoomHistoryVisibility: 100,
  583. EventTypes.CanonicalAlias: 50,
  584. EventTypes.RoomAvatar: 50,
  585. },
  586. "events_default": 0,
  587. "state_default": 50,
  588. "ban": 50,
  589. "kick": 50,
  590. "redact": 50,
  591. "invite": 0,
  592. }
  593. if config["original_invitees_have_ops"]:
  594. for invitee in invite_list:
  595. power_level_content["users"][invitee] = 100
  596. if power_level_content_override:
  597. power_level_content.update(power_level_content_override)
  598. yield send(
  599. etype=EventTypes.PowerLevels,
  600. content=power_level_content,
  601. )
  602. if room_alias and (EventTypes.CanonicalAlias, '') not in initial_state:
  603. yield send(
  604. etype=EventTypes.CanonicalAlias,
  605. content={"alias": room_alias.to_string()},
  606. )
  607. if (EventTypes.JoinRules, '') not in initial_state:
  608. yield send(
  609. etype=EventTypes.JoinRules,
  610. content={"join_rule": config["join_rules"]},
  611. )
  612. if (EventTypes.RoomHistoryVisibility, '') not in initial_state:
  613. yield send(
  614. etype=EventTypes.RoomHistoryVisibility,
  615. content={"history_visibility": config["history_visibility"]}
  616. )
  617. if config["guest_can_join"]:
  618. if (EventTypes.GuestAccess, '') not in initial_state:
  619. yield send(
  620. etype=EventTypes.GuestAccess,
  621. content={"guest_access": "can_join"}
  622. )
  623. for (etype, state_key), content in initial_state.items():
  624. yield send(
  625. etype=etype,
  626. state_key=state_key,
  627. content=content,
  628. )
  629. @defer.inlineCallbacks
  630. def _generate_room_id(self, creator_id, is_public):
  631. # autogen room IDs and try to create it. We may clash, so just
  632. # try a few times till one goes through, giving up eventually.
  633. attempts = 0
  634. while attempts < 5:
  635. try:
  636. random_string = stringutils.random_string(18)
  637. gen_room_id = RoomID(
  638. random_string,
  639. self.hs.hostname,
  640. ).to_string()
  641. if isinstance(gen_room_id, bytes):
  642. gen_room_id = gen_room_id.decode('utf-8')
  643. yield self.store.store_room(
  644. room_id=gen_room_id,
  645. room_creator_user_id=creator_id,
  646. is_public=is_public,
  647. )
  648. defer.returnValue(gen_room_id)
  649. except StoreError:
  650. attempts += 1
  651. raise StoreError(500, "Couldn't generate a room ID.")
  652. class RoomContextHandler(object):
  653. def __init__(self, hs):
  654. self.hs = hs
  655. self.store = hs.get_datastore()
  656. @defer.inlineCallbacks
  657. def get_event_context(self, user, room_id, event_id, limit, event_filter):
  658. """Retrieves events, pagination tokens and state around a given event
  659. in a room.
  660. Args:
  661. user (UserID)
  662. room_id (str)
  663. event_id (str)
  664. limit (int): The maximum number of events to return in total
  665. (excluding state).
  666. event_filter (Filter|None): the filter to apply to the events returned
  667. (excluding the target event_id)
  668. Returns:
  669. dict, or None if the event isn't found
  670. """
  671. before_limit = math.floor(limit / 2.)
  672. after_limit = limit - before_limit
  673. users = yield self.store.get_users_in_room(room_id)
  674. is_peeking = user.to_string() not in users
  675. def filter_evts(events):
  676. return filter_events_for_client(
  677. self.store,
  678. user.to_string(),
  679. events,
  680. is_peeking=is_peeking
  681. )
  682. event = yield self.store.get_event(event_id, get_prev_content=True,
  683. allow_none=True)
  684. if not event:
  685. defer.returnValue(None)
  686. return
  687. filtered = yield(filter_evts([event]))
  688. if not filtered:
  689. raise AuthError(
  690. 403,
  691. "You don't have permission to access that event."
  692. )
  693. results = yield self.store.get_events_around(
  694. room_id, event_id, before_limit, after_limit, event_filter
  695. )
  696. results["events_before"] = yield filter_evts(results["events_before"])
  697. results["events_after"] = yield filter_evts(results["events_after"])
  698. results["event"] = event
  699. if results["events_after"]:
  700. last_event_id = results["events_after"][-1].event_id
  701. else:
  702. last_event_id = event_id
  703. if event_filter and event_filter.lazy_load_members():
  704. state_filter = StateFilter.from_lazy_load_member_list(
  705. ev.sender
  706. for ev in itertools.chain(
  707. results["events_before"],
  708. (results["event"],),
  709. results["events_after"],
  710. )
  711. )
  712. else:
  713. state_filter = StateFilter.all()
  714. # XXX: why do we return the state as of the last event rather than the
  715. # first? Shouldn't we be consistent with /sync?
  716. # https://github.com/matrix-org/matrix-doc/issues/687
  717. state = yield self.store.get_state_for_events(
  718. [last_event_id], state_filter=state_filter,
  719. )
  720. results["state"] = list(state[last_event_id].values())
  721. # We use a dummy token here as we only care about the room portion of
  722. # the token, which we replace.
  723. token = StreamToken.START
  724. results["start"] = token.copy_and_replace(
  725. "room_key", results["start"]
  726. ).to_string()
  727. results["end"] = token.copy_and_replace(
  728. "room_key", results["end"]
  729. ).to_string()
  730. defer.returnValue(results)
  731. class RoomEventSource(object):
  732. def __init__(self, hs):
  733. self.store = hs.get_datastore()
  734. @defer.inlineCallbacks
  735. def get_new_events(
  736. self,
  737. user,
  738. from_key,
  739. limit,
  740. room_ids,
  741. is_guest,
  742. explicit_room_id=None,
  743. ):
  744. # We just ignore the key for now.
  745. to_key = yield self.get_current_key()
  746. from_token = RoomStreamToken.parse(from_key)
  747. if from_token.topological:
  748. logger.warn("Stream has topological part!!!! %r", from_key)
  749. from_key = "s%s" % (from_token.stream,)
  750. app_service = self.store.get_app_service_by_user_id(
  751. user.to_string()
  752. )
  753. if app_service:
  754. # We no longer support AS users using /sync directly.
  755. # See https://github.com/matrix-org/matrix-doc/issues/1144
  756. raise NotImplementedError()
  757. else:
  758. room_events = yield self.store.get_membership_changes_for_user(
  759. user.to_string(), from_key, to_key
  760. )
  761. room_to_events = yield self.store.get_room_events_stream_for_rooms(
  762. room_ids=room_ids,
  763. from_key=from_key,
  764. to_key=to_key,
  765. limit=limit or 10,
  766. order='ASC',
  767. )
  768. events = list(room_events)
  769. events.extend(e for evs, _ in room_to_events.values() for e in evs)
  770. events.sort(key=lambda e: e.internal_metadata.order)
  771. if limit:
  772. events[:] = events[:limit]
  773. if events:
  774. end_key = events[-1].internal_metadata.after
  775. else:
  776. end_key = to_key
  777. defer.returnValue((events, end_key))
  778. def get_current_key(self):
  779. return self.store.get_room_events_max_id()
  780. def get_current_key_for_room(self, room_id):
  781. return self.store.get_room_events_max_id(room_id)
  782. @defer.inlineCallbacks
  783. def get_pagination_rows(self, user, config, key):
  784. events, next_key = yield self.store.paginate_room_events(
  785. room_id=key,
  786. from_key=config.from_key,
  787. to_key=config.to_key,
  788. direction=config.direction,
  789. limit=config.limit,
  790. )
  791. defer.returnValue((events, next_key))