feat(s3-service)!: move to AWS-SDK v3 (#216)

jeremydaly · Mar 9, 2023 · a99b8bb · a99b8bb
1 parent e78a26b
commit a99b8bb
Show file tree

Hide file tree

Showing 7 changed files with 8,342 additions and 5,809 deletions.
diff --git a/__tests__/download.unit.js b/__tests__/download.unit.js
@@ -7,7 +7,6 @@ const fs = require('fs') // Require Node.js file system
 // Require Sinon.js library
 const sinon = require('sinon')
 
-const AWS = require('aws-sdk') // AWS SDK (automatically available in Lambda)
 const S3 = require('../lib/s3-service') // Init S3 Service
 
 // Init API instance

diff --git a/__tests__/getLink.unit.js b/__tests__/getLink.unit.js
@@ -5,9 +5,6 @@ const delay = ms => new Promise(res => setTimeout(res, ms))
 // Require Sinon.js library
 const sinon = require('sinon')
 
-const AWS = require('aws-sdk') // AWS SDK (automatically available in Lambda)
-// AWS.config.credentials = new AWS.SharedIniFileCredentials({profile: 'madlucas'})
-
 const S3 = require('../lib/s3-service') // Init S3 Service
 
 // Init API instance

diff --git a/__tests__/responses.unit.js b/__tests__/responses.unit.js
@@ -1,7 +1,6 @@
 'use strict';
 
 const sinon = require('sinon') // Require Sinon.js library
-const AWS = require('aws-sdk') // AWS SDK (automatically available in Lambda)
 const S3 = require('../lib/s3-service') // Init S3 Service
 const { gzipSync, brotliCompressSync, deflateSync } = require('zlib')
 

diff --git a/__tests__/sendFile.unit.js b/__tests__/sendFile.unit.js
@@ -7,9 +7,6 @@ const fs = require('fs') // Require Node.js file system
 // Require Sinon.js library
 const sinon = require('sinon')
 
-const AWS = require('aws-sdk') // AWS SDK (automatically available in Lambda)
-// AWS.config.credentials = new AWS.SharedIniFileCredentials({profile: 'madlucas'})
-
 const S3 = require('../lib/s3-service'); // Init S3 Service
 
 // Init API instance

diff --git a/lib/s3-service.js b/lib/s3-service.js
@@ -6,7 +6,39 @@
  */
 
 // Require AWS SDK
-const AWS = require('aws-sdk'); // AWS SDK
+const { S3Client, GetObjectCommand } = require('@aws-sdk/client-s3'); // AWS SDK
+const { getSignedUrl } = require('@aws-sdk/s3-request-presigner');
 
 // Export
-module.exports = new AWS.S3();
+exports.client = new S3Client();
+
+exports.getObject = (params) => {
+  const cmd = new GetObjectCommand(params);
+  const promise = this.client.send(cmd);
+  return {
+    promise: () => promise,
+  };
+};
+
+exports.getSignedUrl = async (
+  type,
+  { Expires, ...params },
+  callback = () => {}
+) => {
+  let command;
+  switch (type) {
+    case 'getObject':
+      command = new GetObjectCommand(params);
+      break;
+    default:
+      throw new Error('Invalid command type');
+  }
+  return getSignedUrl(this.client, command, { expiresIn: Expires })
+    .then((url) => {
+      callback(null, url);
+      return url;
+    })
+    .catch((err) => {
+      callback(err);
+    });
+};