@finos/legend-graph 32.3.7 → 32.3.9

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

@@ -161,6 +161,22 @@ export abstract class V1_ContractUserEventPayload {
   eventTimestamp!: string;
 }
 
+export class V1_TerminalOrderItem {
+  providerName!: string;
+  productName!: string;
+  category!: string;
+  price!: number;
+  id!: number;
+  perm_id?: number;
+}
+
+export class V1_TerminalProvisionPayload {
+  ordered_by!: string;
+  kerberos!: string;
+  order_items!: Record<number, V1_TerminalOrderItem[]>;
+  business_justification!: string;
+}
+
 export class V1_ContractUserEventPrivilegeManagerPayload extends V1_ContractUserEventPayload {
   candidateIdentity!: string;
   managerIdentity!: string;

package/src/graph-manager/protocol/pure/v1/lakehouse/helpers/V1_CoreLakehouseHelper.ts

@@ -0,0 +1,40 @@
+/**
+ * Copyright (c) 2025-present, Goldman Sachs
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+import { V1_IngestEnvironmentClassification } from '../ingest/V1_LakehouseIngestEnvironment.js';
+import { V1_EntitlementsLakehouseEnvironmentType } from '../entitlements/V1_EntitlementsDataProduct.js';
+
+export const V1_getEntitlementsEnvFromIngestEnv = (
+  ingestEnv: V1_IngestEnvironmentClassification,
+): V1_EntitlementsLakehouseEnvironmentType => {
+  switch (ingestEnv) {
+    case V1_IngestEnvironmentClassification.DEV:
+      return V1_EntitlementsLakehouseEnvironmentType.DEVELOPMENT;
+    case V1_IngestEnvironmentClassification.PROD_PARALLEL:
+      return V1_EntitlementsLakehouseEnvironmentType.PRODUCTION_PARALLEL;
+    case V1_IngestEnvironmentClassification.PROD:
+      return V1_EntitlementsLakehouseEnvironmentType.PRODUCTION;
+    default:
+      return V1_EntitlementsLakehouseEnvironmentType.PRODUCTION;
+  }
+};
+
+export const V1_isIngestEnvsCompatibleWithEntitlements = (
+  ingestEnv: V1_IngestEnvironmentClassification,
+  entitlementEnv: V1_EntitlementsLakehouseEnvironmentType,
+): boolean => {
+  return V1_getEntitlementsEnvFromIngestEnv(ingestEnv) === entitlementEnv;
+};

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

@@ -41,6 +41,7 @@ export enum V1_DataProductTypeValue {
 
 export abstract class V1_AccessPoint implements Hashable {
   id!: string;
+  title: string | undefined;
   description: string | undefined;
 
   get hashCode(): string {
@@ -98,6 +99,7 @@ export class V1_FunctionAccessPoint extends V1_AccessPoint implements Hashable {
 
 export class V1_AccessPointGroup implements Hashable {
   id!: string;
+  title: string | undefined;
   description: string | undefined;
   stereotypes: V1_StereotypePtr[] = [];
   accessPoints: V1_AccessPoint[] = [];

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

@@ -17,30 +17,30 @@
 import { hashArray, type Hashable } from '@finos/legend-shared';
 
 export class V1_Terminal implements Hashable {
-  id: string | undefined;
-  providerName: string | undefined;
-  productName: string | undefined;
-  category: string | undefined;
+  id!: string;
+  providerName!: string;
+  productName!: string;
+  category!: string;
   vendorProfileId: string | undefined;
   modelName: string | undefined;
   description: string | undefined;
   applicationName: string | undefined;
   tieredPrice: string | undefined;
-  price: string | undefined;
+  price!: string;
   totalFirmPrice: string | undefined;
 
   get hashCode(): string {
     return hashArray([
-      this.id ?? '',
-      this.providerName ?? '',
-      this.productName ?? '',
-      this.category ?? '',
+      this.id,
+      this.providerName,
+      this.productName,
+      this.category,
       this.vendorProfileId ?? '',
       this.modelName ?? '',
       this.description ?? '',
       this.applicationName ?? '',
       this.tieredPrice ?? '',
-      this.price ?? '',
+      this.price,
       this.totalFirmPrice ?? '',
     ]);
   }

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

@@ -73,17 +73,21 @@ const transformAccessPoint = (
     lake.classification = ap.classification;
     lake.reproducible = ap.reproducible;
     lake.description = ap.description;
+    lake.title = ap.title;
     return lake;
   } else if (ap instanceof FunctionAccessPoint) {
     const func = new V1_FunctionAccessPoint();
     func.id = ap.id;
     func.description = ap.description;
+    func.title = func.title;
     func.query = V1_transformRawLambda(ap.query, context);
     return func;
   } else if (ap instanceof UnknownAccessPoint) {
     const un = new V1_UnknownAccessPoint();
     un.content = ap.content;
     un.id = ap.id;
+    un.description = ap.title;
+    un.title = ap.title;
     return un;
   }
   throw new UnsupportedOperationError(
@@ -164,6 +168,7 @@ export const V1_transformDataProduct = (
       if (metamodelGroup instanceof ModelAccessPointGroup) {
         const modelGroup = new V1_ModelAccessPointGroup();
         modelGroup.id = metamodelGroup.id;
+        modelGroup.title = metamodelGroup.title;
         modelGroup.description = metamodelGroup.description;
         modelGroup.stereotypes = metamodelGroup.stereotypes.map(
           V1_transformStereotype,
@@ -216,6 +221,7 @@ export const V1_transformDataProduct = (
       }
       const group = new V1_AccessPointGroup();
       group.id = metamodelGroup.id;
+      group.title = metamodelGroup.title;
       group.description = metamodelGroup.description;
       group.stereotypes = metamodelGroup.stereotypes.map(
         V1_transformStereotype,

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

@@ -84,6 +84,7 @@ export const V1_buildAccessPoint = (
     lakeAccessPoint.reproducible = ap.reproducible;
     lakeAccessPoint.classification = ap.classification;
     lakeAccessPoint.description = ap.description;
+    lakeAccessPoint.title = ap.title;
     return lakeAccessPoint;
   } else if (ap instanceof V1_FunctionAccessPoint) {
     const functionAccessPoint = new FunctionAccessPoint(
@@ -95,11 +96,14 @@ export const V1_buildAccessPoint = (
       ),
     );
     functionAccessPoint.description = ap.description;
+    functionAccessPoint.title = ap.title;
     return functionAccessPoint;
   } else if (ap instanceof V1_UnknownAccessPoint) {
-    const unkown = new UnknownAccessPoint(ap.id);
-    unkown.content = ap.content;
-    return unkown;
+    const unknown = new UnknownAccessPoint(ap.id);
+    unknown.description = ap.description;
+    unknown.title = ap.title;
+    unknown.content = ap.content;
+    return unknown;
   }
   throw new UnsupportedOperationError(
     `Unsupported data product access type ${ap}`,
@@ -137,6 +141,7 @@ export const V1_buildAccessPointGroup = (
   if (elementGroup instanceof V1_ModelAccessPointGroup) {
     const group = new ModelAccessPointGroup();
     group.id = elementGroup.id;
+    group.title = elementGroup.title;
     group.description = elementGroup.description;
     group.accessPoints = elementGroup.accessPoints.map((ep) =>
       V1_buildAccessPoint(ep, context),
@@ -189,6 +194,7 @@ export const V1_buildAccessPointGroup = (
   } else {
     const group = new AccessPointGroup();
     group.id = elementGroup.id;
+    group.title = elementGroup.title;
     group.description = elementGroup.description;
     group.accessPoints = elementGroup.accessPoints.map((ep) =>
       V1_buildAccessPoint(ep, context),

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

@@ -98,6 +98,7 @@ export const V1_lakehouseAccessPointModelSchema = createModelSchema(
     id: primitive(),
     reproducible: optional(primitive()),
     targetEnvironment: primitive(),
+    title: optional(primitive()),
   },
 );
 
@@ -108,6 +109,7 @@ export const V1_functionAccessPointModelSchema = createModelSchema(
     description: optional(primitive()),
     id: primitive(),
     query: usingModelSchema(V1_rawLambdaModelSchema),
+    title: optional(primitive()),
   },
 );
 
@@ -139,6 +141,8 @@ const V1_deserializeAccessPoint = (
       const unknown = new V1_UnknownAccessPoint();
       unknown.content = json;
       unknown.id = json.id as string;
+      unknown.description = json.description as string;
+      unknown.title = json.title as string;
       return unknown;
     }
   }
@@ -185,6 +189,7 @@ export const V1_ModelAccessPointGroupModelSchema = createModelSchema(
     diagrams: customListWithSchema(V1_dataProductDiagramModelSchema),
     featuredElements: customListWithSchema(V1_ElementScopeModelSchema),
     id: primitive(),
+    title: optional(primitive()),
     stereotypes: customListWithSchema(V1_stereotypePtrModelSchema),
     mapping: usingModelSchema(V1_packageableElementPointerModelSchema),
   },
@@ -199,6 +204,7 @@ export const V1_DefaultAccessPointGroupModelSchema = createModelSchema(
     ),
     description: optional(primitive()),
     id: primitive(),
+    title: optional(primitive()),
     stereotypes: customListWithSchema(V1_stereotypePtrModelSchema, {
       INTERNAL__forceReturnEmptyInTest: true,
     }),

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

@@ -44,6 +44,8 @@ import {
   V1_LiteDataContractsResponse,
   V1_LiteDataContract,
   V1_DataContractApprovedUsersResponse,
+  V1_TerminalOrderItem,
+  V1_TerminalProvisionPayload,
 } from '../../../lakehouse/entitlements/V1_ConsumerEntitlements.js';
 import {
   createModelSchema,
@@ -437,6 +439,66 @@ export const V1_createContractPayloadModelSchema = (
     ),
   });
 
+export const V1_terminalOrderItemModelSchema = createModelSchema(
+  V1_TerminalOrderItem,
+  {
+    providerName: primitive(),
+    productName: primitive(),
+    category: primitive(),
+    price: primitive(),
+    id: primitive(),
+    perm_id: optional(primitive()),
+  },
+);
+
+export const V1_serializeOrderItems = (
+  val: Record<number, V1_TerminalOrderItem[]>,
+): Record<number, PlainObject<V1_TerminalOrderItem>[]> => {
+  const result: Record<number, PlainObject<V1_TerminalOrderItem>[]> = {};
+  for (const [key, items] of Object.entries(val)) {
+    result[Number(key)] = items.map(
+      (item: V1_TerminalOrderItem): PlainObject<V1_TerminalOrderItem> =>
+        serialize(V1_terminalOrderItemModelSchema, item),
+    );
+  }
+  return result;
+};
+
+export const V1_deserializeOrderItems = (
+  val: Record<string, PlainObject<V1_TerminalOrderItem>[]>,
+): Record<number, V1_TerminalOrderItem[]> => {
+  const result: Record<number, V1_TerminalOrderItem[]> = {};
+  for (const [key, items] of Object.entries(val)) {
+    result[Number(key)] = items.map(
+      (item: PlainObject<V1_TerminalOrderItem>): V1_TerminalOrderItem =>
+        deserialize(V1_terminalOrderItemModelSchema, item),
+    );
+  }
+  return result;
+};
+
+export const V1_terminalProvisionPayloadModelSchema = createModelSchema(
+  V1_TerminalProvisionPayload,
+  {
+    ordered_by: primitive(),
+    kerberos: primitive(),
+    order_items: custom(V1_serializeOrderItems, V1_deserializeOrderItems),
+    business_justification: primitive(),
+  },
+);
+
+export const V1_serializeTerminalProvisionPayload = (
+  payload: V1_TerminalProvisionPayload,
+): PlainObject<V1_TerminalProvisionPayload> => {
+  return serialize(V1_terminalProvisionPayloadModelSchema, payload);
+};
+
+export const V1_deserializeTerminalProvisionPayload = (
+  json: PlainObject<V1_TerminalProvisionPayload>,
+): V1_TerminalProvisionPayload => {
+  return deserialize(V1_terminalProvisionPayloadModelSchema, json);
+};
+
 export const V1_ContractUserStatusResponseModelSchema = createModelSchema(
   V1_ContractUserStatusResponse,
   {

package/src/index.ts

@@ -308,8 +308,14 @@ export {
   V1_TaskResponse,
   V1_TaskStatus,
   V1_TaskStatusChangeResponse,
+  V1_TerminalOrderItem,
+  V1_TerminalProvisionPayload,
   V1_UserApprovalStatus,
 } from './graph-manager/protocol/pure/v1/lakehouse/entitlements/V1_ConsumerEntitlements.js';
+export {
+  V1_getEntitlementsEnvFromIngestEnv,
+  V1_isIngestEnvsCompatibleWithEntitlements,
+} from './graph-manager/protocol/pure/v1/lakehouse/helpers/V1_CoreLakehouseHelper.js';
 export {
   V1_AdhocTeam,
   V1_AppDirLevel,
@@ -360,6 +366,7 @@ export {
   V1_pendingTasksResponseModelSchema,
   V1_SdlcDeploymentDataProductOriginModelSchema,
   V1_TaskStatusChangeResponseModelSchema,
+  V1_terminalProvisionPayloadModelSchema,
 } from './graph-manager/protocol/pure/v1/transformation/pureProtocol/serializationHelpers/V1_EntitlementSerializationHelper.js';
 export {
   V1_CreateSubscriptionInput,
@@ -737,8 +744,10 @@ export {
   UnknownDataProductIcon,
   DataProductElementScope,
   DataProductDiagram,
+  DataProductType,
   InternalDataProductType,
   Expertise,
+  ExternalDataProductType,
   type DataProductElement,
 } from './graph/metamodel/pure/dataProduct/DataProduct.js';
 export * from './graph-manager/action/changeDetection/DataProductObserveHelper.js';

package/tsconfig.json

@@ -514,6 +514,7 @@
     "./src/graph-manager/protocol/pure/v1/lakehouse/entitlements/V1_EntitlementsTasks.ts",
     "./src/graph-manager/protocol/pure/v1/lakehouse/helpers/V1_ConsumerEntitlementsObserver.ts",
     "./src/graph-manager/protocol/pure/v1/lakehouse/helpers/V1_ConsumerEntitlementsUtils.ts",
+    "./src/graph-manager/protocol/pure/v1/lakehouse/helpers/V1_CoreLakehouseHelper.ts",
     "./src/graph-manager/protocol/pure/v1/lakehouse/ingest/V1_LakehouseIngestEnvironment.ts",
     "./src/graph-manager/protocol/pure/v1/lakehouse/ingest/V1_LakehouseProducerEnvironment.ts",
     "./src/graph-manager/protocol/pure/v1/lakehouse/subscriptions/V1_ConsumerSubscriptions.ts",