diff --git a/src/server.js b/src/server.js index 9514689..8409960 100644 --- a/src/server.js +++ b/src/server.js @@ -89,6 +89,19 @@ module.exports = async function server(req, res) { sent = true; }); } + sent = true; + + // clean up the download if told to + res.on('finish', () => { + const { REMOVE_DOWNLOAD } = process.env; + if (!REMOVE_DOWNLOAD) return; + + const shouldRemove = REMOVE_DOWNLOAD === '1' || REMOVE_DOWNLOAD.toLowerCase() === 'true'; + if (shouldRemove) { + logger.info('Removing local file:', dlFilepath); + fs.unlinkSync(dlFilepath); + } + }); }); };