@finos/legend-graph 32.5.2 → 32.6.0

package/src/graph-manager/protocol/pure/v1/transformation/pureGraph/to/V1_ElementSecondPassBuilder.ts

@@ -144,6 +144,7 @@ import {
   V1_buildDataProductLink,
   V1_buildDataProductOperationalMetadata,
   V1_buildNativeModelAccess,
+  V1_buildDataProductOwner,
 } from './helpers/V1_DataProductBuilder.js';
 import type { V1_IngestDefinition } from '../../../model/packageableElements/ingest/V1_IngestDefinition.js';
 import { IncludeStore } from '../../../../../../../graph/metamodel/pure/packageableElements/store/relational/model/IncludeStore.js';
@@ -808,5 +809,8 @@ export class V1_ElementSecondPassBuilder
     dataProduct.taggedValues = element.taggedValues
       .map((taggedValue) => V1_buildTaggedValue(taggedValue, this.context))
       .filter(isNonNullable);
+    if (element.owner) {
+      dataProduct.owner = V1_buildDataProductOwner(element.owner);
+    }
   }
 }

package/src/graph-manager/protocol/pure/v1/transformation/pureGraph/to/helpers/V1_DataProductBuilder.ts

@@ -43,7 +43,10 @@ import {
   InLineSampleQuery,
   DataProductDiagram,
   NativeModelExecutionContext,
+  type DataProductOwner,
+  AppDirOwner,
 } from '../../../../../../../../graph/metamodel/pure/dataProduct/DataProduct.js';
+import { AppDirNode } from '../../../../../../../../graph/metamodel/pure/packageableElements/ingest/IngestDefinition.js';
 import {
   type V1_AccessPoint,
   type V1_AccessPointGroup,
@@ -63,6 +66,8 @@ import {
   V1_PackageableElementSampleQuery,
   V1_InLineSampleQuery,
   type V1_NativeModelExecutionContext,
+  V1_AppDirOwner,
+  type V1_DataProductOwner,
 } from '../../../../model/packageableElements/dataProduct/V1_DataProduct.js';
 import type { V1_GraphBuilderContext } from '../V1_GraphBuilderContext.js';
 import { V1_buildRawLambdaWithResolvedPaths } from './V1_ValueSpecificationPathResolver.js';
@@ -390,3 +395,28 @@ export const V1_buildNativeModelAccess = (
 
   return metamodelNativeModelAccess;
 };
+
+export const V1_buildDataProductOwner = (
+  v1Owner: V1_DataProductOwner,
+): DataProductOwner => {
+  if (v1Owner instanceof V1_AppDirOwner) {
+    const owner = new AppDirOwner();
+    if (v1Owner.production) {
+      const production = new AppDirNode();
+      production.appDirId = v1Owner.production.appDirId;
+      production.level = v1Owner.production.level;
+      owner.production = production;
+    }
+    if (v1Owner.prodParallel) {
+      const prodParallel = new AppDirNode();
+      prodParallel.appDirId = v1Owner.prodParallel.appDirId;
+      prodParallel.level = v1Owner.prodParallel.level;
+      owner.prodParallel = prodParallel;
+    }
+    return owner;
+  }
+  throw new UnsupportedOperationError(
+    `Unsupported data product owner type`,
+    v1Owner,
+  );
+};

package/src/graph-manager/protocol/pure/v1/transformation/pureProtocol/serializationHelpers/V1_DataProductSerializationHelper.ts

@@ -53,6 +53,9 @@ import {
   V1_PackageableElementSampleQuery,
   V1_InLineSampleQuery,
   V1_NativeModelExecutionContext,
+  V1_AppDirOwner,
+  V1_DataProductOwnerType,
+  type V1_DataProductOwner,
 } from '../../../model/packageableElements/dataProduct/V1_DataProduct.js';
 import {
   UnsupportedOperationError,
@@ -78,6 +81,7 @@ import {
   V1_serializeEmbeddedDataType,
 } from './V1_DataElementSerializationHelper.js';
 import type { PureProtocolProcessorPlugin } from '../../../../PureProtocolProcessorPlugin.js';
+import { V1_AppDirNodeModelSchema } from './lakehouse/V1_CoreEntitlementsSerializationHelper.js';
 import type { V1_EmbeddedData } from '../../../model/data/V1_EmbeddedData.js';
 
 export enum V1_AccessPointType {
@@ -458,6 +462,32 @@ export const V1_serializeDataProductType = (
   throw new Error(`Unknown V1_DataProductType instance: ${value}`);
 };
 
+// ---------------------------------------- Owner -----------------------------------------
+
+const V1_AppDirOwnerModelSchema = createModelSchema(V1_AppDirOwner, {
+  _type: usingConstantValueSchema(V1_DataProductOwnerType.APP_DIR),
+  prodParallel: optionalCustomUsingModelSchema(V1_AppDirNodeModelSchema),
+  production: optionalCustomUsingModelSchema(V1_AppDirNodeModelSchema),
+});
+
+const V1_deserializeDataProductOwner = (
+  json: PlainObject<V1_DataProductOwner>,
+): V1_DataProductOwner => {
+  if (json._type === V1_DataProductOwnerType.APP_DIR) {
+    return deserialize(V1_AppDirOwnerModelSchema, json);
+  }
+  throw new Error(`Unknown V1_DataProductOwner type: ${json._type}`);
+};
+
+const V1_serializeDataProductOwner = (
+  value: V1_DataProductOwner,
+): PlainObject<V1_DataProductOwner> => {
+  if (value instanceof V1_AppDirOwner) {
+    return serialize(V1_AppDirOwnerModelSchema, value);
+  }
+  throw new Error(`Unknown V1_DataProductOwner instance`);
+};
+
 export const V1_dataProductModelSchema = (
   plugins: PureProtocolProcessorPlugin[],
 ): ModelSchema<V1_DataProduct> =>
@@ -484,6 +514,10 @@ export const V1_dataProductModelSchema = (
     operationalMetadata: optionalCustomUsingModelSchema(
       V1_DataProductOperationalMetadataModelSchema,
     ),
+    owner: optionalCustom(
+      V1_serializeDataProductOwner,
+      V1_deserializeDataProductOwner,
+    ),
     package: primitive(),
     type: optionalCustom(
       V1_serializeDataProductType,

package/src/index.ts

@@ -212,6 +212,9 @@ export {
   V1_ExternalDataProductType,
   V1_SupportInfo,
   V1_UnknownDataProductIcon,
+  V1_DataProductOwner,
+  V1_AppDirOwner,
+  V1_DataProductOwnerType,
 } from './graph-manager/protocol/pure/v1/model/packageableElements/dataProduct/V1_DataProduct.js';
 export {
   V1_ExecuteInput,
@@ -747,6 +750,8 @@ export {
   NativeModelExecutionContext,
   DataProductAccessType,
   DataProductOperationalMetadata,
+  DataProductOwner,
+  AppDirOwner,
 } from './graph/metamodel/pure/dataProduct/DataProduct.js';
 export * from './graph-manager/action/changeDetection/DataProductObserveHelper.js';