zudello-integration-sdk 1.0.14 → 1.0.15

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "zudello-integration-sdk",
-	"version": "1.0.14",
+	"version": "1.0.15",
 	"description": "Zudello Integrations SDK",
 	"main": "./src/index.js",
 	"repository": {

package/src/index.js

@@ -22,6 +22,7 @@ const Trigger = require('./utils/trigger')
 const Properties = require('./utils/properties')
 const MiscHelper = require('./utils/miscHelper')
 const ModelHelper = require('./utils/modelHelper')
+const DatasetHelper = require('./utils/datasetHelper')
 const {
   getFileContent,
   uploadFile,
@@ -60,5 +61,6 @@ module.exports = {
   Properties,
   MiscHelper,
   ModelHelper,
+  DatasetHelper,
   S3Client
 }

package/src/utils/apiInstance.js

@@ -9,6 +9,7 @@ class ApiInstance {
    */
   constructor () {
     this.bearerToken = null
+    this.retryStatuses = [502, 504]
 
     this.responseHandler = new ResponseHandler()
 
@@ -39,7 +40,7 @@ class ApiInstance {
    * @param {object} headers Request Headers.
    * @returns {object} Response Data.
    */
-  async get (url, params, headers, returnHeaders = false, callback = () => {}) {
+  async get (url, params, headers, returnHeaders = false, callback = () => {}, retryCount = 0) {
     try {
       const requestBody = { params, headers }
 
@@ -67,6 +68,11 @@ class ApiInstance {
         headers
       })
 
+      if (this.retryStatuses.includes(error?.response?.status) && retryCount === 0) {
+        await this.sleep(20000)
+        return await this.get(url, params, headers, returnHeaders, callback, retryCount + 1)
+      }
+
       if (error?.response?.status === 302) {
         return this.responseHandler.success(error?.response?.headers?.location)
       }
@@ -93,7 +99,7 @@ class ApiInstance {
    * @param {object} params Request Params.
    * @returns {object} Response Data.
    */
-  async post (url, data, headers, params = {}, callback = () => {}) {
+  async post (url, data, headers, params = {}, callback = () => {}, retryCount = 0) {
     try {
       const response = await this.axiosInstance.post(url, data, { headers, params })
 
@@ -119,6 +125,11 @@ class ApiInstance {
         body: data
       })
 
+      if (this.retryStatuses.includes(error?.response?.status) && retryCount === 0) {
+        await this.sleep(20000)
+        return await this.post(url, data, headers, params, callback, retryCount + 1)
+      }
+
       if (error?.response?.status === 500) {
         return this.responseHandler.error(['Something went wrong.'])
       }
@@ -135,7 +146,7 @@ class ApiInstance {
    * @param {object} data PUT Data.
    * @returns {object} Response Data.
    */
-  async put (url, data, headers, params = {}, callback = () => {}) {
+  async put (url, data, headers, params = {}, callback = () => {}, retryCount = 0) {
     try {
       const response = await this.axiosInstance.put(url, data, { headers, params })
 
@@ -157,6 +168,11 @@ class ApiInstance {
         body: data
       })
 
+      if (this.retryStatuses.includes(error?.response?.status) && retryCount === 0) {
+        await this.sleep(20000)
+        return await this.put(url, data, headers, params, callback, retryCount + 1)
+      }
+
       if (error?.response?.data) {
         return this.responseHandler.error([error?.response?.data])
       }
@@ -170,7 +186,7 @@ class ApiInstance {
    * @param {string} url Request URL.
    * @returns {object} Response Data.
    */
-  async delete (url, callback = () => {}) {
+  async delete (url, callback = () => {}, retryCount = 0) {
     try {
       const response = await this.axiosInstance.delete(url)
 
@@ -186,6 +202,11 @@ class ApiInstance {
         response: error?.response?.data
       })
 
+      if (this.retryStatuses.includes(error?.response?.status) && retryCount === 0) {
+        await this.sleep(20000)
+        return await this.delete(url, callback, retryCount + 1)
+      }
+
       if (error?.response?.data) {
         return this.responseHandler.error([error?.response?.data])
       }
@@ -193,6 +214,10 @@ class ApiInstance {
       return this.responseHandler.error(['Something went wrong.'])
     }
   }
+
+  async sleep(ms) {
+    return new Promise(resolve => setTimeout(resolve, ms))
+  }
 }
 
 module.exports = ApiInstance

package/src/utils/datasetHelper.js

@@ -0,0 +1,81 @@
+'use strict'
+
+const _ = require('lodash')
+
+class DatasetHelper {
+  constructor(zudello, logger, data = null, uuid = null) {
+    this.zudello = zudello
+    this.logger = logger
+
+    this.data = data
+    this.uuid = uuid
+  }
+
+  async load(uuid, failCallback) {
+    failCallback = failCallback || ((data) => this.defaultFailCallback(data))
+
+    this.logger.info('DatasetUUID', uuid)
+
+    return await this.get(uuid, failCallback)
+  }
+
+  async get(uuid, failCallback) {
+    failCallback = failCallback || ((data) => this.defaultFailCallback(data))
+
+    if (!uuid) {
+      return failCallback({ uuid })
+    }
+
+    const fetchedDataset = await this.zudello.search({
+      model: 'DataSetRow',
+      offset: 0,
+      limit: 10000000,
+      select: ['uuid', 'custom'],
+      order_by: ['ordering'],
+      filter: {
+        dataset_uuid: uuid
+      }
+    })
+
+    this.logger.debug('Fetched Dataset', fetchedDataset)
+
+    if (fetchedDataset && fetchedDataset.data) {
+      const data = fetchedDataset.data?.data || null
+
+      if (data) {
+        return new DatasetHelper(this.zudello, this.logger, data, uuid)
+      }
+    }
+
+    return failCallback({ uuid })
+  }
+
+  length() {
+    return this.data ? this.data.length : 0
+  }
+
+  rows(mapping = null, filter = null) {
+    let data = this.data
+
+    if (!data) {
+      return []
+    }
+
+    if (filter) {
+      data = _.filter(data, filter)
+    }
+
+    if (mapping) {
+      data = _.map(data, mapping)
+    }
+
+    return data
+  }
+
+  defaultFailCallback(data) {
+    this.logger.error('Unable to find Dataset', data)
+    throw new Error('Unable to find Dataset')
+  }
+}
+
+module.exports = DatasetHelper