var _excluded = ["primaryKey", "_primaryKey"];
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 _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 _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 _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; }
/**
 * 处理表格 rowSelection hooks
 * 除了基本 rowSelection 之外返回平铺后的 selectedRowKeys、selectedRecords
 */
import React, { useEffect, useState } from 'react';
import { getDeepValue } from '@teamix/utils';
import { Balloon, Checkbox } from '@alicloudfe/components';
var Tooltip = Balloon.Tooltip;
function useTableSelection() {
  var config = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : {};
  var records = arguments.length > 1 ? arguments[1] : undefined;
  var primaryKey = arguments.length > 2 ? arguments[2] : undefined;
  var _config$defaultSelect = config.defaultSelectedRecords,
    defaultSelectedRecords = _config$defaultSelect === void 0 ? [] : _config$defaultSelect,
    _config$defaultSelect2 = config.defaultSelectedRowKeys,
    defaultSelectedRowKeys = _config$defaultSelect2 === void 0 ? [] : _config$defaultSelect2;
  var _useState = useState(config.selectedRowKeys || defaultSelectedRowKeys),
    _useState2 = _slicedToArray(_useState, 2),
    selectedRowKeys = _useState2[0],
    setSelectedRowKeys = _useState2[1];
  var _useState3 = useState(defaultSelectedRecords),
    _useState4 = _slicedToArray(_useState3, 2),
    selectedRecords = _useState4[0],
    setSelectedRecords = _useState4[1];
  useEffect(function () {
    var _config$selectedRowKe;
    setSelectedRowKeys((_config$selectedRowKe = config === null || config === void 0 ? void 0 : config.selectedRowKeys) !== null && _config$selectedRowKe !== void 0 ? _config$selectedRowKe : []);
  }, [config.selectedRowKeys]);
  function _onChange(value, records) {
    if (primaryKey) {
      var newValue = value.filter(function (v) {
        return !selectedRowKeys.includes(v);
      });
      var deleteValue = selectedRowKeys.filter(function (v) {
        return !value.includes(v);
      });
      var nextSelectedRecords = _toConsumableArray(selectedRecords);
      if (newValue.length > 0) {
        newValue.forEach(function (v) {
          var nextRecords = records.find(function (n) {
            return getDeepValue(primaryKey, n) === v;
          });
          if (nextRecords) {
            nextSelectedRecords.push(nextRecords);
          }
        });
      }
      if (deleteValue.length > 0) {
        deleteValue.forEach(function (v) {
          nextSelectedRecords = nextSelectedRecords.filter(function (n) {
            return getDeepValue(primaryKey, n) !== v;
          });
        });
      }
      setSelectedRecords(nextSelectedRecords);
    } else {
      setSelectedRecords(records);
    }
    setSelectedRowKeys(value);
    if (config.onChange) {
      config.onChange(value, records);
    }
  }
  var renderCheckBox = function renderCheckBox(props, record) {
    var primaryKey = props.primaryKey,
      _primaryKey = props._primaryKey,
      others = _objectWithoutProperties(props, _excluded);
    return /*#__PURE__*/React.createElement(Checkbox, _objectSpread({
      checked: selectedRowKeys.includes(primaryKey),
      onChange: function onChange(checked) {
        var _config$onSelect;
        config === null || config === void 0 ? void 0 : (_config$onSelect = config.onSelect) === null || _config$onSelect === void 0 ? void 0 : _config$onSelect.call(config, checked, record, records);
        if (checked) {
          _onChange === null || _onChange === void 0 ? void 0 : _onChange([].concat(_toConsumableArray(selectedRowKeys), [primaryKey]), [record]);
        } else {
          _onChange === null || _onChange === void 0 ? void 0 : _onChange(selectedRowKeys.filter(function (item) {
            return item !== primaryKey;
          }), [record]);
        }
      }
    }, others));
  };
  var getProps = function getProps(record, index) {
    var _config$getProps;
    var propResult = config === null || config === void 0 ? void 0 : (_config$getProps = config.getProps) === null || _config$getProps === void 0 ? void 0 : _config$getProps.call(config, record, index);
    // console.log('propResult', propResult, selectedRowKeys);
    var resultProps = _objectSpread(_objectSpread({}, record), propResult);
    var primaryKey = record === null || record === void 0 ? void 0 : record[resultProps === null || resultProps === void 0 ? void 0 : resultProps['_primaryKey']];
    if ((config === null || config === void 0 ? void 0 : config.mode) === 'single') {
      return resultProps;
    }
    return _objectSpread(_objectSpread({}, resultProps), {}, {
      isPreview: true,
      renderPreview: function renderPreview() {
        if (resultProps.disabled === true && resultProps.disabledTooltip) {
          return /*#__PURE__*/React.createElement(Tooltip, {
            trigger: renderCheckBox(_objectSpread(_objectSpread({}, propResult), {}, {
              primaryKey: primaryKey
            }), record),
            align: "t"
          }, propResult.disabledTooltip);
        }
        return renderCheckBox(_objectSpread(_objectSpread({}, propResult), {}, {
          primaryKey: primaryKey
        }), record);
      }
    });
  };
  var rowSelection = _objectSpread(_objectSpread({
    selectedRowKeys: selectedRowKeys,
    selectedRecords: selectedRecords
  }, config), {}, {
    onChange: _onChange,
    getProps: getProps
  });
  return {
    rowSelection: rowSelection,
    selectedRowKeys: selectedRowKeys,
    setSelectedRowKeys: setSelectedRowKeys,
    selectedRecords: selectedRecords,
    setSelectedRecords: setSelectedRecords
  };
}
export default useTableSelection;