From fd53ce6258d4cf0d7d63ef55e0a1dd4e0a27cac0 Mon Sep 17 00:00:00 2001
From: MadCcc <1075746765@qq.com>
Date: Mon, 3 Apr 2023 11:59:25 +0800
Subject: [PATCH] chore: share component render in scripts (#41603)

* chore: share component render in scripts

* chore: code clean
---
 package.json                       |  2 +-
 scripts/check-cssinjs.js           | 49 +++++++------------------
 scripts/collect-token-statistic.js | 59 +++++++++---------------------
 scripts/generate-cssinjs.js        | 42 +++++++++++++++++++++
 4 files changed, 75 insertions(+), 77 deletions(-)
 create mode 100644 scripts/generate-cssinjs.js

diff --git a/package.json b/package.json
index 8ddcd3f905..1fb8602321 100644
--- a/package.json
+++ b/package.json
@@ -75,7 +75,7 @@
     "lint:demo": "eslint components/*/demo/*.md",
     "lint:deps": "antd-tools run deps-lint",
     "lint:md": "remark . -f -q",
-    "lint:style": "ts-node --project tsconfig.node.json scripts/check-cssinjs.js",
+    "lint:style": "ts-node --project tsconfig.node.json scripts/check-cssinjs",
     "lint:script": "eslint . --ext .js,.jsx,.ts,.tsx --cache",
     "pre-publish": "npm run test-all -- --skip-build",
     "prettier": "prettier -c --write **/* --cache",
diff --git a/scripts/check-cssinjs.js b/scripts/check-cssinjs.js
index 45b1b853d5..2433b8fa33 100644
--- a/scripts/check-cssinjs.js
+++ b/scripts/check-cssinjs.js
@@ -1,15 +1,14 @@
 /* eslint-disable import/no-unresolved,no-console,global-require,import/no-dynamic-require */
 const chalk = require('chalk');
-const React = require('react');
-const ReactDOMServer = require('react-dom/server');
-const glob = require('glob');
-const path = require('path');
 const {
   StyleProvider,
   logicalPropertiesLinter,
   legacyNotSelectorLinter,
   parentSelectorLinter,
 } = require('@ant-design/cssinjs');
+const React = require('react');
+const ReactDOMServer = require('react-dom/server');
+const { generateCssinjs } = require('./generate-cssinjs');
 
 console.log(chalk.green(`🔥 Checking CSS-in-JS...`));
 
@@ -24,37 +23,17 @@ console.error = (msg) => {
   }
 };
 
