@finos/legend-graph 32.5.3 → 32.6.1

package/src/graph-manager/protocol/pure/v1/lakehouse/entitlements/V1_CoreEntitlements.ts

@@ -14,8 +14,16 @@
  * limitations under the License.
  */
 
-import type { PlainObject } from '@finos/legend-shared';
+import {
+  hashArray,
+  type Hashable,
+  type PlainObject,
+} from '@finos/legend-shared';
 import type { V1_StereotypePtr } from '../../model/packageableElements/domain/V1_StereotypePtr.js';
+import { AppDirLevel } from '../../../../../../graph/metamodel/pure/packageableElements/ingest/IngestDefinition.js';
+import { CORE_HASH_STRUCTURE } from '../../../../../../graph/Core_HashUtils.js';
+
+export { AppDirLevel as V1_AppDirLevel };
 
 // ---------------------------------------- Users & App Directory --------------------------------------
 
@@ -29,17 +37,17 @@ export enum V1_UserType {
   SYSTEM_ACCOUNT = 'SYSTEM_ACCOUNT',
 }
 
-export class V1_AppDirNode {
+export class V1_AppDirNode implements Hashable {
   appDirId!: number;
-  level!: V1_AppDirLevel;
-}
-
-export enum V1_AppDirLevel {
-  BUSINESS_UNIT = 'BUSINESS_UNIT',
-  SUB_BUSINESS_UNIT = 'SUB_BUSINESS_UNIT',
-  FAMILY = 'FAMILY',
-  APPLICATION = 'APPLICATION',
-  DEPLOYMENT = 'DEPLOYMENT',
+  level!: AppDirLevel;
+
+  get hashCode(): string {
+    return hashArray([
+      CORE_HASH_STRUCTURE.APP_DIR_NODE,
+      this.appDirId.toString(),
+      this.level,
+    ]);
+  }
 }
 
 // -------------------------------------------- Pagination ---------------------------------------------

package/src/graph-manager/protocol/pure/v1/model/packageableElements/dataProduct/V1_DataProduct.ts

@@ -33,6 +33,7 @@ import {
 import type { V1_StereotypePtr } from '../domain/V1_StereotypePtr.js';
 import type { V1_TaggedValue } from '../domain/V1_TaggedValue.js';
 import type { V1_EmbeddedData } from '../../data/V1_EmbeddedData.js';
+import type { V1_AppDirNode } from '../../../lakehouse/entitlements/V1_CoreEntitlements.js';
 
 export const V1_DATA_PRODUCT_ELEMENT_PROTOCOL_TYPE = 'dataProduct';
 
@@ -369,12 +370,9 @@ export abstract class V1_DataProductType implements Hashable {
   }
 }
 
-export class V1_InternalDataProductType extends V1_DataProductType {
-  _type = V1_DataProductTypeValue.INTERNAL;
-}
+export class V1_InternalDataProductType extends V1_DataProductType {}
 
 export class V1_ExternalDataProductType extends V1_DataProductType {
-  _type = V1_DataProductTypeValue.EXTERNAL;
   link!: V1_DataProductLink;
 
   override get hashCode(): string {
@@ -409,6 +407,27 @@ export class V1_DataProductOperationalMetadata implements Hashable {
   }
 }
 
+export enum V1_DataProductOwnerType {
+  APP_DIR = 'appDir',
+}
+
+export abstract class V1_DataProductOwner implements Hashable {
+  abstract get hashCode(): string;
+}
+
+export class V1_AppDirOwner extends V1_DataProductOwner implements Hashable {
+  production: V1_AppDirNode | undefined;
+  prodParallel: V1_AppDirNode | undefined;
+
+  override get hashCode(): string {
+    return hashArray([
+      CORE_HASH_STRUCTURE.DATA_PRODUCT_OWNER,
+      this.production?.hashCode ?? '',
+      this.prodParallel?.hashCode ?? '',
+    ]);
+  }
+}
+
 export class V1_DataProduct extends V1_PackageableElement implements Hashable {
   title: string | undefined;
   description: string | undefined;
@@ -421,6 +440,7 @@ export class V1_DataProduct extends V1_PackageableElement implements Hashable {
   taggedValues: V1_TaggedValue[] = [];
   sampleValues: V1_EmbeddedData[] | undefined;
   operationalMetadata: V1_DataProductOperationalMetadata | undefined;
+  owner: V1_DataProductOwner | undefined;
 
   override get hashCode(): string {
     return hashArray([
@@ -436,6 +456,7 @@ export class V1_DataProduct extends V1_PackageableElement implements Hashable {
       hashArray(this.taggedValues),
       hashArray(this.sampleValues ?? []),
       this.operationalMetadata ?? '',
+      this.owner ?? '',
     ]);
   }
 

package/src/graph-manager/protocol/pure/v1/transformation/pureGraph/from/V1_DataProductTransformer.ts

@@ -30,6 +30,7 @@ import {
   FunctionAccessPoint,
   PackageableElementSampleQuery,
   InLineSampleQuery,
+  AppDirOwner,
 } from '../../../../../../../graph/metamodel/pure/dataProduct/DataProduct.js';
 import {
   type V1_AccessPoint,
@@ -57,6 +58,7 @@ import {
   V1_PackageableElementSampleQuery,
   V1_InLineSampleQuery,
   V1_NativeModelExecutionContext,
+  V1_AppDirOwner,
 } from '../../../model/packageableElements/dataProduct/V1_DataProduct.js';
 import { V1_initPackageableElement } from './V1_CoreTransformerHelper.js';
 import { V1_transformRawLambda } from './V1_RawValueSpecificationTransformer.js';
@@ -69,6 +71,7 @@ import { V1_PackageableElementPointer } from '../../../model/packageableElements
 import { V1_transformEmbeddedData } from './V1_DataElementTransformer.js';
 import { ConcreteFunctionDefinition } from '../../../../../../../graph/metamodel/pure/packageableElements/function/ConcreteFunctionDefinition.js';
 import { generateFunctionPrettyName } from '../../../../../../../graph/helpers/PureLanguageHelper.js';
+import { V1_AppDirNode } from '../../../lakehouse/entitlements/V1_CoreEntitlements.js';
 
 const transformAccessPoint = (
   ap: AccessPoint,
@@ -343,5 +346,26 @@ export const V1_transformDataProduct = (
     V1_transformTaggedValue(taggedValue),
   );
 
+  if (element.owner instanceof AppDirOwner) {
+    const v1Owner = new V1_AppDirOwner();
+    if (element.owner.production) {
+      const v1Node = new V1_AppDirNode();
+      v1Node.appDirId = element.owner.production.appDirId;
+      v1Node.level = element.owner.production.level;
+      v1Owner.production = v1Node;
+    }
+    if (element.owner.prodParallel) {
+      const v1Node = new V1_AppDirNode();
+      v1Node.appDirId = element.owner.prodParallel.appDirId;
+      v1Node.level = element.owner.prodParallel.level;
+      v1Owner.prodParallel = v1Node;
+    }
+    product.owner = v1Owner;
+  } else if (element.owner !== undefined) {
+    throw new UnsupportedOperationError(
+      `Unable to transform data product owner`,
+    );
+  }
+
   return product;
 };

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';