Delete PDF file from S3

zotero · Nov 28, 2018 · 1233cef · 1233cef
1 parent 15519c2
commit 1233cef
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/src/recognizer.js b/src/recognizer.js
@@ -48,6 +48,10 @@ let Recognizer = module.exports = {
 		return uploadID;
 	},
 
+	remove: async function(uploadID) {
+		await S3.deleteObject({Key: uploadID}).promise();
+	},
+
 	/**
 	 * Recognize the uploaded PDF by invoking recognizer Lambda function
 	 *
@@ -192,5 +196,6 @@ let Recognizer = module.exports = {
 		let uploadID = ctx.request.body;
 		let item = await this.recognize(uploadID);
 		ctx.body = Zotero.Utilities.itemToAPIJSON(item);
+		await this.remove(uploadID);
 	}
 };
diff --git a/src/webSession.js b/src/webSession.js
@@ -163,6 +163,7 @@ WebSession.prototype.handleURL = async function () {
 				let uploadID = await Recognizer.upload(req.response);
 				let item = await Recognizer.recognize(uploadID);
 				this.ctx.response.body = Zotero.Utilities.itemToAPIJSON(item);
+				await Recognizer.remove(uploadID);
 				resolve();
 			}