Added AWS sdk functionality and instance parameters

Author: 9trocode

index.js

@@ -1,14 +1,18 @@
 const GCLOUD = require('./lib/GoogleCloudStorageSystem');
 const NFS = require('./lib/NFSStorageSystem');
+const AWS3 = require('./lib/S3StorageSystem');
 
 class StorageFactory {
-    static getInstance(storageMode) {
+    static getInstance(storageMode, Region) {
         if (!storageMode) {
             throw new Error('Please add a storage Mode');
         }
         if (storageMode === 'GCLOUD') {
             return new GCLOUD();
         }
+        if (storageMode === 'AWS') {
+            return new AWS3(Region);
+        }
         if (storageMode === 'NFS') {
             return new NFS();
         }

lib/GoogleCloudStorageSystem.js

@@ -88,6 +88,22 @@ class GoogleCloudStorageSystem extends StorageSystem {
             throw new Error(err)
         }
     }
+
+    async listBuckets(){
+        try {
+            return await storage.getBuckets();
+        }catch (err) {
+            throw new Error(err)
+        }
+    }
+
+    async listFiles(bucketName){
+        try {
+            return await storage.bucket(bucketName).getFiles();
+        }catch (err) {
+            throw new Error(err)
+        }
+    }
 }
 
 module.exports = GoogleCloudStorageSystem;

lib/S3StorageSystem.js

@@ -0,0 +1,89 @@
+const StorageSystem = require('./storage-system');
+const AWS = require('aws-sdk');
+const fs = require('fs');
+
+class S3StorageSystem extends StorageSystem {
+    constructor(Region) {
+        super();
+        this.Region = Region;
+        AWS.config.update({region: this.Region});
+        this.s3 = new AWS.S3({apiVersion: '2006-03-01'});
+    }
+
+    async deleteBucket(bucketName) {
+        this.s3.deleteBucket({Bucket: bucketName}, function (err, data) {
+            if (err) {
+                return err
+            } else {
+                return data
+            }
+        });
+    }
+
+    async listBuckets() {
+        // Call S3 to list the buckets
+        this.s3.listBuckets(function (err, data) {
+            if (err) {
+                return err
+            } else {
+                return data.Buckets;
+            }
+        });
+    }
+
+    async listFiles(bucketName) {
+        // Call S3 to obtain a list of the objects in the bucket
+        this.s3.listObjects({Bucket: bucketName}, function (err, data) {
+            if (err) {
+                return err
+            } else {
+                return data
+            }
+        });
+    }
+
+    async createBucket(bucketName, ACL) {
+        // call S3 to create the bucket
+        this.s3.createBucket({Bucket: bucketName, ACL: ACL}, function (err, data) {
+            if (err) {
+                return (err);
+            } else {
+                return (data.Location);
+            }
+        });
+    }
+
+    async upload(bucketName, filename, key) {
+        fs.readFile(filename, (err, data) => {
+            if (err) throw err;
+            this.s3.upload({Bucket: bucketName, Key: key, Body: JSON.stringify(data, null, 2)}, function (s3Err, data) {
+                if (s3Err) throw s3Err;
+                return (`File uploaded successfully at ${data.Location}`)
+            });
+        });
+    }
+
+
+    /**
+     * @memberof GoogleCloudStorageSystem
+     * @name Download
+     * @params filename, destination
+     * @description Serves as General Download SDK for GCLOUD Storage
+     */
+    async download(bucketName, filename, destination) {
+        this.s3.getObject({Bucket: bucketName, Key: filename}, (err, data) => {
+            if (err) console.error(err);
+            fs.writeFileSync(destination, data.Body.toString());
+            return (`${filename} has been Downloaded!`);
+        });
+    }
+
+    async deleteFile(buckName, filename) {
+        this.s3.deleteObject({Bucket: buckName, Key: filename}, function (err, data) {
+            if (err) return (err.stack); // an error occurred
+            else return (data);           // successful response
+        });
+    }
+}
+
+module.exports = S3StorageSystem;

lib/storage-system.js

@@ -18,6 +18,12 @@ class StorageSystem {
     deleteBucket(){
         throw new Error('concrete implementation should be used');
     }
+    listBuckets(){
+        throw new Error('concrete implementation should be used');
+    }
+    listFiles(){
+        throw new Error('concrete implementation should be used');
+    }
 }
 
 module.exports = StorageSystem;

package-lock.json

@@ -29,6 +29,7 @@
   "homepage": "https://github.com/9trocode/node_storage_manager#readme",
   "dependencies": {
     "@google-cloud/storage": "^4.3.1",
+    "aws-sdk": "^2.626.0",
     "chai": "^4.2.0",
     "fs-extra": "^8.1.0",
     "path": "^0.12.7"

package.json

@@ -6,7 +6,7 @@ describe('Simple Storage Test', () => {
 
     //First stage Unit Test
     it('should run', () => {
-      let StorageInstance =  storageSystem.getInstance('GCLOUD');
+      let StorageInstance =  storageSystem.getInstance('AWS', '');
           StorageInstance.download('smc-v1-dev', 'dev_key.key', '/Users/nitrocode/tmp/');
       console.log(StorageInstance);
     });