aws-v3-sdk 1.0.15 → 1.0.16

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/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.16",
   "description": "AWS v3 SDK Utils",
   "main": "index.js",
   "scripts": {