zarm/es/picker-view/utils.js

基于 React 的移动端UI库

import _defineProperty from "@babel/runtime/helpers/defineProperty";

function ownKeys(e, r) { var t = Object.keys(e); if (Object.getOwnPropertySymbols) { var o = Object.getOwnPropertySymbols(e); r && (o = o.filter(function (r) { return Object.getOwnPropertyDescriptor(e, r).enumerable; })), t.push.apply(t, o); } return t; }

function _objectSpread(e) { for (var r = 1; r < arguments.length; r++) { var t = null != arguments[r] ? arguments[r] : {}; r % 2 ? ownKeys(Object(t), !0).forEach(function (r) { _defineProperty(e, r, t[r]); }) : Object.getOwnPropertyDescriptors ? Object.defineProperties(e, Object.getOwnPropertyDescriptors(t)) : ownKeys(Object(t)).forEach(function (r) { Object.defineProperty(e, r, Object.getOwnPropertyDescriptor(t, r)); }); } return e; }

import omit from 'lodash/omit';
import { toArray } from '../utils';
var DEFAULT_FIELD_NAMES = {
  value: 'value',
  label: 'label',
  children: 'children'
};
export var resolvedFieldNames = function resolvedFieldNames(left) {
  var merged = _objectSpread({}, DEFAULT_FIELD_NAMES);

  left && Object.keys(left).forEach(function (key) {
    merged[key] = (left === null || left === void 0 ? void 0 : left[key]) || DEFAULT_FIELD_NAMES[key];
  });
  return merged;
};
export var isCascader = function isCascader(dataSource) {
  return Array.isArray(dataSource) && dataSource[0] && !Array.isArray(dataSource[0]);
};
export var isColumn = function isColumn(dataSource) {
  return Array.isArray(dataSource) && dataSource[0] && Array.isArray(dataSource[0]);
};
export var isValidValue = function isValidValue(value) {
  var currentValue = toArray(value);
  return currentValue.some(function (item) {
    return !!item || item === 0 || item === false;
  });
};

var resolvedValue = function resolvedValue(props, initialValue) {
  var value = props.value,
      defaultValue = props.defaultValue,
      wheelDefaultValue = props.wheelDefaultValue;

  if ('value' in props && isValidValue(value)) {
    return toArray(value);
  }

  if ('defaultValue' in props && isValidValue(defaultValue)) {
    return toArray(defaultValue);
  }

  if ('wheelDefaultValue' in props && isValidValue(wheelDefaultValue)) {
    return toArray(wheelDefaultValue);
  }

  return toArray(initialValue);
};

export var resolveColumn = function resolveColumn(props) {
  var columns = toArray(props.dataSource);
  var fieldNames = resolvedFieldNames(props.fieldNames);
  var value = resolvedValue(props, columns.map(function (item) {
    var _item$;

    return item === null || item === void 0 ? void 0 : (_item$ = item[0]) === null || _item$ === void 0 ? void 0 : _item$[fieldNames === null || fieldNames === void 0 ? void 0 : fieldNames.value];
  }));
  return {
    value: value,
    columns: columns,
    items: columns.map(function (column, index) {
      return column.filter(function (option) {
        return (option === null || option === void 0 ? void 0 : option[fieldNames.value]) === value[index];
      })[0];
    })
  };
};

var resolveCascade = function resolveCascade(props) {
  var cols = props.cols;
  var fieldNames = resolvedFieldNames(props.fieldNames);
  var value = resolvedValue(props, []);
  var columns = [];
  var items = [];

  var traverse = function traverse(options) {
    var depth = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : 0;
    columns[depth] = options.map(function (option, index) {
      var rest = omit(option, fieldNames.children);
      var children = option[fieldNames.children];
      var currentValue = value[depth];

      if (isValidValue(currentValue) && rest[fieldNames.value] === currentValue || !isValidValue(currentValue) && index === 0) {
        value[depth] = rest[fieldNames.value];
        items[depth] = rest;

        if (Array.isArray(children) && children.length > 0 && depth + 1 < cols) {
          traverse(children, depth + 1);
        }
      }

      return rest;
    });
  };

  traverse(props.dataSource);
  return {
    value: value,
    items: items,
    columns: columns
  };
};

export var resolved = function resolved(props) {
  return isCascader(props.dataSource) ? resolveCascade(props) : resolveColumn(props);
};