mirror of
https://git.anonymousland.org/anonymousland/synapse.git
synced 2024-09-09 15:02:37 +00:00
Only count real users when checking for auto-creation of auto-join room
Previously if the first registered user was a "support" or "bot" user, when the first real user registers, the auto-join rooms were not created. Fix to exclude non-real (ie users with a special user type) users when counting how many users there are to determine whether we should auto-create a room. Signed-off-by: Jason Robinson <jasonr@matrix.org>
This commit is contained in:
parent
55d5b3af88
commit
be618e0551
1
changelog.d/6004.bugfix
Normal file
1
changelog.d/6004.bugfix
Normal file
@ -0,0 +1 @@
|
|||||||
|
Only count real users when checking for auto-creation of auto-join room.
|
@ -275,16 +275,12 @@ class RegistrationHandler(BaseHandler):
|
|||||||
fake_requester = create_requester(user_id)
|
fake_requester = create_requester(user_id)
|
||||||
|
|
||||||
# try to create the room if we're the first real user on the server. Note
|
# try to create the room if we're the first real user on the server. Note
|
||||||
# that an auto-generated support user is not a real user and will never be
|
# that an auto-generated support or bot user is not a real user and will never be
|
||||||
# the user to create the room
|
# the user to create the room
|
||||||
should_auto_create_rooms = False
|
should_auto_create_rooms = False
|
||||||
is_support = yield self.store.is_support_user(user_id)
|
is_real_user = yield self.store.is_real_user(user_id)
|
||||||
# There is an edge case where the first user is the support user, then
|
if self.hs.config.autocreate_auto_join_rooms and is_real_user:
|
||||||
# the room is never created, though this seems unlikely and
|
count = yield self.store.count_real_users()
|
||||||
# recoverable from given the support user being involved in the first
|
|
||||||
# place.
|
|
||||||
if self.hs.config.autocreate_auto_join_rooms and not is_support:
|
|
||||||
count = yield self.store.count_all_users()
|
|
||||||
should_auto_create_rooms = count == 1
|
should_auto_create_rooms = count == 1
|
||||||
for r in self.hs.config.auto_join_rooms:
|
for r in self.hs.config.auto_join_rooms:
|
||||||
logger.info("Auto-joining %s to %s", user_id, r)
|
logger.info("Auto-joining %s to %s", user_id, r)
|
||||||
|
@ -322,6 +322,21 @@ class RegistrationWorkerStore(SQLBaseStore):
|
|||||||
|
|
||||||
return None
|
return None
|
||||||
|
|
||||||
|
@cachedInlineCallbacks()
|
||||||
|
def is_real_user(self, user_id):
|
||||||
|
"""Determines if the user is a real user, ie does not have a 'user_type'.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
user_id (str): user id to test
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Deferred[bool]: True if user 'user_type' is null or empty string
|
||||||
|
"""
|
||||||
|
res = yield self.runInteraction(
|
||||||
|
"is_real_user", self.is_real_user_txn, user_id
|
||||||
|
)
|
||||||
|
return res
|
||||||
|
|
||||||
@cachedInlineCallbacks()
|
@cachedInlineCallbacks()
|
||||||
def is_support_user(self, user_id):
|
def is_support_user(self, user_id):
|
||||||
"""Determines if the user is of type UserTypes.SUPPORT
|
"""Determines if the user is of type UserTypes.SUPPORT
|
||||||
@ -337,6 +352,16 @@ class RegistrationWorkerStore(SQLBaseStore):
|
|||||||
)
|
)
|
||||||
return res
|
return res
|
||||||
|
|
||||||
|
def is_real_user_txn(self, txn, user_id):
|
||||||
|
res = self._simple_select_one_onecol_txn(
|
||||||
|
txn=txn,
|
||||||
|
table="users",
|
||||||
|
keyvalues={"name": user_id},
|
||||||
|
retcol="user_type",
|
||||||
|
allow_none=True,
|
||||||
|
)
|
||||||
|
return True if res is None or res == "" else False
|
||||||
|
|
||||||
def is_support_user_txn(self, txn, user_id):
|
def is_support_user_txn(self, txn, user_id):
|
||||||
res = self._simple_select_one_onecol_txn(
|
res = self._simple_select_one_onecol_txn(
|
||||||
txn=txn,
|
txn=txn,
|
||||||
@ -421,6 +446,20 @@ class RegistrationWorkerStore(SQLBaseStore):
|
|||||||
ret = yield self.runInteraction("count_users", _count_users)
|
ret = yield self.runInteraction("count_users", _count_users)
|
||||||
return ret
|
return ret
|
||||||
|
|
||||||
|
@defer.inlineCallbacks
|
||||||
|
def count_real_users(self):
|
||||||
|
"""Counts all users without a special user_type registered on the homeserver."""
|
||||||
|
|
||||||
|
def _count_users(txn):
|
||||||
|
txn.execute("SELECT COUNT(*) AS users FROM users where user_type is null or user_type = ''")
|
||||||
|
rows = self.cursor_to_dict(txn)
|
||||||
|
if rows:
|
||||||
|
return rows[0]["users"]
|
||||||
|
return 0
|
||||||
|
|
||||||
|
ret = yield self.runInteraction("count_real_users", _count_users)
|
||||||
|
return ret
|
||||||
|
|
||||||
@defer.inlineCallbacks
|
@defer.inlineCallbacks
|
||||||
def find_next_generated_user_id_localpart(self):
|
def find_next_generated_user_id_localpart(self):
|
||||||
"""
|
"""
|
||||||
|
@ -171,11 +171,11 @@ class RegistrationTestCase(unittest.HomeserverTestCase):
|
|||||||
rooms = self.get_success(self.store.get_rooms_for_user(user_id))
|
rooms = self.get_success(self.store.get_rooms_for_user(user_id))
|
||||||
self.assertEqual(len(rooms), 0)
|
self.assertEqual(len(rooms), 0)
|
||||||
|
|
||||||
def test_auto_create_auto_join_rooms_when_support_user_exists(self):
|
def test_auto_create_auto_join_rooms_when_user_is_not_a_real_user(self):
|
||||||
room_alias_str = "#room:test"
|
room_alias_str = "#room:test"
|
||||||
self.hs.config.auto_join_rooms = [room_alias_str]
|
self.hs.config.auto_join_rooms = [room_alias_str]
|
||||||
|
|
||||||
self.store.is_support_user = Mock(return_value=True)
|
self.store.is_real_user = Mock(return_value=False)
|
||||||
user_id = self.get_success(self.handler.register_user(localpart="support"))
|
user_id = self.get_success(self.handler.register_user(localpart="support"))
|
||||||
rooms = self.get_success(self.store.get_rooms_for_user(user_id))
|
rooms = self.get_success(self.store.get_rooms_for_user(user_id))
|
||||||
self.assertEqual(len(rooms), 0)
|
self.assertEqual(len(rooms), 0)
|
||||||
@ -183,6 +183,31 @@ class RegistrationTestCase(unittest.HomeserverTestCase):
|
|||||||
room_alias = RoomAlias.from_string(room_alias_str)
|
room_alias = RoomAlias.from_string(room_alias_str)
|
||||||
self.get_failure(directory_handler.get_association(room_alias), SynapseError)
|
self.get_failure(directory_handler.get_association(room_alias), SynapseError)
|
||||||
|
|
||||||
|
def test_auto_create_auto_join_rooms_when_user_is_the_first_real_user(self):
|
||||||
|
room_alias_str = "#room:test"
|
||||||
|
self.hs.config.auto_join_rooms = [room_alias_str]
|
||||||
|
|
||||||
|
self.store.count_real_users = Mock(return_value=1)
|
||||||
|
self.store.is_real_user = Mock(return_value=True)
|
||||||
|
user_id = self.get_success(self.handler.register_user(localpart="real"))
|
||||||
|
rooms = self.get_success(self.store.get_rooms_for_user(user_id))
|
||||||
|
directory_handler = self.hs.get_handlers().directory_handler
|
||||||
|
room_alias = RoomAlias.from_string(room_alias_str)
|
||||||
|
room_id = self.get_success(directory_handler.get_association(room_alias))
|
||||||
|
|
||||||
|
self.assertTrue(room_id["room_id"] in rooms)
|
||||||
|
self.assertEqual(len(rooms), 1)
|
||||||
|
|
||||||
|
def test_auto_create_auto_join_rooms_when_user_is_not_the_first_real_user(self):
|
||||||
|
room_alias_str = "#room:test"
|
||||||
|
self.hs.config.auto_join_rooms = [room_alias_str]
|
||||||
|
|
||||||
|
self.store.count_real_users = Mock(return_value=2)
|
||||||
|
self.store.is_real_user = Mock(return_value=True)
|
||||||
|
user_id = self.get_success(self.handler.register_user(localpart="real"))
|
||||||
|
rooms = self.get_success(self.store.get_rooms_for_user(user_id))
|
||||||
|
self.assertEqual(len(rooms), 0)
|
||||||
|
|
||||||
def test_auto_create_auto_join_where_no_consent(self):
|
def test_auto_create_auto_join_where_no_consent(self):
|
||||||
"""Test to ensure that the first user is not auto-joined to a room if
|
"""Test to ensure that the first user is not auto-joined to a room if
|
||||||
they have not given general consent.
|
they have not given general consent.
|
||||||
|
Loading…
Reference in New Issue
Block a user