getNodes & getEdges can be called without specifying ids

2024-01-22 16:38:00 +01:00
parent aee1b6b9f8
commit 08bb803ef2
1 changed files with 14 additions and 4 deletions
@@ -30,9 +30,9 @@ export function useSvelteFlow(): {
  zoomIn: ZoomInOut;
  zoomOut: ZoomInOut;
  getNode: (id: string) => Node | undefined;
-  getNodes: (ids: string[]) => (Node | undefined)[];
+  getNodes: (ids?: string[]) => (Node | undefined)[];
  getEdge: (id: string) => Edge | undefined;
-  getEdges: (ids: string[]) => (Edge | undefined)[];
+  getEdges: (ids?: string[]) => (Edge | undefined)[];
  setZoom: (zoomLevel: number, options?: ViewportHelperFunctionOptions) => void;
  getZoom: () => number;
  setCenter: (x: number, y: number, options?: SetCenterOptions) => void;
@@ -128,9 +128,19 @@ export function useSvelteFlow(): {
    zoomIn,
    zoomOut,
    getNode: (id) => get(nodeLookup).get(id),
-    getNodes: (ids) => ids.map((id) => get(nodeLookup).get(id)),
+    getNodes: (ids) => {
+      if (!ids) {
+        return get(nodes);
+      }
+      return ids.map((id) => get(nodeLookup).get(id));
+    },
    getEdge: (id) => get(edgeLookup).get(id),
-    getEdges: (ids) => ids.map((id) => get(edgeLookup).get(id)),
+    getEdges: (ids) => {
+      if (!ids) {
+        return get(edges);
+      }
+      return ids.map((id) => get(edgeLookup).get(id));
+    },
    setZoom: (zoomLevel, options) => {
      get(panZoom)?.scaleTo(zoomLevel, { duration: options?.duration });
    },