✨ remove

Author: tangweikun

src/array/__tests__/remove.spec.js

@@ -0,0 +1,33 @@
+import { remove } from '..'
+
+test('remove', () => {
+ let arr = [1, 2, 3, 4]
+ const fn = n => n % 2 === 0
+ const removed = remove(arr, fn)
+ expect(arr).toEqual([1, 3])
+ expect(removed).toEqual([2, 4])
+})
+
+test('remove', () => {
+ let arr = [1, 2, 3, 4, 4, 1]
+ const fn = n => n === 4
+ const removed = remove(arr, fn)
+ expect(arr).toEqual([1, 2, 3, 1])
+ expect(removed).toEqual([4, 4])
+})
+
+test('remove', () => {
+ let arr = [1, 2, 3, 4]
+ const fn = n => n > 0
+ const removed = remove(arr, fn)
+ expect(arr).toEqual([])
+ expect(removed).toEqual([1, 2, 3, 4])
+})
+
+test('remove', () => {
+ let arr = [1, 2, 3, 4]
+ const fn = n => n < 2
+ const removed = remove(arr, fn)
+ expect(arr).toEqual([2, 3, 4])
+ expect(removed).toEqual([1])
+})

src/array/index.js

@@ -26,3 +26,4 @@ export * from './concat'
 export * from './flattenDepth'
 export * from './fromPairs'
 export * from './pullAt'
+export * from './remove'

src/array/remove.js

@@ -0,0 +1,11 @@
+export const remove = (arr, fn) => {
+ const removed = []
+ let i = 0
+
+ while (i < arr.length) {
+ if (fn(arr[i])) removed.push(...arr.splice(i, 1))
+ else i++
+ }
+
+ return removed
+}