0ct0pu5/ente
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Document

Browse source
This commit is contained in:
Manav Rathi 2024-03-22 11:21:05 +05:30
parent b1f45c8602
commit 18deac3a41
No known key found for this signature in database
1 changed files with 24 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

40
desktop/src/preload.ts
View file

@ -75,10 +75,27 @@ export function logError(error: Error, message: string, info?: string): void {
// -
export const convertBrowserStreamToNode = (
fileStream: ReadableStream<Uint8Array>,
) => {
const reader = fileStream.getReader();
/* preload: duplicated writeStream */
/**
* Write a (web) ReadableStream to a file at the given {@link filePath}.
*
* The returned promise resolves when the write completes.
*
* @param filePath The local filesystem path where the file should be written.
* @param readableStream A [web
* ReadableStream](https://developer.mozilla.org/en-US/docs/Web/API/ReadableStream)
*/
const writeStream = (filePath: string, readableStream: ReadableStream) =>
writeNodeStream(filePath, convertWebReadableStreamToNode(readableStream));
/**
* Convert a Web ReadableStream into a Node.js ReadableStream
*
* This can be used to, for example, write a ReadableStream obtained via
* `net.fetch` into a file using the Node.js `fs` APIs
*/
const convertWebReadableStreamToNode = (readableStream: ReadableStream) => {
const reader = readableStream.getReader();
const rs = new Readable();
rs._read = async () => {
@ -99,10 +116,10 @@ export const convertBrowserStreamToNode = (
return rs;
};
export async function writeNodeStream(
const writeNodeStream = async (
filePath: string,
fileStream: NodeJS.ReadableStream,
) {
) => {
const writeable = fs.createWriteStream(filePath);
fileStream.on("error", (error) => {
@ -120,16 +137,7 @@ export async function writeNodeStream(
reject(e);
});
});
}
/* preload: duplicated writeStream */
export async function writeStream(
filePath: string,
fileStream: ReadableStream<Uint8Array>,
) {
const readable = convertBrowserStreamToNode(fileStream);
await writeNodeStream(filePath, readable);
}
};
// - Export