dumi 2.3.0-alpha.9 → 2.3.0-beta.0

package/dist/assetParsers/block.d.ts

@@ -3,7 +3,13 @@ import type { ExampleBlockAsset } from 'dumi-assets-types';
 import type { sync } from 'enhanced-resolve';
 export interface IParsedBlockAsset {
     asset: ExampleBlockAsset;
-    sources: Record<string, string>;
+    /**
+     * resolveMap: {
+     *   '@/constants': '/path/to/constants.ts',
+     *   'antd/es/button': '/path/to/antd/es/button/index.jsx',
+     * }
+     */
+    resolveMap: Record<string, string>;
     frontmatter: ReturnType<typeof parseCodeFrontmatter>['frontmatter'];
 }
 declare function parseBlockAsset(opts: {

package/dist/assetParsers/block.js

@@ -47,7 +47,7 @@ async function parseBlockAsset(opts) {
   };
   const result = {
     asset,
-    sources: {},
+    resolveMap: {},
     frontmatter: null
   };
   const deferrer = (0, import_esbuild.build)({
@@ -81,12 +81,17 @@ async function parseBlockAsset(opts) {
                   type: "NPM",
                   value: pkg.version
                 };
+                result.resolveMap[args.path] = resolved;
               }
               return { path: args.path, external: true };
             }
             return {
               path: args.kind !== "entry-point" ? opts.resolver(args.resolveDir, args.path) : import_path.default.join(args.resolveDir, args.path),
-              pluginData: { kind: args.kind, resolveDir: args.resolveDir }
+              pluginData: {
+                kind: args.kind,
+                resolveDir: args.resolveDir,
+                source: args.path
+              }
             };
           });
           builder.onLoad({ filter: /.*/ }, (args) => {
@@ -101,9 +106,12 @@ async function parseBlockAsset(opts) {
               ".json"
             ].includes(ext);
             const isEntryPoint = args.pluginData.kind === "entry-point";
-            const filename = isEntryPoint ? `index${ext}` : (0, import_plugin_utils.winPath)(
-              import_path.default.relative(import_path.default.dirname(opts.fileAbsPath), args.path)
-            ).replace(/^(\.?\.\/)+/g, "");
+            const filename = `${isEntryPoint ? "index" : (0, import_plugin_utils.winPath)(
+              import_path.default.format({
+                ...import_path.default.parse(args.pluginData.source),
+                ext: ""
+              })
+            )}${ext}`;
             if (isModule || isPlainText) {
               asset.dependencies[filename] = {
                 type: "FILE",
@@ -124,7 +132,7 @@ async function parseBlockAsset(opts) {
                 }
               }
               if (!isEntryPoint || !opts.entryPointCode) {
-                result.sources[filename] = args.path;
+                result.resolveMap[filename] = args.path;
               }
               return {
                 // only continue to load for module files

package/dist/client/pages/Demo/index.js

@@ -1,12 +1,26 @@
-import { useDemoData, useParams } from 'dumi';
-import { createElement } from 'react';
+import { useDemo, useLiveDemo, useParams } from 'dumi';
+import { createElement, useEffect } from 'react';
 import "./index.less";
 var DemoRenderPage = function DemoRenderPage() {
   var _useParams = useParams(),
     id = _useParams.id;
-  var demoInfo = useDemoData(id);
-  var _ref = demoInfo,
+  var _ref = useDemo(id) || {},
     component = _ref.component;
-  return component && /*#__PURE__*/createElement(component);
+  var _useLiveDemo = useLiveDemo(id),
+    liveDemoNode = _useLiveDemo.node,
+    setSources = _useLiveDemo.setSources;
+  var finalNode = liveDemoNode || component && /*#__PURE__*/createElement(component);
+  useEffect(function () {
+    var handler = function handler(ev) {
+      if (ev.data.type === 'dumi.liveDemo.setSources') {
+        setSources(ev.data.value);
+      }
+    };
+    window.addEventListener('message', handler);
+    return function () {
+      return window.removeEventListener('message', handler);
+    };
+  }, [setSources]);
+  return finalNode;
 };
 export default DemoRenderPage;

package/dist/client/pages/Loading.d.ts

@@ -0,0 +1 @@
+export { default } from 'dumi/theme/slots/Loading';

package/dist/client/pages/Loading.js

@@ -0,0 +1,2 @@
+// allow customize loading via theme slots or ./dumi/loading.tsx
+export { default } from 'dumi/theme/slots/Loading';

package/dist/client/theme-api/DumiDemo/index.js

@@ -1,6 +1,6 @@
 function _extends() { _extends = Object.assign ? Object.assign.bind() : function (target) { for (var i = 1; i < arguments.length; i++) { var source = arguments[i]; for (var key in source) { if (Object.prototype.hasOwnProperty.call(source, key)) { target[key] = source[key]; } } } return target; }; return _extends.apply(this, arguments); }
 import { SP_ROUTE_PREFIX } from "../../../constants";
-import { useAppData, useDemoData, useSiteData } from 'dumi';
+import { useAppData, useDemo, useSiteData } from 'dumi';
 import React, { createElement } from 'react';
 import Previewer from 'dumi/theme/builtins/Previewer';
 import DemoErrorBoundary from "./DemoErrorBoundary";
@@ -9,12 +9,12 @@ var InternalDumiDemo = function InternalDumiDemo(props) {
     historyType = _useSiteData.historyType;
   var _useAppData = useAppData(),
     basename = _useAppData.basename;
-  var demoInfo = useDemoData(props.demo.id);
+  var _ref = useDemo(props.demo.id),
+    component = _ref.component,
+    asset = _ref.asset;
 
   // hide debug demo in production
   if (process.env.NODE_ENV === 'production' && props.previewerProps.debug) return null;
-  var component = demoInfo.component,
-    asset = demoInfo.asset;
   var demoNode = /*#__PURE__*/React.createElement(DemoErrorBoundary, null, /*#__PURE__*/createElement(component));
   if (props.demo.inline) {
     return demoNode;

package/dist/client/theme-api/{context/index.d.ts → context.d.ts}

@@ -1,19 +1,13 @@
-import type { PICKED_PKG_FIELDS } from "../../../constants";
+/// <reference types="react" />
+import type { PICKED_PKG_FIELDS } from "../../constants";
 import type { AtomComponentAsset } from 'dumi-assets-types';
-import { type ComponentType } from 'react';
-import type { ILocalesConfig, IPreviewerProps, IRouteMeta, IThemeConfig } from '../types';
-export type DemoInfo = {
-    component: ComponentType;
-    asset: IPreviewerProps['asset'];
-    routeId: string;
-};
+import type { IDemoData, ILocalesConfig, IThemeConfig } from './types';
 interface ISiteContext {
     pkg: Partial<Record<keyof typeof PICKED_PKG_FIELDS, any>>;
     historyType: 'browser' | 'hash' | 'memory';
     entryExports: Record<string, any>;
-    demos: Record<string, DemoInfo>;
+    demos: Record<string, IDemoData>;
     components: Record<string, AtomComponentAsset>;
-    tabs: IRouteMeta['tabs'];
     locales: ILocalesConfig;
     themeConfig: IThemeConfig;
     hostname?: string;
@@ -23,9 +17,7 @@ interface ISiteContext {
      * private field, do not use it in your code
      */
     _2_level_nav_available: boolean;
-    getDemoById: (id: string) => Promise<DemoInfo | null>;
 }
 export declare const SiteContext: import("react").Context<ISiteContext>;
 export declare const useSiteData: () => ISiteContext;
-export declare function useDemoData(demoId: string): DemoInfo | null;
 export {};

package/dist/client/theme-api/context.js

@@ -0,0 +1,16 @@
+import { createContext, useContext } from 'react';
+export var SiteContext = /*#__PURE__*/createContext({
+  pkg: {},
+  historyType: 'browser',
+  entryExports: {},
+  demos: {},
+  components: {},
+  locales: [],
+  themeConfig: {},
+  loading: false,
+  setLoading: function setLoading() {},
+  _2_level_nav_available: true
+});
+export var useSiteData = function useSiteData() {
+  return useContext(SiteContext);
+};

package/dist/client/theme-api/index.d.ts

@@ -3,13 +3,12 @@ export { AtomRenderer } from './AtomRenderer';
 export { DumiDemo } from './DumiDemo';
 export { DumiDemoGrid } from './DumiDemoGrid';
 export { DumiPage } from './DumiPage';
-export { useDemoData, useSiteData } from './context';
-export { evalCode } from './evalCode';
-export { LiveContext, LiveProvider, isLiveEnabled } from './live';
+export { useSiteData } from './context';
 export { openCodeSandbox } from './openCodeSandbox';
 export { openStackBlitz } from './openStackBlitz';
 export type { IPreviewerProps } from './types';
 export { useAtomAssets } from './useAtomAssets';
+export { useLiveDemo } from './useLiveDemo';
 export { useLocale } from './useLocale';
 export { useNavData } from './useNavData';
 export { usePrefersColor } from './usePrefersColor';

package/dist/client/theme-api/index.js

@@ -3,12 +3,11 @@ export { AtomRenderer } from "./AtomRenderer";
 export { DumiDemo } from "./DumiDemo";
 export { DumiDemoGrid } from "./DumiDemoGrid";
 export { DumiPage } from "./DumiPage";
-export { useDemoData, useSiteData } from "./context";
-export { evalCode } from "./evalCode";
-export { LiveContext, LiveProvider, isLiveEnabled } from "./live";
+export { useSiteData } from "./context";
 export { openCodeSandbox } from "./openCodeSandbox";
 export { openStackBlitz } from "./openStackBlitz";
 export { useAtomAssets } from "./useAtomAssets";
+export { useLiveDemo } from "./useLiveDemo";
 export { useLocale } from "./useLocale";
 export { useNavData } from "./useNavData";
 export { usePrefersColor } from "./usePrefersColor";

package/dist/client/theme-api/types.d.ts

@@ -41,10 +41,6 @@ export interface IPreviewerProps {
      * background color for demo content
      */
     background?: string;
-    /**
-     * enable live demo
-     */
-    live?: boolean;
     /**
      * asset metadata of current demo
      */
@@ -223,4 +219,10 @@ export type IRoutesById = Record<string, {
     redirect?: string;
     [key: string]: any;
 }>;
+export type IDemoData = {
+    component: ComponentType;
+    asset: IPreviewerProps['asset'];
+    routeId: string;
+    context?: Record<string, unknown>;
+};
 export {};

package/dist/client/theme-api/useLiveDemo.d.ts

@@ -0,0 +1,6 @@
+import { type ReactNode } from 'react';
+export declare const useLiveDemo: (id: string) => {
+    node: ReactNode;
+    error: Error | null;
+    setSources: (sources: Record<string, string>) => void;
+};

package/dist/client/theme-api/useLiveDemo.js

@@ -0,0 +1,70 @@
+function _slicedToArray(arr, i) { return _arrayWithHoles(arr) || _iterableToArrayLimit(arr, i) || _unsupportedIterableToArray(arr, i) || _nonIterableRest(); }
+function _nonIterableRest() { throw new TypeError("Invalid attempt to destructure non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); }
+function _unsupportedIterableToArray(o, minLen) { if (!o) return; if (typeof o === "string") return _arrayLikeToArray(o, minLen); var n = Object.prototype.toString.call(o).slice(8, -1); if (n === "Object" && o.constructor) n = o.constructor.name; if (n === "Map" || n === "Set") return Array.from(o); if (n === "Arguments" || /^(?:Ui|I)nt(?:8|16|32)(?:Clamped)?Array$/.test(n)) return _arrayLikeToArray(o, minLen); }
+function _arrayLikeToArray(arr, len) { if (len == null || len > arr.length) len = arr.length; for (var i = 0, arr2 = new Array(len); i < len; i++) arr2[i] = arr[i]; return arr2; }
+function _iterableToArrayLimit(arr, i) { var _i = null == arr ? null : "undefined" != typeof Symbol && arr[Symbol.iterator] || arr["@@iterator"]; if (null != _i) { var _s, _e, _x, _r, _arr = [], _n = !0, _d = !1; try { if (_x = (_i = _i.call(arr)).next, 0 === i) { if (Object(_i) !== _i) return; _n = !1; } else for (; !(_n = (_s = _x.call(_i)).done) && (_arr.push(_s.value), _arr.length !== i); _n = !0); } catch (err) { _d = !0, _e = err; } finally { try { if (!_n && null != _i.return && (_r = _i.return(), Object(_r) !== _r)) return; } finally { if (_d) throw _e; } } return _arr; } }
+function _arrayWithHoles(arr) { if (Array.isArray(arr)) return arr; }
+import { useDemo } from 'dumi';
+import { createElement, useCallback, useState } from 'react';
+import DemoErrorBoundary from "./DumiDemo/DemoErrorBoundary";
+export var useLiveDemo = function useLiveDemo(id) {
+  var _ref = useDemo(id),
+    context = _ref.context,
+    asset = _ref.asset;
+  var _useState = useState(),
+    _useState2 = _slicedToArray(_useState, 2),
+    demoNode = _useState2[0],
+    setDemoNode = _useState2[1];
+  var _useState3 = useState(null),
+    _useState4 = _slicedToArray(_useState3, 2),
+    error = _useState4[0],
+    setError = _useState4[1];
+  var setSources = useCallback(function (sources) {
+    var entryFileName = Object.keys(asset.dependencies).find(function (k) {
+      return asset.dependencies[k].type === 'FILE';
+    });
+    var entryFileCode = sources[entryFileName];
+    var require = function require(v) {
+      if (v in context) return context[v];
+      throw new Error("Cannot find module: ".concat(v));
+    };
+    var exports = {};
+    var module = {
+      exports: exports
+    };
+
+    // lazy load react-dom/server
+    import('react-dom/server').then(function (_ref2) {
+      var renderToStaticMarkup = _ref2.renderToStaticMarkup;
+      try {
+        // initial component with fake runtime
+        new Function('module', 'exports', 'require', entryFileCode)(module, exports, require);
+        var newDemoNode = /*#__PURE__*/createElement(DemoErrorBoundary, null, /*#__PURE__*/createElement(exports.default));
+        var oError = console.error;
+
+        // hijack console.error to avoid useLayoutEffect error
+        console.error = function () {
+          for (var _len = arguments.length, args = new Array(_len), _key = 0; _key < _len; _key++) {
+            args[_key] = arguments[_key];
+          }
+          return !args[0].includes('useLayoutEffect does nothing on the server') && oError.apply(console, args);
+        };
+
+        // check component is renderable, to avoid show react overlay error
+        renderToStaticMarkup(newDemoNode);
+        console.error = oError;
+
+        // set new demo node with passing sources
+        setDemoNode(newDemoNode);
+        setError(null);
+      } catch (err) {
+        setError(err);
+      }
+    });
+  }, [context]);
+  return {
+    node: demoNode,
+    error: error,
+    setSources: setSources
+  };
+};

package/dist/client/theme-api/useRouteMeta.js

@@ -1,19 +1,69 @@
-function _typeof(obj) { "@babel/helpers - typeof"; return _typeof = "function" == typeof Symbol && "symbol" == typeof Symbol.iterator ? function (obj) { return typeof obj; } : function (obj) { return obj && "function" == typeof Symbol && obj.constructor === Symbol && obj !== Symbol.prototype ? "symbol" : typeof obj; }, _typeof(obj); }
+function _slicedToArray(arr, i) { return _arrayWithHoles(arr) || _iterableToArrayLimit(arr, i) || _unsupportedIterableToArray(arr, i) || _nonIterableRest(); }
+function _nonIterableRest() { throw new TypeError("Invalid attempt to destructure non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); }
+function _unsupportedIterableToArray(o, minLen) { if (!o) return; if (typeof o === "string") return _arrayLikeToArray(o, minLen); var n = Object.prototype.toString.call(o).slice(8, -1); if (n === "Object" && o.constructor) n = o.constructor.name; if (n === "Map" || n === "Set") return Array.from(o); if (n === "Arguments" || /^(?:Ui|I)nt(?:8|16|32)(?:Clamped)?Array$/.test(n)) return _arrayLikeToArray(o, minLen); }
+function _arrayLikeToArray(arr, len) { if (len == null || len > arr.length) len = arr.length; for (var i = 0, arr2 = new Array(len); i < len; i++) arr2[i] = arr[i]; return arr2; }
+function _iterableToArrayLimit(arr, i) { var _i = null == arr ? null : "undefined" != typeof Symbol && arr[Symbol.iterator] || arr["@@iterator"]; if (null != _i) { var _s, _e, _x, _r, _arr = [], _n = !0, _d = !1; try { if (_x = (_i = _i.call(arr)).next, 0 === i) { if (Object(_i) !== _i) return; _n = !1; } else for (; !(_n = (_s = _x.call(_i)).done) && (_arr.push(_s.value), _arr.length !== i); _n = !0); } catch (err) { _d = !0, _e = err; } finally { try { if (!_n && null != _i.return && (_r = _i.return(), Object(_r) !== _r)) return; } finally { if (_d) throw _e; } } return _arr; } }
+function _arrayWithHoles(arr) { if (Array.isArray(arr)) return arr; }
 import { getRouteMetaById, matchRoutes, useAppData, useLocation, useRouteData } from 'dumi';
-import { useMemo } from 'react';
-import use from "./context/use";
+import { useCallback, useState } from 'react';
+import { use, useIsomorphicLayoutEffect } from "./utils";
 var cache = new Map();
-var useAsyncRouteMeta = function useAsyncRouteMeta(id) {
-  if (!cache.has(id)) {
-    cache.set(id, getRouteMetaById(id));
-  }
-  return use(cache.get(id));
-};
-var emptyMeta = {
+var EMPTY_META = {
   frontmatter: {},
   toc: [],
   texts: []
 };
+var ASYNC_META_PROPS = ['texts'];
+function getCachedRouteMeta(route) {
+  var cacheKey = route.id;
+  var pendingCacheKey = "".concat(cacheKey, ":pending");
+  if (!cache.get(cacheKey)) {
+    var _meta$tabs;
+    var merge = function merge() {
+      var meta = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : EMPTY_META;
+      if (route.meta) {
+        Object.keys(route.meta).forEach(function (key) {
+          var _ref, _ref$key;
+          (_ref$key = (_ref = meta)[key]) !== null && _ref$key !== void 0 ? _ref$key : _ref[key] = route.meta[key];
+        });
+      }
+      return meta;
+    };
+    var meta = merge(getRouteMetaById(route.id, {
+      syncOnly: true
+    }));
+    var ret = Promise.resolve(meta);
+    var proxyGetter = function proxyGetter(target, prop) {
+      if (ASYNC_META_PROPS.includes(prop)) {
+        if (!cache.get(pendingCacheKey)) {
+          // load async meta then replace cache
+          cache.set(pendingCacheKey, getRouteMetaById(route.id).then(function (full) {
+            return cache.set(cacheKey, Promise.resolve(merge(full))).get(cacheKey);
+          }));
+        }
+
+        // throw promise to trigger suspense
+        throw cache.get(pendingCacheKey);
+      }
+      return target[prop];
+    };
+
+    // return sync meta by default
+    ret.status = 'fulfilled';
+
+    // load async meta if property accessed
+    (_meta$tabs = meta.tabs) === null || _meta$tabs === void 0 ? void 0 : _meta$tabs.forEach(function (tab) {
+      tab.meta = new Proxy(tab.meta, {
+        get: proxyGetter
+      });
+    });
+    ret.value = new Proxy(meta, {
+      get: proxyGetter
+    });
+    cache.set(cacheKey, ret);
+  }
+  return cache.get(cacheKey);
+}
 
 /**
  * hook for get matched route meta
@@ -25,23 +75,26 @@ export var useRouteMeta = function useRouteMeta() {
     pathname = _useLocation.pathname;
   var _useAppData = useAppData(),
     clientRoutes = _useAppData.clientRoutes;
-  var curRoute = useMemo(function () {
+  var getter = useCallback(function () {
+    var ret;
     if (route.path === pathname && !('isLayout' in route)) {
       // use `useRouteData` result if matched, for performance
-      return route;
+      ret = route;
     } else {
       var _matchRoutes;
       // match manually for dynamic route & layout component
       var matched = (_matchRoutes = matchRoutes(clientRoutes, pathname)) === null || _matchRoutes === void 0 ? void 0 : _matchRoutes.pop();
-      return matched === null || matched === void 0 ? void 0 : matched.route;
+      ret = matched === null || matched === void 0 ? void 0 : matched.route;
     }
+    return ret;
+  }, [clientRoutes.length, pathname]);
+  var _useState = useState(getter),
+    _useState2 = _slicedToArray(_useState, 2),
+    matchedRoute = _useState2[0],
+    setMatchedRoute = _useState2[1];
+  var meta = use(getCachedRouteMeta(matchedRoute));
+  useIsomorphicLayoutEffect(function () {
+    setMatchedRoute(getter);
   }, [clientRoutes.length, pathname]);
-  var meta = useAsyncRouteMeta(curRoute === null || curRoute === void 0 ? void 0 : curRoute.id) || emptyMeta;
-  if (curRoute && 'meta' in curRoute && _typeof(curRoute.meta) === 'object') {
-    Object.keys(curRoute.meta).forEach(function (key) {
-      var _ref, _ref$key;
-      (_ref$key = (_ref = meta)[key]) !== null && _ref$key !== void 0 ? _ref$key : _ref[key] = curRoute.meta[key];
-    });
-  }
   return meta;
 };

package/dist/client/theme-api/useSiteSearch/index.d.ts

@@ -20,5 +20,5 @@ export declare const useSiteSearch: () => {
     setKeywords: (val: string) => void;
     result: ISearchResult;
     loading: boolean;
-    loadSearchData: () => void;
+    load: () => Promise<void>;
 };

package/dist/client/theme-api/useSiteSearch/index.js

@@ -1,9 +1,3 @@
-function _typeof(obj) { "@babel/helpers - typeof"; return _typeof = "function" == typeof Symbol && "symbol" == typeof Symbol.iterator ? function (obj) { return typeof obj; } : function (obj) { return obj && "function" == typeof Symbol && obj.constructor === Symbol && obj !== Symbol.prototype ? "symbol" : typeof obj; }, _typeof(obj); }
-function ownKeys(object, enumerableOnly) { var keys = Object.keys(object); if (Object.getOwnPropertySymbols) { var symbols = Object.getOwnPropertySymbols(object); enumerableOnly && (symbols = symbols.filter(function (sym) { return Object.getOwnPropertyDescriptor(object, sym).enumerable; })), keys.push.apply(keys, symbols); } return keys; }
-function _objectSpread(target) { for (var i = 1; i < arguments.length; i++) { var source = null != arguments[i] ? arguments[i] : {}; i % 2 ? ownKeys(Object(source), !0).forEach(function (key) { _defineProperty(target, key, source[key]); }) : Object.getOwnPropertyDescriptors ? Object.defineProperties(target, Object.getOwnPropertyDescriptors(source)) : ownKeys(Object(source)).forEach(function (key) { Object.defineProperty(target, key, Object.getOwnPropertyDescriptor(source, key)); }); } return target; }
-function _defineProperty(obj, key, value) { key = _toPropertyKey(key); if (key in obj) { Object.defineProperty(obj, key, { value: value, enumerable: true, configurable: true, writable: true }); } else { obj[key] = value; } return obj; }
-function _toPropertyKey(arg) { var key = _toPrimitive(arg, "string"); return _typeof(key) === "symbol" ? key : String(key); }
-function _toPrimitive(input, hint) { if (_typeof(input) !== "object" || input === null) return input; var prim = input[Symbol.toPrimitive]; if (prim !== undefined) { var res = prim.call(input, hint || "default"); if (_typeof(res) !== "object") return res; throw new TypeError("@@toPrimitive must return a primitive value."); } return (hint === "string" ? String : Number)(input); }
 function _slicedToArray(arr, i) { return _arrayWithHoles(arr) || _iterableToArrayLimit(arr, i) || _unsupportedIterableToArray(arr, i) || _nonIterableRest(); }
 function _nonIterableRest() { throw new TypeError("Invalid attempt to destructure non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); }
 function _unsupportedIterableToArray(o, minLen) { if (!o) return; if (typeof o === "string") return _arrayLikeToArray(o, minLen); var n = Object.prototype.toString.call(o).slice(8, -1); if (n === "Object" && o.constructor) n = o.constructor.name; if (n === "Map" || n === "Set") return Array.from(o); if (n === "Arguments" || /^(?:Ui|I)nt(?:8|16|32)(?:Clamped)?Array$/.test(n)) return _arrayLikeToArray(o, minLen); }
@@ -26,7 +20,7 @@ if (typeof window !== 'undefined') {
 }
 export var useSiteSearch = function useSiteSearch() {
   var debounceTimer = useRef();
-  var _useState = useState(true),
+  var _useState = useState(false),
     _useState2 = _slicedToArray(_useState, 2),
     loading = _useState2[0],
     setLoading = _useState2[1];
@@ -34,27 +28,22 @@ export var useSiteSearch = function useSiteSearch() {
     _useState4 = _slicedToArray(_useState3, 2),
     keywords = _useState4[0],
     setKeywords = _useState4[1];
-  var _useState5 = useState(false),
-    _useState6 = _slicedToArray(_useState5, 2),
-    enabled = _useState6[0],
-    setEnabled = _useState6[1];
   var navData = useNavData();
-  var _useState7 = useState([]),
-    _useState8 = _slicedToArray(_useState7, 2),
-    result = _useState8[0],
-    setResult = _useState8[1];
+  var _useState5 = useState([]),
+    _useState6 = _slicedToArray(_useState5, 2),
+    result = _useState6[0],
+    setResult = _useState6[1];
+  var _useSearchData = useSearchData(),
+    _useSearchData2 = _slicedToArray(_useSearchData, 2),
+    data = _useSearchData2[0],
+    load = _useSearchData2[1];
   var setter = useCallback(function (val) {
+    load();
     setLoading(true);
     setKeywords(val);
   }, []);
-  var loadSearchData = function loadSearchData() {
-    return setEnabled(true);
-  };
-  var _useSearchData = useSearchData(enabled),
-    _useSearchData2 = _slicedToArray(_useSearchData, 2),
-    filledRoutes = _useSearchData2[0],
-    demos = _useSearchData2[1];
-  var mergedLoading = loading || !filledRoutes;
+  var routes = data === null || data === void 0 ? void 0 : data[0];
+  var demos = data === null || data === void 0 ? void 0 : data[1];
   useEffect(function () {
     worker.onmessage = function (e) {
       setResult(e.data);
@@ -62,35 +51,18 @@ export var useSiteSearch = function useSiteSearch() {
     };
   }, []);
   useEffect(function () {
-    if (!filledRoutes || !demos) {
-      return;
-    }
-
-    // omit demo component for postmessage
-    var demoData = Object.entries(demos).reduce(function (acc, _ref) {
-      var _ref2 = _slicedToArray(_ref, 2),
-        key = _ref2[0],
-        _ref2$ = _ref2[1],
-        asset = _ref2$.asset,
-        routeId = _ref2$.routeId;
-      return _objectSpread(_objectSpread({}, acc), {}, _defineProperty({}, key, {
-        asset: asset,
-        routeId: routeId
-      }));
-    }, {});
+    if (!routes || !demos) return;
     worker.postMessage({
       action: 'generate-metadata',
       args: {
-        routes: JSON.parse(JSON.stringify(filledRoutes)),
+        routes: JSON.parse(JSON.stringify(routes)),
         nav: navData,
-        demos: demoData
+        demos: demos
       }
     });
-  }, [demos, navData, filledRoutes]);
+  }, [routes, demos, navData]);
   useEffect(function () {
-    if (!filledRoutes) {
-      return;
-    }
+    if (!routes) return;
     var str = keywords.trim();
     if (str) {
       clearTimeout(debounceTimer.current);
@@ -105,12 +77,12 @@ export var useSiteSearch = function useSiteSearch() {
     } else {
       setResult([]);
     }
-  }, [keywords, filledRoutes]);
+  }, [keywords, routes]);
   return {
     keywords: keywords,
     setKeywords: setter,
     result: result,
-    loading: mergedLoading,
-    loadSearchData: loadSearchData
+    loading: loading,
+    load: load
   };
 };

package/dist/client/theme-api/useSiteSearch/useSearchData.d.ts

@@ -1,9 +1,12 @@
-import type { DemoInfo } from '../context';
+import { getFullDemos } from 'dumi';
 import { useLocaleDocRoutes } from '../utils';
-type Demos = Record<string, DemoInfo>;
-type ReturnData = [
-    routes: ReturnType<typeof useLocaleDocRoutes> | null,
-    demo: Demos | null
+type IDemosData = Record<string, Partial<Awaited<ReturnType<typeof getFullDemos>>[string]>>;
+type ISearchData = [
+    routes: ReturnType<typeof useLocaleDocRoutes>,
+    demos: IDemosData
+];
+export default function useSearchData(): [
+    ISearchData | null,
+    () => Promise<void>
 ];
-export default function useSearchData(enabled: boolean): ReturnData;
 export {};