aws-v3-sdk 1.0.5 → 1.0.6

package/clients/all.js

@@ -0,0 +1,8 @@
+module.exports = {
+  Lambda: require('./lambda'),
+  DynamoDB: require('./dynamoDB'),
+  EnvironmentCredentials: require('./credential'),
+  S3: require('./s3'),
+  STS: require('./sts'),
+  CognitoIdentityServiceProvider: require('./cognito')
+}

package/clients/cognito.js

@@ -0,0 +1,18 @@
+const { CognitoIdentityProviderClient, ListUsersCommand } = require('@aws-sdk/client-cognito-identity-provider')
+
+class CognitoIdentityServiceProvider {
+  constructor (region) {
+    this.client = new CognitoIdentityProviderClient({ region }) // Replace with your desired region
+  }
+
+  async listUsers (params, callback) {
+    const command = new ListUsersCommand(params)
+    this.client.send(command).then(data => {
+      callback(null, data)
+    }).catch(err => {
+      callback(err)
+    })
+  }
+}
+
+module.exports = CognitoIdentityServiceProvider

package/clients/dynamo/DocumentClient.js

@@ -0,0 +1,100 @@
+const { DynamoDB, DynamoDBClient, BatchGetItemCommand } = require('@aws-sdk/client-dynamodb')
+const { DynamoDBDocument, DynamoDBDocumentClient, GetCommand, BatchWriteCommand, PutCommand, QueryCommand, ScanCommand, DeleteCommand } = require('@aws-sdk/lib-dynamodb')
+const ddb = new DynamoDBClient()
+const documentClient = DynamoDBDocumentClient.from(ddb)
+
+class DocumentClient {
+  get (params, callback) {
+    const getCommand = new GetCommand(params)
+    documentClient.send(getCommand, (err, response) => {
+      if (err || !response) {
+        return callback(err)
+      }
+      if (response.Item) {
+        return callback(null, response)
+      } else if (!response.Item) {
+        return callback(null, {})
+      }
+    })
+  }
+
+  query (params, callback) {
+    const getCommand = new QueryCommand(params)
+    documentClient.send(getCommand, (err, response) => {
+      if (err || !response) {
+        return callback(err)
+      }
+      return callback(null, response)
+    })
+  }
+
+  scan (params, callback) {
+    const getCommand = new ScanCommand(params)
+    documentClient.send(getCommand, (err, response) => {
+      if (err || !response) {
+        return callback(err)
+      }
+      if (response.Items) {
+        return callback(null, response)
+      } else if (!response.Items) {
+        return callback(null)
+      }
+    })
+  }
+
+  put (params, callback) {
+    const putCommand = new PutCommand(params)
+    documentClient.send(putCommand, (err, response) => {
+      if (err || !response) {
+        return callback(err)
+      }
+      return callback(null, response)
+    })
+  }
+
+  delete (params, callback) {
+    const putCommand = new DeleteCommand(params)
+    documentClient.send(putCommand, (err, response) => {
+      if (err) {
+        return callback(err)
+      }
+      return callback(null, response)
+    })
+  }
+
+  update (params, callback) {
+    const docClient = DynamoDBDocument.from(new DynamoDB())
+    docClient.update(params, (err, response) => {
+      if (err) {
+        return callback(err)
+      } else {
+        return callback(null, response)
+      }
+    })
+  }
+
+  batchWrite (params, callback) {
+    const command = new BatchWriteCommand(params)
+    documentClient.send(command, (err, response) => {
+      if (err || !response) {
+        return callback(err)
+      }
+      if (response) {
+        return callback(null, response)
+      }
+    })
+  }
+
+  batchGet (params, callback) {
+    const docClient = DynamoDBDocument.from(new DynamoDB())
+
+    docClient.batchGet(params, function (err, data) {
+      if (err) callback(err)
+      else {
+        callback(null, data)
+      }
+    })
+  }
+}
+
+module.exports = DocumentClient

package/{lib/clients → clients}/dynamoDB.js

