@usebruno/js 0.3.0 → 0.4.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@usebruno/js",
-  "version": "0.3.0",
+  "version": "0.4.0",
   "main": "src/index.js",
   "files": [
     "src",
@@ -16,12 +16,14 @@
     "@usebruno/query": "0.1.0",
     "ajv": "^8.12.0",
     "atob": "^2.1.2",
+    "axios": "^0.26.0",
     "btoa": "^1.2.1",
     "crypto-js": "^4.1.1",
     "json-query": "^2.2.2",
     "lodash": "^4.17.21",
     "moment": "^2.29.4",
     "nanoid": "3.3.4",
+    "node-fetch": "2.*",
     "uuid": "^9.0.0"
   }
 }

package/src/bru.js

@@ -5,6 +5,10 @@ class Bru {
     this._collectionVariables = collectionVariables;
   }
 
+  getProcessEnv(key) {
+    return process.env[key];
+  }
+
   getEnvVar(key) {
     return this._envVariables[key];
   }
@@ -16,7 +20,7 @@ class Bru {
 
     // gracefully ignore if key is not present in environment
     if(!this._envVariables.hasOwnProperty(key)) {
-      return;
+       return;
     }
 
     this._envVariables[key] = value;
@@ -35,4 +39,4 @@ class Bru {
   }
 }
 
-module.exports = Bru;
+module.exports = Bru;

package/src/bruno-response.js

@@ -1,26 +1,26 @@
 class BrunoResponse {
   constructor(res) {
     this.res = res;
-    this.status = res.status;
-    this.statusText = res.statusText;
-    this.headers = res.headers;
-    this.body = res.data;
+    this.status = res ? res.status : null;
+    this.statusText = res ? res.statusText : null;
+    this.headers = res ? res.headers : null;
+    this.body = res ? res.data : null;
   }
 
   getStatus() {
-    return this.res.status;
+    return this.res ? this.res.status : null;
   }
 
   getHeader(name) {
-    return this.res.header[name];
+    return (this.res && this.res.headers) ? this.res.headers[name] : null;
   }
 
   getHeaders() {
-    return this.res.headers;
+    return this.res ? this.res.headers : null;
   }
 
   getBody() {
-    return this.res.data;
+    return this.res ? this.res.data : null;
   }
 }
 

package/src/runtime/script-runtime.js

@@ -1,5 +1,12 @@
 const { NodeVM } = require('vm2');
 const path = require('path');
+const http = require('http');
+const https = require('https');
+const stream = require('stream');
+const util = require('util');
+const zlib = require('zlib');
+const url = require('url');
+const punycode = require('punycode');
 const Bru = require('../bru');
 const BrunoRequest = require('../bruno-request');
 const BrunoResponse = require('../bruno-response');
@@ -11,16 +18,17 @@ const lodash = require('lodash');
 const moment = require('moment');
 const uuid = require('uuid');
 const nanoid = require('nanoid');
+const axios = require('axios');
+const fetch = require('node-fetch');
 const CryptoJS = require('crypto-js');
 
 class ScriptRuntime {
   constructor() {
   }
 
-  runRequestScript(script, request, envVariables, collectionVariables, collectionPath) {
+  async runRequestScript(script, request, envVariables, collectionVariables, collectionPath){
     const bru = new Bru(envVariables, collectionVariables);
     const req = new BrunoRequest(request);
-
     const context = {
       bru,
       req
@@ -32,19 +40,30 @@ class ScriptRuntime {
         external: true,
         root: [collectionPath],
         mock: {
+          // node libs
+          path,
+          stream,
+          util,
+          url,
+          http,
+          https,
+          punycode,
+          zlib,
+          // 3rd party libs
           atob,
           btoa,
           lodash,
           moment,
           uuid,
           nanoid,
+          axios,
+          'node-fetch': fetch,
           'crypto-js': CryptoJS
         }
       }
     });
-
-    vm.run(script, path.join(collectionPath, 'vm.js'));
-
+    const asyncVM = vm.run(`module.exports = async () => { ${script} }`,  path.join(collectionPath, 'vm.js'));
+    await asyncVM();
     return {
       request,
       envVariables,
@@ -52,7 +71,7 @@ class ScriptRuntime {
     };
   }
 
-  runResponseScript(script, request, response, envVariables, collectionVariables, collectionPath) {
+  async runResponseScript(script, request, response, envVariables, collectionVariables, collectionPath) {
     const bru = new Bru(envVariables, collectionVariables);
     const req = new BrunoRequest(request);
     const res = new BrunoResponse(response);
@@ -75,12 +94,15 @@ class ScriptRuntime {
           moment,
           uuid,
           nanoid,
+          axios,
+          'node-fetch': fetch,
           'crypto-js': CryptoJS
         }
       }
     });
 
-    vm.run(script, path.join(collectionPath, 'vm.js'));
+    const asyncVM = vm.run(`module.exports = async () => { ${script} }`,  path.join(collectionPath, 'vm.js'));
+    await asyncVM();
 
     return {
       response,