industrial-model 0.11.2 → 0.11.3

package/dist/index.js

@@ -17,6 +17,11 @@ var CogniteSdkAdapter = class {
       }))
     };
   }
+  async retrieveViews(ids) {
+    const cleanIds = ids.map(({ space, externalId, version }) => ({ space, externalId, version }));
+    const response = await this.client.views.retrieve(cleanIds);
+    return { items: response.items };
+  }
   async queryInstances(request) {
     const response = await this.client.instances.query(
       request
@@ -2033,9 +2038,44 @@ var ViewMapper = class {
     for (const view of dm.views ?? []) {
       views.set(view.externalId, view);
     }
+    await this.loadDependencyViews(views);
     return views;
   }
+  async loadDependencyViews(views) {
+    const pending = /* @__PURE__ */ new Map();
+    for (const view of views.values()) {
+      for (const property of Object.values(view.properties)) {
+        for (const ref of collectPropertyRefs(property)) {
+          if (!views.has(ref.externalId) && !pending.has(ref.externalId)) {
+            pending.set(ref.externalId, ref);
+          }
+        }
+      }
+    }
+    if (pending.size === 0) return;
+    const sizeBefore = views.size;
+    const fetched = await this.cognite.retrieveViews(Array.from(pending.values()));
+    for (const view of fetched.items) {
+      views.set(view.externalId, view);
+    }
+    if (views.size > sizeBefore) {
+      await this.loadDependencyViews(views);
+    }
+  }
 };
+function collectPropertyRefs(property) {
+  if (isViewPropertyDefinition(property)) {
+    const source = getDirectRelationSource(property);
+    return source ? [source] : [];
+  }
+  if (isReverseDirectRelation(property)) {
+    return [property.source, property.through.source];
+  }
+  if (isEdgeConnection(property)) {
+    return [property.source];
+  }
+  return [];
+}
 
 // src/client.ts
 var APPLY_ITEM_LIMIT = 1e3;