fix: vite dev

hengwangsay · hengwangsay · commit a2e86e5c1d56 · 2022-11-11T13:46:40.000+08:00
diff --git a/ui/.env b/ui/.env
@@ -16,8 +16,7 @@ ESBUILD=false
 # feature
 SSR=true
 CSR=false
-MIDDLEWARE=false
-MIDDELWARE_VITE = false
+MIDDLEWARE=true
 ANIMATE_ROUTER=false
 # entry
 SERVER_ENTRY=./src/server/entry.ts
diff --git a/ui/src/server/middleware/render.ts b/ui/src/server/middleware/render.ts
@@ -4,7 +4,7 @@ import { webpackRenderCSR, webpackRenderP_CSR, webpackRenderSSR, viteRenderCSR,
 
 import type { RenderType } from "@server/type";
 
-const useVITE = false;
+const useVITE = true;
 
 // 渲染函数
 const render: RenderType = async ({ req, res }) => {
diff --git a/ui/src/shared/template/Body.tsx b/ui/src/shared/template/Body.tsx
@@ -1,22 +1,11 @@
+import { ViteInject } from "./ViteInject";
+
 import type { HTMLProps } from ".";
 
 export const Body = ({ children, script = [] }: HTMLProps) => (
   <body>
     <div id="__content__">{children}</div>
     {script.filter(Boolean).map((ele) => ele)}
-    <script type="module" src="/@vite/client"></script>
-    <script
-      type="module"
-      dangerouslySetInnerHTML={{
-        __html: `
-import RefreshRuntime from "/@react-refresh"
-RefreshRuntime.injectIntoGlobalHook(window)
-window.$RefreshReg$ = () => {}
-window.$RefreshSig$ = () => (type) => type
-window.__vite_plugin_react_preamble_installed__ = true    
-    `,
-      }}
-    ></script>
-    <script type="module" src="/src/client/entry.tsx"></script>
+    <ViteInject />
   </body>
 );
diff --git a/ui/src/shared/template/ViteInject.tsx b/ui/src/shared/template/ViteInject.tsx
@@ -0,0 +1,26 @@
+import { Children, cloneElement } from "react";
+
+export const ViteInject = () => {
+  const elements = [];
+  if (__DEVELOPMENT__) {
+    elements.push(
+      <script type="module" src="/@vite/client" />,
+      <script
+        type="module"
+        dangerouslySetInnerHTML={{
+          __html: `
+  import RefreshRuntime from "/@react-refresh"
+  RefreshRuntime.injectIntoGlobalHook(window)
+  window.$RefreshReg$ = () => {}
+  window.$RefreshSig$ = () => (type) => type
+  window.__vite_plugin_react_preamble_installed__ = true    
+    `,
+        }}
+      />
+    );
+  }
+
+  elements.push(<script type="module" src="/src/client/entry.tsx" />);
+
+  return Children.map(elements, (child, index) => cloneElement(child, { key: index }));
+};