-const EmptyElement = React.createElement('div');
-
-const styleFiles = glob.globSync(
-  path.join(
-    process.cwd(),
-    'components/!(version|config-provider|icon|auto-complete|col|row|time-picker)/style/index.?(ts|tsx)',
-  ),
-);
-
-styleFiles.forEach((file) => {
-  let useStyle = () => {};
-  if (file.includes('grid')) {
-    const { useColStyle, useRowStyle } = require(file);
-    useStyle = () => {
-      useRowStyle();
-      useColStyle();
-    };
-  } else {
-    useStyle = require(file).default;
-  }
-  const Component = () => {
-    useStyle('check');
-    return EmptyElement;
-  };
-  ReactDOMServer.renderToString(
-    React.createElement(
-      StyleProvider,
-      { linters: [logicalPropertiesLinter, legacyNotSelectorLinter, parentSelectorLinter] },
-      React.createElement(Component),
-    ),
-  );
+generateCssinjs({
+  key: 'check',
+  render: (Component) => {
+    ReactDOMServer.renderToString(
+      React.createElement(
+        StyleProvider,
+        { linters: [logicalPropertiesLinter, legacyNotSelectorLinter, parentSelectorLinter] },
+        React.createElement(Component),
+      ),
+    );
+  },
 });
 
 if (errorCount > 0) {
diff --git a/scripts/collect-token-statistic.js b/scripts/collect-token-statistic.js
index 8a204ccc81..2ec15e76ef 100644
--- a/scripts/collect-token-statistic.js
+++ b/scripts/collect-token-statistic.js
@@ -1,61 +1,38 @@
 /* eslint-disable import/no-unresolved,no-console,global-require,import/no-dynamic-require */
-
 const chalk = require('chalk');
 const React = require('react');
 const ReactDOMServer = require('react-dom/server');
 const fs = require('fs-extra');
-const glob = require('glob');
-const path = require('path');
 const ProgressBar = require('progress');
 const { statistic } = require('../components/theme/util/statistic');
 const { DesignTokenContext } = require('../components/theme/internal');
 const seedToken = require('../components/theme/themes/seed');
+const { generateCssinjs, filenames } = require('./generate-cssinjs');
 
 console.log(chalk.green(`🔥 Collecting token statistics...`));
 
-const EmptyElement = React.createElement('div');
-
-const styleFiles = glob.globSync(
-  path.join(
-    process.cwd(),
-    'components/!(version|config-provider|icon|auto-complete|col|row|time-picker)/style/index.?(ts|tsx)',
-  ),
-);
-
 const bar = new ProgressBar('🚀 Collecting by component: [:bar] :component (:current/:total)', {
   complete: '=',
   incomplete: ' ',
-  total: styleFiles.length,
+  total: filenames.length,
 });
 
-styleFiles.forEach((file) => {
-  const pathArr = file.split('/');
-  const styleIndex = pathArr.lastIndexOf('style');
-  const componentName = pathArr[styleIndex - 1];
-  bar.tick(1, { component: componentName });
-  let useStyle = () => {};
-  if (file.includes('grid')) {
-    const { useColStyle, useRowStyle } = require(file);
-    useStyle = () => {
-      useRowStyle();
-      useColStyle();
-    };
-  } else {
-    useStyle = require(file).default;
-  }
-  const Component = () => {
-    useStyle('file');
-    return EmptyElement;
-  };
-  ReactDOMServer.renderToString(React.createElement(Component));
-  // Render wireframe
-  ReactDOMServer.renderToString(
-    React.createElement(
-      DesignTokenContext.Provider,
-      { value: { token: { ...seedToken, wireframe: true } } },
-      React.createElement(Component),
-    ),
-  );
+generateCssinjs({
+  key: 'file',
+  beforeRender: (componentName) => {
+    bar.tick(1, { component: componentName });
+  },
+  render: (Component) => {
+    ReactDOMServer.renderToString(React.createElement(Component));
+    // Render wireframe
+    ReactDOMServer.renderToString(
+      React.createElement(
+        DesignTokenContext.Provider,
+        { value: { token: { ...seedToken, wireframe: true } } },
+        React.createElement(Component),
+      ),
+    );
+  },
 });
 
 (() => {
diff --git a/scripts/generate-cssinjs.js b/scripts/generate-cssinjs.js
new file mode 100644
index 0000000000..32577a209d
--- /dev/null
+++ b/scripts/generate-cssinjs.js
@@ -0,0 +1,42 @@
+const React = require('react');
+const glob = require('glob');
+const path = require('path');
+
+const styleFiles = glob.globSync(
+  path.join(
+    process.cwd(),
+    'components/!(version|config-provider|icon|auto-complete|col|row|time-picker)/style/index.?(ts|tsx)',
+  ),
+);
+
+module.exports = {
+  generateCssinjs({ key, beforeRender, render }) {
+    const EmptyElement = React.createElement('div');
+
+    styleFiles.forEach((file) => {
+      let useStyle = () => {};
+      if (file.includes('grid')) {
+        // eslint-disable-next-line global-require,import/no-dynamic-require
+        const { useColStyle, useRowStyle } = require(file);
+        useStyle = () => {
+          useRowStyle();
+          useColStyle();
+        };
+      } else {
+        // eslint-disable-next-line global-require,import/no-dynamic-require
+        useStyle = require(file).default;
+      }
+      const Component = () => {
+        useStyle(key);
+        return EmptyElement;
+      };
+
+      const pathArr = file.split('/');
+      const styleIndex = pathArr.lastIndexOf('style');
+      const componentName = pathArr[styleIndex - 1];
+      beforeRender?.(componentName);
+      render(Component);
+    });
+  },
+  filenames: styleFiles,
+};