Parameterising resource acquisitions

src/types.ts

@@ -1,10 +1,10 @@
-type ResourceAcquire<Resource = void> = () => Promise<
-  readonly [ResourceRelease, Resource?]
->;
+type ResourceAcquire<Resource> = (
+  resources: readonly any[],
+) => Promise<readonly [ResourceRelease, Resource?]>;
 
 type ResourceRelease = (e?: Error) => Promise<void>;
 
-type Resources<T extends readonly ResourceAcquire<any>[]> = {
+type Resources<T extends readonly ResourceAcquire<unknown>[]> = {
   [K in keyof T]: T[K] extends ResourceAcquire<infer R> ? R : never;
 };
 

src/utils.ts

@@ -17,7 +17,7 @@ async function withF<
   let e_: Error | undefined;
   try {
     for (const acquire of acquires) {
-      const [release, resource] = await acquire();
+      const [release, resource] = await acquire(resources);
       releases.push(release);
       resources.push(resource);
     }
@@ -54,7 +54,7 @@ async function* withG<
   let e_: Error | undefined;
   try {
     for (const acquire of acquires) {
-      const [release, resource] = await acquire();
+      const [release, resource] = await acquire(resources);
       releases.push(release);
       resources.push(resource);
     }

tests/index.test.ts

@@ -82,7 +82,7 @@ describe('index', () => {
     // Multiple resources outside requires type declaration
     const resourceAcquires6: [
       ResourceAcquire<number>,
-      ResourceAcquire,
+      ResourceAcquire<void>,
       ResourceAcquire<string>,
     ] = [
       async () => {
@@ -246,4 +246,37 @@ describe('index', () => {
     expect(acquireOrder).toStrictEqual([lock1, lock2]);
     expect(releaseOrder).toStrictEqual([lock2, lock1]);
   });
+  test('withF parameterised resources', async () => {
+    await withF(
+      [
+        async () => [async () => {}, 1],
+        async ([a]) => [async () => {}, a + 1],
+        async ([, b]) => [async () => {}, b + 1],
+      ],
+      async ([a, b, c]) => {
+        expect(a).toBe(1);
+        expect(b).toBe(2);
+        expect(c).toBe(3);
+      },
+    );
+  });
+  test('withG parameterised resources', async () => {
+    const g = withG(
+      [
+        async () => [async () => {}, 1],
+        async ([a]) => [async () => {}, a + 1],
+        async ([, b]) => [async () => {}, b + 1],
+      ],
+      async function* ([a, b, c]): AsyncGenerator<number> {
+        yield a;
+        yield b;
+        yield c;
+      },
+    );
+    let counter = 1;
+    for await (const r of g) {
+      expect(r).toBe(counter);
+      counter++;
+    }
+  });
 });