LendAndRegret/node_modules/es-toolkit/dist/compat/object/mergeWith.mjs
2026-05-02 17:27:43 +08:00

107 lines
3.9 KiB
JavaScript

import { cloneDeep } from './cloneDeep.mjs';
import { isUnsafeProperty } from '../../_internal/isUnsafeProperty.mjs';
import { clone } from '../../object/clone.mjs';
import { isBuffer } from '../../predicate/isBuffer.mjs';
import { isPrimitive } from '../../predicate/isPrimitive.mjs';
import { getSymbols } from '../_internal/getSymbols.mjs';
import { isArguments } from '../predicate/isArguments.mjs';
import { isArrayLikeObject } from '../predicate/isArrayLikeObject.mjs';
import { isObjectLike } from '../predicate/isObjectLike.mjs';
import { isPlainObject } from '../predicate/isPlainObject.mjs';
import { isTypedArray } from '../predicate/isTypedArray.mjs';
function mergeWith(object, ...otherArgs) {
const sources = otherArgs.slice(0, -1);
const merge = otherArgs[otherArgs.length - 1];
let result = object;
for (let i = 0; i < sources.length; i++) {
const source = sources[i];
result = mergeWithDeep(result, source, merge, new Map());
}
return result;
}
function mergeWithDeep(target, source, merge, stack) {
if (isPrimitive(target)) {
target = Object(target);
}
if (source == null || typeof source !== 'object') {
return target;
}
if (stack.has(source)) {
return clone(stack.get(source));
}
stack.set(source, target);
if (Array.isArray(source)) {
source = source.slice();
for (let i = 0; i < source.length; i++) {
source[i] = source[i] ?? undefined;
}
}
const sourceKeys = [...Object.keys(source), ...getSymbols(source)];
for (let i = 0; i < sourceKeys.length; i++) {
const key = sourceKeys[i];
if (isUnsafeProperty(key)) {
continue;
}
let sourceValue = source[key];
let targetValue = target[key];
if (isArguments(sourceValue)) {
sourceValue = { ...sourceValue };
}
if (isArguments(targetValue)) {
targetValue = { ...targetValue };
}
if (isBuffer(sourceValue)) {
sourceValue = cloneDeep(sourceValue);
}
if (Array.isArray(sourceValue)) {
if (Array.isArray(targetValue)) {
const cloned = [];
const targetKeys = Reflect.ownKeys(targetValue);
for (let i = 0; i < targetKeys.length; i++) {
const targetKey = targetKeys[i];
cloned[targetKey] = targetValue[targetKey];
}
targetValue = cloned;
}
else if (isArrayLikeObject(targetValue)) {
const cloned = [];
for (let i = 0; i < targetValue.length; i++) {
cloned[i] = targetValue[i];
}
targetValue = cloned;
}
else {
targetValue = [];
}
}
const merged = merge(targetValue, sourceValue, key, target, source, stack);
if (merged !== undefined) {
target[key] = merged;
}
else if (Array.isArray(sourceValue)) {
target[key] = mergeWithDeep(targetValue, sourceValue, merge, stack);
}
else if (isObjectLike(targetValue) &&
isObjectLike(sourceValue) &&
(isPlainObject(targetValue) ||
isPlainObject(sourceValue) ||
isTypedArray(targetValue) ||
isTypedArray(sourceValue))) {
target[key] = mergeWithDeep(targetValue, sourceValue, merge, stack);
}
else if (targetValue == null && isPlainObject(sourceValue)) {
target[key] = mergeWithDeep({}, sourceValue, merge, stack);
}
else if (targetValue == null && isTypedArray(sourceValue)) {
target[key] = cloneDeep(sourceValue);
}
else if (targetValue === undefined || sourceValue !== undefined) {
target[key] = sourceValue;
}
}
return target;
}
export { mergeWith };