@finos/legend-graph 32.1.22 → 32.1.24

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

@@ -75,6 +75,8 @@ import {
   V1_serializeOwnership,
   V1_deserializeOwnership,
   V1_PostDeploymentActionSchema,
+  V1_serializePostDeploymentProperties,
+  V1_deserializePostDeploymentProperties,
 } from './V1_FunctionActivatorSerializationHelper.js';
 import {
   V1_deserializeTestSuite,
@@ -90,6 +92,7 @@ import {
 } from './V1_TypeSerializationHelper.js';
 import { PackageableElementPointerType } from '../../../../../../../graph/MetaModelConst.js';
 import type { V1_PackageableElementPointer } from '../../../model/packageableElements/V1_PackageableElement.js';
+import { V1_PostDeploymentAction } from '../../../engine/functionActivator/V1_PostDeploymentAction.js';
 
 export const V1_CLASS_ELEMENT_PROTOCOL_TYPE = 'class';
 export const V1_PROFILE_ELEMENT_PROTOCOL_TYPE = 'profile';
@@ -196,56 +199,72 @@ export const V1_measureModelSchema = createModelSchema(V1_Measure, {
   package: primitive(),
 });
 
-export const V1_snowflakeAppModelSchema = createModelSchema(V1_SnowflakeApp, {
-  _type: usingConstantValueSchema(V1_SNOWFLAKE_APP_TYPE),
-  description: optional(primitive()),
-  applicationName: primitive(),
-  function: usingModelSchema(V1_packageableElementPointerModelSchema),
-  name: primitive(),
-  package: primitive(),
-  permissionScheme: optional(primitive()),
-  actions: list(usingModelSchema(V1_PostDeploymentActionSchema)),
-  usageRole: optional(primitive()),
-  stereotypes: customListWithSchema(V1_stereotypePtrModelSchema, {
-    INTERNAL__forceReturnEmptyInTest: true,
-  }),
-  taggedValues: customListWithSchema(V1_taggedValueModelSchema, {
-    INTERNAL__forceReturnEmptyInTest: true,
-  }),
-  activationConfiguration: usingModelSchema(
-    V1_SnowflakeAppDeploymentConfigurationAppModelSchema,
-  ),
-  ownership: optionalCustom(
-    (val) => V1_serializeDeploymentOwership(val),
-    (val) => V1_deserializeDeploymentOwnership(val),
-  ),
-});
+export const V1_snowflakeAppModelSchema = (
+  plugins: PureProtocolProcessorPlugin[],
+): ModelSchema<V1_SnowflakeApp> =>
+  createModelSchema(V1_SnowflakeApp, {
+    _type: usingConstantValueSchema(V1_SNOWFLAKE_APP_TYPE),
+    description: optional(primitive()),
+    applicationName: primitive(),
+    function: usingModelSchema(V1_packageableElementPointerModelSchema),
+    name: primitive(),
+    package: primitive(),
+    permissionScheme: optional(primitive()),
+    actions: list(usingModelSchema(V1_PostDeploymentActionSchema(plugins))),
+    usageRole: optional(primitive()),
+    stereotypes: customListWithSchema(V1_stereotypePtrModelSchema, {
+      INTERNAL__forceReturnEmptyInTest: true,
+    }),
+    taggedValues: customListWithSchema(V1_taggedValueModelSchema, {
+      INTERNAL__forceReturnEmptyInTest: true,
+    }),
+    activationConfiguration: usingModelSchema(
+      V1_SnowflakeAppDeploymentConfigurationAppModelSchema,
+    ),
+    ownership: optionalCustom(
+      (val) => V1_serializeDeploymentOwership(val),
+      (val) => V1_deserializeDeploymentOwnership(val),
+    ),
+  });
 
-export const V1_HostedServiceModelSchema = createModelSchema(V1_HostedService, {
-  _type: usingConstantValueSchema(V1_HOSTED_SERVICE_TYPE),
-  documentation: optional(primitive()),
-  pattern: primitive(),
-  autoActivateUpdates: primitive(),
-  storeModel: primitive(),
-  generateLineage: primitive(),
-  function: usingModelSchema(V1_packageableElementPointerModelSchema),
-  name: primitive(),
-  package: primitive(),
-  stereotypes: customListWithSchema(V1_stereotypePtrModelSchema, {
-    INTERNAL__forceReturnEmptyInTest: true,
-  }),
-  taggedValues: customListWithSchema(V1_taggedValueModelSchema, {
-    INTERNAL__forceReturnEmptyInTest: true,
-  }),
-  ownership: optionalCustom(
-    (val) => V1_serializeOwnership(val),
-    (val) => V1_deserializeOwnership(val),
-  ),
-  actions: list(usingModelSchema(V1_PostDeploymentActionSchema)),
-  activationConfiguration: optional(
-    usingModelSchema(V1_HostedServiceDeploymentConfigurationAppModelSchema),
-  ),
-});
+export const V1_HostedServiceModelSchema = (
+  plugins: PureProtocolProcessorPlugin[],
+): ModelSchema<V1_HostedService> =>
+  createModelSchema(V1_HostedService, {
+    _type: usingConstantValueSchema(V1_HOSTED_SERVICE_TYPE),
+    documentation: optional(primitive()),
+    pattern: primitive(),
+    autoActivateUpdates: primitive(),
+    storeModel: primitive(),
+    generateLineage: primitive(),
+    function: usingModelSchema(V1_packageableElementPointerModelSchema),
+    name: primitive(),
+    package: primitive(),
+    stereotypes: customListWithSchema(V1_stereotypePtrModelSchema, {
+      INTERNAL__forceReturnEmptyInTest: true,
+    }),
+    taggedValues: customListWithSchema(V1_taggedValueModelSchema, {
+      INTERNAL__forceReturnEmptyInTest: true,
+    }),
+    ownership: optionalCustom(
+      (val) => V1_serializeOwnership(val),
+      (val) => V1_deserializeOwnership(val),
+    ),
+    actions: list(
+      usingModelSchema(
+        createModelSchema(V1_PostDeploymentAction, {
+          automated: optional(primitive()),
+          properties: optionalCustom(
+            (val) => V1_serializePostDeploymentProperties(val, plugins),
+            (val) => V1_deserializePostDeploymentProperties(val, plugins),
+          ),
+        }),
+      ),
+    ),
+    activationConfiguration: optional(
+      usingModelSchema(V1_HostedServiceDeploymentConfigurationAppModelSchema),
+    ),
+  });
 
 // ------------------------------------- Class -------------------------------------
 

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

@@ -29,6 +29,7 @@ import {
   optional,
   primitive,
   serialize,
+  type ModelSchema,
 } from 'serializr';
 import { V1_connectionPointerModelSchema } from './V1_ConnectionSerializationHelper.js';
 import {
@@ -39,9 +40,11 @@ import {
 import { V1_HostedServiceDeploymentConfiguration } from '../../../engine/functionActivator/V1_HostedServiceDeploymentConfiguration.js';
 import { V1_PostDeploymentAction } from '../../../engine/functionActivator/V1_PostDeploymentAction.js';
 import {
+  type V1_PostDeploymentProperties,
   V1_INTERNAL__UnknownPostDeploymentProperties,
-  V1_PostDeploymentProperties,
 } from '../../../engine/functionActivator/V1_PostDeploymentProperties.js';
+import type { PureProtocolProcessorPlugin } from '../../../../PureProtocolProcessorPlugin.js';
+import type { DSL_FunctionActivator_PureProtocolProcessorPlugin_Extension } from '../../../../extensions/DSL_FunctionActivator_PureProtocolProcessorPlugin_Extension.js';
 
 const V1_SNOWFLAKE_APP_DEPLOYMENT_CONFIGURATION_APP_TYPE =
   'snowflakeDeploymentConfiguration';
@@ -118,41 +121,67 @@ export const V1_deserializeOwnership = (
       );
   }
 };
-
 export const V1_deserializePostDeploymentProperties = (
   json: PlainObject<V1_PostDeploymentProperties>,
+  plugins: PureProtocolProcessorPlugin[],
 ): V1_PostDeploymentProperties => {
   switch (json._type) {
-    // we can add known post actions here
-    default:
+    default: {
+      const extraConnectionDatasourceSpecificationProtocolDeserializers =
+        plugins.flatMap(
+          (plugin) =>
+            (
+              plugin as DSL_FunctionActivator_PureProtocolProcessorPlugin_Extension
+            ).V1_getPostDeploymentPropertiesDeserializers?.() ?? [],
+        );
+      for (const deserializer of extraConnectionDatasourceSpecificationProtocolDeserializers) {
+        const protocol = deserializer(json);
+        if (protocol) {
+          return protocol;
+        }
+      }
       // Fall back to unknown
       const protocol = new V1_INTERNAL__UnknownPostDeploymentProperties();
       protocol.content = json;
       return protocol;
+    }
   }
 };
 
 export const V1_PostDeploymentPropertiesSchema = createModelSchema(
-  V1_PostDeploymentProperties,
+  V1_INTERNAL__UnknownPostDeploymentProperties,
   {},
 );
 
 export const V1_serializePostDeploymentProperties = (
   protocol: V1_PostDeploymentProperties,
+  plugins: PureProtocolProcessorPlugin[],
 ): PlainObject<V1_PostDeploymentProperties> => {
   if (protocol instanceof V1_INTERNAL__UnknownPostDeploymentProperties) {
     return protocol.content;
   }
+  const extraPostDeploymentPropertiesSerializers = plugins.flatMap(
+    (plugin) =>
+      (
+        plugin as DSL_FunctionActivator_PureProtocolProcessorPlugin_Extension
+      ).V1_getPostDeploymentPropertiesSerializers?.() ?? [],
+  );
+  for (const serializer of extraPostDeploymentPropertiesSerializers) {
+    const json = serializer(protocol);
+    if (json) {
+      return json;
+    }
+  }
   return serialize(V1_PostDeploymentPropertiesSchema, undefined);
 };
 
-export const V1_PostDeploymentActionSchema = createModelSchema(
-  V1_PostDeploymentAction,
-  {
+export const V1_PostDeploymentActionSchema = (
+  plugins: PureProtocolProcessorPlugin[],
+): ModelSchema<V1_PostDeploymentAction> =>
+  createModelSchema(V1_PostDeploymentAction, {
     automated: optional(primitive()),
     properties: optionalCustom(
-      (val) => V1_serializePostDeploymentProperties(val),
-      (val) => V1_deserializePostDeploymentProperties(val),
+      (val) => V1_serializePostDeploymentProperties(val, plugins),
+      (val) => V1_deserializePostDeploymentProperties(val, plugins),
     ),
-  },
-);
+  });

package/src/index.ts

@@ -95,6 +95,20 @@ export {
   DEFAULT_HOSTED_SERVICE_PATTERN,
 } from './graph/metamodel/pure/packageableElements/function/HostedService.js';
 export { HostedServiceDeploymentConfiguration } from './graph/metamodel/pure/functionActivator/HostedServiceDeploymentConfiguration.js';
+export { PostDeploymentAction } from './graph/metamodel/pure/functionActivator/PostDeploymentAction.js';
+export {
+  PostDeploymentProperties,
+  INTERNAL__UnknownPostDeploymentProperties,
+} from './graph/metamodel/pure/functionActivator/PostDeploymentProperties.js';
+export {
+  V1_PostDeploymentProperties,
+  V1_INTERNAL__UnknownPostDeploymentProperties,
+} from './graph-manager/protocol/pure/v1/engine/functionActivator/V1_PostDeploymentProperties.js';
+export {
+  type V1_PostDeploymentPropertiesTransformer,
+  type V1_PostDeploymentPropertiesDeserializer,
+  type V1_PostDeploymentPropertiesSerializer,
+} from './graph-manager/protocol/pure/extensions/DSL_FunctionActivator_PureProtocolProcessorPlugin_Extension.js';
 
 // --------------------------------------------- VALUE SPECIFICATION --------------------------------------------------
 
@@ -244,7 +258,12 @@ export { V1_DuckDBDatasourceSpecification } from './graph-manager/protocol/pure/
 export { V1_TestAuthenticationStrategy } from './graph-manager/protocol/pure/v1/model/packageableElements/store/relational/connection/V1_AuthenticationStrategy.js';
 export { V1_RelationStoreAccessor } from './graph-manager/protocol/pure/v1/model/valueSpecification/raw/classInstance/relation/V1_RelationStoreAccessor.js';
 export { V1_PackageableConnection } from './graph-manager/protocol/pure/v1/model/packageableElements/connection/V1_PackageableConnection.js';
-
+export {
+  V1_DataContract,
+  V1_DataContractsRecord,
+  V1_TaskStatus,
+  V1_ContractCreate_LegendDataProduct,
+} from './graph-manager/protocol/pure/v1/entitlements/V1_ConsumerEntitlements.js';
 // --------------------------------------------- EXECUTION PLAN --------------------------------------------------
 
 export { INTERNAL__UnknownExecutionResult } from './graph-manager/action/execution/INTERNAL__UnknownExecutionResult.js';
@@ -840,6 +859,7 @@ export { V1_FunctionActivatorInfo } from './graph-manager/protocol/pure/v1/engin
 export { V1_FunctionActivatorInput } from './graph-manager/protocol/pure/v1/engine/functionActivator/V1_FunctionActivatorInput.js';
 export { V1_DatabaseToModelGenerationInput } from './graph-manager/protocol/pure/v1/engine/relational/V1_DatabaseToModelGenerationInput.js';
 export { V1_RelationalConnectionBuilder } from './graph-manager/protocol/pure/v1/engine/relational/V1_RelationalConnectionBuilder.js';
+export * from './graph-manager/protocol/pure/extensions/DSL_FunctionActivator_PureProtocolProcessorPlugin_Extension.js';
 
 // ------------------------------------- TO BE MODULARIZED --------------------------------------------
 /**

package/tsconfig.json

@@ -415,6 +415,7 @@
     "./src/graph-manager/extensions/Core_PureGraphManagerPlugin.ts",
     "./src/graph-manager/extensions/DSL_Data_PureGraphManagerPlugin_Extension.ts",
     "./src/graph-manager/extensions/DSL_ExternalFormat_PureGraphManagerPlugin.ts",
+    "./src/graph-manager/extensions/DSL_FunctionActivator_PureGraphManager_Extension.ts",
     "./src/graph-manager/extensions/DSL_Generation_PureGraphManagerPlugin_Extension.ts",
     "./src/graph-manager/extensions/DSL_Mapping_PureGraphManagerPlugin_Extension.ts",
     "./src/graph-manager/extensions/STO_Relational_PureGraphManagerPlugin_Extension.ts",
@@ -426,6 +427,7 @@
     "./src/graph-manager/protocol/pure/PureProtocolProcessorPlugin.ts",
     "./src/graph-manager/protocol/pure/extensions/DSL_Data_PureProtocolProcessorPlugin_Extension.ts",
     "./src/graph-manager/protocol/pure/extensions/DSL_ExternalFormat_PureProtocolProcessorPlugin.ts",
+    "./src/graph-manager/protocol/pure/extensions/DSL_FunctionActivator_PureProtocolProcessorPlugin_Extension.ts",
     "./src/graph-manager/protocol/pure/extensions/DSL_Generation_PureProtocolProcessorPlugin_Extension.ts",
     "./src/graph-manager/protocol/pure/extensions/DSL_Mapping_PureProtocolProcessorPlugin_Extension.ts",
     "./src/graph-manager/protocol/pure/extensions/STO_Relational_PureProtocolProcessorPlugin_Extension.ts",
@@ -488,6 +490,9 @@
     "./src/graph-manager/protocol/pure/v1/engine/test/V1_DebugTestsResult.ts",
     "./src/graph-manager/protocol/pure/v1/engine/test/V1_RunTestsInput.ts",
     "./src/graph-manager/protocol/pure/v1/engine/test/V1_RunTestsResult.ts",
+    "./src/graph-manager/protocol/pure/v1/entitlements/V1_ConsumerEntitlements.ts",
+    "./src/graph-manager/protocol/pure/v1/entitlements/V1_CoreEntitlements.ts",
+    "./src/graph-manager/protocol/pure/v1/entitlements/V1_EntitlementsTasks.ts",
     "./src/graph-manager/protocol/pure/v1/helpers/V1_DomainHelper.ts",
     "./src/graph-manager/protocol/pure/v1/helpers/V1_DomainObserverHelper.ts",
     "./src/graph-manager/protocol/pure/v1/helpers/V1_ValueSpecificationObserver.ts",