edits

Author: artyom-activeloop

lib/plugins/s3Cache.js

@@ -1,52 +1,59 @@
-var s3 = new (require('aws-sdk')).S3({params:{Bucket: process.env.S3_BUCKET_NAME}});
+const { S3Client, GetObjectCommand, PutObjectCommand } = require('@aws-sdk/client-s3');
+
+const s3Client = new S3Client({});
+const bucketName = process.env.S3_BUCKET_NAME;
 
 module.exports = {
 
-	requestReceived: function(req, res, next) {
+	requestReceived: async function(req, res, next) {
 		if(req.method !== 'GET') {
 			return next();
 		}
 
-		var key = req.prerender.url;
+		let key = req.prerender.url;
 
 		if (process.env.S3_PREFIX_KEY) {
 			key = process.env.S3_PREFIX_KEY + '/' + key;
 		}
 
-		s3.getObject({
+		try {
+			const command = new GetObjectCommand({
+				Bucket: bucketName,
 				Key: key
-		}, function (err, result) {
-
-			if (!err && result) {
-				return res.send(200, result.Body);
-			}
-
+			});
+			const result = await s3Client.send(command);
+			const body = await result.Body.transformToString();
+			return res.send(200, body);
+		} catch (err) {
 			next();
-		});
+		}
 	},
 
-	pageLoaded: function(req, res, next) {
+	pageLoaded: async function(req, res, next) {
 		if(req.prerender.statusCode !== 200) {
 			return next();
 		}
 
-		var key = req.prerender.url;
-		
+		let key = req.prerender.url;
+
 		if (process.env.S3_PREFIX_KEY) {
 			key = process.env.S3_PREFIX_KEY + '/' + key;
 		}
 
-		s3.putObject({
-			Key: key,
-			ContentType: 'text/html;charset=UTF-8',
-			StorageClass: 'REDUCED_REDUNDANCY',
-			Body: req.prerender.content
-		}, function(err, result) {
-
-			if (err) console.error(err);
+		try {
+			const command = new PutObjectCommand({
+				Bucket: bucketName,
+				Key: key,
+				ContentType: 'text/html;charset=UTF-8',
+				StorageClass: 'REDUCED_REDUNDANCY',
+				Body: req.prerender.content
+			});
+			await s3Client.send(command);
+		} catch (err) {
+			console.error(err);
+		}
 
-			next();
-		});
+		next();
 	}
 };