@@ -1,4 +1,3 @@
-
 const DocumentClient = require('./dynamo/DocumentClient')
 
 class DynamoDB {

package/{lib/clients → clients}/lambda.js

@@ -2,10 +2,22 @@ const { LambdaClient, InvokeCommand, InvokeAsyncCommand } = require('@aws-sdk/cl
 const { toUtf8, fromUtf8 } = require('@aws-sdk/util-utf8-node')
 
 class Lambda {
+  constructor (region, session) {
+    if (region && session) {
+      this.client = new LambdaClient({ region, credentials: session })
+    } else {
+      this.client = new LambdaClient()
+    }
+  }
+
   async invoke (params, callback) {
-    const client = new LambdaClient()
+    if (params.Payload) {
+      params.InvokeArgs = fromUtf8(params.Payload)
+      delete params.Payload
+    }
+
     const command = new InvokeCommand(params)
-    client.send(command).then(response => {
+    this.client.send(command).then(response => {
       response.Payload = toUtf8(response.Payload)
       callback(null, response)
     }).catch(err => {

package/clients/s3.js

@@ -0,0 +1,35 @@
+const { S3Client, GetObjectCommand, PutObjectCommand, ListObjectsCommand, DeleteObjectsCommand } = require('@aws-sdk/client-s3')
+const { getSignedUrl } = require('@aws-sdk/s3-request-presigner')
+const client = new S3Client({})
+
+class S3 {
+  getObject (params, callback) {
+    return new Promise((resolve, reject) => {
+      const getObject = new GetObjectCommand(params)
+      client.send(getObject).then(response => {
+        const chunks = []
+        const stream = response.Body
+        stream.on('data', chunk => chunks.push(chunk))
+        stream.on('end', () => {
+          const imageBuffer = Buffer.concat(chunks)
+          response.Body = imageBuffer
+          resolve(response)
+        })
+      }).catch(err => {
+        resolve(err)
+      })
+    }).then(result => {
+      if (callback && typeof callback === 'function') {
+        callback(null, result)
+      }
+      return result
+    }).catch(error => {
+      if (callback && typeof callback === 'function') {
+        callback(error, null)
+      }
+      return Promise.reject(error)
+    })
+  }
+}
+
+module.exports = S3

package/clients/sts.js

@@ -0,0 +1,19 @@
+const { STSClient, AssumeRoleCommand } = require('@aws-sdk/client-sts')
+
+class STS {
+  constructor (AWS_REGION) {
+    this.client = new STSClient({ region: AWS_REGION })
+  }
+
+  assumeRole (params, callback) {
+    console.log('params', params)
+    const command = new AssumeRoleCommand(params)
+    this.client.send(command).then(role => {
+      return callback(null, role)
+    }).catch(err => {
+      return callback(err)
+    })
+  }
+}
+
+module.exports = STS

package/lib/aws.js

@@ -1,5 +1,5 @@
 // Load all service classes
-const AWS = require('./clients/all')
+const AWS = require('../clients/all')
 
 /**
  * @api private

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "aws-v3-sdk",
-  "version": "1.0.5",
+  "version": "1.0.6",
   "description": "AWS v3 SDK Utils",
   "main": "index.js",
   "scripts": {

package/lib/clients/all.js

@@ -1,5 +0,0 @@
-module.exports = {
-  Lambda: require('./lambda'),
-  DynamoDB: require('./dynamoDB'),
-  EnvironmentCredentials: require('./credential')
-}

package/lib/clients/dynamo/DocumentClient.js

@@ -1,23 +0,0 @@
-const { DynamoDBClient } = require('@aws-sdk/client-dynamodb')
-const { DynamoDBDocumentClient, GetCommand } = require('@aws-sdk/lib-dynamodb')
-const ddb = new DynamoDBClient()
-const documentClient = DynamoDBDocumentClient.from(ddb)
-
-class DocumentClient {
-  get (params, callback) {
-    const getCommand = new GetCommand(params)
-    documentClient.send(getCommand, (err, response) => {
-      console.log(response)
-      if (err || !response) {
-        return callback(err)
-      }
-      if (response.Item) {
-        return callback(null, response)
-      } else if (!response.Item) {
-        return callback(response)
-      }
-    })
-  }
-}
-
-module.exports = DocumentClient