|
1 | | -import React from 'react'; |
| 1 | +import React, { useState } from 'react'; |
2 | 2 | import { Tooltip } from 'antd'; |
3 | 3 | import { NavLink, useSidebarData } from 'dumi'; |
4 | 4 |
|
5 | 5 | import { useMobile } from '../../hooks'; |
6 | 6 | import 'dumi/theme-default/slots/Sidebar/index.less'; |
7 | 7 | import './index.scss'; |
8 | 8 |
|
| 9 | +function filterSideBar(arr: ReturnType<typeof useSidebarData>, search: string) { |
| 10 | + if (!search) return arr; |
| 11 | + return arr |
| 12 | + .map((item) => { |
| 13 | + const children = item.children.filter((child) => |
| 14 | + child.title.toLocaleLowerCase().includes(search.toLocaleLowerCase()) |
| 15 | + ); |
| 16 | + return { ...item, children }; |
| 17 | + }) |
| 18 | + .filter((i) => i.children.length); |
| 19 | +} |
| 20 | + |
9 | 21 | export default function Sidebar() { |
10 | 22 | const sidebar = useSidebarData(); |
11 | 23 | const isMobile = useMobile(); |
| 24 | + const [search, setSearch] = useState(''); |
| 25 | + |
| 26 | + const isRenderSearch = sidebar.some((s) => s.title === '组件'); |
12 | 27 |
|
13 | 28 | if (!sidebar) return null; |
14 | 29 |
|
15 | 30 | return ( |
16 | 31 | <div className="dumi-default-sidebar"> |
17 | | - {sidebar.map((item, i) => ( |
| 32 | + {isRenderSearch && ( |
| 33 | + <input |
| 34 | + className="dt-theme-filter" |
| 35 | + placeholder="搜索组件..." |
| 36 | + value={search} |
| 37 | + onChange={(e) => setSearch(e.target.value)} |
| 38 | + onKeyDown={(e) => { |
| 39 | + if (e.key === 'Enter') { |
| 40 | + const firstLink = document.querySelector( |
| 41 | + '.dumi-default-sidebar-group dd a' |
| 42 | + ); |
| 43 | + if (firstLink) { |
| 44 | + (firstLink as HTMLAnchorElement).click(); |
| 45 | + } |
| 46 | + } |
| 47 | + }} |
| 48 | + /> |
| 49 | + )} |
| 50 | + {filterSideBar(sidebar, isRenderSearch ? search : '').map((item, i) => ( |
18 | 51 | <dl className="dumi-default-sidebar-group" key={String(i)}> |
19 | 52 | {item.title && <dt>{item.title}</dt>} |
20 | 53 | {item.children.map((child) => { |
21 | 54 | const link = ( |
22 | 55 | <dd key={child.link}> |
23 | | - <NavLink to={child.link} title={child.title} end> |
| 56 | + <NavLink |
| 57 | + to={child.link} |
| 58 | + title={child.title} |
| 59 | + end |
| 60 | + onClick={() => setSearch('')} |
| 61 | + > |
24 | 62 | {child.title} |
25 | 63 | </NavLink> |
26 | 64 | </dd> |
|
0 commit comments