Fixes

lambda_template.yaml.j2

@@ -14,13 +14,25 @@ Resources:
       MemorySize: 2048
       Timeout: 30
       Policies:
+        - Statement:
+           Action:
+            - s3:PutObject
+            - s3:DeleteObject
+           Effect: Allow
+           Resource: !ImportValue RecognizerUploadBucketArn
         - LambdaInvokePolicy:
            FunctionName: {{ identifier_search_function_name }}
         - LambdaInvokePolicy:
            FunctionName: {{ recognizer_function_name }}
       Events:
         # API Gateway
-        API:
+        GetAPI:
+          Type: Api
+          Properties:
+            # Proxy all GET requests to Lambda function
+            Path: /{proxy+}
+            Method: get
+        PostAPI:
           Type: Api
           Properties:
             # Proxy all POST requests to Lambda function

src/http.js

@@ -153,7 +153,7 @@ Zotero.HTTP = new function() {
 					return reject(error);
 				}
 
-				if(!response.headers['content-type']) {
+				if (!response.headers['content-type']) {
 					return reject(new Error('Missing content-type header'));
 				}
 
@@ -223,7 +223,7 @@ Zotero.HTTP = new function() {
 							}
 						}
 					}
-					else if(
+					else if (
 						options.fallbackToPDF &&
 						mimeType.essence === 'application/pdf'
 					) {

src/lambda.js

@@ -43,7 +43,7 @@ app.use(bodyParser({enableTypes: ['text', 'json']}));
 app.use(_.post('/web', WebEndpoint.handle.bind(WebEndpoint)));
 app.use(_.post('/search', SearchEndpoint.handle.bind(SearchEndpoint)));
 app.use(_.post('/export', ExportEndpoint.handle.bind(ExportEndpoint)));
-app.use(_.post('/recognize/getUploadParams', Recognizer.handleUpload.bind(Recognizer)));
+app.use(_.get('/recognize/getUploadParams', Recognizer.handleUpload.bind(Recognizer)));
 app.use(_.post('/recognize/process', Recognizer.handleProcess.bind(Recognizer)));
 
 Debug.init(1);

src/recognizer.js

@@ -194,8 +194,20 @@ let Recognizer = module.exports = {
 	 */
 	handleProcess: async function (ctx) {
 		let uploadID = ctx.request.body;
-		let item = await this.recognize(uploadID);
-		ctx.body = Zotero.Utilities.itemToAPIJSON(item);
-		await this.remove(uploadID);
+
+		if (!uploadID) {
+			ctx.throw(400, "uploadID not provided\n");
+		}
+
+		try {
+			let item = await this.recognize(uploadID);
+			ctx.body = Zotero.Utilities.itemToAPIJSON(item);
+		}
+		catch(e) {
+			throw e;
+		}
+		finally {
+			await this.remove(uploadID);
+		}
 	}
 };

src/server.js

@@ -44,7 +44,7 @@ app.use(bodyParser({ enableTypes: ['text', 'json']}));
 app.use(_.post('/web', WebEndpoint.handle.bind(WebEndpoint)));
 app.use(_.post('/search', SearchEndpoint.handle.bind(SearchEndpoint)));
 app.use(_.post('/export', ExportEndpoint.handle.bind(ExportEndpoint)));
-app.use(_.post('/recognize/getUploadParams', Recognizer.handleUpload.bind(Recognizer)));
+app.use(_.get('/recognize/getUploadParams', Recognizer.handleUpload.bind(Recognizer)));
 app.use(_.post('/recognize/process', Recognizer.handleProcess.bind(Recognizer)));
 
 Debug.init(1);

src/webSession.js

@@ -152,19 +152,27 @@ WebSession.prototype.handleURL = async function () {
 					cookieSandbox: this._cookieSandbox
 				}
 			);
-			if(req.type === 'document') {
+			if (req.type === 'document') {
 				translate.setDocument(req.response);
 				// This could be optimized by only running detect on secondary translators
 				// if the first fails, but for now just run detect on all
 				translate.getTranslators(true);
 			}
 			// If PDF received
 			else {
-				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();
+				let uploadID;
+				try {
+					uploadID = await Recognizer.upload(req.response);
+					let item = await Recognizer.recognize(uploadID);
+					this.ctx.response.body = Zotero.Utilities.itemToAPIJSON(item);
+					resolve();
+				}
+				catch (e) {
+					throw e;
+				}
+				finally {
+					if (uploadID) await Recognizer.remove(uploadID);
+				}
 			}
 
 			return promise;