From c37d10bb9dfd96c833a43ca2a4f201ba28b62e80 Mon Sep 17 00:00:00 2001
From: rkfg <rkfg@rkfg.me>
Date: Sun, 25 Dec 2022 01:02:37 +0300
Subject: [PATCH] Fix resolveClose

---
 relay.ts | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/relay.ts b/relay.ts
index d9c2b87..4970ef0 100644
--- a/relay.ts
+++ b/relay.ts
@@ -73,7 +73,7 @@ export function relayInit(url: string): Relay {
       }
       ws.onclose = async () => {
         listeners.disconnect.forEach(cb => cb())
-        resolveClose()
+        resolveClose && resolveClose()
       }
 
       ws.onmessage = async e => {
@@ -257,11 +257,10 @@ export function relayInit(url: string): Relay {
     },
     connect,
     close(): Promise<void> {
-      const result = new Promise<void>(resolve => {
+      ws.close()
+      return new Promise<void>(resolve => {
         resolveClose = resolve
       })
-      ws.close()
-      return result
     },
     get status() {
       return ws?.readyState ?? 3