add includeDependencies in listDiagrams (chartdb#270)

thanh4000 · Oct 11, 2024 · 492c90f · 492c90f
1 parent c393b28
commit 492c90f
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/src/context/storage-context/storage-provider.tsx b/src/context/storage-context/storage-provider.tsx
@@ -190,7 +190,12 @@ export const StorageProvider: React.FC<React.PropsWithChildren> = ({
         options: {
             includeTables?: boolean;
             includeRelationships?: boolean;
-        } = { includeRelationships: false, includeTables: false }
+            includeDependencies?: boolean;
+        } = {
+            includeRelationships: false,
+            includeTables: false,
+            includeDependencies: false,
+        }
     ): Promise<Diagram[]> => {
         let diagrams = await db.diagrams.toArray();
 
@@ -212,6 +217,15 @@ export const StorageProvider: React.FC<React.PropsWithChildren> = ({
             );
         }
 
+        if (options.includeDependencies) {
+            diagrams = await Promise.all(
+                diagrams.map(async (diagram) => {
+                    diagram.dependencies = await listDependencies(diagram.id);
+                    return diagram;
+                })
+            );
+        }
+
         return diagrams;
     };