polestar-ui-kit/lib/Field/TreeSelect.js

## Install

"use strict";

require("core-js/modules/es.array.filter");

require("core-js/modules/es.array.for-each");

require("core-js/modules/es.array.is-array");

require("core-js/modules/es.array.map");

require("core-js/modules/es.array.some");

require("core-js/modules/es.date.to-string");

require("core-js/modules/es.object.get-prototype-of");

require("core-js/modules/es.object.keys");

require("core-js/modules/es.object.set-prototype-of");

require("core-js/modules/es.object.to-string");

require("core-js/modules/es.promise");

require("core-js/modules/es.reflect.construct");

require("core-js/modules/es.regexp.exec");

require("core-js/modules/es.regexp.to-string");

require("core-js/modules/es.string.split");

require("core-js/modules/web.dom-collections.for-each");

Object.defineProperty(exports, "__esModule", {
  value: true
});
exports["default"] = void 0;

var _react = _interopRequireDefault(require("react"));

var _propTypes = _interopRequireDefault(require("prop-types"));

var _polestarIcons = _interopRequireDefault(require("polestar-icons"));

var _treeSelect = _interopRequireDefault(require("antd/lib/tree-select"));

var _request = _interopRequireDefault(require("../helpers/request"));

function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { "default": obj }; }

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 _typeof(obj) { "@babel/helpers - typeof"; if (typeof Symbol === "function" && typeof Symbol.iterator === "symbol") { _typeof = function _typeof(obj) { return typeof obj; }; } else { _typeof = function _typeof(obj) { return obj && typeof Symbol === "function" && obj.constructor === Symbol && obj !== Symbol.prototype ? "symbol" : typeof obj; }; } return _typeof(obj); }

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); 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 } }); 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; } 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 { Date.prototype.toString.call(Reflect.construct(Date, [], 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); }

var TreeNode = _treeSelect["default"].TreeNode,
    SHOW_ALL = _treeSelect["default"].SHOW_ALL,
    SHOW_PARENT = _treeSelect["default"].SHOW_PARENT,
    SHOW_CHILD = _treeSelect["default"].SHOW_CHILD;

var TreeSelect = function (_React$Component) {
  _inherits(TreeSelect, _React$Component);

  var _super = _createSuper(TreeSelect);

  function TreeSelect(props) {
    var _this;

    _classCallCheck(this, TreeSelect);

    _this = _super.call(this, props);

    _this.onChange = function (value, label, extra) {
      if (_this.props.onChange) {
        _this.props.onChange(value, label, extra);
      }
    };

    _this.onSelect = function (value, node, extra) {
      if (_this.props.onSelect) {
        _this.props.onSelect(value, node, extra);
      }
    };

    _this.fetchObject = function (nodeItem) {
      var fetchObj;

      if (typeof _this.props.request === 'function') {
        fetchObj = _this.props.request(nodeItem);
      } else if (_this.props.request && _typeof(_this.props.request) === 'object') {
        var _this$props$request = _this.props.request,
            method = _this$props$request.method,
            url = _this$props$request.url,
            params = _this$props$request.params,
            headers = _this$props$request.headers;
        var reqParams = params;

        if (typeof params === 'function') {
          reqParams = params(nodeItem);
        }

        fetchObj = (0, _request["default"])({
          headers: headers,
          method: method,
          url: typeof url === 'function' ? url(nodeItem) : url,
          params: _extends({}, reqParams, {
            nodeItem: nodeItem
          })
        });
      }

      return fetchObj;
    };

    _this.childrenRequest = function (nodeData) {
      return new Promise(function (resolve) {
        var _this$props = _this.props,
            itemsKey = _this$props.itemsKey,
            dataType = _this$props.dataType;

        var fetchObj = _this.fetchObject(nodeData);

        fetchObj.then(function (response) {
          var items = response.data;

          if (itemsKey !== null && itemsKey.length > 0) {
            var itemsKeys = itemsKey.split('.');
            itemsKeys.forEach(function (key) {
              items = items[key];
            });
          }

          if (dataType === 'list') {
            resolve(_this.convertTreeForList(items));
          } else {
            resolve(items);
          }
        })["catch"](function (error) {
          throw new Error(error);
        });
      });
    };

    _this.handleLoadData = function (treeNode) {
      return new Promise(function (resolve) {
        if (treeNode.props.children !== null) {
          resolve();
          return;
        }

        _this.childrenRequest(treeNode.props.dataRef).then(function (item) {
          _this.appendNodes(_this.state.items, treeNode.props, item);

          _this.setState({
            items: _this.state.items
          });

          resolve();
        });
      });
    };

    _this.appendNodes = function (items, parentProps, addNodes) {
      var idKey = _this.props.idKey;
      items.some(function (item) {
        if (item.children && item.children.length > 0) {
          _this.appendNodes(item.children, parentProps, addNodes);
        }

        if (parentProps.dataRef[idKey] === item[idKey]) {
          item.children = addNodes;
          return true;
        }

        return false;
      });
    };

    _this.convertTreeForList = function (item) {
      var _this$props2 = _this.props,
          idKey = _this$props2.idKey,
          parentKey = _this$props2.parentKey,
          childrenKey = _this$props2.childrenKey;
      var nodeList = [];
      var lookup = {};
      item.forEach(function (data) {
        lookup[data[idKey]] = data;
        data.key = data[idKey];
        data[childrenKey] = [];
      });
      item.forEach(function (data) {
        if (data[parentKey] != null && data[parentKey] !== 'null') {
          if (lookup[data[parentKey]]) {
            lookup[data[parentKey]][childrenKey].push(data);
          } else {
            nodeList.push(data);
          }
        } else {
          nodeList.push(data);
        }
      });
      return nodeList;
    };

    _this.renderTreeItems = function (items) {
      if (items === undefined || items === null) {
        return null;
      }

      if (Array.isArray(items)) {
        return items.map(function (item) {
          return _this.renderTreeForHierarchy(item);
        });
      }

      return _this.renderTreeForHierarchy(items);
    };

    _this.renderTreeForHierarchy = function (item) {
      var _this$props3 = _this.props,
          idKey = _this$props3.idKey,
          labelKey = _this$props3.labelKey,
          valueKey = _this$props3.valueKey,
          childrenKey = _this$props3.childrenKey;

      if (item === null || Object.keys(item).length === 0) {
        return null;
      }

      var title = _react["default"].createElement("div", null, item.icon ? _react["default"].createElement(_polestarIcons["default"], {
        name: item.icon
      }) : null, item[labelKey]);

      return _react["default"].createElement(TreeNode, {
        key: item[idKey],
        title: title,
        value: String(item[valueKey]),
        isLeaf: item.isLeaf ? item.isLeaf : false,
        dataRef: item
      }, item[childrenKey] && item[childrenKey].length > 0 ? item[childrenKey].map(function (childItem) {
        return _this.renderTreeForHierarchy(childItem);
      }) : null);
    };

    var _items = _this.props.items;

    if (_items) {
      if (_this.props.dataType === 'list') {
        _items = _this.convertTreeForList(_items);
      }
    }

    _this.state = {
      items: _items
    };
    return _this;
  }

  _createClass(TreeSelect, [{
    key: "componentDidMount",
    value: function componentDidMount() {
      var _this2 = this;

      var fetchObj = this.fetchObject();

      if (fetchObj) {
        var _this$props4 = this.props,
            itemsKey = _this$props4.itemsKey,
            dataType = _this$props4.dataType;
        fetchObj.then(function (response) {
          var items = response.data;

          if (itemsKey !== null && itemsKey.length > 0) {
            var itemsKeys = itemsKey.split('.');
            itemsKeys.forEach(function (key) {
              items = items[key];
            });
          }

          if (dataType === 'list') {
            var convertList = _this2.convertTreeForList(items);

            _this2.setState({
              items: convertList
            });
          } else {
            _this2.setState({
              items: items
            });
          }
        })["catch"](function (error) {
          throw new Error(error);
        });
      }
    }
  }, {
    key: "render",
    value: function render() {
      var _this$props5 = this.props,
          ondemand = _this$props5.ondemand,
          showSearch = _this$props5.showSearch,
          treeDefaultExpandAll = _this$props5.treeDefaultExpandAll,
          dropdownStyleMaxHeight = _this$props5.dropdownStyleMaxHeight,
          allowClear = _this$props5.allowClear,
          multiple = _this$props5.multiple,
          placeholder = _this$props5.placeholder,
          searchPlaceholder = _this$props5.searchPlaceholder,
          treeCheckable = _this$props5.treeCheckable,
          showCheckedStrategy = _this$props5.showCheckedStrategy;

      var _showCheckedStrategy;

      if (showCheckedStrategy === 'SHOW_PARENT') {
        _showCheckedStrategy = SHOW_PARENT;
      } else if (showCheckedStrategy === 'SHOW_CHILD') {
        _showCheckedStrategy = SHOW_CHILD;
      } else {
        _showCheckedStrategy = SHOW_ALL;
      }

      return _react["default"].createElement("div", null, _react["default"].createElement(_treeSelect["default"], {
        showSearch: showSearch,
        loadData: ondemand ? this.handleLoadData : null,
        onChange: this.onChange,
        dropdownStyle: {
          maxHeight: dropdownStyleMaxHeight,
          overflow: 'auto'
        },
        treeDefaultExpandAll: treeDefaultExpandAll,
        allowClear: allowClear,
        multiple: multiple,
        placeholder: placeholder,
        searchPlaceholder: searchPlaceholder,
        treeCheckable: treeCheckable,
        showCheckedStrategy: _showCheckedStrategy
      }, this.renderTreeItems(this.state.filter ? this.state.filterNodes : this.state.items)));
    }
  }]);

  return TreeSelect;
}(_react["default"].Component);

TreeSelect.displayName = 'TreeSelect';
TreeSelect.propTypes = {
  request: _propTypes["default"].oneOfType([_propTypes["default"].func, _propTypes["default"].shape({
    headers: _propTypes["default"].object,
    url: _propTypes["default"].oneOfType([_propTypes["default"].string, _propTypes["default"].func]),
    method: _propTypes["default"].oneOf(['get', 'post', 'GET', 'POST']),
    params: _propTypes["default"].oneOfType([_propTypes["default"].object, _propTypes["default"].func]),
    responseType: _propTypes["default"].oneOf(['json', 'text'])
  })]),
  items: _propTypes["default"].array,
  itemsKey: _propTypes["default"].string,
  idKey: _propTypes["default"].string,
  labelKey: _propTypes["default"].string,
  valueKey: _propTypes["default"].string,
  childrenKey: _propTypes["default"].string,
  parentKey: _propTypes["default"].string,
  dataType: _propTypes["default"].oneOf(['list', 'hierarchy']),
  ondemand: _propTypes["default"].bool,
  treeDefaultExpandAll: _propTypes["default"].bool,
  dropdownStyleMaxHeight: _propTypes["default"].number,
  allowClear: _propTypes["default"].bool,
  multiple: _propTypes["default"].bool,
  placeholder: _propTypes["default"].string,
  searchPlaceholder: _propTypes["default"].string,
  treeCheckable: _propTypes["default"].bool,
  showCheckedStrategy: _propTypes["default"].string,
  onChange: _propTypes["default"].func,
  onSelect: _propTypes["default"].func
};
TreeSelect.defaultProps = {
  itemsKey: 'data',
  idKey: 'id',
  labelKey: 'label',
  childrenKey: 'children',
  parentKey: 'parentId',
  dataType: 'hierarchy',
  ondemand: false,
  showSearch: true,
  treeDefaultExpandAll: true,
  dropdownStyleMaxHeight: 400,
  allowClear: false,
  multiple: false,
  placeholder: null,
  searchPlaceholder: null,
  treeCheckable: false,
  showCheckedStrategy: 'SHOW_ALL'
};
var _default = TreeSelect;
exports["default"] = _default;