@finos/legend-extension-dsl-data-space 10.4.252 → 10.4.253

package/src/graph-manager/protocol/pure/v1/model/packageableElements/dataSpace/V1_DSL_DataSpace_DataSpace.ts

@@ -22,6 +22,7 @@ import {
   type V1_PackageableElementVisitor,
   type V1_StereotypePtr,
   type V1_TaggedValue,
+  type V1_DataProductOperationalMetadata,
   V1_PackageableElement,
   type V1_DataElementReference,
 } from '@finos/legend-graph';
@@ -196,6 +197,7 @@ export class V1_DataSpace extends V1_PackageableElement implements Hashable {
   executables?: V1_DataSpaceExecutable[] | undefined;
   diagrams?: V1_DataSpaceDiagram[] | undefined;
   supportInfo?: V1_DataSpaceSupportInfo | undefined;
+  operationalMetadata?: V1_DataProductOperationalMetadata | undefined;
 
   override get hashCode(): string {
     return hashArray([
@@ -210,6 +212,7 @@ export class V1_DataSpace extends V1_PackageableElement implements Hashable {
       hashArray(this.executables ?? []),
       hashArray(this.diagrams ?? []),
       this.supportInfo ?? '',
+      this.operationalMetadata ?? '',
     ]);
   }
 

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

@@ -32,12 +32,15 @@ import {
   V1_PackageableElementPointer,
   V1_rawLambdaModelSchema,
   V1_dataElementReferenceModelSchema,
+  V1_DataProductOperationalMetadata,
 } from '@finos/legend-graph';
 import {
   type PlainObject,
   UnsupportedOperationError,
   usingConstantValueSchema,
   usingModelSchema,
+  optionalCustom,
+  optionalCustomList,
   optionalCustomListWithSchema,
   customListWithSchema,
   isString,
@@ -218,6 +221,20 @@ const V1_dataSpaceDiagramModelSchema = createModelSchema(V1_DataSpaceDiagram, {
   title: primitive(),
 });
 
+const V1_dataSpaceOperationalMetadataModelSchema = createModelSchema(
+  V1_DataProductOperationalMetadata,
+  {
+    coverageRegions: optionalCustomList(
+      (val) => (val as string).toLowerCase(),
+      (val) => (val as unknown as string).toUpperCase(),
+    ),
+    updateFrequency: optionalCustom(
+      (val) => (val as string).toLowerCase(),
+      (val) => (val as string).toUpperCase(),
+    ),
+  },
+);
+
 const V1_dataSpaceModelSchema = createModelSchema(V1_DataSpace, {
   _type: usingConstantValueSchema(V1_DATA_SPACE_ELEMENT_PROTOCOL_TYPE),
   defaultExecutionContext: optional(primitive()),
@@ -237,6 +254,9 @@ const V1_dataSpaceModelSchema = createModelSchema(V1_DataSpace, {
     V1_packageableElementPointerModelSchema,
   ),
   name: primitive(),
+  operationalMetadata: optional(
+    usingModelSchema(V1_dataSpaceOperationalMetadataModelSchema),
+  ),
   package: primitive(),
   stereotypes: customListWithSchema(V1_stereotypePtrModelSchema, {
     INTERNAL__forceReturnEmptyInTest: true,

package/src/stores/DataSpaceLayoutState.ts

@@ -41,6 +41,45 @@ type DataSpacePageNavigationCommand = {
   anchor: string;
 };
 
+export class CollapseState {
+  private readonly collapsedKeys = new Set<string>();
+
+  constructor() {
+    makeObservable<CollapseState, 'collapsedKeys'>(this, {
+      collapsedKeys: observable,
+      toggleSectionCollapse: action,
+      toggleAllSectionsCollapse: action,
+    });
+  }
+
+  isSectionCollapsed(key: string): boolean {
+    return this.collapsedKeys.has(key);
+  }
+
+  toggleSectionCollapse(key: string): void {
+    if (this.collapsedKeys.has(key)) {
+      this.collapsedKeys.delete(key);
+    } else {
+      this.collapsedKeys.add(key);
+    }
+  }
+
+  areAllSectionsCollapsed(keys: string[]): boolean {
+    return keys.length > 0 && keys.every((key) => this.collapsedKeys.has(key));
+  }
+
+  toggleAllSectionsCollapse(keys: string[]): void {
+    const shouldCollapse = !this.areAllSectionsCollapsed(keys);
+    keys.forEach((key) => {
+      if (shouldCollapse) {
+        this.collapsedKeys.add(key);
+      } else {
+        this.collapsedKeys.delete(key);
+      }
+    });
+  }
+}
+
 export class DataSpaceLayoutState {
   readonly dataSpaceViewerState: DataSpaceViewerState;
 
@@ -56,6 +95,8 @@ export class DataSpaceLayoutState {
   private wikiPageVisibleAnchors: string[] = [];
   private wikiPageScrollIntersectionObserver?: IntersectionObserver | undefined;
 
+  readonly sectionCollapseState = new CollapseState();
+
   constructor(dataSpaceViewerState: DataSpaceViewerState) {
     makeObservable<
       DataSpaceLayoutState,

package/src/stores/DataSpaceModelsDocumentationState.ts

@@ -27,7 +27,13 @@ export class DataSpaceViewerModelsDocumentationState
   readonly dataSpaceViewerState: DataSpaceViewerState;
 
   constructor(dataSpaceViewerState: DataSpaceViewerState) {
-    super(dataSpaceViewerState.dataSpaceAnalysisResult.elementDocs);
+    super(
+      dataSpaceViewerState.applicationStore,
+      dataSpaceViewerState.dataSpaceAnalysisResult.elementDocs,
+      {
+        collapseState: dataSpaceViewerState.layoutState.sectionCollapseState,
+      },
+    );
 
     this.dataSpaceViewerState = dataSpaceViewerState;
   }

package/src/stores/DataSpaceViewerDiagramViewerState.ts

@@ -28,8 +28,12 @@ export class DataSpaceViewerDiagramViewerState
 
   constructor(dataSpaceViewerState: DataSpaceViewerState) {
     super(
+      dataSpaceViewerState.applicationStore,
       dataSpaceViewerState.dataSpaceAnalysisResult.diagrams,
-      dataSpaceViewerState.queryClass,
+      {
+        queryClass: dataSpaceViewerState.queryClass,
+        collapseState: dataSpaceViewerState.layoutState.sectionCollapseState,
+      },
     );
     this.dataSpaceViewerState = dataSpaceViewerState;
   }

package/src/stores/DataSpaceViewerExecutableState.ts

@@ -32,6 +32,7 @@ import {
 import type { DataSpaceViewerState } from './DataSpaceViewerState.js';
 import { action, flow, makeObservable, observable } from 'mobx';
 import {
+  extractAnnotatedElementDocumentation,
   RelationType,
   TDSExecutionResult,
   type ExecutionResultWithMetadata,
@@ -212,6 +213,7 @@ export class DataSpaceExecutableRelationResultState {
       id: uuid(),
       name: col.name,
       type: col.genericType.value.rawType.name,
+      documentation: extractAnnotatedElementDocumentation(col),
     }));
   }
 }

package/src/stores/query-builder/DataSpaceQueryBuilderState.ts

@@ -23,6 +23,8 @@ import {
   type QueryBuilderExtraFunctionAnalysisInfo,
   QUERY_BUILDER_LAMBDA_WRITER_MODE,
   type ExtraOptionsConfig,
+  QueryBuilderEmbeddedFromExecutionContextState,
+  type QueryBuilderExecutionContextState,
 } from '@finos/legend-query-builder';
 import {
   type Class,
@@ -69,6 +71,7 @@ import type {
 } from '@finos/legend-storage';
 import { buildDataSpaceExecutableAnalysisResultFromExecutable } from '../../graph-manager/action/analytics/DataSpaceAnalysisHelper.js';
 import { compareLabelFn } from '@finos/legend-art';
+import { resolveExecutionContextMapping } from '../../graph-manager/DSL_DataSpace_GraphManagerHelper.js';
 
 const matchesDataElement = (
   _class: Class,
@@ -367,6 +370,23 @@ export class DataSpaceQueryBuilderState extends QueryBuilderState {
     return this.buildQueryLambdaWithoutExecutionContext();
   }
 
+  override setExecutionContextState(
+    val: QueryBuilderExecutionContextState,
+  ): void {
+    if (
+      val instanceof QueryBuilderEmbeddedFromExecutionContextState &&
+      val.mapping === undefined
+    ) {
+      const resolvedMapping = resolveExecutionContextMapping(
+        this.executionContext,
+      );
+      if (resolvedMapping) {
+        val.setMapping(resolvedMapping);
+      }
+    }
+    super.setExecutionContextState(val);
+  }
+
   override buildExecutionContextExpression(
     lambdaFunction: LambdaFunction,
   ): LambdaFunction {