untested stab at autocreating autojoin rooms

This commit is contained in:
Matthew Hodgson 2018-09-28 01:42:53 +01:00
parent 36c62a67c4
commit 07340cdaca
2 changed files with 22 additions and 0 deletions

View file

@ -44,6 +44,7 @@ class RegistrationConfig(Config):
)
self.auto_join_rooms = config.get("auto_join_rooms", [])
self.autocreate_auto_join_rooms = config.get("autocreate_auto_join_rooms", true)
def default_config(self, **kwargs):
registration_shared_secret = random_string_with_symbols(50)
@ -98,6 +99,9 @@ class RegistrationConfig(Config):
# to these rooms
#auto_join_rooms:
# - "#example:example.com"
# Have first user on server autocreate autojoin rooms
autocreate_auto_join_rooms: true
""" % locals()
def add_arguments(self, parser):

View file

@ -50,6 +50,8 @@ class RegistrationHandler(BaseHandler):
self._auth_handler = hs.get_auth_handler()
self.profile_handler = hs.get_profile_handler()
self.user_directory_handler = hs.get_user_directory_handler()
self._room_creation_handler = hs.get_room_creation_handler()
self._directory_handler = hs.get_handlers().directory_handler
self.captcha_client = CaptchaServerHttpClient(hs)
self._next_generated_user_id = None
@ -513,6 +515,22 @@ class RegistrationHandler(BaseHandler):
@defer.inlineCallbacks
def _join_user_to_room(self, requester, room_identifier):
# try to create the room if we're the first user on the server
if self.config.autocreate_auto_join_rooms:
count = yield self.store.count_all_users()
if count == 1 and RoomAlias.is_valid(room_identifier):
info = yield self._room_creation_handler.create_room(
requester,
config={
"preset": "public_chat",
},
ratelimit=False,
)
room_id = info["room_id"]
yield create_association(self, requester.user, room_identifier, room_id)
room_id = None
room_member_handler = self.hs.get_room_member_handler()
if RoomID.is_valid(room_identifier):