@jetbrains/ring-ui/components/table/smart-table.js

import React, {PureComponent} from 'react';
import PropTypes from 'prop-types';

import Table from './table';
import Selection from './selection';

/* eslint-disable no-unused-vars */
const {
  selection: __selection__,
  onSelect: __onSelect__,
  ...restPropTypes
} = Table.propTypes;
/* eslint-enable */

// eslint-disable-next-line react/no-deprecated
class SmartTable extends PureComponent {
  static propTypes = {
    onSelectionChange: PropTypes.func,
    isItemSelectable: PropTypes.func,
    ...restPropTypes
  };

  static defaultProps = {
    onSelectionChange: () => {}
  };

  state = {
    selection: new Selection({
      data: this.props.data,
      isItemSelectable: this.props.isItemSelectable
    })
  };

  componentWillReceiveProps(nextProps) {
    const {data, isItemSelectable} = nextProps;
    if (this.props.data !== data || this.props.isItemSelectable !== isItemSelectable) {
      const selection = new Selection({data, isItemSelectable});
      this.setState({selection});
    }
  }

  onSelect = selection => {
    this.setState({selection});
    this.props.onSelectionChange(selection);
  };

  render() {
    return (
      <Table
        {...this.props}
        selection={this.state.selection}
        onSelect={this.onSelect}
      />
    );
  }
}

export default SmartTable;