import Menu, { SubMenu, Item as MenuItem } from '../vc-menu'; import closest from 'dom-closest'; import classNames from 'classnames'; import shallowequal from 'shallowequal'; import Dropdown from '../dropdown'; import Icon from '../icon'; import Checkbox from '../checkbox'; import Radio from '../radio'; import FilterDropdownMenuWrapper from './FilterDropdownMenuWrapper'; import { FilterMenuProps } from './interface'; import { initDefaultProps, getOptionProps, isValidElement } from '../_util/props-util'; import { cloneElement } from '../_util/vnode'; import BaseMixin from '../_util/BaseMixin'; import { generateValueMaps } from './util'; function stopPropagation(e) { e.stopPropagation(); } export default { name: 'FilterMenu', mixins: [BaseMixin], props: initDefaultProps(FilterMenuProps, { handleFilter() {}, column: {}, }), data() { const visible = 'filterDropdownVisible' in this.column ? this.column.filterDropdownVisible : false; this.preProps = { ...getOptionProps(this) }; return { sSelectedKeys: this.selectedKeys, sKeyPathOfSelectedItem: {}, // 记录所有有选中子菜单的祖先菜单 sVisible: visible, sValueKeys: generateValueMaps(this.column.filters), }; }, watch: { _propsSymbol() { const nextProps = getOptionProps(this); const { column } = nextProps; const newState = {}; /** * if the state is visible the component should ignore updates on selectedKeys prop to avoid * that the user selection is lost * this happens frequently when a table is connected on some sort of realtime data * Fixes https://github.com/ant-design/ant-design/issues/10289 and * https://github.com/ant-design/ant-design/issues/10209 */ if ( 'selectedKeys' in nextProps && !shallowequal(this.preProps.selectedKeys, nextProps.selectedKeys) ) { newState.sSelectedKeys = nextProps.selectedKeys; } if (!shallowequal((this.preProps.column || {}).filters, (nextProps.column || {}).filters)) { newState.sValueKeys = generateValueMaps(nextProps.column.filters); } if ('filterDropdownVisible' in column) { newState.sVisible = column.filterDropdownVisible; } if (Object.keys(newState).length > 0) { this.setState(newState); } this.preProps = { ...nextProps }; }, // 'column.fixed': function (val) { // this.setNeverShown(this.column) // }, // column (val) { // if ('filterDropdownVisible' in val) { // this.sVisible = val.filterDropdownVisible // } // }, // selectedKeys (val) { // this.sSelectedKeys = val // }, }, mounted() { const { column } = this; this.$nextTick(() => { this.setNeverShown(column); }); }, updated() { const { column } = this; this.$nextTick(() => { this.setNeverShown(column); }); }, methods: { getDropdownVisible() { return this.neverShown ? false : this.sVisible; }, setNeverShown(column) { const rootNode = this.$el; const filterBelongToScrollBody = !!closest(rootNode, `.ant-table-scroll`); if (filterBelongToScrollBody) { // When fixed column have filters, there will be two dropdown menus // Filter dropdown menu inside scroll body should never be shown // To fix https://github.com/ant-design/ant-design/issues/5010 and // https://github.com/ant-design/ant-design/issues/7909 this.neverShown = !!column.fixed; } }, setSelectedKeys({ selectedKeys }) { this.setState({ sSelectedKeys: selectedKeys }); }, setVisible(visible) { const { column } = this; if (!('filterDropdownVisible' in column)) { this.setState({ sVisible: visible }); } if (column.onFilterDropdownVisibleChange) { column.onFilterDropdownVisibleChange(visible); } }, handleClearFilters() { this.setState( { sSelectedKeys: [], }, this.handleConfirm, ); }, handleConfirm() { this.setVisible(false); this.confirmFilter2(); // Call `setSelectedKeys` & `confirm` in the same time will make filter data not up to date // https://github.com/ant-design/ant-design/issues/12284 this.$forceUpdate(); this.$nextTick(() => { this.confirmFilter; }); }, onVisibleChange(visible) { this.setVisible(visible); const { column } = this.$props; // https://github.com/ant-design/ant-design/issues/17833 if (!visible && !(column.filterDropdown instanceof Function)) { this.confirmFilter2(); } }, handleMenuItemClick(info) { const { sSelectedKeys: selectedKeys } = this.$data; if (!info.keyPath || info.keyPath.length <= 1) { return; } const { sKeyPathOfSelectedItem: keyPathOfSelectedItem } = this.$data; if (selectedKeys && selectedKeys.indexOf(info.key) >= 0) { // deselect SubMenu child delete keyPathOfSelectedItem[info.key]; } else { // select SubMenu child keyPathOfSelectedItem[info.key] = info.keyPath; } this.setState({ sKeyPathOfSelectedItem: keyPathOfSelectedItem }); }, hasSubMenu() { const { column: { filters = [] }, } = this; return filters.some(item => !!(item.children && item.children.length > 0)); }, confirmFilter2() { const { column, selectedKeys: propSelectedKeys, confirmFilter } = this.$props; const { sSelectedKeys: selectedKeys, sValueKeys: valueKeys } = this.$data; const { filterDropdown } = column; if (!shallowequal(selectedKeys, propSelectedKeys)) { confirmFilter( column, filterDropdown ? selectedKeys : selectedKeys.map(key => valueKeys[key]).filter(key => key !== undefined), ); } }, renderMenus(items) { const { dropdownPrefixCls, prefixCls } = this.$props; return items.map(item => { if (item.children && item.children.length > 0) { const { sKeyPathOfSelectedItem } = this; const containSelected = Object.keys(sKeyPathOfSelectedItem).some( key => sKeyPathOfSelectedItem[key].indexOf(item.value) >= 0, ); const subMenuCls = classNames(`${prefixCls}-dropdown-submenu`, { [`${dropdownPrefixCls}-submenu-contain-selected`]: containSelected, }); return ( {this.renderMenus(item.children)} ); } return this.renderMenuItem(item); }); }, renderFilterIcon() { const { column, locale, prefixCls, selectedKeys } = this; const filtered = selectedKeys && selectedKeys.length > 0; let filterIcon = column.filterIcon; if (typeof filterIcon === 'function') { filterIcon = filterIcon(filtered, column); } const dropdownIconClass = classNames({ [`${prefixCls}-selected`]: filtered, [`${prefixCls}-open`]: this.getDropdownVisible(), }); if (!filterIcon) { return ( ); } if (filterIcon.length === 1 && isValidElement(filterIcon[0])) { return cloneElement(filterIcon[0], { on: { click: stopPropagation, }, class: classNames(`${prefixCls}-icon`, dropdownIconClass), }); } return {filterIcon}; }, renderMenuItem(item) { const { column } = this; const { sSelectedKeys: selectedKeys } = this.$data; const multiple = 'filterMultiple' in column ? column.filterMultiple : true; // We still need trade key as string since Menu render need string // const internalSelectedKeys = (selectedKeys || []).map(key => key.toString()); const input = multiple ? ( = 0} /> ) : ( = 0} /> ); return ( {input} {item.text} ); }, }, render() { const { sSelectedKeys: originSelectedKeys } = this.$data; const { column, locale, prefixCls, dropdownPrefixCls, getPopupContainer } = this; // default multiple selection in filter dropdown const multiple = 'filterMultiple' in column ? column.filterMultiple : true; const dropdownMenuClass = classNames({ [`${dropdownPrefixCls}-menu-without-submenu`]: !this.hasSubMenu(), }); let { filterDropdown } = column; if (filterDropdown instanceof Function) { filterDropdown = filterDropdown({ prefixCls: `${dropdownPrefixCls}-custom`, setSelectedKeys: selectedKeys => this.setSelectedKeys({ selectedKeys }), selectedKeys: originSelectedKeys, confirm: this.handleConfirm, clearFilters: this.handleClearFilters, filters: column.filters, visible: this.getDropdownVisible(), column, }); } const menus = filterDropdown ? ( {filterDropdown} ) : ( val.toString())} getPopupContainer={getPopupContainer} > {this.renderMenus(column.filters)} ); return ( {this.renderFilterIcon()} ); }, };