drop.chapril.org-firefoxsend/app/streams.js
2018-07-18 16:39:14 -07:00

37 lines
816 B
JavaScript

/* global ReadableStream */
export function transform(readable, transformer) {
const reader = readable.getReader();
const tstream = new ReadableStream({
start(controller) {
if (transformer.start) {
return transformer.start(controller);
}
},
async pull(controller) {
let enqueued = false;
const c = {
enqueue(d) {
enqueued = true;
controller.enqueue(d);
}
};
while (!enqueued) {
const x = await reader.read();
if (x.done) {
if (transformer.flush) {
await transformer.flush(controller);
}
return controller.close();
}
await transformer.transform(x.value, c);
}
},
cancel() {
readable.cancel();
}
});
return tstream;
}