Add event bus to pool for socket open/close

package-lock.json

@@ -1,12 +1,12 @@
 {
   "name": "paravel",
-  "version": "0.1.5",
+  "version": "0.1.7",
   "lockfileVersion": 2,
   "requires": true,
   "packages": {
     "": {
       "name": "paravel",
-      "version": "0.1.5",
+      "version": "0.1.7",
       "license": "MIT",
       "dependencies": {
         "husky": "^8.0.3",

package.json

@@ -1,6 +1,6 @@
 {
   "name": "paravel",
-  "version": "0.1.6",
+  "version": "0.1.7",
   "description": "Yet another toolkit for nostr",
   "repository": {
     "type": "git",

src/Pool.ts

@@ -1,16 +1,25 @@
 import {Socket} from "./util/Socket"
+import {EventBus} from "./util/EventBus"
 
 export class Pool {
   data: Map<string, Socket>
   constructor() {
     this.data = new Map()
+    this.bus = new EventBus()
   }
   has(url) {
     return this.data.has(url)
   }
   get(url) {
     if (!this.data.has(url)) {
-      this.data.set(url, new Socket(url))
+      const socket = new Socket(url)
+
+      this.data.set(url, socket)
+
+      socket.bus.addListeners({
+        open: () => this.bus.emit('open', {url}),
+        close: () => this.bus.emit('close', {url}),
+      })
     }
 
     return this.data.get(url)
@@ -19,7 +28,7 @@ export class Pool {
     const socket = this.data.get(url)
 
     if (socket) {
-      socket.disconnect()
+      socket.cleanup()
       this.data.delete(url)
     }
   }

src/util/Socket.ts

@@ -31,6 +31,7 @@ export class Socket {
     this._onOpen = e => {
       this.status = Socket.STATUS.READY
       this.ready?.resolve()
+      this.bus.emit('open')
     }
 
     this._onMessage = e => {
@@ -45,6 +46,7 @@ export class Socket {
       this.disconnect()
       this.ready?.reject()
       this.status = Socket.STATUS.CLOSED
+      this.bus.emit('close')
     }
   }
   async connect() {
@@ -65,14 +67,15 @@ export class Socket {
     await this.ready?.catch(() => null)
   }
   disconnect() {
-    if (this.ws) {
-      this.ws.close()
-      this.ws.removeEventListener("open", this._onOpen)
-      this.ws.removeEventListener("message", this._onMessage)
-      this.ws.removeEventListener("error", this._onClose)
-      this.ws.removeEventListener("close", this._onClose)
-      this.ws = undefined
-    }
+    this.ws?.close()
+    this.ws?.removeEventListener("open", this._onOpen)
+    this.ws?.removeEventListener("message", this._onMessage)
+    this.ws?.removeEventListener("close", this._onClose)
+    this.ws = undefined
+  }
+  cleanup() {
+    this.disconnect()
+    this.bus.clear()
   }
   handleMessages() {
     for (const json of this.queue.splice(0, 10)) {