tntd/es/table/resizableTable/index.js

tntd是基于 TNT Design 设计体系的 React UI 组件库，主要用于研发企业级中后台产品。

"use strict";

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); }
Object.defineProperty(exports, "__esModule", {
  value: true
});
exports["default"] = void 0;
var _throttle2 = _interopRequireDefault(require("lodash/throttle"));
var _react = _interopRequireWildcard(require("react"));
var _classnames = _interopRequireDefault(require("classnames"));
var _reactResizable = require("react-resizable");
var _table = require("../table");
require("./index.less");
function _getRequireWildcardCache(e) { if ("function" != typeof WeakMap) return null; var r = new WeakMap(), t = new WeakMap(); return (_getRequireWildcardCache = function _getRequireWildcardCache(e) { return e ? t : r; })(e); }
function _interopRequireWildcard(e, r) { if (!r && e && e.__esModule) return e; if (null === e || "object" != _typeof(e) && "function" != typeof e) return { "default": e }; var t = _getRequireWildcardCache(r); if (t && t.has(e)) return t.get(e); var n = { __proto__: null }, a = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var u in e) if ("default" !== u && {}.hasOwnProperty.call(e, u)) { var i = a ? Object.getOwnPropertyDescriptor(e, u) : null; i && (i.get || i.set) ? Object.defineProperty(n, u, i) : n[u] = e[u]; } return n["default"] = e, t && t.set(e, n), n; }
function _interopRequireDefault(e) { return e && e.__esModule ? e : { "default": 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); }
function _slicedToArray(r, e) { return _arrayWithHoles(r) || _iterableToArrayLimit(r, e) || _unsupportedIterableToArray(r, e) || _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(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 _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 _iterableToArrayLimit(r, l) { var t = null == r ? null : "undefined" != typeof Symbol && r[Symbol.iterator] || r["@@iterator"]; if (null != t) { var e, n, i, u, a = [], f = !0, o = !1; try { if (i = (t = t.call(r)).next, 0 === l) { if (Object(t) !== t) return; f = !1; } else for (; !(f = (e = i.call(t)).done) && (a.push(e.value), a.length !== l); f = !0); } catch (r) { o = !0, n = r; } finally { try { if (!f && null != t["return"] && (u = t["return"](), Object(u) !== u)) return; } finally { if (o) throw n; } } return a; } }
function _arrayWithHoles(r) { if (Array.isArray(r)) return r; }
var __rest = void 0 && (void 0).__rest || function (s, e) {
  var t = {};
  for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p) && e.indexOf(p) < 0) t[p] = s[p];
  if (s != null && typeof Object.getOwnPropertySymbols === "function") for (var i = 0, p = Object.getOwnPropertySymbols(s); i < p.length; i++) {
    if (e.indexOf(p[i]) < 0 && Object.prototype.propertyIsEnumerable.call(s, p[i])) t[p[i]] = s[p[i]];
  }
  return t;
};
var ResizableTitle = function ResizableTitle(props) {
  var onResize = props.onResize,
    width = props.width,
    titleTip = props.titleTip,
    restProps = __rest(props, ["onResize", "width", "titleTip"]);
  if (!width) {
    return _react["default"].createElement("th", Object.assign({
      title: titleTip
    }, restProps));
  }
  return _react["default"].createElement(_reactResizable.Resizable, {
    width: width,
    height: 0,
    handle: _react["default"].createElement("span", {
      className: "react-resizable-handle",
      onClick: function onClick(e) {
        e.stopPropagation();
      }
    }),
    onResize: onResize,
    draggableOpts: {
      enableUserSelectHack: false
    }
  }, _react["default"].createElement("th", Object.assign({
    title: titleTip
  }, restProps)));
};
var MyComponent = function MyComponent(props, ref) {
  var _useState = (0, _react.useState)([]),
    _useState2 = _slicedToArray(_useState, 2),
    columns = _useState2[0],
    setColumns = _useState2[1];
  var dataRef = (0, _react.useRef)([]);
  var _useState3 = (0, _react.useState)(0),
    _useState4 = _slicedToArray(_useState3, 2),
    renderKey = _useState4[0],
    setRenderKey = _useState4[1];
  var renderKeyRef = (0, _react.useRef)(renderKey);
  var handleResize = function handleResize(index, minWidth) {
    return (0, _throttle2["default"])(function (e, _ref) {
      var size = _ref.size;
      var updatedColumns = dataRef.current || [];
      var width = Math.max(size.width, minWidth || 52);
      setColumns(updatedColumns.map(function (column, idx) {
        return idx === index ? Object.assign(Object.assign({}, column), {
          width: width,
          resizeKey: width
        }) : column;
      }));
    }, 25);
  };
  var buildColumns = function buildColumns(columns) {
    return columns === null || columns === void 0 ? void 0 : columns.map(function (column, index) {
      var _a;
      if (typeof (column === null || column === void 0 ? void 0 : column.width) === 'string' && ((_a = column === null || column === void 0 ? void 0 : column.width) === null || _a === void 0 ? void 0 : _a.includes('%'))) {
        return column;
      }
      var titleTip;
      if (column && typeof column.title === 'string') {
        titleTip = column.title;
      }
      return Object.assign(Object.assign({}, column), {
        render: typeof (column === null || column === void 0 ? void 0 : column.render) === 'function' ? function () {
          var _a2;
          var _a;
          for (var _len = arguments.length, args = new Array(_len), _key = 0; _key < _len; _key++) {
            args[_key] = arguments[_key];
          }
          return (_a = column === null || column === void 0 ? void 0 : column.render) === null || _a === void 0 ? void 0 : (_a2 = _a).call.apply(_a2, [column].concat(args, [renderKeyRef.current]));
        } : column === null || column === void 0 ? void 0 : column.render,
        onHeaderCell: function onHeaderCell(column) {
          var width = column.width;
          var minWidth = column.minWidth;
          if (typeof width === 'string' && !width.includes('%')) {
            width = Number(width.replace(/[^\d]/g, ''));
          }
          if (minWidth && typeof minWidth === 'string' && !minWidth.includes('%')) {
            minWidth = Number(minWidth.replace(/[^\d]/g, ''));
          }
          return {
            width: width,
            titleTip: titleTip,
            onResize: handleResize(index, minWidth),
            onMouseUp: function onMouseUp() {
              renderKeyRef.current += 1;
              setRenderKey(renderKeyRef.current);
            }
          };
        }
      });
    });
  };
  (0, _react.useEffect)(function () {
    if (Array.isArray(props === null || props === void 0 ? void 0 : props.columns)) {
      var curColumns = buildColumns(props.columns);
      setColumns(curColumns || []);
    }
  }, [props === null || props === void 0 ? void 0 : props.columns]);
  (0, _react.useEffect)(function () {
    dataRef.current = columns;
  }, [columns]);
  return _react["default"].createElement(_table.Table, Object.assign({}, props, Array.isArray(props === null || props === void 0 ? void 0 : props.columns) ? {
    columns: columns
  } : {}, {
    className: (0, _classnames["default"])('tnt-resizable-table', _defineProperty({}, props.className, props.className)),
    components: {
      header: {
        cell: ResizableTitle
      }
    }
  }));
};
var _default = exports["default"] = (0, _react.forwardRef)(MyComponent);