feat(tree): make one init for everything (#564)

* feat(tree): make one init for everything * refactor: maybe fix test --------- Co-authored-by: Henrik Bossart <[email protected]>
equinor · Oct 2, 2023 · 8510162 · 8510162
1 parent abaa082
commit 8510162
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/packages/dm-core/src/domain/Tree.tsx b/packages/dm-core/src/domain/Tree.tsx
@@ -1,4 +1,5 @@
 import { AxiosResponse } from 'axios'
+import { splitAddress } from '../utils/addressUtilities'
 import { EBlueprint } from '../Enums'
 import { useDMSS } from '../context/DMSSContext'
 import { DmssAPI } from '../services'
@@ -268,6 +269,18 @@ export class Tree {
     this.updateCallback = updateCallback
   }
 
+  async init(path?: string[]) {
+    if (!path) {
+      this.initFromDataSources()
+      return
+    }
+    path
+      .filter((x) => splitAddress(x).documentPath)
+      .forEach((x) => this.initFromPath(x))
+    const dataSources = path.filter((x) => !splitAddress(x).documentPath)
+    if (dataSources.length) this.initFromDataSources(dataSources)
+  }
+
   async initFromDataSources(dataSources?: string[]) {
     // Add the dataSources as the top-level nodes
     const allDataSources = await this.dmssApi