zudello-execute-local 1.1.44 → 1.1.45

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "zudello-execute-local",
-  "version": "1.1.44",
+  "version": "1.1.45",
   "description": "Zudello Execute tool for local runs",
   "main": "index.js",
   "bin": {
@@ -29,7 +29,7 @@
     "xml2js": "^0.6.2",
     "yargs": "^17.7.2",
     "zudello-integration-sdk": "^1.0.95",
-    "zudello-integration-sdk-v1": "1.0.87"
+    "zudello-integration-sdk-v1": "1.0.91"
   },
   "overrides": {
     "e2b": {

package/src/utils/nodeVM.js

@@ -2,25 +2,22 @@ const { NodeVM, makeResolverFromLegacyOptions } = require('vm2')
 const path = require('path')
 const fs = require('fs')
 
-const createCustomResolver = (sdkOverridePath) => {
+const createSdkMock = (sdkOverridePath) => {
   if (!sdkOverridePath) {
-    return undefined
+    return {}
   }
 
   const absoluteSdkPath = path.resolve(sdkOverridePath)
 
   if (!fs.existsSync(absoluteSdkPath)) {
     console.warn(`[WARN] INTEGRATION_SDK_PATH does not exist: ${absoluteSdkPath}`)
-    return undefined
+    return {}
   }
 
   console.log(`[INFO] Using SDK override path: ${absoluteSdkPath}`)
 
-  return (moduleName, fromPath) => {
-    if (moduleName === 'zudello-integration-sdk-v1') {
-      return absoluteSdkPath
-    }
-    return undefined
+  return {
+    'zudello-integration-sdk-v1': require(absoluteSdkPath)
   }
 }
 
@@ -173,17 +170,17 @@ const connectionSdkMapping = {
     variable: 'SunSystems',
     class: 'SunSystemsSDK',
     isExternal: true
-  }]
+  }],
+  unleashed: [{
+    variable: 'Unleashed',
+    class: 'UnleashedSDK',
+    isExternal: true
+  }],
 }
 
 const initiateNodeVM = (dirNamePath) => {
   const sdkOverridePath = process.env.INTEGRATION_SDK_PATH
 
-  const rootPaths = [dirNamePath]
-  if (sdkOverridePath) {
-    rootPaths.push(path.resolve(sdkOverridePath))
-  }
-
   return new NodeVM({
     console: 'inherit',
     sandbox: {
@@ -201,7 +198,7 @@ const initiateNodeVM = (dirNamePath) => {
         ],
       },
       builtin: ['https'],
-      root: rootPaths,
+      root: [dirNamePath],
       import: [
         'lodash',
         'performance-now',
@@ -210,8 +207,8 @@ const initiateNodeVM = (dirNamePath) => {
         'he',
         'zudello-integration-sdk-v1',
       ],
-      resolve: createCustomResolver(sdkOverridePath),
       mock: {
+        ...createSdkMock(sdkOverridePath),
         fs: {
           readFileSync() {
             return 'Nice try!'
@@ -240,7 +237,8 @@ const runScript = async (handler, source, object, loggerMode, dirNamePath, dirNa
 
   const classesInUse = [
     'Logger', 'Message', 'Metadata', 'Tags', 'AI', 'FileBucket', 'EDIBucket', 'Trigger', 'Properties', 'MiscHelper',
-    'ModelHelper', 'DatasetHelper', 'S3Client', 'Auth', 'ZudelloSDK', 'ZudelloAuthSDK', 'ZudelloStoreSDK'
+    'ModelHelper', 'DatasetHelper', 'S3Client', 'Auth', 'ZudelloSDK', 'ZudelloAuthSDK', 'ZudelloStoreSDK',
+    'ClickHouse', 'Checkpoint'
   ]
 
   const initAuth = `
@@ -301,6 +299,7 @@ const runScript = async (handler, source, object, loggerMode, dirNamePath, dirNa
         let File
         let EDI
         let triggerNode
+        let clickhouse
 
         if (auth) {
           message = new Message(auth.getApiInstance(), auth.teamUUID)
@@ -308,8 +307,22 @@ const runScript = async (handler, source, object, loggerMode, dirNamePath, dirNa
           File = new FileBucket(auth.getApiInstance(), auth.teamUUID)
           EDI = new EDIBucket(auth.getApiInstance(), auth.teamUUID)
           triggerNode = new Trigger(auth.getApiInstance(), auth.teamUUID)
+          clickhouse = new ClickHouse(auth.getApiInstance(), auth.teamUUID)
+          metadata.init({
+            apiInstance: auth.getApiInstance(),
+            apiURL: '${authData.api_url}',
+            teamUUID: auth.teamUUID,
+            triggerUUID: ${process.env.TRIGGER_UUID ? `'${process.env.TRIGGER_UUID}'` : 'null'},
+            connectionUUID: ${externalConnectionUUID ? `'${externalConnectionUUID}'` : 'null'}
+          })
         }
 
+        const checkpoint = new Checkpoint({
+          logger,
+          metadata,
+          s3Config: {}
+        })
+
         ${source}
 
         const startTime = now()
@@ -359,4 +372,4 @@ const runScript = async (handler, source, object, loggerMode, dirNamePath, dirNa
 
 module.exports = {
   runScript
-}
+}