@nmtjs/protocol 0.15.0-beta.9 → 0.15.1

package/src/server/stream.ts

@@ -28,6 +28,8 @@ export class ProtocolClientStream extends PassThrough {
 export class ProtocolServerStream extends PassThrough {
   public readonly id: number
   public readonly metadata: ProtocolBlobMetadata
+  readonly #source: Readable
+  #piped = false
 
   constructor(id: number, blob: ProtocolBlob) {
     let readable: Readable
@@ -43,9 +45,28 @@ export class ProtocolServerStream extends PassThrough {
     super()
 
     this.pause()
-    readable.pipe(this)
+    this.#source = readable
+    this.#source.on('error', (error) => {
+      this.destroy(error)
+    })
 
     this.id = id
     this.metadata = blob.metadata
   }
+
+  override resume(): this {
+    if (!this.#piped) {
+      this.#piped = true
+      this.#source.pipe(this)
+    }
+    return super.resume()
+  }
+
+  override destroy(error?: Error | null) {
+    if (!this.#piped) {
+      this.#piped = true
+    }
+    this.#source.destroy?.(error ?? undefined)
+    return super.destroy(error ?? undefined)
+  }
 }

package/src/server/utils.ts

@@ -11,12 +11,12 @@ export const getFormat = (
   format: ProtocolFormats,
   { accept, contentType }: ResolveFormatParams,
 ) => {
-  const encoder = contentType ? format.supportsEncoder(contentType) : undefined
-  if (!encoder)
-    throw new UnsupportedContentTypeError('Unsupported Content type')
+  const encoder = accept ? format.supportsEncoder(accept) : undefined
+  if (!encoder) throw new UnsupportedAcceptTypeError('Unsupported Accept type')
 
-  const decoder = accept ? format.supportsDecoder(accept) : undefined
-  if (!decoder) throw new UnsupportedAcceptTypeError('Unsupported Accept type')
+  const decoder = contentType ? format.supportsDecoder(contentType) : undefined
+  if (!decoder)
+    throw new UnsupportedContentTypeError('Unsupported Content type')
 
   return { encoder, decoder }
 }

package/src/server/versions/v1.ts

@@ -38,10 +38,6 @@ export class ProtocolVersion1 extends ProtocolVersionInterface {
 
         return { type: messageType, rpc: { callId, procedure, payload } }
       }
-      case ClientMessageType.RpcPull: {
-        const callId = messagePayload.readUInt32LE(0)
-        return { type: messageType, callId }
-      }
       case ClientMessageType.RpcAbort: {
         const callId = messagePayload.readUInt32LE(0)
         const reasonPayload = messagePayload.subarray(MessageByteLength.CallId)
@@ -49,6 +45,14 @@ export class ProtocolVersion1 extends ProtocolVersionInterface {
           reasonPayload.byteLength > 0 ? decodeText(reasonPayload) : undefined
         return { type: messageType, callId, reason }
       }
+      case ClientMessageType.Ping: {
+        const nonce = messagePayload.readUInt32LE(0)
+        return { type: messageType, nonce }
+      }
+      case ClientMessageType.Pong: {
+        const nonce = messagePayload.readUInt32LE(0)
+        return { type: messageType, nonce }
+      }
       case ClientMessageType.ServerStreamAbort: {
         const streamId = messagePayload.readUInt32LE(0)
         const reasonPayload = messagePayload.subarray(
@@ -146,6 +150,22 @@ export class ProtocolVersion1 extends ProtocolVersionInterface {
           reason ? encodeText(reason) : Buffer.alloc(0),
         )
       }
+      case ServerMessageType.Pong: {
+        const { nonce } =
+          payload as ServerMessageTypePayload[ServerMessageType.Pong]
+        return this.encode(
+          encodeNumber(messageType, 'Uint8'),
+          encodeNumber(nonce, 'Uint32'),
+        )
+      }
+      case ServerMessageType.Ping: {
+        const { nonce } =
+          payload as ServerMessageTypePayload[ServerMessageType.Ping]
+        return this.encode(
+          encodeNumber(messageType, 'Uint8'),
+          encodeNumber(nonce, 'Uint32'),
+        )
+      }
       case ServerMessageType.ClientStreamPull: {
         const { size, streamId } =
           payload as ServerMessageTypePayload[ServerMessageType.ClientStreamPull]