@perfma/heaven/es/table/column-setting/index.js

An UI Framework base [Ant Design V4](https://ant.design/components/overview-cn/) for React.

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 _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 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 _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; }

function _slicedToArray(arr, i) { return _arrayWithHoles(arr) || _iterableToArrayLimit(arr, i) || _unsupportedIterableToArray(arr, i) || _nonIterableRest(); }

function _nonIterableRest() { throw new TypeError("Invalid attempt to destructure 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 _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 _iterableToArrayLimit(arr, i) { var _i = arr == null ? null : typeof Symbol !== "undefined" && arr[Symbol.iterator] || arr["@@iterator"]; if (_i == null) return; var _arr = []; var _n = true; var _d = false; var _s, _e; try { for (_i = _i.call(arr); !(_n = (_s = _i.next()).done); _n = true) { _arr.push(_s.value); if (i && _arr.length === i) break; } } catch (err) { _d = true; _e = err; } finally { try { if (!_n && _i["return"] != null) _i["return"](); } finally { if (_d) throw _e; } } return _arr; }

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

import React, { useState, useContext, useEffect } from 'react';
import { Checkbox, Button, Tree } from 'antd';
import Icon from '../../icon';
import TableContext from '../store';
import './index.less';

var ColumnSetting = function ColumnSetting(props) {
  var onOk = props.onOk,
      onColumnSettingChange = props.onChange;

  var _useContext = useContext(TableContext),
      columns = _useContext.columns,
      setColumns = _useContext.setColumns;

  var _useState = useState([]),
      _useState2 = _slicedToArray(_useState, 2),
      checkedList = _useState2[0],
      setCheckedList = _useState2[1];

  var _useState3 = useState(true),
      _useState4 = _slicedToArray(_useState3, 2),
      indeterminate = _useState4[0],
      setIndeterminate = _useState4[1];

  var _useState5 = useState(false),
      _useState6 = _slicedToArray(_useState5, 2),
      checkAll = _useState6[0],
      setCheckAll = _useState6[1];

  var onChange = function onChange(list) {
    var newColums = columns.map(function (col) {
      return _objectSpread(_objectSpread({}, col), {}, {
        hidden: !list.includes(col.key)
      });
    });
    setColumns(newColums);
    setCheckedList(list);
    setIndeterminate(!!list.length && list.length < columns.length);
    setCheckAll(list.length === columns.length);
    onColumnSettingChange === null || onColumnSettingChange === void 0 ? void 0 : onColumnSettingChange(list);
  };

  var onCheckAllChange = function onCheckAllChange(e) {
    var ret = e.target.checked ? columns.map(function (col) {
      return col.key;
    }) : [];
    var newColums = columns.map(function (col) {
      return _objectSpread(_objectSpread({}, col), {}, {
        hidden: ret.length === 0
      });
    });
    setColumns(newColums);
    setCheckedList(ret);
    setIndeterminate(false);
    setCheckAll(e.target.checked);
    onColumnSettingChange === null || onColumnSettingChange === void 0 ? void 0 : onColumnSettingChange(ret);
  };

  useEffect(function () {
    var defaultColumns = columns.filter(function (col) {
      return !col.hidden;
    }).map(function (col) {
      return col.key;
    });
    setCheckedList(defaultColumns);
    setIndeterminate(!!defaultColumns.length && defaultColumns.length < columns.length);
    setCheckAll(!!defaultColumns.length && defaultColumns.length === columns.length);
  }, [columns]);

  var move = function move(id, targetId, dropPosition) {
    var newColumns = _toConsumableArray(columns);

    var findIndex = newColumns.findIndex(function (col) {
      return col.key === id;
    });
    var targetIndex = newColumns.findIndex(function (col) {
      return col.key === targetId;
    });
    var isDownWord = dropPosition > findIndex;

    if (findIndex < 0) {
      return;
    }

    var targetItem = newColumns[findIndex];
    newColumns.splice(findIndex, 1);

    if (dropPosition === 0) {
      newColumns.unshift(targetItem);
    } else {
      newColumns.splice(isDownWord ? targetIndex : targetIndex + 1, 0, targetItem);
    } // 更新数组


    setColumns(newColumns);
  };

  var onDrop = function onDrop(info) {
    var dropKey = info.node.key;
    var dragKey = info.dragNode.key;
    var dropPosition = info.dropPosition,
        dropToGap = info.dropToGap;
    var position = dropPosition === -1 || !dropToGap ? dropPosition + 1 : dropPosition;
    move(dragKey, dropKey, position);
  };

  var NodeItem = function NodeItem(_ref) {
    var node = _ref.node;
    return /*#__PURE__*/React.createElement("div", _objectSpread({
      style: {
        display: 'flex',
        alignItems: 'center',
        justifyContent: 'space-between',
        flexWrap: 'nowrap'
      }
    }, node), /*#__PURE__*/React.createElement("div", {
      style: {
        overflow: 'hidden',
        textOverflow: 'ellipsis',
        whiteSpace: 'nowrap'
      }
    }, node.title), /*#__PURE__*/React.createElement(Icon, {
      type: "Menu",
      style: {
        marginLeft: 4,
        color: '#b6b6b6'
      }
    }));
  };

  var treeData = columns.map(function (col) {
    return {
      key: col.key,
      title: col.title,
      disableCheckbox: col.disableCheckbox
    };
  });
  var disabled = treeData.find(function (col) {
    return col.disableCheckbox;
  });
  return (
    /*#__PURE__*/
    // <Popover
    //   // ? antd bug
    //   arrowPointAtCenter
    //   placement="bottomRight"
    //   // ----
    //   overlayClassName="column-setting-overlay"
    //   overlayStyle={{ width: 240 }}
    //   content={
    //   }
    //   title="自定义表格"
    //   trigger="click"
    //   visible={visible}
    //   onVisibleChange={(status) => setVisible(status)}
    // >
    //   <span style={{ cursor: 'pointer' }}>
    //     <SettingOutlined style={{ fontSize: 14, marginRight: 4 }} />
    //     字段设置
    //   </span>
    // </Popover>
    React.createElement("div", {
      className: "column-setting-overlay"
    }, /*#__PURE__*/React.createElement("div", {
      className: "column-setting-overlay-header"
    }, /*#__PURE__*/React.createElement("div", null, "\u81EA\u5B9A\u4E49\u8868\u683C"), /*#__PURE__*/React.createElement(Icon, {
      type: "Close",
      className: "column-setting-overlay-close",
      onClick: function onClick() {
        return onOk && onOk();
      }
    })), /*#__PURE__*/React.createElement(Tree, {
      treeData: treeData,
      selectable: false,
      checkable: true,
      checkedKeys: checkedList,
      onCheck: onChange,
      draggable: true,
      blockNode: true,
      titleRender: function titleRender(node) {
        return /*#__PURE__*/React.createElement(NodeItem, {
          node: node
        });
      },
      onDrop: onDrop
    }), /*#__PURE__*/React.createElement("div", {
      className: "column-setting-overlay-footer"
    }, /*#__PURE__*/React.createElement(Checkbox, {
      indeterminate: indeterminate,
      onChange: onCheckAllChange,
      checked: checkAll,
      disabled: !!disabled
    }, "\u663E\u793A\u5168\u90E8"), /*#__PURE__*/React.createElement(Button, {
      type: "primary",
      onClick: function onClick() {
        return onOk && onOk();
      }
    }, " \u786E\u5B9A ")))
  );
};

export default ColumnSetting;