refactor: execute js/py code (#118)

2024-10-26 08:26:52 +08:00
parent fbeaa9cb2c
commit e1a26cb4f2
3 changed files with 26 additions and 10 deletions
@@ -1,5 +1,3 @@
-const vm = require('vm');
-
 /**
 * Execute the javascript code in node.js.
 * @typedef {Object} Args
@@ -7,7 +5,15 @@ const vm = require('vm');
 * @param {Args} args
 */
 exports.run = function run({ code }) {
-  const context = vm.createContext({});
-  const script = new vm.Script(code);
-  return script.runInContext(context);
+  let log = "";
+  const oldStdoutWrite = process.stdout.write.bind(process.stdout);
+  process.stdout.write = (chunk, _encoding, callback) => {
+    log += chunk;
+    if (callback) callback();
+  };
+
+  eval(code);
+  
+  process.stdout.write = oldStdoutWrite;
+  return log;
 }