From e6a1a427673fc2cc53b2cdc8a49addd1d4613443 Mon Sep 17 00:00:00 2001
From: achingbrain <alex@achingbrain.net>
Date: Wed, 19 Apr 2023 11:03:43 +0200
Subject: [PATCH] fix: accept source as a sink input type

To make it a bit more flexible
---
 package.json       | 1 +
 src/index.ts       | 4 ++--
 test/index.spec.ts | 6 +++---
 3 files changed, 6 insertions(+), 5 deletions(-)

diff --git a/package.json b/package.json
index b9ea411..4cf07aa 100644
--- a/package.json
+++ b/package.json
@@ -153,6 +153,7 @@
     ]
   },
   "scripts": {
+    "clean": "aegir clean",
     "lint": "aegir lint",
     "dep-check": "aegir dep-check dist/src/**/*.js dist/test/**/*.js",
     "build": "tsc",
diff --git a/src/index.ts b/src/index.ts
index 5d2fb0e..13a8735 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -139,11 +139,11 @@ export function abortableSource <T> (source: Source<T>, signal: AbortSignal, opt
   return abortable()
 }
 
-export function abortableSink <T, R = Promise<void>> (sink: Sink<AsyncIterable<T>, R>, signal: AbortSignal, options?: Options<T>): Sink<AsyncIterable<T>, R> {
+export function abortableSink <T, R = Promise<void>> (sink: Sink<AsyncIterable<T>, R>, signal: AbortSignal, options?: Options<T>): Sink<Source<T>, R> {
   return (source: Source<T>) => sink(abortableSource(source, signal, options))
 }
 
-export function abortableDuplex <TSource, TSink = TSource, RSink = Promise<void>> (duplex: Duplex<AsyncIterable<TSource>, AsyncIterable<TSink>, RSink>, signal: AbortSignal, options?: Options<TSource>): Duplex<AsyncGenerator<TSource>, AsyncIterable<TSink>, RSink> {
+export function abortableDuplex <TSource, TSink = TSource, RSink = Promise<void>> (duplex: Duplex<AsyncIterable<TSource>, Source<TSink>, RSink>, signal: AbortSignal, options?: Options<TSource>): Duplex<AsyncGenerator<TSource>, Source<TSink>, RSink> {
   return {
     sink: abortableSink(duplex.sink, signal, {
       ...options,
diff --git a/test/index.spec.ts b/test/index.spec.ts
index dbec5b4..0bddbb4 100644
--- a/test/index.spec.ts
+++ b/test/index.spec.ts
@@ -175,7 +175,7 @@ describe('abortable-iterator', () => {
 
   it('should abort a duplex used as a source', async () => {
     const controller = new AbortController()
-    const duplex: Duplex<AsyncIterable<number>> = {
+    const duplex: Duplex<AsyncIterable<number>, Source<number>> = {
       source: forever(),
       sink: async (source) => { await drain(source) }
     }
@@ -192,7 +192,7 @@ describe('abortable-iterator', () => {
 
   it('should abort a duplex used as a transform', async () => {
     const controller = new AbortController()
-    const duplex: Duplex<AsyncIterable<number>> = {
+    const duplex: Duplex<AsyncIterable<number>, Source<number>> = {
       source: forever(),
       sink: drain
     }
@@ -210,7 +210,7 @@ describe('abortable-iterator', () => {
 
   it('should abort a duplex used as a sink', async () => {
     const controller = new AbortController()
-    const duplex: Duplex<AsyncIterable<number>> = {
+    const duplex: Duplex<AsyncIterable<number>, Source<number>> = {
       source: forever(),
       sink: drain
     }