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

一个设计体系组件库

"use strict";

Object.defineProperty(exports, "__esModule", {
  value: true
});
exports["default"] = tree;
var _react = _interopRequireDefault(require("react"));
var _propTypes = _interopRequireDefault(require("prop-types"));
var _row = _interopRequireDefault(require("./tree/row"));
var _cell = _interopRequireDefault(require("./tree/cell"));
var _util = require("./util");
var _excluded = ["components", "isTree", "dataSource", "indent"];
function _interopRequireDefault(e) { return e && e.__esModule ? e : { "default": e }; }
function _typeof(o) { "@babel/helpers - typeof"; return _typeof = "function" == typeof Symbol && "symbol" == typeof Symbol.iterator ? function (o) { return typeof o; } : function (o) { return o && "function" == typeof Symbol && o.constructor === Symbol && o !== Symbol.prototype ? "symbol" : typeof o; }, _typeof(o); }
function _extends() { return _extends = Object.assign ? Object.assign.bind() : function (n) { for (var e = 1; e < arguments.length; e++) { var t = arguments[e]; for (var r in t) ({}).hasOwnProperty.call(t, r) && (n[r] = t[r]); } return n; }, _extends.apply(null, arguments); }
function ownKeys(e, r) { var t = Object.keys(e); if (Object.getOwnPropertySymbols) { var o = Object.getOwnPropertySymbols(e); r && (o = o.filter(function (r) { return Object.getOwnPropertyDescriptor(e, r).enumerable; })), t.push.apply(t, o); } return t; }
function _objectSpread(e) { for (var r = 1; r < arguments.length; r++) { var t = null != arguments[r] ? arguments[r] : {}; r % 2 ? ownKeys(Object(t), !0).forEach(function (r) { _defineProperty(e, r, t[r]); }) : Object.getOwnPropertyDescriptors ? Object.defineProperties(e, Object.getOwnPropertyDescriptors(t)) : ownKeys(Object(t)).forEach(function (r) { Object.defineProperty(e, r, Object.getOwnPropertyDescriptor(t, r)); }); } return e; }
function _objectWithoutProperties(e, t) { if (null == e) return {}; var o, r, i = _objectWithoutPropertiesLoose(e, t); if (Object.getOwnPropertySymbols) { var s = Object.getOwnPropertySymbols(e); for (r = 0; r < s.length; r++) o = s[r], t.includes(o) || {}.propertyIsEnumerable.call(e, o) && (i[o] = e[o]); } return i; }
function _objectWithoutPropertiesLoose(r, e) { if (null == r) return {}; var t = {}; for (var n in r) if ({}.hasOwnProperty.call(r, n)) { if (e.includes(n)) continue; t[n] = r[n]; } return t; }
function _toConsumableArray(r) { return _arrayWithoutHoles(r) || _iterableToArray(r) || _unsupportedIterableToArray(r) || _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(r, a) { if (r) { if ("string" == typeof r) return _arrayLikeToArray(r, a); var t = {}.toString.call(r).slice(8, -1); return "Object" === t && r.constructor && (t = r.constructor.name), "Map" === t || "Set" === t ? Array.from(r) : "Arguments" === t || /^(?:Ui|I)nt(?:8|16|32)(?:Clamped)?Array$/.test(t) ? _arrayLikeToArray(r, a) : void 0; } }
function _iterableToArray(r) { if ("undefined" != typeof Symbol && null != r[Symbol.iterator] || null != r["@@iterator"]) return Array.from(r); }
function _arrayWithoutHoles(r) { if (Array.isArray(r)) return _arrayLikeToArray(r); }
function _arrayLikeToArray(r, a) { (null == a || a > r.length) && (a = r.length); for (var e = 0, n = Array(a); e < a; e++) n[e] = r[e]; return n; }
function _classCallCheck(a, n) { if (!(a instanceof n)) throw new TypeError("Cannot call a class as a function"); }
function _defineProperties(e, r) { for (var t = 0; t < r.length; t++) { var o = r[t]; o.enumerable = o.enumerable || !1, o.configurable = !0, "value" in o && (o.writable = !0), Object.defineProperty(e, _toPropertyKey(o.key), o); } }
function _createClass(e, r, t) { return r && _defineProperties(e.prototype, r), t && _defineProperties(e, t), Object.defineProperty(e, "prototype", { writable: !1 }), e; }
function _callSuper(t, o, e) { return o = _getPrototypeOf(o), _possibleConstructorReturn(t, _isNativeReflectConstruct() ? Reflect.construct(o, e || [], _getPrototypeOf(t).constructor) : o.apply(t, e)); }
function _possibleConstructorReturn(t, e) { if (e && ("object" == _typeof(e) || "function" == typeof e)) return e; if (void 0 !== e) throw new TypeError("Derived constructors may only return object or undefined"); return _assertThisInitialized(t); }
function _assertThisInitialized(e) { if (void 0 === e) throw new ReferenceError("this hasn't been initialised - super() hasn't been called"); return e; }
function _isNativeReflectConstruct() { try { var t = !Boolean.prototype.valueOf.call(Reflect.construct(Boolean, [], function () {})); } catch (t) {} return (_isNativeReflectConstruct = function _isNativeReflectConstruct() { return !!t; })(); }
function _getPrototypeOf(t) { return _getPrototypeOf = Object.setPrototypeOf ? Object.getPrototypeOf.bind() : function (t) { return t.__proto__ || Object.getPrototypeOf(t); }, _getPrototypeOf(t); }
function _inherits(t, e) { if ("function" != typeof e && null !== e) throw new TypeError("Super expression must either be null or a function"); t.prototype = Object.create(e && e.prototype, { constructor: { value: t, writable: !0, configurable: !0 } }), Object.defineProperty(t, "prototype", { writable: !1 }), e && _setPrototypeOf(t, e); }
function _setPrototypeOf(t, e) { return _setPrototypeOf = Object.setPrototypeOf ? Object.setPrototypeOf.bind() : function (t, e) { return t.__proto__ = e, t; }, _setPrototypeOf(t, e); }
function _defineProperty(e, r, t) { return (r = _toPropertyKey(r)) in e ? Object.defineProperty(e, r, { value: t, enumerable: !0, configurable: !0, writable: !0 }) : e[r] = t, e; }
function _toPropertyKey(t) { var i = _toPrimitive(t, "string"); return "symbol" == _typeof(i) ? i : i + ""; }
function _toPrimitive(t, r) { if ("object" != _typeof(t) || !t) return t; var e = t[Symbol.toPrimitive]; if (void 0 !== e) { var i = e.call(t, r || "default"); if ("object" != _typeof(i)) return i; throw new TypeError("@@toPrimitive must return a primitive value."); } return ("string" === r ? String : Number)(t); }
var noop = function noop() {};
function tree(BaseComponent) {
  var TreeTable = /*#__PURE__*/function (_React$Component) {
    function TreeTable(props, context) {
      var _this;
      _classCallCheck(this, TreeTable);
      _this = _callSuper(this, TreeTable, [props, context]);
      _defineProperty(_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;
    }
    _inherits(TreeTable, _React$Component);
    return _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 _loop2 = function loop(dataSource, level) {
          dataSource.forEach(function (item) {
            item.__level = level;
            ret.push(item);
            if (item.children) {
              _loop2(item.children, level + 1);
            }
          });
        };
        _loop2(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 = _row["default"];
          }
          if (!components.Cell) {
            components.Cell = _cell["default"];
          }
          dataSource = this.normalizeDataSource(dataSource);
        }
        return /*#__PURE__*/_react["default"].createElement(BaseComponent, _extends({}, others, {
          dataSource: dataSource,
          components: components
        }));
      }
    }]);
  }(_react["default"].Component);
  _defineProperty(TreeTable, "TreeRow", _row["default"]);
  _defineProperty(TreeTable, "TreeCell", _cell["default"]);
  _defineProperty(TreeTable, "propTypes", _objectSpread({
    /**
     * 默认情况下展开的树形表格，传入了此属性代表tree的展开为受控操作
     */
    openRowKeys: _propTypes["default"].array,
    /**
     * 点击tree展开或者关闭的时候触发的事件
     * @param {Array} openRowKeys tree模式下展开的key
     * @param {String} currentRowKey 当前点击行的key
     * @param {Boolean} opened 当前点击是展开还是收起
     * @param {Object} currentRecord 当前点击行的记录
     */
    onRowOpen: _propTypes["default"].func,
    /**
     * dataSource当中数据的主键，如果给定的数据源中的属性不包含该主键，会造成选择状态全部选中
     */
    primaryKey: _propTypes["default"].string,
    /**
     * 在tree模式下的缩进尺寸， 仅在isTree为true时候有效
     */
    indent: _propTypes["default"].number,
    /**
     * 开启Table的tree模式, 接收的数据格式中包含children则渲染成tree table
     */
    isTree: _propTypes["default"].bool,
    locale: _propTypes["default"].object
  }, BaseComponent.propTypes));
  _defineProperty(TreeTable, "defaultProps", _objectSpread(_objectSpread({}, BaseComponent.defaultProps), {}, {
    primaryKey: 'id',
    onRowOpen: noop,
    components: {},
    indent: 12
  }));
  _defineProperty(TreeTable, "childContextTypes", {
    openTreeRowKeys: _propTypes["default"].array,
    indent: _propTypes["default"].number,
    treeStatus: _propTypes["default"].array,
    onTreeNodeClick: _propTypes["default"].func,
    isTree: _propTypes["default"].bool
  });
  (0, _util.statics)(TreeTable, BaseComponent);
  return TreeTable;
}