@sms-frontend/components/es/TreeSelect/hook/useStateValue.js

SMS Design React UI Library.

var __assign = (this && this.__assign) || function () {
    __assign = Object.assign || function(t) {
        for (var s, i = 1, n = arguments.length; i < n; i++) {
            s = arguments[i];
            for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p))
                t[p] = s[p];
        }
        return t;
    };
    return __assign.apply(this, arguments);
};
var __read = (this && this.__read) || function (o, n) {
    var m = typeof Symbol === "function" && o[Symbol.iterator];
    if (!m) return o;
    var i = m.call(o), r, ar = [], e;
    try {
        while ((n === void 0 || n-- > 0) && !(r = i.next()).done) ar.push(r.value);
    }
    catch (error) { e = { error: error }; }
    finally {
        try {
            if (r && !r.done && (m = i["return"])) m.call(i);
        }
        finally { if (e) throw e.error; }
    }
    return ar;
};
import { useState, useCallback } from 'react';
import isEqualWith from 'lodash/isEqualWith';
import { normalizeValueToArray } from '../utils';
import { isArray, isObject, isUndefined } from '../../_util/is';
import Tree from '../../Tree';
import { getCheckedKeysByInitKeys } from '../../Tree/util';
import useUpdate from '../../_util/hooks/useUpdate';
export var parseValue = function (v, key2nodeProps, valueMap) {
    if (v === undefined || v === null) {
        return [];
    }
    var value = isArray(v) ? v : [v];
    return value.map(function (x) {
        var result = isObject(x) ? __assign({}, x) : { value: x, label: undefined };
        var item = key2nodeProps[result.value];
        if (item) {
            result.label = result.label || item.title;
            result.disabled = item.disabled;
        }
        else if (isArray(valueMap)) {
            var v_1 = valueMap.find(function (y) { return y.value === result.value; });
            result = __assign(__assign({}, result), v_1);
        }
        if (isUndefined(result.label)) {
            result.label = result.value;
        }
        return result;
    });
};
var getInitCheckKeys = function (keys, key2nodeProps, indeterminateKeys, props) {
    if (!props.treeCheckStrictly) {
        var _a = getCheckedKeysByInitKeys(keys, key2nodeProps), allCheckedKeys = _a.checkedKeys, halfKeys = _a.indeterminateKeys;
        var checkedKeys_1 = allCheckedKeys;
        indeterminateKeys.current = halfKeys;
        if (props.treeCheckedStrategy === Tree.SHOW_PARENT) {
            checkedKeys_1 = checkedKeys_1.filter(function (x) {
                var item = key2nodeProps[x];
                if (!item || checkedKeys_1.indexOf(item.parentKey) === -1) {
                    return true;
                }
            });
        }
        else if (props.treeCheckedStrategy === Tree.SHOW_CHILD) {
            checkedKeys_1 = checkedKeys_1.filter(function (x) {
                var item = key2nodeProps[x];
                if (!item || !item.children || !item.children.length) {
                    return true;
                }
            });
        }
        return checkedKeys_1;
    }
    indeterminateKeys.current = [];
    return keys;
};
var useStateValue = function (props, key2nodeProps, indeterminateKeys) {
    var calcValue = function () {
        var propsValue = props.value || props.defaultValue || [];
        if (props.treeCheckable) {
            var initCheckedKeys = getInitCheckKeys(normalizeValueToArray(propsValue), key2nodeProps, indeterminateKeys, props);
            var parsedPropValue = parseValue(propsValue, key2nodeProps);
            var parsedCheckedValue = parseValue(initCheckedKeys, key2nodeProps, parsedPropValue);
            return parsedCheckedValue;
        }
        return parseValue(propsValue, key2nodeProps);
    };
    var _a = __read(useState(calcValue()), 2), value = _a[0], setValue = _a[1];
    useUpdate(function () {
        var nextValue = calcValue();
        if ('value' in props) {
            if (props.labelInValue) {
                // 以外部传入为准,只比较value
                if (!isEqualWith(normalizeValueToArray(value), normalizeValueToArray(nextValue))) {
                    setValue(nextValue);
                }
            }
            else if (!isEqualWith(value, nextValue)) {
                setValue(nextValue);
            }
        }
    }, [
        props.treeCheckedStrategy,
        props.treeCheckStrictly,
        props.treeCheckable,
        props.value,
        key2nodeProps,
    ]);
    var setStateValue = useCallback(function (newValue) {
        var onChange = props.onChange, labelInValue = props.labelInValue;
        var multiple = props.multiple || props.treeCheckable;
        if (!('value' in props)) {
            setValue(newValue);
        }
        var tmp;
        if (multiple) {
            tmp = newValue.map(function (x) {
                return labelInValue ? { label: x.label, value: x.value } : x.value;
            });
        }
        else {
            tmp = labelInValue ? newValue[0] : newValue[0] && newValue[0].value;
        }
        onChange && onChange(tmp);
    }, [props.onChange, props.labelInValue, props.multiple, props.treeCheckable, props.value]);
    return [value, setStateValue];
};
export default useStateValue;