From bdfa90ca48822e64f4e0e71d7124df8b2f6f8879 Mon Sep 17 00:00:00 2001
From: Fritz Lin <uxfritz@163.com>
Date: Sat, 18 Apr 2020 15:15:18 +0800
Subject: [PATCH] perf: babel-transform for jsx only

---
 package.json                  |  2 +-
 src/playground/Playground.tsx | 31 +++++++++++++++++++------------
 src/playground/babelMaster.ts | 35 +----------------------------------
 src/playground/util.tsx       | 31 +++++++++++++++++++++++++++++++
 4 files changed, 52 insertions(+), 47 deletions(-)

diff --git a/package.json b/package.json
index 3efb893..b893de5 100644
--- a/package.json
+++ b/package.json
@@ -40,7 +40,7 @@
     "typescript": "3.5.2"
   },
   "scripts": {
-    "deploy": "npm run build && gh-pages -d ./build -b master -r https://github.com/coldemo/coldemo.github.io",
+    "deploy": "npm run build && gh-pages -d ./build -b master -r git@github.com:coldemo/coldemo.github.io.git",
     "dev": "npm start",
     "start": "react-scripts start",
     "build": "react-scripts build",
diff --git a/src/playground/Playground.tsx b/src/playground/Playground.tsx
index ad59b4d..616ecba 100644
--- a/src/playground/Playground.tsx
+++ b/src/playground/Playground.tsx
@@ -31,6 +31,7 @@ import {
   loadCss,
   appendJs,
   appendCss,
+  wrapCode,
 } from './util';
 
 Object.assign(window, {
@@ -124,18 +125,24 @@ export let Playground: React.FC = () => {
 
   Object.assign(window, { setRendering }); // without side-effect
 
-  let doPreview = useCallback(async (code: string) => {
-    if (!code) return;
-    setCompiling(true);
-    try {
-      let res = await babelTransform(code);
-      setPreview(res);
-    } catch (err) {
-      displayError(err);
-    } finally {
-      setCompiling(false);
-    }
-  }, []);
+  let doPreview = useCallback(
+    async (code: string) => {
+      if (!code) return;
+      setCompiling(true);
+      try {
+        let res = wrapCode(code);
+        if (file && file.endsWith('.jsx')) {
+          res = await babelTransform(res);
+        }
+        setPreview(res);
+      } catch (err) {
+        displayError(err);
+      } finally {
+        setCompiling(false);
+      }
+    },
+    [file]
+  );
 
   // let persistEditor = useCallback((code: string) => {
   //   localStorage.setItem(storeKeyCode, code);
diff --git a/src/playground/babelMaster.ts b/src/playground/babelMaster.ts
index 50ab68d..3dd2385 100644
--- a/src/playground/babelMaster.ts
+++ b/src/playground/babelMaster.ts
@@ -1,40 +1,7 @@
 let worker = new Worker('./babel-worker.js');
 
-let wrapCode = (code: string) => {
-  return `
-(async () => {
-  setRendering(true)
-  // mountNode.innerHTML = '' // can cause error in react
-  ReactDOM.unmountComponentAtNode(mountNode)
-  try {
-    let { useRef, useMemo, useState, useEffect, useLayoutEffect, useReducer, useContext, useCallback, useImperativeHandle } = React
-    ;;${code};;
-
-    let isVueLike = _.isPlainObject(App)
-    if (isVueLike) {
-      if (!mountNode.children[0]) {
-        let innerNode = document.createElement('div')
-        mountNode.appendChild(innerNode)
-      }
-      let curr = mountNode.children[0]
-      new Vue(App).$mount(curr)
-    } else {
-      ReactDOM.render(<App />, mountNode)
-    }
-  } catch (err) {
-    console.error(['displayError', err])
-    displayError(err)
-  } finally {
-    setRendering(false)
-  }
-})()
-  `.trim();
-};
-
 export let babelTransform = async (code: string): Promise<string> => {
   return new Promise((resolve, reject) => {
-    let fullCode = wrapCode(code);
-
     worker.onerror = e => {
       console.error(['worker onerror', e]);
       reject(new Error(`worker onerror - ${code.slice(0, 100)}`));
@@ -48,6 +15,6 @@ export let babelTransform = async (code: string): Promise<string> => {
         resolve(res.data as string);
       }
     };
-    worker.postMessage(fullCode);
+    worker.postMessage(code);
   });
 };
diff --git a/src/playground/util.tsx b/src/playground/util.tsx
index b70a67f..2859285 100644
--- a/src/playground/util.tsx
+++ b/src/playground/util.tsx
@@ -80,3 +80,34 @@ export let appendCss = (code: string) => {
   el.innerHTML = code;
   window.mountNode.appendChild(el);
 };
+
+export let wrapCode = (code: string) => {
+  return `
+(async () => {
+  setRendering(true)
+  // mountNode.innerHTML = '' // can cause error in react
+  ReactDOM.unmountComponentAtNode(mountNode)
+  try {
+    let { useRef, useMemo, useState, useEffect, useLayoutEffect, useReducer, useContext, useCallback, useImperativeHandle } = React
+    ;;${code};;
+
+    let isVueLike = _.isPlainObject(App)
+    if (isVueLike) {
+      if (!mountNode.children[0]) {
+        let innerNode = document.createElement('div')
+        mountNode.appendChild(innerNode)
+      }
+      let curr = mountNode.children[0]
+      new Vue(App).$mount(curr)
+    } else {
+      ReactDOM.render(React.createElement(App), mountNode)
+    }
+  } catch (err) {
+    console.error(['displayError', err])
+    displayError(err)
+  } finally {
+    setRendering(false)
+  }
+})()
+  `.trim();
+};