23 lines
510 B
TypeScript
23 lines
510 B
TypeScript
|
// Copyright 2024 Signal Messenger, LLC
|
||
|
|
// SPDX-License-Identifier: AGPL-3.0-only
|
||
|
|
|
||
|
|
import { Transform } from 'node:stream';
|
||
|
|
|
||
|
|
export function finalStream(finalizer: () => Promise<void> | void): Transform {
|
||
|
|
return new Transform({
|
||
|
|
transform(data, enc, callback) {
|
||
|
|
this.push(Buffer.from(data, enc));
|
||
|
|
callback();
|
||
|
|
},
|
||
|
|
async final(callback) {
|
||
|
|
try {
|
||
|
|
await finalizer();
|
||
|
|
} catch (error) {
|
||
|
|
callback(error);
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
callback(null);
|
||
|
|
},
|
||
|
|
});
|
||
|
|
}
|