fix(engine): expression can not return right value in browser

didi · Aug 21, 2023 · 9ef9f2e · 9ef9f2e
1 parent a907c87
commit 9ef9f2e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/packages/engine/src/expression/browserVm.ts b/packages/engine/src/expression/browserVm.ts
@@ -11,6 +11,7 @@ const createContext = (globalData) => {
   if (!document || !document.body) {
     console.error(getErrorMsg(ErrorCode.NO_DOCUMENT_BODY));
   }
+  document.body.appendChild(iframe);
   const iframeWindow = iframe.contentWindow as any;
   iframeWindow.parent = null;
   Object.keys(globalData).forEach((key) => {
@@ -22,6 +23,9 @@ const runInContext = (code, context) => {
   try {
     const iframeEval = context.eval;
     iframeEval.call(context, code);
+    if (context.frameElement) {
+      document.body.removeChild(context.frameElement);
+    }
   } catch (e) {
     console.warn(getWarningMsg(WarningCode.EXPRESSION_EXEC_ERROR), { code, context, e });
   }