Added timers cancellation (#7)

theatre/src/theatre/server/room.py

@@ -110,8 +110,15 @@ def cleanup() -> None:
             if len(self._users) == 0:
                 self.emit("empty")
 
+        async def timeout() -> None:
+            if user.id not in self._connected_users:
+                cleanup()
+
+        timer = Timer(60, timeout)
+
         @connection.on("connected")
         def on_connected() -> None:
+            timer.cancel()
             self._connected_users.add(user.id)
             self.handle_connect(user.id)
 
@@ -124,11 +131,5 @@ def on_disconnected() -> None:
         def on_data(message: Union[bytes, str]) -> None:
             self.handle_data(user.id, message)
 
-        async def timeout() -> None:
-            if user.id not in self._connected_users:
-                cleanup()
-
-        Timer(60, timeout)
-
         self._users[user.id] = UserInfo(data=user, connection=connection)
         return user

theatre/src/theatre/server/server.py

@@ -28,15 +28,16 @@ async def close() -> None:
             await room.close()
             self.rooms.pop(code, None)
 
-        @room.on("empty")
-        async def on_empty() -> None:
-            await close()
-
         async def timeout() -> None:
             if len(room.users) == 0:
                 await close()
 
-        Timer(60, timeout)
+        timer = Timer(60, timeout)
+
+        @room.on("empty")
+        async def on_empty() -> None:
+            timer.cancel()
+            await close()
 
         self.rooms[code] = room
         return room