pkgviz 0.7.2 → 0.7.3

package/src/utils/filter/toggleCompoundNodes.ts

@@ -0,0 +1,40 @@
+import { ElementsDefinition } from 'cytoscape';
+
+/**
+ * Toggles display of compound nodes ('parent' attribute) in the node
+ * @param elements - Cytoscape elements (nodes and edges)
+ * @param show - Whether to show compound nodes
+ * @param currentPackage - Current package path to adjust node names when showing compound nodes
+ * @returns Updated Cytoscape elements with compound nodes toggled
+ */
+export function toggleCompoundNodes(
+  { nodes, edges }: ElementsDefinition,
+  show: boolean,
+  currentPackage: string
+): ElementsDefinition {
+  const updatedNodes = nodes.map(node => {
+    if (!node.data.idInactive) node.data.idInactive = node.data.id;
+    // ACTIVE Compound Nodes
+    if (show && node.data.parentInactive) {
+      node.data.parent = node.data.parentInactive;
+      node.data.parentInactive = undefined;
+    } else if (!show && node.data.parent) {
+      node.data.parentInactive = node.data.parent;
+      node.data.parent = undefined;
+    }
+    return node;
+  });
+
+  const labelledNodes = updatedNodes.map(node => {
+    if (show) {
+      node.data.name = node.data.idInactive.split('.').pop();
+    } else {
+      node.data.name = node.data.idInactive.slice(
+        currentPackage.length ? currentPackage.length + 1 : 0
+      );
+    }
+    return node;
+  });
+
+  return { nodes: labelledNodes, edges };
+}

package/src/utils/hasChildren.ts

@@ -0,0 +1,7 @@
+import type { NodeDataDefinition, NodeDefinition } from 'cytoscape';
+
+/**
+ * Checks if a node (package) has children based on id prefix.
+ */
+export const hasChildren = (node: NodeDefinition, nodes: NodeDataDefinition[]) =>
+  nodes.some(n => n.data.id?.startsWith(`${node.data.id}.`));