wip: working on async scripting support

This commit is contained in:
Anoop M D 2023-03-28 14:49:57 +05:30
parent 78251c530c
commit 5fc32d035f
2 changed files with 29 additions and 3 deletions

View File

@ -113,7 +113,7 @@ const registerNetworkIpc = (mainWindow, watcher, lastOpenedCollections) => {
const requestScript = get(request, 'script.req'); const requestScript = get(request, 'script.req');
if(requestScript && requestScript.length) { if(requestScript && requestScript.length) {
const scriptRuntime = new ScriptRuntime(); const scriptRuntime = new ScriptRuntime();
const result = scriptRuntime.runRequestScript(requestScript, request, envVars, collectionVariables, collectionPath); const result = await scriptRuntime.runRequestScript(requestScript, request, envVars, collectionVariables, collectionPath);
mainWindow.webContents.send('main:script-environment-update', { mainWindow.webContents.send('main:script-environment-update', {
envVariables: result.envVariables, envVariables: result.envVariables,

View File

@ -1,5 +1,12 @@
const { NodeVM } = require('vm2'); const { NodeVM } = require('vm2');
const path = require('path'); 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 Bru = require('../bru');
const BrunoRequest = require('../bruno-request'); const BrunoRequest = require('../bruno-request');
const BrunoResponse = require('../bruno-response'); const BrunoResponse = require('../bruno-response');
@ -11,13 +18,14 @@ const lodash = require('lodash');
const moment = require('moment'); const moment = require('moment');
const uuid = require('uuid'); const uuid = require('uuid');
const nanoid = require('nanoid'); const nanoid = require('nanoid');
const axios = require('axios');
const CryptoJS = require('crypto-js'); const CryptoJS = require('crypto-js');
class ScriptRuntime { class ScriptRuntime {
constructor() { constructor() {
} }
runRequestScript(script, request, envVariables, collectionVariables, collectionPath) { async runRequestScript(script, request, envVariables, collectionVariables, collectionPath) {
const bru = new Bru(envVariables, collectionVariables); const bru = new Bru(envVariables, collectionVariables);
const req = new BrunoRequest(request); const req = new BrunoRequest(request);
@ -32,18 +40,36 @@ class ScriptRuntime {
external: true, external: true,
root: [collectionPath], root: [collectionPath],
mock: { mock: {
// node libs
path,
stream,
util,
url,
http,
https,
punycode,
zlib,
// 3rd party libs
atob, atob,
btoa, btoa,
lodash, lodash,
moment, moment,
uuid, uuid,
nanoid, nanoid,
axios,
'crypto-js': CryptoJS 'crypto-js': CryptoJS
} }
} }
}); });
vm.run(script, path.join(collectionPath, 'vm.js')); // wrap script inside a async function that gets called
script = `return (async () => { ${script} })()`;
// bug that needs to be fixed
// vm.run is not awaiting the async function
// created an issue in vm2 repo: https://github.com/patriksimek/vm2/issues/513
const result = await vm.run(script, path.join(collectionPath, 'vm.js'));
console.log(result);
return { return {
request, request,