es-toolkit
Version:
A state-of-the-art, high-performance JavaScript utility library with a small bundle size and strong type annotations.
79 lines (78 loc) • 2.34 kB
JavaScript
import { isPlainObject } from "../predicate/isPlainObject.mjs";
import { isDeepKey } from "../_internal/isDeepKey.mjs";
import { cloneDeepWith } from "./cloneDeepWith.mjs";
import { flatten } from "../array/flatten.mjs";
import { unset } from "./unset.mjs";
import { keysIn } from "./keysIn.mjs";
import { getSymbolsIn } from "../_internal/getSymbolsIn.mjs";
//#region src/compat/object/omit.ts
/**
* Creates a new object with specified keys omitted.
*
* This function takes an object and a variable number of keys, and returns a new object that
* excludes the properties corresponding to the specified keys. Note that keys can be deep.
*
* Deep keys can be specified for keys.
*
* @template T - The type of object.
* @param obj - The object to omit keys from.
* @param keysArr - A variable number of keys to be omitted from the object.
* @returns A new object with the specified keys omitted.
*
* @example
* omit({ a: 1, b: 2, c: 3 }, 'a', 'b');
* // => { c: 3 }
*
* omit({ a: { b: 1, c: 2 }, d: 3 }, 'a.b', 'd');
* // => { a: { c: 2 } }
*/
function omit(obj, ...keysArr) {
if (obj == null) return {};
keysArr = flatten(keysArr);
const result = cloneInOmit(obj, keysArr);
for (let i = 0; i < keysArr.length; i++) {
let keys = keysArr[i];
switch (typeof keys) {
case "object":
if (!Array.isArray(keys)) keys = Array.from(keys);
for (let j = 0; j < keys.length; j++) {
const key = keys[j];
unset(result, key);
}
break;
case "string":
case "symbol":
case "number":
unset(result, keys);
break;
}
}
return result;
}
function cloneInOmit(obj, keys) {
if (keys.some((key) => Array.isArray(key) || isDeepKey(key))) return deepCloneInOmit(obj);
return shallowCloneInOmit(obj);
}
function shallowCloneInOmit(obj) {
const result = {};
const keysToCopy = [...keysIn(obj), ...getSymbolsIn(obj)];
for (let i = 0; i < keysToCopy.length; i++) {
const key = keysToCopy[i];
result[key] = obj[key];
}
return result;
}
function deepCloneInOmit(obj) {
const result = {};
const keysToCopy = [...keysIn(obj), ...getSymbolsIn(obj)];
for (let i = 0; i < keysToCopy.length; i++) {
const key = keysToCopy[i];
result[key] = cloneDeepWith(obj[key], (valueToClone) => {
if (isPlainObject(valueToClone)) return;
return valueToClone;
});
}
return result;
}
//#endregion
export { omit };