aws-v3-sdk 1.0.14 → 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)
@@ -113,6 +118,129 @@ class DocumentClient {
       }
     })
   }
+
+  getAsync (params, callback) {
+    return new Promise((resolve, reject) => {
+      const getCommand = new GetCommand(params)
+      documentClient.send(getCommand, (err, response) => {
+        if (err || !response) {
+          return reject(err)
+        }
+        if (response.Item) {
+          return resolve(response)
+        } else if (!response.Item) {
+          return resolve({})
+        }
+      })
+    }).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)
+    })
+  }
+
+  queryAsync (params, callback) {
+    return new Promise((resolve, reject) => {
+      const getCommand = new QueryCommand(params)
+      documentClient.send(getCommand, (err, response) => {
+        if (err || !response) {
+          return resolve(err)
+        }
+        return resolve(response)
+      })
+    }).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)
+    })
+  }
+
+  putAsync (params, callback) {
+    return new Promise((resolve, reject) => {
+      let item = params.Item
+      try {
+        item = Object.fromEntries(Object.entries(item).filter(([key, value]) => value !== undefined))
+      } catch (error) {
+
+      }
+      params.Item = item
+      const putCommand = new PutCommand(params)
+      documentClient.send(putCommand, (err, response) => {
+        if (err || !response) {
+          return resolve(err)
+        }
+        return resolve(response)
+      })
+    }).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)
+    })
+  }
+
+  deleteAsync (params, callback) {
+    return new Promise((resolve, reject) => {
+      const docClient = DynamoDBDocument.from(new DynamoDB())
+      docClient.delete(params, (err, response) => {
+        if (err) {
+          return resolve(err)
+        } else {
+          return resolve(response)
+        }
+      })
+    }).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)
+    })
+  }
+
+  updateAsync (params, callback) {
+    return new Promise((resolve, reject) => {
+      const docClient = DynamoDBDocument.from(new DynamoDB())
+      docClient.update(params, (err, response) => {
+        if (err) {
+          return resolve(err)
+        } else {
+          return resolve(response)
+        }
+      })
+    }).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 = DocumentClient

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.14",
+  "version": "1.0.16",
   "description": "AWS v3 SDK Utils",
   "main": "index.js",
   "scripts": {