feat: add pickBy and omitBy

src/index.ts

@@ -68,7 +68,9 @@ export {
   mapKeys,
   mapValues,
   omit,
+  omitBy,
   pick,
+  pickBy,
   set,
   shake,
   upperize

src/object.ts

@@ -181,6 +181,21 @@ export const pick = <T extends object, TKeys extends keyof T>(
   }, {} as Pick<T, TKeys>)
 }
 
+/**
+ * Picks properties of an object based on a predicate function.
+ *
+ * @param {T} obj - The object from which to pick properties.
+ * @param {function} predicate - The function used to filter the object's properties.
+ * @return {object} A new object with properties picked based on the predicate.
+ */
+export const pickBy = <T extends object, TKeys extends keyof T>(
+  obj: T,
+  predicate: (value: T[TKeys]) => boolean = () => true
+): Pick<T, TKeys> => {
+  if (!obj) return {} as Pick<T, TKeys>
+  return pick(obj, (Object.keys(obj) as TKeys[]).filter(key => predicate(obj[key])))
+}
+
 /**
  * Omit a list of properties from an object
  * returning a new object with the properties
@@ -205,6 +220,21 @@ export const omit = <T, TKeys extends keyof T>(
   )
 }
 
+/**
+ * Omit properties from an object based on a predicate function.
+ *
+ * @param {T} obj - The object to omit properties from.
+ * @param {function} predicate - The function used to determine which properties to omit.
+ * @return {Omit<T, TKeys>} A new object with properties omitted based on the predicate.
+ */
+export const omitBy = <T, TKeys extends keyof T>(
+  obj: T,
+  predicate: (value: T[TKeys]) => boolean = () => true
+): Omit<T, TKeys> => {
+  if (!obj) return {} as Omit<T, TKeys>
+  return omit(obj, (Object.keys(obj) as TKeys[]).filter(key => predicate(obj[key])))
+}
+
 /**
  * Dynamically get a nested value from an array or
  * object with a string.

src/tests/async.test.ts

@@ -2,6 +2,11 @@ import { assert } from 'chai'
 import * as _ from '..'
 import { AggregateError } from '../async'
 
+// fix error of calling useFakeTimers https://stackoverflow.com/a/77694958/5131623
+Object.defineProperty(global, 'performance', {
+  writable: true,
+});
+
 describe('async module', () => {
   beforeEach(() => jest.useFakeTimers({ advanceTimers: true }))
 

src/tests/object.test.ts

@@ -282,6 +282,23 @@ describe('object module', () => {
     })
   })
 
+  describe('pickBy function', () => {
+    test('handles null input', () => {
+      const result = _.pickBy(null as unknown as Record<string, unknown>)
+      assert.deepEqual(result, {})
+    })
+
+    test('handles no identity function', () => {
+      const result = _.pickBy({ a: 2 })
+      assert.deepEqual(result, { a: 2 })
+    })
+
+    test('returns picked properties only', () => {
+      const result = _.pickBy({ a: 1, b: 2, c: 3, d: 4 }, v => v % 2 === 0)
+      assert.deepEqual(result, { b: 2, d: 4 } as any)
+    })
+  })
+
   describe('omit function', () => {
     const person = {
       name: 'jay',
@@ -309,6 +326,32 @@ describe('object module', () => {
     })
   })
 
+  describe('omitBy function', () => {
+    const person = {
+      name: 'jay',
+      age: 20,
+      active: true
+    }
+
+    test('handles null input', () => {
+      const result = _.omitBy(null)
+      assert.deepEqual(result, {})
+    })
+
+    test('handles no identity function', () => {
+      const result = _.omitBy({ a: 2 })
+      assert.deepEqual(result, {})
+    })
+
+    test('returns object without omitted properties', () => {
+      const result = _.omitBy(person, v => typeof v === 'string')
+      assert.deepEqual(result, {
+        age: 20,
+        active: true
+      })
+    })
+  })
+
   describe('get function', () => {
     type Person = {
       name: string