1

2026-05-25 17:08:18 +08:00
parent df501f6151
commit f1259b71b5
9178 changed files with 1626125 additions and 0 deletions
@@ -0,0 +1,94 @@
+import get from "./get";
+function internalSet(entity, paths, value, removeIfUndefined) {
+  if (!paths.length) {
+    return value;
+  }
+  const [path, ...restPath] = paths;
+  let clone;
+  if (!entity && typeof path === 'number') {
+    clone = [];
+  } else if (Array.isArray(entity)) {
+    clone = [...entity];
+  } else {
+    clone = {
+      ...entity
+    };
+  }
+
+  // Delete prop if `removeIfUndefined` and value is undefined
+  if (removeIfUndefined && value === undefined && restPath.length === 1) {
+    delete clone[path][restPath[0]];
+  } else {
+    clone[path] = internalSet(clone[path], restPath, value, removeIfUndefined);
+  }
+  return clone;
+}
+export default function set(entity, paths, value, removeIfUndefined = false) {
+  // Do nothing if `removeIfUndefined` and parent object not exist
+  if (paths.length && removeIfUndefined && value === undefined && !get(entity, paths.slice(0, -1))) {
+    return entity;
+  }
+  return internalSet(entity, paths, value, removeIfUndefined);
+}
+function isObject(obj) {
+  return typeof obj === 'object' && obj !== null && Object.getPrototypeOf(obj) === Object.prototype;
+}
+function createEmpty(source) {
+  return Array.isArray(source) ? [] : {};
+}
+const keys = typeof Reflect === 'undefined' ? Object.keys : Reflect.ownKeys;
+
+// ================================ Merge ================================
+
+/**
+ * Merge multiple objects. Support custom merge logic.
+ * @param sources object sources
+ * @param config.prepareArray Customize array prepare function.
+ * It will return empty [] by default.
+ * So when match array, it will auto be override with next array in sources.
+ */
+export function mergeWith(sources, config = {}) {
+  const {
+    prepareArray
+  } = config;
+  const finalPrepareArray = prepareArray || (() => []);
+  let clone = createEmpty(sources[0]);
+  sources.forEach(src => {
+    function internalMerge(path, parentLoopSet) {
+      const loopSet = new Set(parentLoopSet);
+      const value = get(src, path);
+      const isArr = Array.isArray(value);
+      if (isArr || isObject(value)) {
+        // Only add not loop obj
+        if (!loopSet.has(value)) {
+          loopSet.add(value);
+          const originValue = get(clone, path);
+          if (isArr) {
+            // Array will always be override
+            clone = set(clone, path, finalPrepareArray(originValue, value));
+          } else if (!originValue || typeof originValue !== 'object') {
+            // Init container if not exist
+            clone = set(clone, path, createEmpty(value));
+          }
+          keys(value).forEach(key => {
+            if (Object.getOwnPropertyDescriptor(value, key).enumerable) {
+              internalMerge([...path, key], loopSet);
+            }
+          });
+        }
+      } else {
+        clone = set(clone, path, value);
+      }
+    }
+    internalMerge([]);
+  });
+  return clone;
+}
+
+/**
+ * Merge multiple objects into a new single object.
+ * Arrays will be replaced by default.
+ */
+export function merge(...sources) {
+  return mergeWith(sources);
+}