fix: document no exist in ssr

ant-design · Feb 11, 2022 · 19191fc · 19191fc
1 parent 5532138
commit 19191fc
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 6 deletions.
diff --git a/.umirc.ts b/.umirc.ts
@@ -7,8 +7,8 @@ export default defineConfig({
   logo: 'https://avatars0.githubusercontent.com/u/9441414?s=200&v=4',
   outputPath: '.doc',
   exportStatic: {},
-  base: '/',
-  publicPath: '/',
+  base: '/cssinjs/',
+  publicPath: '/cssinjs/',
   hash: true,
   styles: [
     `

diff --git a/src/useCacheToken.tsx b/src/useCacheToken.tsx
@@ -17,11 +17,13 @@ function recordCleanToken(tokenKey: string) {
 }
 
 function removeStyleTags(key: string) {
-  const styles = document.querySelectorAll(`style[data-token-key="${key}"]`);
+  if (typeof document !== 'undefined') {
+    const styles = document.querySelectorAll(`style[data-token-key="${key}"]`);
 
-  styles.forEach((style) => {
-    style.parentNode?.removeChild(style);
-  });
+    styles.forEach((style) => {
+      style.parentNode?.removeChild(style);
+    });
+  }
 }
 
 // Remove will check current keys first