@finos/legend-application-marketplace 0.2.26 → 0.2.27

package/src/utils/EntitlementsUtils.tsx

@@ -25,6 +25,7 @@ import {
 } from '@finos/legend-extension-dsl-data-product';
 import {
   GraphManagerState,
+  V1_LiteDataContract,
   V1_LiteDataContractWithUserStatus,
   V1_ResourceType,
   V1_AccessPointGroupReference,
@@ -104,8 +105,15 @@ export type EntitlementsRow =
     }
   | { kind: 'request'; data: V1_DataRequestWithWorkflow };
 
-export const getContractData = (row: EntitlementsRow) =>
-  row.kind === ROW_KIND_CONTRACT ? row.data.contractResultLite : undefined;
+export const getContractData = (
+  row: EntitlementsRow,
+): V1_LiteDataContract | undefined => {
+  if (row.kind === ROW_KIND_CONTRACT) {
+    const lite = row.data.contractResultLite;
+    return lite instanceof V1_LiteDataContract ? lite : undefined;
+  }
+  return undefined;
+};
 
 export const getRequestData = (row: EntitlementsRow) =>
   row.kind === ROW_KIND_REQUEST ? row.data.dataRequest : undefined;
@@ -402,6 +410,9 @@ export const useSelectedViewerState = (
     }
     if (selectedRow.kind === ROW_KIND_CONTRACT) {
       const contract = selectedRow.data.contractResultLite;
+      if (!(contract instanceof V1_LiteDataContract)) {
+        return undefined;
+      }
       return new DataContractViewerState(
         contract,
         (contractId: string, taskId: string) =>

package/src/utils/LakehouseUtils.tsx

@@ -18,15 +18,23 @@ import {
   type GraphManagerState,
   type V1_EntitlementsDataProductDetails,
   type V1_PureGraphManager,
+  type V1_DataProduct,
   V1_AdHocDeploymentDataProductOrigin,
+  V1_dataProductModelSchema,
   V1_SdlcDeploymentDataProductOrigin,
+  CORE_PURE_PATH,
 } from '@finos/legend-graph';
-import { ActionState, type PlainObject } from '@finos/legend-shared';
+import {
+  ActionState,
+  guaranteeNonNullable,
+  type PlainObject,
+} from '@finos/legend-shared';
 import {
   type ProjectVersionEntities,
   resolveVersion,
 } from '@finos/legend-server-depot';
 import type { Entity } from '@finos/legend-storage';
+import { deserialize } from 'serializr';
 import type { LegendMarketplaceBaseStore } from '../stores/LegendMarketplaceBaseStore.js';
 
 export const buildGraphForDataProduct = async (
@@ -69,3 +77,74 @@ export const buildGraphForDataProduct = async (
     );
   }
 };
+
+export const getDataProductFromDetails = async (
+  details: V1_EntitlementsDataProductDetails,
+  graphManager: V1_PureGraphManager,
+  marketplaceBaseStore: LegendMarketplaceBaseStore,
+): Promise<V1_DataProduct | undefined> => {
+  if (details.origin instanceof V1_SdlcDeploymentDataProductOrigin) {
+    const rawEntities =
+      (await marketplaceBaseStore.depotServerClient.getVersionEntities(
+        details.origin.group,
+        details.origin.artifact,
+        resolveVersion(details.origin.version),
+        CORE_PURE_PATH.DATA_PRODUCT,
+      )) as {
+        artifactId: string;
+        entity: Entity;
+        groupId: string;
+        versionId: string;
+        versionedEntity: boolean;
+      }[];
+    const entities = rawEntities.map((entity) =>
+      deserialize(
+        V1_dataProductModelSchema(
+          graphManager.pluginManager.getPureProtocolProcessorPlugins(),
+        ),
+        entity.entity.content,
+      ),
+    );
+    const matchingEntities = entities.filter(
+      (entity) => entity.name.toLowerCase() === details.id.toLowerCase(),
+    );
+    if (matchingEntities.length === 0) {
+      throw new Error(
+        `No data product found with name ${details.id} in project`,
+      );
+    } else if (matchingEntities.length > 1) {
+      throw new Error(
+        `Multiple data products found with name ${details.id} in project`,
+      );
+    }
+    return matchingEntities[0];
+  } else if (details.origin instanceof V1_AdHocDeploymentDataProductOrigin) {
+    const entities: Entity[] = await graphManager.pureCodeToEntities(
+      details.origin.definition,
+    );
+    const elements = entities
+      .filter((e) => e.classifierPath === CORE_PURE_PATH.DATA_PRODUCT)
+      .map((entity) =>
+        deserialize(
+          V1_dataProductModelSchema(
+            graphManager.pluginManager.getPureProtocolProcessorPlugins(),
+          ),
+          entity.content,
+        ),
+      );
+    const matchingEntities = elements.filter(
+      (element) => element.name.toLowerCase() === details.id.toLowerCase(),
+    );
+    if (matchingEntities.length > 1) {
+      throw new Error(
+        `Multiple data products found with name ${details.id} in deployed definition`,
+      );
+    }
+    return guaranteeNonNullable(
+      matchingEntities[0],
+      `No data product found with name ${details.id} in deployed definition`,
+    );
+  } else {
+    return undefined;
+  }
+};