From 185b2abae425ac885b285d5fefc752d838c44d9c Mon Sep 17 00:00:00 2001
From: break27 <buzybreakerbear@gmail.com>
Date: Wed, 10 Sep 2025 18:25:42 +0800
Subject: [PATCH] minor fix

---
 main.py | 10 ++++------
 1 file changed, 4 insertions(+), 6 deletions(-)

diff --git a/main.py b/main.py
index c5b4d61..e372274 100644
--- a/main.py
+++ b/main.py
@@ -45,8 +45,7 @@ parser.add_argument('-r', '--attempts', type=int, default=3)
 
 args = parser.parse_args()
 inbox, outbox = Queue(), Queue()
-connection = None
-socket = None
+connection, server = None, None
 
 class Greetings:
     def __init__(self, locale: str, timezone: str, default: str, morning=None, afternoon=None, evening=None, predicate=None):
@@ -93,10 +92,10 @@ def main(driver: WebDriver):
     try:
         html = str(Path('index.html').resolve())
         driver.get(html)
-        host, port = socket.getsockname()
+        endpoint = server.listeners[0]
         parameters = vars(args)
         locales = json.dumps(Greetings.presets(), default=lambda o: o.__dict__)
-        driver.execute_script(f"main(...arguments);", f'ws://{host}:{port}', parameters, locales)
+        driver.execute_script(f"main(...arguments);", f'ws://{endpoint.address}:{endpoint.port}', parameters, locales)
     except Exception as e:
         tell('载入初始页面时发生错误', e, level=0)
         return 1
@@ -562,10 +561,9 @@ async def handler(request: ws.WebSocketRequest):
     inbox.shutdown(immediate=True)
 
 async def backend(listen='127.0.0.1', port=0):
-    global socket
+    global server
     listeners = await trio.open_tcp_listeners(port, host=listen)
     server = ws.WebSocketServer(handler, listeners, max_message_size=125_000_000)
-    socket = listeners[0].socket
     await server.run()
 
 if __name__ == '__main__':