Popup.Menu declared as a forwardable ref component

This commit is contained in:
NikolaBorislavovHristov 2019-01-18 14:44:21 +02:00
parent d2287598cd
commit b6621b0620
2 changed files with 6 additions and 6 deletions

View file

@ -1,7 +1,9 @@
import React from 'react';
const Menu = ({ children }) => {
return React.Children.only(children);
};
const Menu = React.forwardRef(({ children }, ref) => (
<div ref={ref}>
{children}
</div>
));
export default Menu;

View file

@ -228,9 +228,7 @@ class Popup extends Component {
<Modal className={classnames('modal-container', this.props.className)} onClick={this.close}>
<div ref={this.menuContainerRef} className={styles['menu-container']} onClick={this.menuContainerOnClick}>
<div ref={this.menuScrollRef} className={styles['menu-scroll-container']}>
<div ref={this.menuChildrenRef}>
{children}
</div>
{React.cloneElement(children, { ref: this.menuChildrenRef })}
</div>
<div ref={this.menuBorderTopRef} className={classnames(styles['border'], styles['border-top'])} />
<div ref={this.menuBorderRightRef} className={classnames(styles['border'], styles['border-right'])} />