@finos/legend-application-studio 28.18.85 → 28.18.86

package/src/components/editor/editor-group/uml-editor/ClassQueryBuilder.tsx

@@ -25,6 +25,8 @@ import {
   ELEMENT_PATH_DELIMITER,
   RawVariableExpression,
   getFunctionSignature,
+  GenericTypeExplicitReference,
+  GenericType,
 } from '@finos/legend-graph';
 import {
   type QueryBuilderState,
@@ -118,7 +120,7 @@ export const promoteQueryToFunction = async (
     const returnType = queryBuilderState.getQueryReturnType();
     const _function = new ConcreteFunctionDefinition(
       functionName, // use functionName for now and it will be reset after composing _function.parameters and _function.returnType
-      PackageableElementExplicitReference.create(returnType),
+      GenericTypeExplicitReference.create(new GenericType(returnType)),
       Multiplicity.ONE,
     );
     // we will copy the body of the query to the body of the function and extract the parameters out

package/src/stores/editor/NewElementState.ts

@@ -82,6 +82,8 @@ import {
   generateIdentifiedConnectionId,
   getMappingCompatibleRuntimes,
   RuntimePointer,
+  GenericTypeExplicitReference,
+  GenericType,
 } from '@finos/legend-graph';
 import type { DSL_Mapping_LegendStudioApplicationPlugin_Extension } from '../extensions/DSL_Mapping_LegendStudioApplicationPlugin_Extension.js';
 import {
@@ -949,7 +951,9 @@ export class NewElementState {
       case PACKAGEABLE_ELEMENT_TYPE.FUNCTION: {
         const fn = new ConcreteFunctionDefinition(
           name,
-          PackageableElementExplicitReference.create(PrimitiveType.STRING),
+          GenericTypeExplicitReference.create(
+            new GenericType(PrimitiveType.STRING),
+          ),
           Multiplicity.ONE,
         );
         // default to empty string

package/src/stores/editor/editor-state/element-editor-state/function-activator/testable/FunctionTestableState.ts

@@ -591,7 +591,7 @@ export const createFunctionTest = (
   } else {
     const equalTo = new EqualTo();
     equalTo.id = DEFAULT_TEST_ASSERTION_ID;
-    const type = functionDefinition.returnType.value;
+    const type = functionDefinition.returnType.value.rawType;
     const valSpec = buildDefaultInstanceValue(
       editorStore.graphManagerState.graph,
       type,

package/src/stores/editor/editor-state/element-editor-state/mapping/MappingEditorState.ts

@@ -102,6 +102,7 @@ import {
   type Store,
   ModelStore,
   INTERNAL__UnknownSetImplementation,
+  RelationFunctionInstanceSetImplementation,
 } from '@finos/legend-graph';
 import type {
   DSL_Mapping_LegendStudioApplicationPlugin_Extension,
@@ -312,6 +313,12 @@ export const getMappingElementSource = (
       plugins,
     );
   }
+  // TODO: We could probably return the relation function used for the mapping here once we implement the form mode support for it
+  else if (
+    mappingElement instanceof RelationFunctionInstanceSetImplementation
+  ) {
+    return undefined;
+  }
   const extraMappingElementSourceExtractors = plugins.flatMap(
     (plugin) =>
       (
@@ -900,6 +907,12 @@ export class MappingEditorState extends ElementEditorState {
       );
       return;
     }
+    if (mappingElement instanceof RelationFunctionInstanceSetImplementation) {
+      this.editorStore.applicationStore.notificationService.notifyUnsupportedFeature(
+        'Relation Function mapping editor',
+      );
+      return;
+    }
     // Open mapping element from included mapping in another mapping editor tab
     if (mappingElement._PARENT !== this.element) {
       this.editorStore.graphEditorMode.openElement(mappingElement._PARENT);

package/src/stores/editor/editor-state/element-editor-state/mapping/MappingElementDecorator.ts

@@ -65,6 +65,7 @@ import {
   EnumerationMappingExplicitReference,
   SetImplementationExplicitReference,
   type INTERNAL__UnknownSetImplementation,
+  type RelationFunctionInstanceSetImplementation,
 } from '@finos/legend-graph';
 import type { EditorStore } from '../../../EditorStore.js';
 import {
@@ -657,6 +658,12 @@ export class MappingElementDecorator implements SetImplementationVisitor<void> {
   ): void {
     return;
   }
+
+  visit_RelationFunctionInstanceSetImplementation(
+    setImplementation: RelationFunctionInstanceSetImplementation,
+  ): void {
+    return;
+  }
 }
 
 /**
@@ -803,4 +810,10 @@ export class MappingElementDecorationCleaner
   ): void {
     return;
   }
+
+  visit_RelationFunctionInstanceSetImplementation(
+    setImplementation: RelationFunctionInstanceSetImplementation,
+  ): void {
+    return;
+  }
 }

package/src/stores/graph-modifier/DomainGraphModifierHelper.ts

@@ -27,7 +27,6 @@ import {
   type GenericTypeReference,
   type TaggedValue,
   type StereotypeReference,
-  type Type,
   type Multiplicity,
   type Stereotype,
   type Tag,
@@ -68,7 +67,6 @@ import {
   observe_TaggedValue,
   observe_Constraint,
   observe_GenericType,
-  observe_Type,
   observe_Unit,
   observe_RawLambda,
   isStubbed_PackageableElement,
@@ -365,9 +363,9 @@ export const function_addParameter = action(
     addUniqueEntry(_func.parameters, observe_RawVariableExpression(val));
   },
 );
-export const function_setReturnType = action(
-  (_func: ConcreteFunctionDefinition, val: Type): void => {
-    packageableElementReference_setValue(_func.returnType, observe_Type(val));
+export const function_setReturnGenericType = action(
+  (_func: ConcreteFunctionDefinition, val: GenericType): void => {
+    setGenericTypeReferenceValue(_func.returnType, observe_GenericType(val));
   },
 );
 export const function_setReturnMultiplicity = action(