1
0
mirror of https://github.com/robonen/tools.git synced 2026-03-20 10:54:44 +00:00

feat(packages/stdlib): add omit object tool

This commit is contained in:
2024-10-31 00:48:01 +07:00
parent a1747ea535
commit 3daa47dc83
2 changed files with 88 additions and 0 deletions

View File

@@ -0,0 +1,50 @@
import { describe, it, expect } from 'vitest';
import { omit } from '.';
describe('omit', () => {
it('omit a single key from the object', () => {
const result = omit({ a: 1, b: 2, c: 3 }, 'a');
expect(result).toEqual({ b: 2, c: 3 });
});
it('omit multiple keys from the object', () => {
const result = omit({ a: 1, b: 2, c: 3 }, ['a', 'b']);
expect(result).toEqual({ c: 3 });
});
it('return the same object if no keys are omitted', () => {
const result = omit({ a: 1, b: 2, c: 3 }, []);
expect(result).toEqual({ a: 1, b: 2, c: 3 });
});
it('not modify the original object', () => {
const obj = { a: 1, b: 2, c: 3 };
const result = omit(obj, 'a');
expect(obj).toEqual({ a: 1, b: 2, c: 3 });
expect(result).toEqual({ b: 2, c: 3 });
});
it('handle an empty object', () => {
const result = omit({}, 'a' as any);
expect(result).toEqual({});
});
it('handle non-existent keys gracefully', () => {
const result = omit({ a: 1, b: 2, c: 3 } as Record<string, number>, 'd');
expect(result).toEqual({ a: 1, b: 2, c: 3 });
});
it('handle null gracefully', () => {
const emptyTarget = omit(null as any, 'a');
const emptyKeys = omit({ a: 1 }, null as any);
expect(emptyTarget).toEqual({});
expect(emptyKeys).toEqual({ a: 1 });
});
});

View File

@@ -0,0 +1,38 @@
import { isArray, type Arrayable } from '../../types';
/**
* @name omit
* @category Objects
* @description Returns a new object with the specified keys omitted
*
* @param {object} target - The object to omit keys from
* @param {Arrayable<keyof Target>} keys - The keys to omit
* @returns {Omit<Target, Key>} The new object with the specified keys omitted
*
* @example
* omit({ a: 1, b: 2, c: 3 }, 'a') // => { b: 2, c: 3 }
*
* @example
* omit({ a: 1, b: 2, c: 3 }, ['a', 'b']) // => { c: 3 }
*
* @since 0.0.3
*/
export function omit<Target extends object, Key extends keyof Target>(
target: Target,
keys: Arrayable<Key>
): Omit<Target, Key> {
const result = { ...target };
if (!target || !keys)
return result;
if (isArray(keys)) {
for (const key of keys) {
delete result[key];
}
} else {
delete result[keys];
}
return result;
}