From 964b2cf3bfd454733da3f9daad4debd9ddc6e885 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=99=88=E5=98=89=E6=B6=B5?= <chenjiahan@youzan.com>
Date: Fri, 29 Nov 2019 10:55:11 +0800
Subject: [PATCH] feat(cli): gen style deps map

---
 packages/vant-cli/src/common/constant.ts      |   2 +-
 .../src/compiler/gen-style-deps-map.ts        | 113 +++++++++++-------
 2 files changed, 70 insertions(+), 45 deletions(-)

diff --git a/packages/vant-cli/src/common/constant.ts b/packages/vant-cli/src/common/constant.ts
index c683ef21e..0ea996a88 100644
--- a/packages/vant-cli/src/common/constant.ts
+++ b/packages/vant-cli/src/common/constant.ts
@@ -15,7 +15,7 @@ export const CONFIG_DIR = join(__dirname, '../config');
 export const PACKAGE_ENTRY_FILE = join(DIST_DIR, 'package-entry.js');
 export const MOBILE_ENTRY_FILE = join(DIST_DIR, 'mobile-entry.js');
 export const DESKTOP_ENTRY_FILE = join(DIST_DIR, 'desktop-entry.js');
-export const STYPE_DEPS_MAP_FILE = join(DIST_DIR, 'style-deps-map.json');
+export const STYPE_DEPS_JSON_FILE = join(DIST_DIR, 'style-deps.json');
 
 export const JEST_CONFIG_FILE = join(CONFIG_DIR, 'jest.config.js');
 export const BABEL_CONFIG_FILE = join(CONFIG_DIR, 'babel.config.js');
diff --git a/packages/vant-cli/src/compiler/gen-style-deps-map.ts b/packages/vant-cli/src/compiler/gen-style-deps-map.ts
index 436471d6e..2f858ddf4 100644
--- a/packages/vant-cli/src/compiler/gen-style-deps-map.ts
+++ b/packages/vant-cli/src/compiler/gen-style-deps-map.ts
@@ -2,38 +2,17 @@
  * Build style entry of all components
  */
 
-import { join } from 'path';
-import { existsSync, writeFileSync } from 'fs-extra';
-import { getComponents } from '../common';
-import { ES_DIR, STYPE_DEPS_MAP_FILE } from '../common/constant';
 import dependencyTree from 'dependency-tree';
+import { join } from 'path';
+import { existsSync, writeFileSync } from 'fs';
+import { getComponents } from '../common';
+import { ES_DIR, STYPE_DEPS_JSON_FILE } from '../common/constant';
 
 interface DependencyObj {
   [k: string]: DependencyObj;
 }
 
 const components = getComponents();
-const STYLE_EXTS = ['.css', '.less', '.scss'];
-const JS_EXTS = ['.js', '.jsx', '.ts', '.tsx', '.vue'];
-
-function getStylePath(component: string, ext = '.css'): string {
-  return join(ES_DIR, `${component}/index${ext}`);
-}
-
-function checkStyleExists(component: string): boolean {
-  return STYLE_EXTS.some(ext => existsSync(getStylePath(component, ext)));
-}
-
-function getScriptPath(component: string): string {
-  for (let i = 0; i < JS_EXTS.length; i++) {
-    const path = join(ES_DIR, component, `index${JS_EXTS[i]}`);
-    if (existsSync(path)) {
-      return path;
-    }
-  }
-
-  return '';
-}
 
 function matchPath(path: string, component: string): boolean {
   return path
@@ -55,42 +34,88 @@ function search(tree: DependencyObj, component: string, checkList: string[]) {
   });
 }
 
+function getStylePath(component: string) {
+  return join(ES_DIR, `${component}/index.css`);
+}
+
+function checkStyleExists(component: string) {
+  return existsSync(getStylePath(component));
+}
+
 // analyze component dependencies
 function analyzeDeps(component: string) {
   const checkList: string[] = [];
-  const filename = getScriptPath(component);
-
-  if (!filename) {
-    return [];
-  }
 
   search(
     dependencyTree({
-      filename,
       directory: ES_DIR,
-      filter: path => !~path.indexOf('node_modules'),
-      detective: {
-        es6: {
-          mixedImports: true
-        }
-      }
+      filename: join(ES_DIR, component, 'index.js'),
+      filter: path => !~path.indexOf('node_modules')
     }),
     component,
     checkList
   );
 
-  checkList.push(component);
-
   return checkList.filter(checkStyleExists);
 }
 
-export function genStyleDepsMap() {
-  const map = components.reduce((map, component) => {
+type DepsMap = Record<string, string[]>;
+
+function sortComponentsByDeps(depsMap: DepsMap) {
+  const result: string[] = [];
+  const record: string[] = [];
+
+  function add(item: string) {
+    const deps = depsMap[item];
+
+    if (result.includes(item) || !deps) {
+      return;
+    }
+
+    if (record.includes(item)) {
+      result.push(item);
+      return;
+    }
+
+    record.push(item);
+
+    if (!deps.length) {
+      result.push(item);
+      return;
+    }
+
+    deps.forEach(add);
+
+    if (result.includes(item)) {
+      return;
+    }
+
+    const maxIndex = Math.max(...deps.map(dep => result.indexOf(dep)));
+
+    result.splice(maxIndex + 1, 0, item);
+  }
+
+  components.forEach(add);
+
+  return result;
+}
+
+export function genDepsMap() {
+  const map = components.filter(checkStyleExists).reduce((map, component) => {
     map[component] = analyzeDeps(component);
     return map;
-  }, {} as Record<string, string[]>);
+  }, {} as DepsMap);
 
-  const content = JSON.stringify(map, null, 2);
+  const sequence = sortComponentsByDeps(map);
 
-  writeFileSync(STYPE_DEPS_MAP_FILE, content);
+  Object.keys(map).forEach(key => {
+    map[key] = map[key].sort(
+      (a, b) => sequence.indexOf(a) - sequence.indexOf(b)
+    );
+  });
+
+  writeFileSync(
+    STYPE_DEPS_JSON_FILE,
+    JSON.stringify({ map, sequence }, null, 2)
+  );
 }