refactor(utils): add omit and get methods

Author: John60676

src/__tests__/utils.test.ts

@@ -8,6 +8,8 @@ import {
   isString,
   requestProxy,
   isObject,
+  get,
+  omit,
 } from '../core/utils';
 import limitTrigger from '../core/utils/limitTrigger';
 import fetchMock from 'fetch-mock';
@@ -162,4 +164,18 @@ describe('utils', () => {
       expect(error.message).toBe('Not Found');
     }
   });
+
+  test('get should work', () => {
+    const object = { a: [{ b: { c: 3 } }] };
+
+    expect(get(object, 'a[0].b.c')).toBe(3);
+    // => 3
+
+    expect(get(object, 'a.b.c', 'default')).toBe('default');
+  });
+
+  test('omit should work', () => {
+    const object = { a: 1, b: 2, c: 3 };
+    expect(omit(object, ['a', 'b'])).toStrictEqual({ c: 3 });
+  });
 });

src/core/utils/index.ts

@@ -49,3 +49,31 @@ export const requestProxy = async (...args: [url: string, ...rest: any[]]) => {
   }
   throw new Error(res.statusText);
 };
+
+export const get = (
+  source: Record<string, any>,
+  path: string,
+  defaultValue: any = undefined,
+) => {
+  // a[3].b -> a.3.b
+  const paths = path.replace(/\[(\d+)\]/g, '.$1').split('.');
+  let result = source;
+  for (const p of paths) {
+    result = Object(result)[p];
+    if (result === undefined) {
+      return defaultValue;
+    }
+  }
+  return result;
+};
+
+export function omit<T, K extends keyof T>(
+  object: T,
+  keys: Array<K>,
+): Pick<T, Exclude<keyof T, K>> {
+  const result = Object.assign({}, object);
+  for (const key of keys) {
+    delete result[key];
+  }
+  return result;
+}