aws-v3-sdk 1.0.15 → 1.0.17

package/clients/all.js

@@ -4,5 +4,7 @@ module.exports = {
   EnvironmentCredentials: require('./credential'),
   S3: require('./s3'),
   STS: require('./sts'),
+  SQS: require('./sqs'),
+  SNS: require('./sns'),
   CognitoIdentityServiceProvider: require('./cognito')
 }

package/clients/cognito.js

@@ -1,4 +1,4 @@
-const { CognitoIdentityProviderClient, ListUsersCommand } = require('@aws-sdk/client-cognito-identity-provider')
+const { CognitoIdentityProviderClient, ListUsersCommand, AdminCreateUserCommand, AdminEnableUserCommand, AdminDisableUserCommand, AdminSetUserMFAPreferenceCommand, AdminUserGlobalSignOutCommand, AdminDeleteUserCommand } = require('@aws-sdk/client-cognito-identity-provider')
 
 class CognitoIdentityServiceProvider {
   constructor (region) {
@@ -13,6 +13,62 @@ class CognitoIdentityServiceProvider {
       callback(err)
     })
   }
+
+  async adminCreateUser (params, callback) {
+    const command = new AdminCreateUserCommand(params);
+    this.client.send(command).then(data => {
+      callback(null, data)
+    }).catch(err => {
+      callback(err)
+    })
+  }
+
+  async adminSetUserMFAPreference (params, callback) {
+    const command = new AdminSetUserMFAPreferenceCommand(params);
+    this.client.send(command).then(data => {
+      callback(null, data)
+    }).catch(err => {
+      callback(err)
+    })
+  }
+
+  async adminDeleteUser (params, callback) {
+    const command = new AdminDeleteUserCommand(params);
+    this.client.send(command).then(data => {
+      callback(null, data)
+    }).catch(err => {
+      callback(err)
+    })
+  }
+
+  async adminEnableUser (params, callback) {
+    const command = new AdminEnableUserCommand(params);
+    this.client.send(command).then(data => {
+      callback(null, data)
+    }).catch(err => {
+      callback(err)
+    })
+  }
+
+
+  async adminDisableUser (params, callback) {
+    const command = new AdminDisableUserCommand(params);
+    this.client.send(command).then(data => {
+      callback(null, data)
+    }).catch(err => {
+      callback(err)
+    })
+  }
+
+  async adminUserGlobalSignOut (params, callback) {
+    const command = new AdminUserGlobalSignOutCommand(params);
+    this.client.send(command).then(data => {
+      callback(null, data)
+    }).catch(err => {
+      callback(err)
+    })
+  }
+
 }
 
 module.exports = CognitoIdentityServiceProvider

package/clients/dynamo/DocumentClient.js

@@ -81,7 +81,12 @@ class DocumentClient {
   }
 
   update (params, callback) {
-    const docClient = DynamoDBDocument.from(new DynamoDB())
+    const options = {
+      marshallOptions: {
+        removeUndefinedValues: true
+      }
+    }
+    const docClient = DynamoDBDocument.from(ddb, options)
     docClient.update(params, (err, response) => {
       if (err) {
         return callback(err)

package/clients/sns.js

@@ -0,0 +1,27 @@
+const { SNSClient, PublishCommand } = require("@aws-sdk/client-sns");
+const snsClient = new SNSClient({});
+
+class SNS {
+  publish (params, callback) {
+    return new Promise((resolve, reject) => {
+      let publishCommand = new PublishCommand(params)
+      const response =  snsClient.send(publishCommand).then(response=>{
+        resolve(response)
+      }).catch(err=>{
+        reject(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 = SNS

package/clients/sqs.js

@@ -0,0 +1,69 @@
+const { SQSClient, ReceiveMessageCommand, DeleteMessageCommand, SendMessageCommand } = require("@aws-sdk/client-sqs");
+const snsClient = new SQSClient();
+
+class SQS {
+  sendMessage (params, callback) {
+    return new Promise((resolve, reject) => {
+      const command = new SendMessageCommand(params)
+      const response =  snsClient.send(command).then(response=>{
+        resolve(response)
+      }).catch(err=>{
+        reject(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)
+    })
+  }
+
+  receiveMessage (params, callback) {
+    return new Promise((resolve, reject) => {
+      const command = new ReceiveMessageCommand(params)
+      const response =  snsClient.send(command).then(response=>{
+        resolve(response)
+      }).catch(err=>{
+        reject(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)
+    })
+  }
+
+  deleteMessage (params, callback) {
+    return new Promise((resolve, reject) => {
+      const command = new DeleteMessageCommand(params)
+      const response =  snsClient.send(command).then(response=>{
+        resolve(response)
+      }).catch(err=>{
+        reject(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 = SQS

package/es-handler/index.js

@@ -1,9 +1,12 @@
-const { sendGetRequest } = require('./sendGetRequest')
+const { sendGetRequest, sendGetRequestCallback, getCount, multiSearch } = require('./sendGetRequest')
 const { sendAuditsReqToES } = require('./es-audits-wrapper')
 const { sendReqToES } = require('./es-wrapper')
 
 module.exports = {
   sendGetRequest,
+  sendGetRequestCallback,
+  getCount,
+  multiSearch,
   sendReqToES,
   sendAuditsReqToES
 }

package/es-handler/sendGetRequest.js

@@ -55,3 +55,62 @@ exports.sendGetRequest = async (index, query, from, size, sort, _source) => {
     })
   })
 }
+
+exports.sendGetRequestCallback = async (index, query, from, size, sort, callback) => {
+  const client = await getClient()
+  return new Promise((resolve, reject) => {
+    let body = { query }
+    if (sort) {
+      body = { query, sort }
+    }
+    client.search({
+      index,
+      from,
+      size,
+      body
+    }).then(response => {
+      const data = response.body.hits.hits
+      const totalCount = response.body.hits.total
+      const mappedData = data.map(item => {
+        return item._source
+      })
+      callback(null, { items: mappedData, data: mappedData, totalCount })
+    }).catch(e => {
+      callback(null, { items: [], data: [], totalCount: 0 })
+    })
+  })
+}
+
+exports.getCount = async (index, query) => {
+  const client = await getClient()
+  return new Promise((resolve, reject) => {
+    client.count({
+      index,
+      body: { query }
+    }).then(response => {
+      const data = response.body
+      resolve({ data })
+    }).catch(e => {
+      console.log(e);
+      resolve({ })
+    })
+  })
+}
+
+exports.multiSearch = async (query) => {
+  const client = await getClient()
+  return new Promise((resolve, reject) => {
+    let body = query 
+    client.msearch({
+      body
+    }).then(response => {
+      resolve(response);
+    }).catch(e => {
+      console.log(e);
+      resolve({ items: [], data: [], totalCount: 0 })
+    })
+  })
+}
+
+
+

package/package.json

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