Strip TransferDescriptor type from args and return values

Author: andywer

src/types/master.ts

@@ -6,6 +6,7 @@
 import { Observable } from "observable-fns"
 import { ObservablePromise } from "../observable-promise"
 import { $errors, $events, $terminate, $worker } from "../symbols"
+import { TransferDescriptor } from "../transferable"
 
 interface ObservableLikeSubscription {
   unsubscribe(): any
@@ -26,12 +27,21 @@ export type StripAsync<Type> =
   ? ObservableBaseType
   : Type
 
+export type StripTransfer<Type> =
+  Type extends TransferDescriptor<infer BaseType>
+  ? BaseType
+  : Type
+
 export type ModuleMethods = { [methodName: string]: (...args: any) => any }
 
+export type ProxyableArgs<Args extends any[]> = Args extends [arg0: infer Arg0, ...rest: infer RestArgs]
+  ? [Arg0 extends Transferable ? Arg0 | TransferDescriptor<Arg0> : Arg0, ...RestArgs]
+  : Args
+
 export type ProxyableFunction<Args extends any[], ReturnType> =
   Args extends []
-    ? () => ObservablePromise<StripAsync<ReturnType>>
-  : (...args: Args) => ObservablePromise<StripAsync<ReturnType>>
+    ? () => ObservablePromise<StripTransfer<StripAsync<ReturnType>>>
+  : (...args: ProxyableArgs<Args>) => ObservablePromise<StripTransfer<StripAsync<ReturnType>>>
 
 export type ModuleProxy<Methods extends ModuleMethods> = {
   [method in keyof Methods]: ProxyableFunction<Parameters<Methods[method]>, ReturnType<Methods[method]>>

test/transferables.test.ts

@@ -1,5 +1,6 @@
 import test from "ava"
 import { spawn, Thread, Transfer, TransferDescriptor, Worker } from "../src/index"
+import { XorBuffer } from "./workers/arraybuffer-xor"
 
 type SpyInit<Args extends any[], OriginalReturn, NewReturn> =
   (originalFn: (...args: Args) => OriginalReturn) =>
@@ -43,8 +44,10 @@ test("can pass transferable objects on thread call", async t => {
     return postMessage(...args)
   })
 
-  const xorBuffer = await spawn<(buffer: ArrayBuffer | TransferDescriptor<ArrayBuffer>, value: number) => ArrayBuffer>(worker)
-  await xorBuffer(Transfer(testData), 15)
+  const xorBuffer = await spawn<XorBuffer>(worker)
+  const returnedBuffer = await xorBuffer(Transfer(testData), 15)
+
+  t.is(returnedBuffer.byteLength, 64)
 
   t.is(postMessageCalls.length, 1)
   t.is(postMessageCalls[0].length, 2)

test/workers/arraybuffer-xor.ts

@@ -1,7 +1,10 @@
 import { expose, Transfer } from "../../src/worker"
 
-expose(function xor(buffer: ArrayBuffer, value: number) {
+function xor(buffer: ArrayBuffer, value: number) {
   const view = new Uint8Array(buffer)
   view.forEach((byte, offset) => view.set([byte ^ value], offset))
   return Transfer(buffer)
-})
+}
+
+expose(xor)
+export type XorBuffer = typeof xor