feat: add pick method

Author: li-jia-nan

src/omit.ts

@@ -1,14 +1,16 @@
-export default function omit<T extends object, K extends keyof T>(
+function omit<T extends object, K extends keyof T>(
   obj: T,
   fields: K[] | readonly K[],
 ): Omit<T, K> {
   const clone = Object.assign({}, obj);
 
   if (Array.isArray(fields)) {
-    fields.forEach(key => {
+    for (const key of fields) {
       delete clone[key];
-    });
+    }
   }
 
   return clone;
 }
+
+export default omit;

src/pick.ts

@@ -0,0 +1,16 @@
+const pick = <T extends object, K extends keyof T>(
+  obj: T,
+  fields: K[] | readonly K[],
+): Pick<T, K> => {
+  const result = {} as Pick<T, K>;
+
+  if (Array.isArray(fields)) {
+    for (const key of fields) {
+      result[key] = obj[key];
+    }
+  }
+
+  return result;
+};
+
+export default pick;

tests/pick.test.ts

@@ -0,0 +1,18 @@
+import pick from '../src/pick';
+
+describe('pick', () => {
+  it('should work', () => {
+    const ret = pick({ a: 1, b: 2, c: 3 }, ['a', 'b']);
+    expect(ret).toEqual({ a: 1, b: 2 });
+  });
+
+  it('invalidate array', () => {
+    const ret = pick({ test: 1 }, null);
+    expect(ret).toEqual({});
+  });
+
+  it('readonly array', () => {
+    const ret = pick({ a: 1, b: 2 }, ['a'] as const);
+    expect(ret).toEqual({ a: 1 });
+  });
+});