function _typeof(obj) { 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 _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); if (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 = arguments[i] != null ? arguments[i] : {}; if (i % 2) { ownKeys(source, true).forEach(function (key) { _defineProperty(target, key, source[key]); }); } else if (Object.getOwnPropertyDescriptors) { Object.defineProperties(target, Object.getOwnPropertyDescriptors(source)); } else { ownKeys(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 _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 _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 _getPrototypeOf(o) { _getPrototypeOf = Object.setPrototypeOf ? Object.getPrototypeOf : function _getPrototypeOf(o) { return o.__proto__ || Object.getPrototypeOf(o); }; return _getPrototypeOf(o); }

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 _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, { Children } from 'react';
import PropTypes from 'prop-types';
import classnames from 'classnames';
import ListHeader from './list-header';
import ListFooter from './list-footer';
import RowComponent from './list/row';
import BodyComponent from './list/body';
import HeaderComponent from './fixed/header';
import WrapperComponent from './fixed/wrapper';
import { statics } from './util';
export default function list(BaseComponent) {
  var ListTable =
  /*#__PURE__*/
  function (_React$Component) {
    _inherits(ListTable, _React$Component);

    function ListTable() {
      _classCallCheck(this, ListTable);

      return _possibleConstructorReturn(this, _getPrototypeOf(ListTable).apply(this, arguments));
    }

    _createClass(ListTable, [{
      key: "getChildContext",
      value: function getChildContext() {
        return {
          listHeader: this.listHeader,
          listFooter: this.listFooter,
          rowSelection: this.rowSelection
        };
      }
    }, {
      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: "render",
      value: function render() {
        var _this = this;

        /* eslint-disable prefer-const */
        var _this$props = this.props,
            components = _this$props.components,
            children = _this$props.children,
            className = _this$props.className,
            prefix = _this$props.prefix,
            others = _objectWithoutProperties(_this$props, ["components", "children", "className", "prefix"]);

        var isList = false,
            ret = [];
        Children.forEach(children, function (child) {
          if (child) {
            if (typeof child.type === 'function') {
              if (child.type._typeMark === 'listHeader') {
                _this.listHeader = child.props;
                isList = true;
              } else if (child.type._typeMark === 'listFooter') {
                _this.listFooter = child.props;
              } else {
                ret.push(child);
              }
            } else {
              ret.push(child);
            }
          }
        });
        this.rowSelection = this.props.rowSelection;

        if (isList) {
          var _classnames;

          components = _objectSpread({}, components);
          components.Row = components.Row || RowComponent;
          components.Body = components.Body || BodyComponent;
          components.Header = components.Header || HeaderComponent;
          components.Wrapper = components.Wrapper || WrapperComponent;
          className = classnames((_classnames = {}, _defineProperty(_classnames, "".concat(prefix, "table-group"), true), _defineProperty(_classnames, className, className), _classnames));
        }

        return React.createElement(BaseComponent, _extends({}, others, {
          components: components,
          children: ret,
          className: className,
          prefix: prefix
        }));
      }
    }]);

    return ListTable;
  }(React.Component);

  _defineProperty(ListTable, "ListHeader", ListHeader);

  _defineProperty(ListTable, "ListFooter", ListFooter);

  _defineProperty(ListTable, "ListRow", RowComponent);

  _defineProperty(ListTable, "ListBody", BodyComponent);

  _defineProperty(ListTable, "propTypes", _objectSpread({}, BaseComponent.propTypes));

  _defineProperty(ListTable, "defaultProps", _objectSpread({}, BaseComponent.defaultProps));

  _defineProperty(ListTable, "childContextTypes", {
    listHeader: PropTypes.any,
    listFooter: PropTypes.any,
    rowSelection: PropTypes.object
  });

  statics(ListTable, BaseComponent);
  return ListTable;
}