67 lines
1.4 KiB
Plaintext
67 lines
1.4 KiB
Plaintext
import { PassThrough } from 'node:stream';
|
|
import { renderToChunks } from './lib/chunked.js';
|
|
|
|
/**
|
|
* @typedef {object} RenderToPipeableStreamOptions
|
|
* @property {() => void} [onShellReady]
|
|
* @property {() => void} [onAllReady]
|
|
* @property {(error) => void} [onError]
|
|
*/
|
|
|
|
/**
|
|
* @typedef {object} PipeableStream
|
|
* @property {() => void} abort
|
|
* @property {(writable: import('stream').Writable) => void} pipe
|
|
*/
|
|
|
|
/**
|
|
* @param {import('preact').VNode} vnode
|
|
* @param {RenderToPipeableStreamOptions} options
|
|
* @param {any} [context]
|
|
* @returns {PipeableStream}
|
|
*/
|
|
export function renderToPipeableStream(vnode, options, context) {
|
|
const encoder = new TextEncoder('utf-8');
|
|
|
|
const controller = new AbortController();
|
|
const stream = new PassThrough();
|
|
|
|
renderToChunks(vnode, {
|
|
context,
|
|
abortSignal: controller.signal,
|
|
onError: (error) => {
|
|
if (options.onError) {
|
|
options.onError(error);
|
|
}
|
|
controller.abort(error);
|
|
},
|
|
onWrite(s) {
|
|
stream.write(encoder.encode(s));
|
|
}
|
|
})
|
|
.then(() => {
|
|
options.onAllReady && options.onAllReady();
|
|
stream.end();
|
|
})
|
|
.catch((error) => {
|
|
stream.destroy(error);
|
|
});
|
|
|
|
Promise.resolve().then(() => {
|
|
options.onShellReady && options.onShellReady();
|
|
});
|
|
|
|
return {
|
|
abort() {
|
|
controller.abort();
|
|
stream.destroy(new Error('aborted'));
|
|
},
|
|
/**
|
|
* @param {import("stream").Writable} writable
|
|
*/
|
|
pipe(writable) {
|
|
stream.pipe(writable, { end: true });
|
|
}
|
|
};
|
|
}
|