feat: execute API

Author: antfu

src/cli.js

@@ -1,6 +1,6 @@
 const fs = require("fs");
 const version = require("./version");
-const { compile } = require("./parser");
+const { compile, evalCompiled } = require("./parser");
 const { render, unrender } = require("./render");
 const path = require("path");
 const commander = require("commander");
@@ -38,6 +38,7 @@ program
     "--roman [method]",
     'Romanize identifiers. The method can be "pinyin", "baxter" or "unicode"'
   )
+  .option("--outputHanzi", "Convert output to hanzi", true)
   .option("--log <file>", "Save log to file")
   .option("--title <title>", "Override title in rendering")
   .helpOption("-h, --help", "Display help");
@@ -192,17 +193,11 @@ function exec() {
     );
     process.exit(1);
   }
-  if (program.lang === "js") {
-    eval(getCompiled());
-  } else if (program.lang === "py") {
-    var execSync = require("child_process").execSync;
-    fs.writeFileSync("tmp.py", out);
-    var ret = execSync(
-      "which python3; if [ $? == 0 ]; then python3 tmp.py; else python tmp.py; fi; rm tmp.py",
-      { encoding: "utf-8" }
-    );
-    console.log(ret);
-  }
+
+  evalCompiled(getCompiled(), {
+    outputHanzi: program.outputHanzi,
+    lang: program.lang
+  });
 }
 
 function replscope() {

src/parser.js

@@ -1,5 +1,10 @@
 try {
-  var { hanzi2num, hanzi2numstr, num2hanzi, bool2hanzi } = require("./hanzi2num");
+  var {
+    hanzi2num,
+    hanzi2numstr,
+    num2hanzi,
+    bool2hanzi
+  } = require("./hanzi2num");
   var hanzi2pinyin = require("./hanzi2pinyin");
   var STDLIB = require("./stdlib");
   var { NUMBER_KEYWORDS, KEYWORDS } = require("./keywords");
@@ -753,8 +758,69 @@ function compile(
   return targ;
 }
 
+function isLangSupportedForEval(lang) {
+  if (lang !== "js")
+    throw new Error(
+      `Executing for target language "${lang}" is not supported in current environment`
+    );
+  return true;
+}
+
+function hanzinize(value) {
+  if (typeof value == "number") {
+    return num2hanzi(value);
+  } else if (typeof value == "boolean") {
+    return bool2hanzi(value);
+  } else if (Array.isArray(value)) {
+    return value.map(i => hanzinize(i));
+  } else {
+    return value;
+  }
+}
+
+function outputHanziWrapper(log) {
+  return function output(...args) {
+    log(...args.map(i => hanzinize(i)));
+  };
+}
+
+function evalCompiled(compiledCode, options = {}) {
+  const { outputHanzi = true, scoped = false, lang = "js" } = options;
+
+  isLangSupportedForEval(lang);
+
+  let code = compiledCode;
+
+  (() => {
+    const _console_log = console.log;
+    if (outputHanzi) {
+      console.log = outputHanziWrapper(_console_log);
+    }
+    try {
+      if (!scoped && "window" in this) {
+        window.eval(code);
+      } else {
+        eval(code);
+      }
+    } catch (e) {
+      throw e;
+    } finally {
+      if (outputHanzi) console.log = _console_log;
+    }
+  })();
+}
+
+function execute(source, options = {}) {
+  const { lang = "js" } = options;
+  isLangSupportedForEval(lang);
+  const compiled = compile(options.lang, source, options);
+  evalCompiled(compiled, options);
+}
+
 var parser = {
   compile,
+  evalCompiled,
+  execute,
   version,
   wy2tokens,
   tokens2asc,