Add some backup checks when stopping maubot

maubot/__main__.py

@@ -90,7 +90,10 @@ except KeyboardInterrupt:
     log.debug("Closing websockets")
     loop.run_until_complete(stop_management_api())
     log.debug("Stopping server")
-    loop.run_until_complete(server.stop())
+    try:
+        loop.run_until_complete(asyncio.wait_for(server.stop(), 5, loop=loop))
+    except asyncio.TimeoutError:
+        log.warning("Stopping server timed out")
     log.debug("Closing event loop")
     loop.close()
     log.debug("Everything stopped, shutting down")

maubot/management/api/log.py

@@ -72,7 +72,7 @@ sockets = []
 async def stop_all() -> None:
     for socket in sockets:
         try:
-            await socket.close(1012)
+            await socket.close(code=1012)
         except Exception:
             pass
 
@@ -106,6 +106,9 @@ async def log_websocket(request: web.Request) -> web.WebSocketResponse:
                 await ws.send_json({"auth_success": True})
             elif not authenticated:
                 await ws.send_json({"auth_success": False})
+    except Exception:
+        try:
+            await ws.close()
         except Exception:
             pass
     log_root.removeHandler(handler)

maubot/server.py

@@ -99,6 +99,7 @@ class MaubotServer:
         self.log.info(f"Listening on {site.name}")
 
     async def stop(self) -> None:
+        await self.runner.shutdown()
         await self.runner.cleanup()
 
     @staticmethod