@soratani-code/samtools 1.3.19 → 1.3.20

package/lib/tree.d.ts

@@ -2,3 +2,4 @@ export declare function findNodeFormKey<D = any>(key: string, nodes: D[], id: st
 export declare function updateNodeFormKey<D = any>(key: string, nodes: D[], id: string, value: any): D[] | null;
 export declare function deleteNodeFormKey<D = any>(key: string, nodes: D[], id: string): D[];
 export declare function toList<D = any>(nodes: D[]): any[];
+export declare function slice<T extends Record<string, any>>(tree: T[] | T, level: number, childrenKey?: string): T | T[] | undefined;

package/lib/tree.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.toList = exports.deleteNodeFormKey = exports.updateNodeFormKey = exports.findNodeFormKey = void 0;
+exports.slice = exports.toList = exports.deleteNodeFormKey = exports.updateNodeFormKey = exports.findNodeFormKey = void 0;
 const lodash_1 = require("lodash");
 function findNodeFormKey(key, nodes, id) {
     if (!id)
@@ -77,3 +77,39 @@ function toList(nodes) {
     return result;
 }
 exports.toList = toList;
+function slice(tree, level, childrenKey = 'children') {
+    if (level < 1)
+        return undefined;
+    const process = (nodes, maxLevel) => {
+        let currentLevel = 1;
+        let queue = nodes.map(node => (Object.assign({}, node)));
+        let result = queue;
+        while (currentLevel < maxLevel) {
+            let nextLevelNodes = [];
+            for (let node of queue) {
+                const n = node;
+                if (Array.isArray(n[childrenKey])) {
+                    n[childrenKey] = n[childrenKey].map((child) => (Object.assign({}, child)));
+                    nextLevelNodes.push(...n[childrenKey]);
+                }
+                else {
+                    delete n[childrenKey];
+                }
+            }
+            queue = nextLevelNodes;
+            currentLevel++;
+        }
+        for (let node of queue) {
+            const n = node;
+            if (n[childrenKey]) {
+                delete n[childrenKey];
+            }
+        }
+        return result;
+    };
+    if (Array.isArray(tree)) {
+        return process(tree, level);
+    }
+    return process([tree], level)[0];
+}
+exports.slice = slice;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@soratani-code/samtools",
-  "version": "1.3.19",
+  "version": "1.3.20",
   "description": "",
   "typings": "lib/index.d.ts",
   "main": "lib/index.js",