@txdfe/at/es/table/tree.js

一个设计体系组件库

function _typeof(obj) { "@babel/helpers - typeof"; return _typeof = "function" == typeof Symbol && "symbol" == typeof Symbol.iterator ? function (obj) { return typeof obj; } : function (obj) { return obj && "function" == typeof Symbol && obj.constructor === Symbol && obj !== Symbol.prototype ? "symbol" : typeof obj; }, _typeof(obj); }

var _excluded = ["components", "isTree", "dataSource", "indent"];

function _extends() { _extends = Object.assign || function (target) { for (var i = 1; i < arguments.length; i++) { var source = arguments[i]; for (var key in source) { if (Object.prototype.hasOwnProperty.call(source, key)) { target[key] = source[key]; } } } return target; }; return _extends.apply(this, arguments); }

function ownKeys(object, enumerableOnly) { var keys = Object.keys(object); if (Object.getOwnPropertySymbols) { var symbols = Object.getOwnPropertySymbols(object); enumerableOnly && (symbols = symbols.filter(function (sym) { return Object.getOwnPropertyDescriptor(object, sym).enumerable; })), keys.push.apply(keys, symbols); } return keys; }

function _objectSpread(target) { for (var i = 1; i < arguments.length; i++) { var source = null != arguments[i] ? arguments[i] : {}; i % 2 ? ownKeys(Object(source), !0).forEach(function (key) { _defineProperty(target, key, source[key]); }) : Object.getOwnPropertyDescriptors ? Object.defineProperties(target, Object.getOwnPropertyDescriptors(source)) : ownKeys(Object(source)).forEach(function (key) { Object.defineProperty(target, key, Object.getOwnPropertyDescriptor(source, key)); }); } return target; }

function _objectWithoutProperties(source, excluded) { if (source == null) return {}; var target = _objectWithoutPropertiesLoose(source, excluded); var key, i; if (Object.getOwnPropertySymbols) { var sourceSymbolKeys = Object.getOwnPropertySymbols(source); for (i = 0; i < sourceSymbolKeys.length; i++) { key = sourceSymbolKeys[i]; if (excluded.indexOf(key) >= 0) continue; if (!Object.prototype.propertyIsEnumerable.call(source, key)) continue; target[key] = source[key]; } } return target; }

function _objectWithoutPropertiesLoose(source, excluded) { if (source == null) return {}; var target = {}; var sourceKeys = Object.keys(source); var key, i; for (i = 0; i < sourceKeys.length; i++) { key = sourceKeys[i]; if (excluded.indexOf(key) >= 0) continue; target[key] = source[key]; } return target; }

function _toConsumableArray(arr) { return _arrayWithoutHoles(arr) || _iterableToArray(arr) || _unsupportedIterableToArray(arr) || _nonIterableSpread(); }

