From 3792de3c91f307403eee6acec5a914cdf5ab9809 Mon Sep 17 00:00:00 2001
From: Joe Fleming <joe.fleming@gmail.com>
Date: Wed, 27 Nov 2019 14:30:11 -0700
Subject: [PATCH] feat: allow cleanup via REMOVE_DOWNLOAD env

---
 src/server.js | 13 +++++++++++++
 1 file changed, 13 insertions(+)

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);
+      }
+    });
   });
 };