choerodon-ui/pro/es/table/AggregationButton.js

An enterprise-class UI design language and React-based implementation

import React, { memo, useCallback, useContext } from 'react';
import { action } from 'mobx';
import TableContext from './TableContext';
import Dropdown from '../dropdown/Dropdown';
import Menu from '../menu';
import { $l } from '../locale-context';

var AggregationButton = function AggregationButton(props) {
  var expanded = props.expanded,
      record = props.record,
      columnGroup = props.columnGroup;

  var _useContext = useContext(TableContext),
      tableStore = _useContext.tableStore,
      prefixCls = _useContext.prefixCls;

  var handleMenuClick = useCallback(action(function (_ref) {
    var key = _ref.key;

    switch (key) {
      case 'cell':
        tableStore.setAggregationCellExpanded(record, columnGroup.key, !expanded);
        break;

      case 'row':
        tableStore.columnGroups.allLeafs.forEach(function (_ref2) {
          var col = _ref2.column,
              columnKey = _ref2.key;
          return col.aggregation && tableStore.setAggregationCellExpanded(record, columnKey, !expanded);
        });
        break;

      case 'column':
        {
          var dataSet = record.dataSet;

          if (dataSet) {
            dataSet.forEach(function (r) {
              return tableStore.setAggregationCellExpanded(r, columnGroup.key, !expanded);
            });
          }

          break;
        }

      default:
    }
  }), [tableStore, record, columnGroup, expanded]);
  var handleClick = useCallback(function () {
    handleMenuClick({
      key: tableStore.aggregationExpandType
    });
  }, [handleMenuClick, tableStore]);
  var getOverlay = useCallback(function () {
    return React.createElement(Menu, {
      prefixCls: "".concat(prefixCls, "-dropdown-menu"),
      onClick: handleMenuClick
    }, React.createElement(Menu.Item, {
      key: "cell"
    }, $l('Table', expanded ? 'collapse_cell' : 'expand_cell')), React.createElement(Menu.Item, {
      key: "row"
    }, $l('Table', expanded ? 'collapse_row' : 'expand_row')), React.createElement(Menu.Item, {
      key: "column"
    }, $l('Table', expanded ? 'collapse_column' : 'expand_column')));
  }, [prefixCls, handleMenuClick, expanded]);
  return React.createElement("span", {
    className: "".concat(prefixCls, "-cell-expand")
  }, React.createElement("button", {
    type: "button",
    className: "".concat(prefixCls, "-cell-expand-btn"),
    onClick: handleClick
  }, $l('Table', expanded ? 'collapse' : 'expand_button')), React.createElement(Dropdown, {
    overlay: getOverlay
  }, React.createElement("span", {
    className: "".concat(prefixCls, "-cell-expand-menu-btn")
  })));
};

AggregationButton.displayName = 'AggregationButton';
export default memo(AggregationButton);
//# sourceMappingURL=AggregationButton.js.map