diff options
| -rw-r--r-- | packages/backend/src/misc/create-temp.ts | 11 |
1 files changed, 9 insertions, 2 deletions
diff --git a/packages/backend/src/misc/create-temp.ts b/packages/backend/src/misc/create-temp.ts index f2138abf66..9ba95cff42 100644 --- a/packages/backend/src/misc/create-temp.ts +++ b/packages/backend/src/misc/create-temp.ts @@ -30,13 +30,20 @@ export function createTempDir(): Promise<[string, () => void]> { }); } -export async function saveToTempFile(stream: NodeJS.ReadableStream): Promise<[string, () => void]> { +export async function saveToTempFile(stream: NodeJS.ReadableStream & { truncated?: boolean }): Promise<[string, () => void]> { const [filepath, cleanup] = await createTemp(); + try { await pipeline(stream, fs.createWriteStream(filepath)); - return [filepath, cleanup]; } catch (e) { cleanup(); throw e; } + + if (stream.truncated) { + cleanup(); + throw new Error('Read failed: input stream truncated'); + } + + return [filepath, cleanup]; } |