@dxos/util 0.5.2-next.2408cc7 → 0.5.2-next.41b4fdd

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@dxos/util",
-  "version": "0.5.2-next.2408cc7",
+  "version": "0.5.2-next.41b4fdd",
   "description": "Temporary bucket for misc functions, which should graduate into separate packages.",
   "homepage": "https://dxos.org",
   "bugs": "https://github.com/dxos/dxos/issues",
@@ -16,13 +16,13 @@
     "src"
   ],
   "dependencies": {
-    "@dxos/debug": "0.5.2-next.2408cc7",
-    "@dxos/invariant": "0.5.2-next.2408cc7",
-    "@dxos/keys": "0.5.2-next.2408cc7",
-    "@dxos/node-std": "0.5.2-next.2408cc7"
+    "@dxos/debug": "0.5.2-next.41b4fdd",
+    "@dxos/keys": "0.5.2-next.41b4fdd",
+    "@dxos/node-std": "0.5.2-next.41b4fdd",
+    "@dxos/invariant": "0.5.2-next.41b4fdd"
   },
   "devDependencies": {
-    "@dxos/crypto": "0.5.2-next.2408cc7"
+    "@dxos/crypto": "0.5.2-next.41b4fdd"
   },
   "publishConfig": {
     "access": "public"

package/src/map-values.test.ts

@@ -0,0 +1,44 @@
+//
+// Copyright 2024 DXOS.org
+//
+
+import { test } from 'vitest';
+
+import { deepMapValues } from './map-values';
+
+test('deepMapValues', ({ expect }) => {
+  const obj = {
+    a: 1,
+    b: {
+      c: 2,
+    },
+    d: [
+      3,
+      {
+        e: 4,
+      },
+    ],
+  };
+
+  // Double all numbers.
+  const res = deepMapValues(obj, (value, recurse) => {
+    if (typeof value === 'number') {
+      return value * 2;
+    }
+
+    return recurse(value);
+  });
+
+  expect(res).toEqual({
+    a: 2,
+    b: {
+      c: 4,
+    },
+    d: [
+      6,
+      {
+        e: 8,
+      },
+    ],
+  });
+});

package/src/map-values.ts

@@ -9,3 +9,49 @@ export const mapValues = <T, U>(obj: Record<string, T>, fn: (value: T, key: stri
   });
   return result;
 };
+
+/**
+ * Recursively maps values traversing arrays and objects.
+ * @param fn Function to apply to each value. Second argument is a function to recurse into the value.
+ */
+export const deepMapValues = (value: any, fn: (value: any, recurse: (value: any) => any) => any): any => {
+  return new DeepMapper(fn).map(value);
+};
+
+class DeepMapper {
+  private readonly _cyclic = new Map<any, any>();
+
+  constructor(private readonly _fn: (value: any, recurse: (value: any) => any) => any) {}
+
+  map(value: any): any {
+    if (this._cyclic.has(value)) {
+      return this._cyclic.get(value);
+    }
+
+    return this._fn(value, this._recurse);
+  }
+
+  private _recurse = (value: any) => {
+    if (this._cyclic.has(value)) {
+      return this._cyclic.get(value);
+    }
+
+    if (Array.isArray(value)) {
+      const res = new Array(value.length);
+      this._cyclic.set(value, res);
+      for (let i = 0; i < value.length; i++) {
+        res[i] = this.map(value[i]);
+      }
+      return res;
+    } else if (value !== null && typeof value === 'object') {
+      const res: any = {};
+      this._cyclic.set(value, res);
+      for (const key in value) {
+        res[key] = this.map(value[key]);
+      }
+      return res;
+    } else {
+      return value;
+    }
+  };
+}