function _nonIterableSpread() { throw new TypeError("Invalid attempt to spread non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); }

function _unsupportedIterableToArray(o, minLen) { if (!o) return; if (typeof o === "string") return _arrayLikeToArray(o, minLen); var n = Object.prototype.toString.call(o).slice(8, -1); if (n === "Object" && o.constructor) n = o.constructor.name; if (n === "Map" || n === "Set") return Array.from(o); if (n === "Arguments" || /^(?:Ui|I)nt(?:8|16|32)(?:Clamped)?Array$/.test(n)) return _arrayLikeToArray(o, minLen); }

function _iterableToArray(iter) { if (typeof Symbol !== "undefined" && iter[Symbol.iterator] != null || iter["@@iterator"] != null) return Array.from(iter); }

function _arrayWithoutHoles(arr) { if (Array.isArray(arr)) return _arrayLikeToArray(arr); }

function _arrayLikeToArray(arr, len) { if (len == null || len > arr.length) len = arr.length; for (var i = 0, arr2 = new Array(len); i < len; i++) { arr2[i] = arr[i]; } return arr2; }

function _classCallCheck(instance, Constructor) { if (!(instance instanceof Constructor)) { throw new TypeError("Cannot call a class as a function"); } }

function _defineProperties(target, props) { for (var i = 0; i < props.length; i++) { var descriptor = props[i]; descriptor.enumerable = descriptor.enumerable || false; descriptor.configurable = true; if ("value" in descriptor) descriptor.writable = true; Object.defineProperty(target, descriptor.key, descriptor); } }

function _createClass(Constructor, protoProps, staticProps) { if (protoProps) _defineProperties(Constructor.prototype, protoProps); if (staticProps) _defineProperties(Constructor, staticProps); Object.defineProperty(Constructor, "prototype", { writable: false }); return Constructor; }

function _inherits(subClass, superClass) { if (typeof superClass !== "function" && superClass !== null) { throw new TypeError("Super expression must either be null or a function"); } subClass.prototype = Object.create(superClass && superClass.prototype, { constructor: { value: subClass, writable: true, configurable: true } }); Object.defineProperty(subClass, "prototype", { writable: false }); if (superClass) _setPrototypeOf(subClass, superClass); }

function _setPrototypeOf(o, p) { _setPrototypeOf = Object.setPrototypeOf || function _setPrototypeOf(o, p) { o.__proto__ = p; return o; }; return _setPrototypeOf(o, p); }

function _createSuper(Derived) { var hasNativeReflectConstruct = _isNativeReflectConstruct(); return function _createSuperInternal() { var Super = _getPrototypeOf(Derived), result; if (hasNativeReflectConstruct) { var NewTarget = _getPrototypeOf(this).constructor; result = Reflect.construct(Super, arguments, NewTarget); } else { result = Super.apply(this, arguments); } return _possibleConstructorReturn(this, result); }; }

function _possibleConstructorReturn(self, call) { if (call && (_typeof(call) === "object" || typeof call === "function")) { return call; } else if (call !== void 0) { throw new TypeError("Derived constructors may only return object or undefined"); } return _assertThisInitialized(self); }

function _assertThisInitialized(self) { if (self === void 0) { throw new ReferenceError("this hasn't been initialised - super() hasn't been called"); } return self; }

function _isNativeReflectConstruct() { if (typeof Reflect === "undefined" || !Reflect.construct) return false; if (Reflect.construct.sham) return false; if (typeof Proxy === "function") return true; try { Boolean.prototype.valueOf.call(Reflect.construct(Boolean, [], function () {})); return true; } catch (e) { return false; } }

function _getPrototypeOf(o) { _getPrototypeOf = Object.setPrototypeOf ? Object.getPrototypeOf : function _getPrototypeOf(o) { return o.__proto__ || Object.getPrototypeOf(o); }; return _getPrototypeOf(o); }

function _defineProperty(obj, key, value) { if (key in obj) { Object.defineProperty(obj, key, { value: value, enumerable: true, configurable: true, writable: true }); } else { obj[key] = value; } return obj; }

import React from 'react';
import PropTypes from 'prop-types';
import RowComponent from './tree/row';
import CellComponent from './tree/cell';
import { statics } from './util';

var noop = function noop() {};

export default function tree(BaseComponent) {
  var TreeTable = /*#__PURE__*/function (_React$Component) {
    _inherits(TreeTable, _React$Component);

    var _super = _createSuper(TreeTable);

    function TreeTable(props, context) {
      var _this;

      _classCallCheck(this, TreeTable);

      _this = _super.call(this, props, context);

      _defineProperty(_assertThisInitialized(_this), "onTreeNodeClick", function (record) {
        var primaryKey = _this.props.primaryKey;
        var id = record[primaryKey];
        var dataSource = _this.ds;

        var openRowKeys = _toConsumableArray(_this.state.openRowKeys);

        var index = openRowKeys.indexOf(id);

        var getChildrenKeyById = function getChildrenKeyById(id) {
          var ret = [id];

          var loop = function loop(data) {
            data.forEach(function (item) {
              ret.push(item[primaryKey]);

              if (item.children) {
                loop(item.children);
              }
            });
          };

          dataSource.forEach(function (item) {
            if (item[primaryKey] === id) {
              if (item.children) {
                loop(item.children);
              }
            }
          });
          return ret;
        };

        if (index > -1) {
          // 不仅要删除当前的openRowKey，还需要删除关联子节点的openRowKey
          var ids = getChildrenKeyById(id);
          ids.forEach(function (id) {
            var i = openRowKeys.indexOf(id);

            if (i > -1) {
              openRowKeys.splice(i, 1);
            }
          });
        } else {
          openRowKeys.push(id);
        }

        if (!('openRowKeys' in _this.props)) {
          _this.setState({
            openRowKeys: openRowKeys
          });
        }

        _this.props.onRowOpen(openRowKeys, id, index === -1, record);
      });

      _this.state = {
        openRowKeys: props.openRowKeys || []
      };
      return _this;
    }

    _createClass(TreeTable, [{
      key: "getChildContext",
      value: function getChildContext() {
        return {
          openTreeRowKeys: this.state.openRowKeys,
          indent: this.props.indent,
          treeStatus: this.getTreeNodeStatus(this.ds),
          onTreeNodeClick: this.onTreeNodeClick,
          isTree: this.props.isTree
        };
      }
    }, {
      key: "UNSAFE_componentWillReceiveProps",
      value: function UNSAFE_componentWillReceiveProps(nextProps) {
        if ('openRowKeys' in nextProps) {
          var openRowKeys = nextProps.openRowKeys;
          this.setState({
            openRowKeys: openRowKeys
          });
        }
      }
    }, {
      key: "normalizeDataSource",
      value: function normalizeDataSource(dataSource) {
        var ret = [];

        var loop = function loop(dataSource, level) {
          dataSource.forEach(function (item) {
            item.__level = level;
            ret.push(item);

            if (item.children) {
              loop(item.children, level + 1);
            }
          });
        };

        loop(dataSource, 0);
        this.ds = ret;
        return ret;
      }
    }, {
      key: "getTreeNodeStatus",
      value: function getTreeNodeStatus() {
        var dataSource = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : [];
        var openRowKeys = this.state.openRowKeys;
        var primaryKey = this.props.primaryKey;
        var ret = [];
        openRowKeys.forEach(function (openKey) {
          dataSource.forEach(function (item) {
            if (item[primaryKey] === openKey) {
              if (item.children) {
                item.children.forEach(function (child) {
                  ret.push(child[primaryKey]);
                });
              }
            }
          });
        });
        return ret;
      }
    }, {
      key: "render",
      value: function render() {
        /* eslint-disable no-unused-vars, prefer-const */
        var _this$props = this.props,
            components = _this$props.components,
            isTree = _this$props.isTree,
            dataSource = _this$props.dataSource,
            indent = _this$props.indent,
            others = _objectWithoutProperties(_this$props, _excluded);

        if (isTree) {
          components = _objectSpread({}, components);

          if (!components.Row) {
            components.Row = RowComponent;
          }

          if (!components.Cell) {
            components.Cell = CellComponent;
          }

          dataSource = this.normalizeDataSource(dataSource);
        }

        return /*#__PURE__*/React.createElement(BaseComponent, _extends({}, others, {
          dataSource: dataSource,
          components: components
        }));
      }
    }]);

    return TreeTable;
  }(React.Component);

  _defineProperty(TreeTable, "TreeRow", RowComponent);

  _defineProperty(TreeTable, "TreeCell", CellComponent);

  _defineProperty(TreeTable, "propTypes", _objectSpread({
    /**
     * 默认情况下展开的树形表格，传入了此属性代表tree的展开为受控操作
     */
    openRowKeys: PropTypes.array,

    /**
     * 点击tree展开或者关闭的时候触发的事件
     * @param {Array} openRowKeys tree模式下展开的key
     * @param {String} currentRowKey 当前点击行的key
     * @param {Boolean} opened 当前点击是展开还是收起
     * @param {Object} currentRecord 当前点击行的记录
     */
    onRowOpen: PropTypes.func,

    /**
     * dataSource当中数据的主键，如果给定的数据源中的属性不包含该主键，会造成选择状态全部选中
     */
    primaryKey: PropTypes.string,

    /**
     * 在tree模式下的缩进尺寸， 仅在isTree为true时候有效
     */
    indent: PropTypes.number,

    /**
     * 开启Table的tree模式, 接收的数据格式中包含children则渲染成tree table
     */
    isTree: PropTypes.bool,
    locale: PropTypes.object
  }, BaseComponent.propTypes));

  _defineProperty(TreeTable, "defaultProps", _objectSpread(_objectSpread({}, BaseComponent.defaultProps), {}, {
    primaryKey: 'id',
    onRowOpen: noop,
    components: {},
    indent: 12
  }));

  _defineProperty(TreeTable, "childContextTypes", {
    openTreeRowKeys: PropTypes.array,
    indent: PropTypes.number,
    treeStatus: PropTypes.array,
    onTreeNodeClick: PropTypes.func,
    isTree: PropTypes.bool
  });

  statics(TreeTable, BaseComponent);
  return TreeTable;
}