@finos/legend-application-studio 26.1.7 → 26.1.9

package/src/stores/editor/utils/MockDataUtils.ts

@@ -21,10 +21,12 @@ import {
   Randomizer,
   UnsupportedOperationError,
   type PlainObject,
+  isNonNullable,
+  filterByType,
 } from '@finos/legend-shared';
 import type { EditorStore } from '../EditorStore.js';
 import {
-  type PrimitiveType,
+  PrimitiveType,
   type Enumeration,
   PRIMITIVE_TYPE,
   Class,
@@ -32,6 +34,28 @@ import {
   DATE_FORMAT,
   DATE_TIME_FORMAT,
   classHasCycle,
+  TableAlias,
+  Table,
+  Column,
+  VarChar,
+  type RelationalDataType,
+  Char,
+  VarBinary,
+  TinyInt,
+  Float,
+  Timestamp,
+  Binary,
+  Bit,
+  Other,
+  Numeric,
+  Decimal,
+  Double,
+  Integer,
+  Real,
+  SmallInt,
+  Date as ColumnDate,
+  SemiStructured,
+  Json,
 } from '@finos/legend-graph';
 import {
   CLASS_PROPERTY_TYPE,
@@ -182,6 +206,85 @@ export const createMockDataForClassWithFormat = (
   }
 };
 
+export const getPrimitiveTypeFromRelationalType = (
+  type: RelationalDataType,
+): PrimitiveType | undefined => {
+  if (
+    type instanceof VarChar ||
+    type instanceof Char ||
+    type instanceof VarBinary ||
+    type instanceof Binary ||
+    type instanceof Bit ||
+    type instanceof Other
+  ) {
+    return PrimitiveType.STRING;
+  } else if (type instanceof Numeric) {
+    return PrimitiveType.NUMBER;
+  } else if (type instanceof Decimal) {
+    return PrimitiveType.DECIMAL;
+  } else if (
+    type instanceof Double ||
+    type instanceof Integer ||
+    type instanceof Real ||
+    type instanceof SmallInt ||
+    type instanceof TinyInt
+  ) {
+    return PrimitiveType.INTEGER;
+  } else if (type instanceof Float) {
+    return PrimitiveType.FLOAT;
+  } else if (type instanceof ColumnDate) {
+    return PrimitiveType.DATE;
+  } else if (type instanceof Timestamp) {
+    return PrimitiveType.DATETIME;
+  }
+  return undefined;
+};
+export const createMockDataForColumn = (
+  col: Column,
+  isPrimaryKey: boolean,
+  idx?: number | number,
+): string | undefined => {
+  const type = col.type;
+
+  if (
+    (type instanceof Double ||
+      type instanceof Integer ||
+      type instanceof Real ||
+      type instanceof SmallInt ||
+      type instanceof TinyInt) &&
+    isPrimaryKey &&
+    idx
+  ) {
+    return idx.toString();
+  }
+  const primitive = getPrimitiveTypeFromRelationalType(type);
+  if (primitive) {
+    return createMockPrimitiveProperty(primitive, col.name).toString();
+  } else if (type instanceof Json || type instanceof SemiStructured) {
+    return '{}';
+  }
+  return undefined;
+};
+
+export const createMockDataForTable = (
+  table: Table,
+  ITERATIONS = 1,
+): string => {
+  const targetedCols = table.columns.filter(filterByType(Column));
+  const colNames = targetedCols.map((e) => e.name).join(',');
+  const vals = Array.from(Array(ITERATIONS).keys())
+    .map((idx) =>
+      targetedCols
+        .map((col) =>
+          createMockDataForColumn(col, table.primaryKey.includes(col), idx),
+        )
+        .filter(isNonNullable)
+        .join(','),
+    )
+    .join('\n');
+  return `${colNames}\n${vals}`;
+};
+
 export const createMockDataForMappingElementSource = (
   srcElement: MappingElementSource,
   editorStore: EditorStore,
@@ -200,6 +303,11 @@ export const createMockDataForMappingElementSource = (
       );
       return '';
     }
+  } else if (
+    srcElement instanceof TableAlias &&
+    srcElement.relation instanceof Table
+  ) {
+    return createMockDataForTable(srcElement.relation);
   }
   editorStore.applicationStore.notificationService.notifyWarning(
     new UnsupportedOperationError(

package/src/stores/editor/utils/TestableUtils.ts

@@ -25,19 +25,17 @@ import {
   type RelationalDatabaseConnection,
   type XmlModelConnection,
   type Runtime,
-  type IdentifiedConnection,
   type TestAssertion,
   type AtomicTest,
   type Class,
   ExternalFormatData,
   RelationalCSVData,
   ConnectionTestData,
-  EngineRuntime,
-  RuntimePointer,
   EqualToJson,
   DEFAULT_TEST_ASSERTION_PREFIX,
   RelationalCSVDataTable,
   type INTERNAL__UnknownConnection,
+  getAllIdentifiedConnectionsFromRuntime,
 } from '@finos/legend-graph';
 import {
   assertTrue,
@@ -60,6 +58,9 @@ export const validateTestableId = (
   id: string | undefined,
   possibleIds: string[] | undefined,
 ): string | undefined => {
+  if (id === undefined) {
+    return undefined;
+  }
   if (!id) {
     return 'ID is required';
   } else if (id.includes(' ')) {
@@ -90,22 +91,6 @@ export const createDefaultEqualToJSONTestAssertion = (
   return assertion;
 };
 
-export const getAllIdentifiedConnectionsFromRuntime = (
-  runtime: Runtime,
-): IdentifiedConnection[] => {
-  const resolvedRuntimes: EngineRuntime[] = [];
-  if (runtime instanceof RuntimePointer) {
-    resolvedRuntimes.push(runtime.packageableRuntime.value.runtimeValue);
-  } else if (runtime instanceof EngineRuntime) {
-    resolvedRuntimes.push(runtime);
-  }
-  return resolvedRuntimes
-    .flatMap((e) =>
-      e.connections.map((connection) => connection.storeConnections),
-    )
-    .flat();
-};
-
 export const createEmbeddedDataFromClass = (
   _class: Class,
   editorStore: EditorStore,

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

@@ -24,10 +24,14 @@ import {
   type DataElementReference,
   type RelationalCSVDataTable,
   type RelationalCSVData,
+  type ModelData,
+  type PackageableElementReference,
+  type ModelEmbeddedData,
   observe_EmbeddedData,
   observe_DataElement,
   observe_RelationalDataTable,
   PackageableElementExplicitReference,
+  observe_ModelData,
 } from '@finos/legend-graph';
 import { addUniqueEntry, deleteEntry } from '@finos/legend-shared';
 import { action } from 'mobx';
@@ -54,12 +58,6 @@ export const externalFormatData_setData = action(
   },
 );
 
-export const modelStoreData_setInstance = action(
-  (modelStoreData: ModelStoreData, value: Map<Class, object>): void => {
-    modelStoreData.instances = value;
-  },
-);
-
 export const relationalData_setTableValues = action(
   (data: RelationalCSVDataTable, values: string): void => {
     data.values = values;
@@ -102,3 +100,44 @@ export const dataElementReference_setDataElement = action(
       );
   },
 );
+
+// Model Store
+export const modelStoreData_addModelData = action(
+  (
+    modelStoreData: ModelStoreData,
+    value: ModelData,
+    context: ObserverContext,
+  ): void => {
+    const entries = modelStoreData.modelData ?? [];
+    if (!modelStoreData.modelData) {
+      modelStoreData.modelData = entries;
+    }
+    addUniqueEntry(entries, observe_ModelData(value, context));
+  },
+);
+
+export const modelStoreData_setDataModelModel = action(
+  (modelData: ModelData, val: PackageableElementReference<Class>): void => {
+    modelData.model = val;
+  },
+);
+
+export const modelStoreData_deleteModelData = action(
+  (modelStoreData: ModelStoreData, value: ModelData): void => {
+    if (modelStoreData.modelData) {
+      deleteEntry(modelStoreData.modelData, value);
+    }
+  },
+);
+
+export const modelStoreData_setModelDataClass = action(
+  (modelData: ModelData, val: PackageableElementReference<Class>): void => {
+    modelData.model = val;
+  },
+);
+
+export const modelStoreData_setModelDataData = action(
+  (modelData: ModelEmbeddedData, val: EmbeddedData): void => {
+    modelData.data = val;
+  },
+);

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

@@ -204,7 +204,7 @@ export const mapping_deleteTestSuite = action(
 
 export const mappingTestable_setQuery = action(
   (test: MappingQueryTest | MappingQueryTestSuite, value: RawLambda): void => {
-    test.query = observe_RawLambda(value);
+    test.func = observe_RawLambda(value);
   },
 );
 

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

@@ -309,6 +309,12 @@ export const snowflakeDatasourceSpec_setQuotedIdentifiersIgnoreCase = action(
   },
 );
 
+export const snowflakeDatasourceSpec_setEnableQueryTags = action(
+  (spec: SnowflakeDatasourceSpecification, val: boolean | undefined): void => {
+    spec.enableQueryTags = val;
+  },
+);
+
 export const snowflakeDatasourceSpec_setProxyHost = action(
   (spec: SnowflakeDatasourceSpecification, val: string | undefined): void => {
     spec.proxyHost = val;

package/src/stores/project-view/ProjectViewerStore.ts

@@ -24,10 +24,6 @@ import {
   LogEvent,
   IllegalStateError,
   StopWatch,
-  returnUndefOnError,
-  downloadFileUsingDataURI,
-  formatDate,
-  ContentType,
 } from '@finos/legend-shared';
 import {
   type ProjectViewerPathParams,
@@ -50,10 +46,7 @@ import {
   Workspace,
 } from '@finos/legend-server-sdlc';
 import { LEGEND_STUDIO_APP_EVENT } from '../../__lib__/LegendStudioEvent.js';
-import {
-  DEFAULT_DATE_TIME_FORMAT,
-  DEFAULT_TAB_SIZE,
-} from '@finos/legend-application';
+import { DEFAULT_TAB_SIZE } from '@finos/legend-application';
 import { resolveVersion, StoreProjectData } from '@finos/legend-server-depot';
 import {
   type WorkflowManagerState,
@@ -70,6 +63,7 @@ import {
 } from '@finos/legend-graph';
 import { GRAPH_EDITOR_MODE } from '../editor/EditorConfig.js';
 import { LegendStudioTelemetryHelper } from '../../__lib__/LegendStudioTelemetryHelper.js';
+import { payloadDebugger } from '../editor/panel-group/DevToolPanelState.js';
 
 interface ProjectViewerGraphBuilderMaterial {
   entities: Entity[];
@@ -372,23 +366,7 @@ export class ProjectViewerStore {
             queryBaseUrl:
               this.editorStore.applicationStore.config.engineQueryServerUrl,
             enableCompression: true,
-            payloadDebugger: (payload: unknown, identifier: string) => {
-              const content =
-                returnUndefOnError(() =>
-                  JSON.stringify(payload, undefined, 2),
-                ) ??
-                returnUndefOnError(() => `${payload}`) ??
-                '';
-              // TODO: we can also copy the debug content to clipboard
-              downloadFileUsingDataURI(
-                `PAYLOAD_DEBUG__${identifier}__${formatDate(
-                  new Date(Date.now()),
-                  DEFAULT_DATE_TIME_FORMAT,
-                )}.json`,
-                content,
-                ContentType.TEXT_PLAIN,
-              );
-            },
+            payloadDebugger,
           },
         },
         {