fix: correct return type

package.json

@@ -124,6 +124,7 @@
     ]
   },
   "scripts": {
+    "clean": "aegir clean",
     "lint": "aegir lint",
     "dep-check": "aegir dep-check",
     "build": "aegir build",

src/index.ts

@@ -5,17 +5,17 @@ import type { Duplex, Source } from 'it-stream-types'
 import type { Pushable } from 'it-pushable'
 import type { Uint8ArrayList } from 'uint8arraylist'
 
-export interface Handshake<TSource extends Uint8Array | Uint8ArrayList = Uint8Array, TSink extends Uint8Array | Uint8ArrayList = TSource> {
+export interface Handshake<TSink = Uint8Array | Uint8ArrayList> {
   reader: Reader
   writer: Pushable<TSink>
-  stream: Duplex<TSource, TSink>
+  stream: Duplex<Uint8ArrayList, TSink>
   rest: () => Source<TSink>
   write: (data: TSink) => void
   read: () => Promise<Uint8ArrayList | undefined>
 }
 
 // Convert a duplex stream into a reader and writer and rest stream
-export function handshake<TSource extends Uint8Array | Uint8ArrayList = Uint8Array, TSink extends Uint8Array | Uint8ArrayList = TSource> (stream: Duplex<TSource, TSink>): Handshake<TSource, TSink> {
+export function handshake<TSink extends Uint8ArrayList | Uint8Array = Uint8ArrayList> (stream: Duplex<Uint8ArrayList | Uint8Array, TSink>): Handshake<TSink> {
   const writer = pushable<TSink>() // Write bytes on demand to the sink
   const source = reader(stream.source) // Read bytes on demand from the source
 
@@ -33,7 +33,7 @@ export function handshake<TSource extends Uint8Array | Uint8ArrayList = Uint8Arr
     sinkErr = err
   })
 
-  const rest: Duplex<TSource, TSink> = {
+  const rest: Duplex<Uint8ArrayList, TSink> = {
     sink: async source => {
       if (sinkErr != null) {
         return await Promise.reject(sinkErr)
@@ -42,7 +42,7 @@ export function handshake<TSource extends Uint8Array | Uint8ArrayList = Uint8Arr
       sourcePromise.resolve(source)
       return await sinkPromise
     },
-    source: stream.source
+    source
   }
 
   return {

test/index.spec.ts

@@ -26,15 +26,15 @@ describe('handshake', () => {
       rShake.stream,
       (source) => (async function * () {
         for await (const message of source) {
-          expect(message).to.eql(buffer)
-          yield message
+          expect(message.subarray()).to.eql(buffer)
+          yield message.subarray()
         }
       })(),
       rShake.stream
     )
 
     const data = await pipe([buffer], iShake.stream, async (source) => await all(source))
-    expect(data).to.eql([buffer])
+    expect(data[0].subarray()).to.eql(buffer)
   })
   it('should be able to perform a handshake via Uint8ArrayList', async () => {
     const [initiator, responder] = duplexPair<Uint8ArrayList>()
@@ -96,15 +96,15 @@ describe('handshake', () => {
       rShake2.stream,
       (source) => (async function * () {
         for await (const message of source) {
-          yield message
+          yield message.subarray()
         }
       })(),
       rShake2.stream
     )
 
     const buffer = uint8ArrayFromString('more data')
     const data = await pipe([buffer], iShake2.stream, async (source) => await all(source))
-    expect(data).to.eql([buffer])
+    expect(data[0].subarray()).to.eql(buffer)
   })
 
   it('should persist data across handshakes', async () => {
@@ -141,15 +141,15 @@ describe('handshake', () => {
       rShake2.stream,
       (source) => (async function * () {
         for await (const message of source) {
-          yield message
+          yield message.subarray()
         }
       })(),
       rShake2.stream
     )
 
     const buffer = uint8ArrayFromString('more data')
     const data = await pipe([buffer], iShake2.stream, async (source) => await all(source))
-    expect(data).to.have.nested.property('[0]').that.equalBytes(buffer)
+    expect(data[0].subarray()).to.eql(buffer)
   })
 
   it('should survive an exploding sink while doing other async work', async () => {