Improve spec, clean up connection sender a bit

packages/net/src/ConnectionSender.ts

@@ -1,4 +1,4 @@
-import {Worker} from "@welshman/lib"
+import {Worker, complement, spec} from "@welshman/lib"
 import {AUTH_JOIN} from "@welshman/util"
 import {SocketStatus} from "./Socket.js"
 import type {Message} from "./Socket.js"
@@ -10,8 +10,10 @@ export class ConnectionSender {
 
   constructor(readonly cxn: Connection) {
     this.worker = new Worker({
-      shouldDefer: ([verb, ...extra]: Message) => {
-        // Always send CLOSE to clean up pending requests, even if the connection is closed
+      shouldDefer: (message: Message) => {
+        const verb = message[0]
+
+        // Always send CLOSE to clean up pending requests
         if (verb === "CLOSE") return false
 
         // If we're not connected, nothing we can do
@@ -21,7 +23,7 @@ export class ConnectionSender {
         if (verb === "AUTH") return false
 
         // Always allow sending join requests
-        if (verb === "EVENT" && extra[0].kind === AUTH_JOIN) return false
+        if (verb === "EVENT" && message[1].kind === AUTH_JOIN) return false
 
         // Wait for auth
         if (![AuthStatus.None, AuthStatus.Ok].includes(cxn.auth.status)) return true
@@ -33,20 +35,24 @@ export class ConnectionSender {
       },
     })
 
-    this.worker.addGlobalHandler(([verb, ...extra]: Message) => {
-      // If we ended up handling a CLOSE before we handled the REQ, don't send the REQ
-      if (verb === "CLOSE") {
-        this.worker.buffer = this.worker.buffer.filter(m => !(m[0] === "REQ" && m[1] === extra[0]))
+    this.worker.addGlobalHandler((message: Message) => {
+      const verb = message[0]
+
+      // If we're closing something that never got sent, skip it
+      if (verb === "CLOSE" && !cxn.state.pendingRequests.has(message[1])) {
+        return
       }
 
-      // Re-check socket status since we let CLOSE through
-      if (cxn.socket.status === SocketStatus.Open) {
-        cxn.socket.send([verb, ...extra])
-      }
+      cxn.socket.send(message)
     })
   }
 
   push = (message: Message) => {
+    // If we ended up handling a CLOSE before we sent the REQ, don't send the REQ
+    if (message[0] === "CLOSE") {
+      this.worker.buffer = this.worker.buffer.filter(complement(spec(["REQ", message[1]])))
+    }
+
     this.worker.push(message)
   }
 }