@arco-design/web-react
Version:
Arco Design React UI Library.
203 lines (202 loc) • 11.2 kB
JavaScript
var __assign = (this && this.__assign) || function () {
__assign = Object.assign || function(t) {
for (var s, i = 1, n = arguments.length; i < n; i++) {
s = arguments[i];
for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p))
t[p] = s[p];
}
return t;
};
return __assign.apply(this, arguments);
};
var __rest = (this && this.__rest) || function (s, e) {
var t = {};
for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p) && e.indexOf(p) < 0)
t[p] = s[p];
if (s != null && typeof Object.getOwnPropertySymbols === "function")
for (var i = 0, p = Object.getOwnPropertySymbols(s); i < p.length; i++) {
if (e.indexOf(p[i]) < 0 && Object.prototype.propertyIsEnumerable.call(s, p[i]))
t[p[i]] = s[p[i]];
}
return t;
};
var __read = (this && this.__read) || function (o, n) {
var m = typeof Symbol === "function" && o[Symbol.iterator];
if (!m) return o;
var i = m.call(o), r, ar = [], e;
try {
while ((n === void 0 || n-- > 0) && !(r = i.next()).done) ar.push(r.value);
}
catch (error) { e = { error: error }; }
finally {
try {
if (r && !r.done && (m = i["return"])) m.call(i);
}
finally { if (e) throw e.error; }
}
return ar;
};
var __spreadArray = (this && this.__spreadArray) || function (to, from, pack) {
if (pack || arguments.length === 2) for (var i = 0, l = from.length, ar; i < l; i++) {
if (ar || !(i in from)) {
if (!ar) ar = Array.prototype.slice.call(from, 0, i);
ar[i] = from[i];
}
}
return to.concat(ar || Array.prototype.slice.call(from));
};
import React, { useEffect, useContext, forwardRef, useRef, useMemo } from 'react';
import cs from '../_util/classNames';
import Item from './item';
import ItemGroup from './item-group';
import SubMenu from './sub-menu';
import OverflowWrap from './overflow-wrap';
import omit from '../_util/omit';
import { generateInfoMap, processChildren } from './util';
import { ConfigContext } from '../ConfigProvider';
import { SiderContext } from '../Layout';
import useMergeValue from '../_util/hooks/useMergeValue';
import IconMenuFold from '../../icon/react-icon/IconMenuFold';
import IconMenuUnfold from '../../icon/react-icon/IconMenuUnfold';
import useForceUpdate from '../_util/hooks/useForceUpdate';
import MenuContext from './context';
import useMergeProps from '../_util/hooks/useMergeProps';
import useKeyboardEvent from '../_util/hooks/useKeyboardEvent';
import useId from '../_util/hooks/useId';
import { isObject } from '../_util/is';
var DEFAULT_THEME = 'light';
var defaultProps = {
mode: 'vertical',
selectable: true,
ellipsis: true,
};
function Menu(baseProps, ref) {
var _a;
var _b = useContext(ConfigContext), getPrefixCls = _b.getPrefixCls, componentConfig = _b.componentConfig, rtl = _b.rtl;
var props = useMergeProps(baseProps, defaultProps, componentConfig === null || componentConfig === void 0 ? void 0 : componentConfig.Menu);
var style = props.style, children = props.children, className = props.className, menuPrefixCls = props.prefixCls, mode = props.mode, propTheme = props.theme, icons = props.icons, levelIndent = props.levelIndent, propCollapse = props.collapse, inDropdown = props.inDropdown, selectable = props.selectable, triggerProps = props.triggerProps, tooltipProps = props.tooltipProps, ellipsis = props.ellipsis, accordion = props.accordion, autoOpen = props.autoOpen, autoScrollIntoView = props.autoScrollIntoView, scrollConfig = props.scrollConfig, hasCollapseButton = props.hasCollapseButton, defaultOpenKeys = props.defaultOpenKeys, defaultSelectedKeys = props.defaultSelectedKeys, propOpenKeys = props.openKeys, propSelectedKeys = props.selectedKeys, onClickSubMenu = props.onClickSubMenu, onClickMenuItem = props.onClickMenuItem, onCollapseChange = props.onCollapseChange, onEllipsisChange = props.onEllipsisChange, rest = __rest(props, ["style", "children", "className", "prefixCls", "mode", "theme", "icons", "levelIndent", "collapse", "inDropdown", "selectable", "triggerProps", "tooltipProps", "ellipsis", "accordion", "autoOpen", "autoScrollIntoView", "scrollConfig", "hasCollapseButton", "defaultOpenKeys", "defaultSelectedKeys", "openKeys", "selectedKeys", "onClickSubMenu", "onClickMenuItem", "onCollapseChange", "onEllipsisChange"]);
var _c = __read(useMergeValue([], {
defaultValue: defaultOpenKeys,
value: propOpenKeys,
}), 2), openKeys = _c[0], setOpenKeys = _c[1];
var _d = __read(useMergeValue([], {
defaultValue: defaultSelectedKeys,
value: propSelectedKeys,
}), 2), selectedKeys = _d[0], setSelectedKeys = _d[1];
var _e = __read(useMergeValue(false, {
value: propCollapse,
}), 2), collapse = _e[0], setCollapse = _e[1];
var menuContext = useContext(MenuContext);
var siderCollapsed = useContext(SiderContext).siderCollapsed;
var prefixCls = menuPrefixCls || getPrefixCls('menu');
var mergedCollapse = siderCollapsed || collapse || inDropdown || mode === 'popButton';
var theme = propTheme || menuContext.theme || DEFAULT_THEME;
var refSubMenuKeys = useRef([]);
var refPrevSubMenuKeys = useRef([]);
var forceUpdate = useForceUpdate();
var getKeyboardEvents = useKeyboardEvent();
var menuInfoMap = useMemo(function () {
return generateInfoMap(children);
}, [children]);
// Unique ID of this instance
var _instanceId = useId(prefixCls + "-");
var instanceId = rest.id || _instanceId;
// autoOpen 时，初次渲染展开所有的子菜单
useEffect(function () {
// 从 openKeys 中过滤已经不存在的 subMenuKey
var validOpenKeys = openKeys.filter(function (key) { return refSubMenuKeys.current.indexOf(key) !== -1; });
if (autoOpen) {
var keysAdded = refSubMenuKeys.current.filter(function (key) { return refPrevSubMenuKeys.current.indexOf(key) === -1; });
validOpenKeys = openKeys.concat(keysAdded);
}
setOpenKeys(accordion ? validOpenKeys.slice(0, 1) : validOpenKeys);
refPrevSubMenuKeys.current = refSubMenuKeys.current.slice();
}, [refSubMenuKeys.current.toString()]);
var mergedHasCollapseButton = mode !== 'horizontal' && mode !== 'popButton' && !inDropdown && hasCollapseButton;
var renderChildren = function () {
var childrenList = processChildren(children, { level: 1 });
var collapseIcon = collapse
? (icons && icons.collapseActive) || React.createElement(IconMenuUnfold, null)
: (icons && icons.collapseDefault) || React.createElement(IconMenuFold, null);
var collapseButtonClickHandler = function () {
var newCollapse = !collapse;
setCollapse(newCollapse);
onCollapseChange && onCollapseChange(newCollapse);
};
return (React.createElement(React.Fragment, null,
React.createElement("div", { className: prefixCls + "-inner" }, mode === 'horizontal' && ellipsis !== false ? (React.createElement(OverflowWrap, { ellipsisText: isObject(ellipsis) ? ellipsis.text : '···', onEllipsisChange: onEllipsisChange }, childrenList)) : (childrenList)),
mergedHasCollapseButton && (React.createElement("div", __assign({ tabIndex: 0, role: "button", "aria-controls": instanceId, "aria-expanded": !collapse, className: prefixCls + "-collapse-button", onClick: collapseButtonClickHandler }, getKeyboardEvents({ onPressEnter: collapseButtonClickHandler })), collapseIcon))));
};
var usedStyle = __assign({}, style);
if (mergedCollapse && !inDropdown) {
delete usedStyle.width;
}
return (React.createElement("div", __assign({ id: mergedHasCollapseButton ? instanceId : undefined, role: "menu" }, omit(rest, ['isMenu']), { ref: ref, style: usedStyle, className: cs(prefixCls, prefixCls + "-" + theme, prefixCls + "-" + (mode === 'horizontal' ? 'horizontal' : 'vertical'), (_a = {},
_a[prefixCls + "-collapse"] = mergedCollapse,
// 缩起状态自动变成 pop 模式
_a[prefixCls + "-pop"] = mode === 'pop' || mergedCollapse,
_a[prefixCls + "-pop-button"] = mode === 'popButton',
_a[prefixCls + "-rtl"] = rtl,
_a), className) }),
React.createElement(MenuContext.Provider, { value: {
mode: mode,
theme: theme,
collapse: mergedCollapse,
levelIndent: levelIndent,
inDropdown: inDropdown,
selectedKeys: selectedKeys,
openKeys: openKeys,
icons: icons,
triggerProps: triggerProps,
tooltipProps: tooltipProps,
autoScrollIntoView: autoScrollIntoView,
scrollConfig: scrollConfig,
// pass props directly
id: instanceId,
prefixCls: prefixCls,
collectInlineMenuKeys: function (key, unmount) {
if (unmount) {
refSubMenuKeys.current = refSubMenuKeys.current.filter(function (x) { return x !== key; });
}
else {
refSubMenuKeys.current.push(key);
}
forceUpdate();
},
onClickMenuItem: function (key, event) {
var _a;
selectable && setSelectedKeys([key]);
onClickMenuItem && onClickMenuItem(key, event, (_a = menuInfoMap[key]) === null || _a === void 0 ? void 0 : _a.keyPath);
},
onClickSubMenu: function (key, level, type) {
var _a;
var newOpenKeys = __spreadArray([], __read(openKeys), false);
if (type === 'inline') {
if ((openKeys === null || openKeys === void 0 ? void 0 : openKeys.indexOf(key)) > -1) {
if (accordion && level === 1) {
newOpenKeys = [];
}
else {
newOpenKeys = openKeys.filter(function (item) { return item !== key; });
}
}
else if (accordion && level === 1) {
newOpenKeys = [key];
}
else {
newOpenKeys = openKeys.concat([key]);
}
}
setOpenKeys(newOpenKeys);
onClickSubMenu && onClickSubMenu(key, newOpenKeys, (_a = menuInfoMap[key]) === null || _a === void 0 ? void 0 : _a.keyPath);
},
} }, renderChildren())));
}
var ForwardRefMenu = forwardRef(Menu);
var MenuComponent = ForwardRefMenu;
MenuComponent.displayName = 'Menu';
MenuComponent.Item = Item;
MenuComponent.SubMenu = SubMenu;
MenuComponent.ItemGroup = ItemGroup;
MenuComponent.__ARCO_MENU__ = true;
export default MenuComponent;