@finos/legend-graph 7.0.0 → 8.0.0

package/src/GraphManagerTestUtils.tsx

@@ -14,22 +14,23 @@
  * limitations under the License.
  */
 
-/// <reference types="jest-extended" />
+import { jest, expect } from '@jest/globals';
 import {
   type LoggerPlugin,
+  type TEMPORARRY__JestMatcher,
   Log,
   AbstractPluginManager,
   promisify,
 } from '@finos/legend-shared';
-import type { PureGraphManagerPlugin } from './graphManager/PureGraphManagerPlugin';
-import { GraphManagerState } from './GraphManagerState';
-import { GraphManagerStateProvider } from './GraphManagerStateProvider';
-import type { GraphPluginManager } from './GraphPluginManager';
-import type { PureProtocolProcessorPlugin } from './models/protocols/pure/PureProtocolProcessorPlugin';
+import type { PureGraphManagerPlugin } from './graphManager/PureGraphManagerPlugin.js';
+import { GraphManagerState } from './GraphManagerState.js';
+import { GraphManagerStateProvider } from './GraphManagerStateProvider.js';
+import type { GraphPluginManager } from './GraphPluginManager.js';
+import type { PureProtocolProcessorPlugin } from './models/protocols/pure/PureProtocolProcessorPlugin.js';
 import type { Entity } from '@finos/legend-model-storage';
-import { SECTION_INDEX_ELEMENT_PATH } from './MetaModelConst';
-import type { GraphBuilderOptions } from './graphManager/AbstractPureGraphManager';
-import type { PureGraphPlugin } from './graph/PureGraphPlugin';
+import { SECTION_INDEX_ELEMENT_PATH } from './MetaModelConst.js';
+import type { GraphBuilderOptions } from './graphManager/AbstractPureGraphManager.js';
+import type { PureGraphPlugin } from './graph/PureGraphPlugin.js';
 
 export class TEST__GraphPluginManager
   extends AbstractPluginManager
@@ -91,7 +92,7 @@ export const TEST__provideMockedGraphManagerState = (customization?: {
   const value =
     customization?.mock ??
     TEST__getTestGraphManagerState(customization?.pluginManager);
-  const MockedGraphManagerStateProvider = require('./GraphManagerStateProvider'); // eslint-disable-line @typescript-eslint/no-unsafe-assignment
+  const MockedGraphManagerStateProvider = require('./GraphManagerStateProvider.js'); // eslint-disable-line @typescript-eslint/no-unsafe-assignment
   MockedGraphManagerStateProvider.useGraphManagerState = jest.fn();
   MockedGraphManagerStateProvider.useGraphManagerState.mockReturnValue(value);
   return value;
@@ -194,10 +195,12 @@ export const TEST__checkGraphHashUnchanged = async (
       }),
     ),
   );
-  expect(
-    Array.from(originalHashesIndex.entries()).filter(
-      (entry) => entry[0] !== SECTION_INDEX_ELEMENT_PATH,
-    ),
+  (
+    expect(
+      Array.from(originalHashesIndex.entries()).filter(
+        (entry) => entry[0] !== SECTION_INDEX_ELEMENT_PATH,
+      ),
+    ) as TEMPORARRY__JestMatcher
   ).toIncludeSameMembers(
     Array.from(graphHashesIndex.entries()).filter(
       (entry) => entry[0] !== SECTION_INDEX_ELEMENT_PATH,
@@ -222,7 +225,7 @@ export const TEST__checkBuildingElementsRoundtrip = async (
     TEST__ensureObjectFieldsAreSortedAlphabetically(entity.content),
   );
   // check if the contents are the same (i.e. roundtrip test)
-  expect(transformedEntities).toIncludeSameMembers(
+  (expect(transformedEntities) as TEMPORARRY__JestMatcher).toIncludeSameMembers(
     TEST__excludeSectionIndex(entities),
   );
   await TEST__checkGraphHashUnchanged(graphManagerState, entities);
@@ -247,7 +250,7 @@ export const TEST__checkBuildingResolvedElements = async (
     TEST__ensureObjectFieldsAreSortedAlphabetically(entity.content),
   );
   // check if the contents are the same (i.e. roundtrip test)
-  expect(transformedEntities).toIncludeSameMembers(
+  (expect(transformedEntities) as TEMPORARRY__JestMatcher).toIncludeSameMembers(
     TEST__excludeSectionIndex(resolvedEntities),
   );
 };

package/src/GraphPluginManager.ts

@@ -15,9 +15,9 @@
  */
 
 import type { AbstractPluginManager } from '@finos/legend-shared';
-import type { PureGraphPlugin } from './graph/PureGraphPlugin';
-import type { PureGraphManagerPlugin } from './graphManager/PureGraphManagerPlugin';
-import type { PureProtocolProcessorPlugin } from './models/protocols/pure/PureProtocolProcessorPlugin';
+import type { PureGraphPlugin } from './graph/PureGraphPlugin.js';
+import type { PureGraphManagerPlugin } from './graphManager/PureGraphManagerPlugin.js';
+import type { PureProtocolProcessorPlugin } from './models/protocols/pure/PureProtocolProcessorPlugin.js';
 
 export interface GraphPluginManager extends AbstractPluginManager {
   getPureGraphPlugins(): PureGraphPlugin[];

package/src/MetaModelConst.ts

@@ -262,6 +262,7 @@ export enum CORE_HASH_STRUCTURE {
   GCP_APPLICATION_DEFAULT_CREDENTIALS_AUTHENTICATION_STRATEGY = 'GCP_APPLICATION_DEFAULT_CREDENTIALS_AUTHENTICATION_STRATEGY',
   USERNAME_PASSWORD_AUTHENTICATION_STRATEGY = 'USERNAME_PASSWORD_AUTHENTICATION_STRATEGY',
   OAUTH_AUTHENTICATION_STRATEGY = 'OAUTH_AUTHENTICATION_STRATEGY',
+  GCP_WORKLOAD_IDENTITY_FEDERATION_AUTHENTICATION_STRATEGY = 'GCP_WORKLOAD_IDENTITY_FEDERATION_AUTHENTICATION_STRATEGY',
   // relational database connection post processors
   MAPPER_POST_PROCESSOR = 'MAPPER_POST_PROCESSOR',
   SCHEMA_MAPPER = 'SCHEMA_MAPPER',
@@ -370,3 +371,28 @@ export enum PackageableElementPointerType {
   FILE_GENERATION = 'FILE_GENERATION',
   SERVICE = 'SERVICE',
 }
+
+export const DURATION_UNIT = 'meta::pure::functions::date::DurationUnit';
+export const DAY_OF_WEEK = 'meta::pure::functions::date::DayOfWeek';
+
+export enum SUPPORTED_FUNCTIONS {
+  // date-time value helper functions
+  TODAY = 'meta::pure::functions::date::today',
+  NOW = 'meta::pure::functions::date::now',
+  FIRST_DAY_OF_WEEK = 'meta::pure::functions::date::firstDayOfThisWeek',
+  FIRST_DAY_OF_MONTH = 'meta::pure::functions::date::firstDayOfThisMonth',
+  FIRST_DAY_OF_QUARTER = 'meta::pure::functions::date::firstDayOfThisQuarter',
+  FIRST_DAY_OF_YEAR = 'meta::pure::functions::date::firstDayOfThisYear',
+  PREVIOUS_DAY_OF_WEEK = 'meta::pure::functions::date::previousDayOfWeek',
+  IS_ON_DAY = 'meta::pure::functions::date::isOnDay',
+  IS_ON_OR_AFTER_DAY = 'meta::pure::functions::date::isOnOrAfterDay',
+  IS_AFTER_DAY = 'meta::pure::functions::date::isAfterDay',
+  IS_ON_OR_BEFORE_DAY = 'meta::pure::functions::date::isOnOrBeforeDay',
+  IS_BEFORE_DAY = 'meta::pure::functions::date::isBeforeDay',
+
+  // adjust time
+  MINUS = 'meta::pure::functions::math::minus',
+  ADJUST = 'meta::pure::functions::date::adjust',
+  // variables
+  LET = 'meta::pure::functions::lang::letFunction',
+}

package/src/MetaModelUtils.ts

@@ -27,13 +27,14 @@ import {
   SOURCE_INFORMATION_KEY,
   ELEMENT_PATH_DELIMITER,
   CORE_HASH_STRUCTURE,
-} from './MetaModelConst';
+} from './MetaModelConst.js';
 import {
   findLast,
   guaranteeNonNullable,
   hashArray,
   hashObject,
   hashString,
+  recursiveOmit,
 } from '@finos/legend-shared';
 
 export const extractElementNameFromPath = (fullPath: string): string =>
@@ -85,6 +86,20 @@ export const isValidPath = (path: string): boolean =>
 export const fromElementPathToMappingElementId = (className: string): string =>
   className.split(ELEMENT_PATH_DELIMITER).join('_');
 
+/**
+ * Prune source information from object such as raw lambda, raw value specification, etc.
+ *
+ * NOTE: currently, there is no exhaustive way to do this. Majority of the cases, the source information field
+ * is named `sourceInformation`, however, we have sometimes deviated from this pattern in our protocol model,
+ * so we have fields like `classSourceInformation`, etc. So this is really an optimistic, non-exhaustive prune.
+ * To do this exhaustively, we might need to make use of engine grammar API endpoints or do a full roundtrip
+ * raw object-protocol transformation to prune unrecognized fields, which include source information fields.
+ */
+export const pruneSourceInformation = (
+  object: Record<PropertyKey, unknown>,
+): Record<PropertyKey, unknown> =>
+  recursiveOmit(object, [SOURCE_INFORMATION_KEY]);
+
 // -------------------------------- HASHING -------------------------------------
 // TODO: this should be moved after we refactor `hashing` out of metamodels
 
@@ -93,7 +108,8 @@ export const hashObjectWithoutSourceInformation = (val: object): string =>
   hashObject(val, {
     excludeKeys: (key: string) => key === SOURCE_INFORMATION_KEY,
   });
-export const hashLambda = (
+
+export const hashRawLambda = (
   parameters: object | undefined,
   body: object | undefined,
 ): string =>

package/src/StoreFlatData_Exports.ts

@@ -15,20 +15,20 @@
  */
 
 // metamodels
-export { FlatData } from './models/metamodels/pure/packageableElements/store/flatData/model/FlatData';
-export { FlatDataSection } from './models/metamodels/pure/packageableElements/store/flatData/model/FlatDataSection';
+export { FlatData } from './models/metamodels/pure/packageableElements/store/flatData/model/FlatData.js';
+export { FlatDataSection } from './models/metamodels/pure/packageableElements/store/flatData/model/FlatDataSection.js';
 export {
   FlatDataSectionReference,
   FlatDataSectionExplicitReference,
-} from './models/metamodels/pure/packageableElements/store/flatData/model/FlatDataSectionReference';
-export { RootFlatDataRecordTypeExplicitReference } from './models/metamodels/pure/packageableElements/store/flatData/model/RootFlatDataRecordTypeReference';
-export { FlatDataInstanceSetImplementation } from './models/metamodels/pure/packageableElements/store/flatData/mapping/FlatDataInstanceSetImplementation';
-export { AbstractFlatDataPropertyMapping } from './models/metamodels/pure/packageableElements/store/flatData/mapping/AbstractFlatDataPropertyMapping';
-export { EmbeddedFlatDataPropertyMapping } from './models/metamodels/pure/packageableElements/store/flatData/mapping/EmbeddedFlatDataPropertyMapping';
-export { FlatDataPropertyMapping } from './models/metamodels/pure/packageableElements/store/flatData/mapping/FlatDataPropertyMapping';
-export { FlatDataConnection } from './models/metamodels/pure/packageableElements/store/flatData/connection/FlatDataConnection';
+} from './models/metamodels/pure/packageableElements/store/flatData/model/FlatDataSectionReference.js';
+export { RootFlatDataRecordTypeExplicitReference } from './models/metamodels/pure/packageableElements/store/flatData/model/RootFlatDataRecordTypeReference.js';
+export { FlatDataInstanceSetImplementation } from './models/metamodels/pure/packageableElements/store/flatData/mapping/FlatDataInstanceSetImplementation.js';
+export { AbstractFlatDataPropertyMapping } from './models/metamodels/pure/packageableElements/store/flatData/mapping/AbstractFlatDataPropertyMapping.js';
+export { EmbeddedFlatDataPropertyMapping } from './models/metamodels/pure/packageableElements/store/flatData/mapping/EmbeddedFlatDataPropertyMapping.js';
+export { FlatDataPropertyMapping } from './models/metamodels/pure/packageableElements/store/flatData/mapping/FlatDataPropertyMapping.js';
+export { FlatDataConnection } from './models/metamodels/pure/packageableElements/store/flatData/connection/FlatDataConnection.js';
 export {
   RootFlatDataRecordType,
   FlatDataRecordField,
-} from './models/metamodels/pure/packageableElements/store/flatData/model/FlatDataDataType';
-export { FlatDataInputData } from './models/metamodels/pure/packageableElements/store/flatData/mapping/FlatDataInputData';
+} from './models/metamodels/pure/packageableElements/store/flatData/model/FlatDataDataType.js';
+export { FlatDataInputData } from './models/metamodels/pure/packageableElements/store/flatData/mapping/FlatDataInputData.js';

package/src/StoreRelational_Exports.ts

@@ -15,65 +15,65 @@
  */
 
 // metamodels
-export { Database } from './models/metamodels/pure/packageableElements/store/relational/model/Database';
-export { Table } from './models/metamodels/pure/packageableElements/store/relational/model/Table';
-export { Column } from './models/metamodels/pure/packageableElements/store/relational/model/Column';
-export { Schema } from './models/metamodels/pure/packageableElements/store/relational/model/Schema';
-export { View } from './models/metamodels/pure/packageableElements/store/relational/model/View';
-export { Join } from './models/metamodels/pure/packageableElements/store/relational/model/Join';
+export { Database } from './models/metamodels/pure/packageableElements/store/relational/model/Database.js';
+export { Table } from './models/metamodels/pure/packageableElements/store/relational/model/Table.js';
+export { Column } from './models/metamodels/pure/packageableElements/store/relational/model/Column.js';
+export { Schema } from './models/metamodels/pure/packageableElements/store/relational/model/Schema.js';
+export { View } from './models/metamodels/pure/packageableElements/store/relational/model/View.js';
+export { Join } from './models/metamodels/pure/packageableElements/store/relational/model/Join.js';
 export {
   ViewReference,
   ViewExplicitReference,
-} from './models/metamodels/pure/packageableElements/store/relational/model/ViewReference';
+} from './models/metamodels/pure/packageableElements/store/relational/model/ViewReference.js';
 export {
   TableReference,
   TableExplicitReference,
-} from './models/metamodels/pure/packageableElements/store/relational/model/TableReference';
+} from './models/metamodels/pure/packageableElements/store/relational/model/TableReference.js';
 export {
   ColumnReference,
   ColumnExplicitReference,
-} from './models/metamodels/pure/packageableElements/store/relational/model/ColumnReference';
-export { type Mapper } from './models/metamodels/pure/packageableElements/store/relational/connection/postprocessor/Mapper';
-export { PostProcessor } from './models/metamodels/pure/packageableElements/store/relational/connection/postprocessor/PostProcessor';
-export { Milestoning } from './models/metamodels/pure/packageableElements/store/relational/model/milestoning/Milestoning';
-export * from './models/metamodels/pure/packageableElements/store/relational/model/RelationalDataType';
-export * from './models/metamodels/pure/packageableElements/store/relational/model/RelationalOperationElement';
-export * from './models/metamodels/pure/packageableElements/store/relational/model/RawRelationalOperationElement';
+} from './models/metamodels/pure/packageableElements/store/relational/model/ColumnReference.js';
+export { type Mapper } from './models/metamodels/pure/packageableElements/store/relational/connection/postprocessor/Mapper.js';
+export { PostProcessor } from './models/metamodels/pure/packageableElements/store/relational/connection/postprocessor/PostProcessor.js';
+export { Milestoning } from './models/metamodels/pure/packageableElements/store/relational/model/milestoning/Milestoning.js';
+export * from './models/metamodels/pure/packageableElements/store/relational/model/RelationalDataType.js';
+export * from './models/metamodels/pure/packageableElements/store/relational/model/RelationalOperationElement.js';
+export * from './models/metamodels/pure/packageableElements/store/relational/model/RawRelationalOperationElement.js';
 export {
   RelationalInputData,
   RelationalInputType,
-} from './models/metamodels/pure/packageableElements/store/relational/mapping/RelationalInputData';
-export { RelationalPropertyMapping } from './models/metamodels/pure/packageableElements/store/relational/mapping/RelationalPropertyMapping';
-export { RelationalInstanceSetImplementation } from './models/metamodels/pure/packageableElements/store/relational/mapping/RelationalInstanceSetImplementation';
-export { RootRelationalInstanceSetImplementation } from './models/metamodels/pure/packageableElements/store/relational/mapping/RootRelationalInstanceSetImplementation';
-export { EmbeddedRelationalInstanceSetImplementation } from './models/metamodels/pure/packageableElements/store/relational/mapping/EmbeddedRelationalInstanceSetImplementation';
-export { OtherwiseEmbeddedRelationalInstanceSetImplementation } from './models/metamodels/pure/packageableElements/store/relational/mapping/OtherwiseEmbeddedRelationalInstanceSetImplementation';
+} from './models/metamodels/pure/packageableElements/store/relational/mapping/RelationalInputData.js';
+export { RelationalPropertyMapping } from './models/metamodels/pure/packageableElements/store/relational/mapping/RelationalPropertyMapping.js';
+export { RelationalInstanceSetImplementation } from './models/metamodels/pure/packageableElements/store/relational/mapping/RelationalInstanceSetImplementation.js';
+export { RootRelationalInstanceSetImplementation } from './models/metamodels/pure/packageableElements/store/relational/mapping/RootRelationalInstanceSetImplementation.js';
+export { EmbeddedRelationalInstanceSetImplementation } from './models/metamodels/pure/packageableElements/store/relational/mapping/EmbeddedRelationalInstanceSetImplementation.js';
+export { OtherwiseEmbeddedRelationalInstanceSetImplementation } from './models/metamodels/pure/packageableElements/store/relational/mapping/OtherwiseEmbeddedRelationalInstanceSetImplementation.js';
 export {
   DatabaseConnection,
   DatabaseType,
   RelationalDatabaseConnection,
-} from './models/metamodels/pure/packageableElements/store/relational/connection/RelationalDatabaseConnection';
-export * from './models/metamodels/pure/packageableElements/store/relational/connection/AuthenticationStrategy';
-export * from './models/metamodels/pure/packageableElements/store/relational/connection/DatasourceSpecification';
+} from './models/metamodels/pure/packageableElements/store/relational/connection/RelationalDatabaseConnection.js';
+export * from './models/metamodels/pure/packageableElements/store/relational/connection/AuthenticationStrategy.js';
+export * from './models/metamodels/pure/packageableElements/store/relational/connection/DatasourceSpecification.js';
 
 // protocols
-export * from './models/protocols/pure/StoreRelational_PureProtocolProcessorPlugin_Extension';
-export * from './models/protocols/pure/v1/model/packageableElements/store/relational/model/V1_TablePtr';
-export { V1_transformTableToTablePointer } from './models/protocols/pure/v1/transformation/pureGraph/from/V1_DatabaseTransformer';
+export * from './models/protocols/pure/StoreRelational_PureProtocolProcessorPlugin_Extension.js';
+export * from './models/protocols/pure/v1/model/packageableElements/store/relational/model/V1_TablePtr.js';
+export { V1_transformTableToTablePointer } from './models/protocols/pure/v1/transformation/pureGraph/from/V1_DatabaseTransformer.js';
 export {
   V1_transformMapper,
   V1_transformRelation,
-} from './models/protocols/pure/v1/transformation/pureGraph/from/V1_PostProcessorTransformer';
-export { V1_getRelation } from './models/protocols/pure/v1/transformation/pureGraph/to/helpers/V1_DatabaseBuilderHelper';
-export { V1_tablePtrModelSchema } from './models/protocols/pure/v1/transformation/pureProtocol/serializationHelpers/V1_DatabaseSerializationHelper';
-export { type V1_Mapper } from './models/protocols/pure/v1/model/packageableElements/store/relational/connection/postprocessor/V1_Mapper';
-export { V1_PostProcessor } from './models/protocols/pure/v1/model/packageableElements/store/relational/connection/postprocessor/V1_PostProcessor';
-export { V1_DatasourceSpecification } from './models/protocols/pure/v1/model/packageableElements/store/relational/connection/V1_DatasourceSpecification';
-export { V1_AuthenticationStrategy } from './models/protocols/pure/v1/model/packageableElements/store/relational/connection/V1_AuthenticationStrategy';
-export { V1_Milestoning } from './models/protocols/pure/v1/model/packageableElements/store/relational/model/milestoning/V1_Milestoning';
-export { V1_buildMapper } from './models/protocols/pure/v1/transformation/pureGraph/to/helpers/V1_PostProcessorBuilderHelper';
+} from './models/protocols/pure/v1/transformation/pureGraph/from/V1_PostProcessorTransformer.js';
+export { V1_getRelation } from './models/protocols/pure/v1/transformation/pureGraph/to/helpers/V1_DatabaseBuilderHelper.js';
+export { V1_tablePtrModelSchema } from './models/protocols/pure/v1/transformation/pureProtocol/serializationHelpers/V1_DatabaseSerializationHelper.js';
+export { type V1_Mapper } from './models/protocols/pure/v1/model/packageableElements/store/relational/connection/postprocessor/V1_Mapper.js';
+export { V1_PostProcessor } from './models/protocols/pure/v1/model/packageableElements/store/relational/connection/postprocessor/V1_PostProcessor.js';
+export { V1_DatasourceSpecification } from './models/protocols/pure/v1/model/packageableElements/store/relational/connection/V1_DatasourceSpecification.js';
+export { V1_AuthenticationStrategy } from './models/protocols/pure/v1/model/packageableElements/store/relational/connection/V1_AuthenticationStrategy.js';
+export { V1_Milestoning } from './models/protocols/pure/v1/model/packageableElements/store/relational/model/milestoning/V1_Milestoning.js';
+export { V1_buildMapper } from './models/protocols/pure/v1/transformation/pureGraph/to/helpers/V1_PostProcessorBuilderHelper.js';
 export {
   V1_serializeMapper,
   V1_deserializeMapper,
-} from './models/protocols/pure/v1/transformation/pureProtocol/serializationHelpers/V1_PostProcessorSerializationHelper';
-export * from './graphManager/StoreRelational_PureGraphManagerPlugin_Extension';
+} from './models/protocols/pure/v1/transformation/pureProtocol/serializationHelpers/V1_PostProcessorSerializationHelper.js';
+export * from './graphManager/StoreRelational_PureGraphManagerPlugin_Extension.js';

package/src/graph/BasicModel.ts

@@ -25,46 +25,46 @@ import {
   guaranteeNonNullable,
   guaranteeType,
 } from '@finos/legend-shared';
-import type { ROOT_PACKAGE_NAME } from '../MetaModelConst';
-import { Package } from '../models/metamodels/pure/packageableElements/domain/Package';
-import { Type } from '../models/metamodels/pure/packageableElements/domain/Type';
-import { Association } from '../models/metamodels/pure/packageableElements/domain/Association';
-import { Mapping } from '../models/metamodels/pure/packageableElements/mapping/Mapping';
-import { Class } from '../models/metamodels/pure/packageableElements/domain/Class';
-import { Enumeration } from '../models/metamodels/pure/packageableElements/domain/Enumeration';
-import { PackageableElement } from '../models/metamodels/pure/packageableElements/PackageableElement';
-import { Profile } from '../models/metamodels/pure/packageableElements/domain/Profile';
-import { Service } from '../models/metamodels/pure/packageableElements/service/Service';
-import { ConcreteFunctionDefinition } from '../models/metamodels/pure/packageableElements/domain/ConcreteFunctionDefinition';
-import { Store } from '../models/metamodels/pure/packageableElements/store/Store';
-import { FlatData } from '../models/metamodels/pure/packageableElements/store/flatData/model/FlatData';
-import { PackageableRuntime } from '../models/metamodels/pure/packageableElements/runtime/PackageableRuntime';
-import { PackageableConnection } from '../models/metamodels/pure/packageableElements/connection/PackageableConnection';
-import { FileGenerationSpecification } from '../models/metamodels/pure/packageableElements/fileGeneration/FileGenerationSpecification';
-import { GenerationSpecification } from '../models/metamodels/pure/packageableElements/generationSpecification/GenerationSpecification';
+import type { ROOT_PACKAGE_NAME } from '../MetaModelConst.js';
+import { Package } from '../models/metamodels/pure/packageableElements/domain/Package.js';
+import { Type } from '../models/metamodels/pure/packageableElements/domain/Type.js';
+import { Association } from '../models/metamodels/pure/packageableElements/domain/Association.js';
+import { Mapping } from '../models/metamodels/pure/packageableElements/mapping/Mapping.js';
+import { Class } from '../models/metamodels/pure/packageableElements/domain/Class.js';
+import { Enumeration } from '../models/metamodels/pure/packageableElements/domain/Enumeration.js';
+import { PackageableElement } from '../models/metamodels/pure/packageableElements/PackageableElement.js';
+import { Profile } from '../models/metamodels/pure/packageableElements/domain/Profile.js';
+import { Service } from '../models/metamodels/pure/packageableElements/service/Service.js';
+import { ConcreteFunctionDefinition } from '../models/metamodels/pure/packageableElements/domain/ConcreteFunctionDefinition.js';
+import { Store } from '../models/metamodels/pure/packageableElements/store/Store.js';
+import { FlatData } from '../models/metamodels/pure/packageableElements/store/flatData/model/FlatData.js';
+import { PackageableRuntime } from '../models/metamodels/pure/packageableElements/runtime/PackageableRuntime.js';
+import { PackageableConnection } from '../models/metamodels/pure/packageableElements/connection/PackageableConnection.js';
+import { FileGenerationSpecification } from '../models/metamodels/pure/packageableElements/fileGeneration/FileGenerationSpecification.js';
+import { GenerationSpecification } from '../models/metamodels/pure/packageableElements/generationSpecification/GenerationSpecification.js';
 import {
   Unit,
   Measure,
-} from '../models/metamodels/pure/packageableElements/domain/Measure';
-import { Database } from '../models/metamodels/pure/packageableElements/store/relational/model/Database';
-import { SectionIndex } from '../models/metamodels/pure/packageableElements/section/SectionIndex';
-import type { Section } from '../models/metamodels/pure/packageableElements/section/Section';
-import { PureGraphExtension } from './PureGraphExtension';
-import { PrimitiveType } from '../models/metamodels/pure/packageableElements/domain/PrimitiveType';
-import { DataType } from '../models/metamodels/pure/packageableElements/domain/DataType';
+} from '../models/metamodels/pure/packageableElements/domain/Measure.js';
+import { Database } from '../models/metamodels/pure/packageableElements/store/relational/model/Database.js';
+import { SectionIndex } from '../models/metamodels/pure/packageableElements/section/SectionIndex.js';
+import type { Section } from '../models/metamodels/pure/packageableElements/section/Section.js';
+import { PureGraphExtension } from './PureGraphExtension.js';
+import { PrimitiveType } from '../models/metamodels/pure/packageableElements/domain/PrimitiveType.js';
+import { DataType } from '../models/metamodels/pure/packageableElements/domain/DataType.js';
 import {
   isValidFullPath,
   isValidPath,
   resolvePackagePathAndElementName,
-} from '../MetaModelUtils';
+} from '../MetaModelUtils.js';
 import {
   addElementToPackage,
   deleteElementFromPackage,
   getOrCreateGraphPackage,
   getOrCreatePackage,
-} from '../helpers/DomainHelper';
-import { DataElement } from '../models/metamodels/pure/packageableElements/data/DataElement';
-import type { Testable } from '../models/metamodels/pure/test/Testable';
+} from '../helpers/DomainHelper.js';
+import { DataElement } from '../models/metamodels/pure/packageableElements/data/DataElement.js';
+import type { Testable } from '../models/metamodels/pure/test/Testable.js';
 
 const FORBIDDEN_EXTENSION_ELEMENT_CLASS = new Set([
   PackageableElement,

package/src/graph/DSLExternalFormat_Extension.ts

@@ -16,22 +16,20 @@
 
 import packageJson from '../../package.json';
 import { AbstractPreset } from '@finos/legend-shared';
-import { DSLExternalFormat_PureGraphManagerPlugin } from '../graphManager/DSLExternalFormat_PureGraphManagerPlugin';
-import type { GraphPluginManager } from '../GraphPluginManager';
-import { DSLExternalFormat_PureProtocolProcessorPlugin } from '../models/protocols/pure/DSLExternalFormat_PureProtocolProcessorPlugin';
-import { DSLExternalFormat_PureGraphPlugin } from './DSLExternalFormat_PureGraphPlugin';
+import { DSLExternalFormat_PureGraphManagerPlugin } from '../graphManager/DSLExternalFormat_PureGraphManagerPlugin.js';
+import { DSLExternalFormat_PureProtocolProcessorPlugin } from '../models/protocols/pure/DSLExternalFormat_PureProtocolProcessorPlugin.js';
+import { DSLExternalFormat_PureGraphPlugin } from './DSLExternalFormat_PureGraphPlugin.js';
 
 export class DSLExternalFormat_GraphPreset extends AbstractPreset {
   constructor() {
     super(
       packageJson.extensions.dsl_external_format_graphPreset,
       packageJson.version,
+      [
+        new DSLExternalFormat_PureGraphPlugin(),
+        new DSLExternalFormat_PureGraphManagerPlugin(),
+        new DSLExternalFormat_PureProtocolProcessorPlugin(),
+      ],
     );
   }
-
-  install(pluginManager: GraphPluginManager): void {
-    new DSLExternalFormat_PureGraphPlugin().install(pluginManager);
-    new DSLExternalFormat_PureGraphManagerPlugin().install(pluginManager);
-    new DSLExternalFormat_PureProtocolProcessorPlugin().install(pluginManager);
-  }
 }

package/src/graph/DSLExternalFormat_PureGraphPlugin.ts

@@ -16,10 +16,10 @@
 
 import packageJson from '../../package.json';
 import type { Clazz } from '@finos/legend-shared';
-import type { PackageableElement } from '../models/metamodels/pure/packageableElements/PackageableElement';
-import { PureGraphPlugin } from './PureGraphPlugin';
-import { Binding } from '../models/metamodels/pure/packageableElements/externalFormat/store/DSLExternalFormat_Binding';
-import { SchemaSet } from '../models/metamodels/pure/packageableElements/externalFormat/schemaSet/DSLExternalFormat_SchemaSet';
+import type { PackageableElement } from '../models/metamodels/pure/packageableElements/PackageableElement.js';
+import { PureGraphPlugin } from './PureGraphPlugin.js';
+import { Binding } from '../models/metamodels/pure/packageableElements/externalFormat/store/DSLExternalFormat_Binding.js';
+import { SchemaSet } from '../models/metamodels/pure/packageableElements/externalFormat/schemaSet/DSLExternalFormat_SchemaSet.js';
 
 export class DSLExternalFormat_PureGraphPlugin extends PureGraphPlugin {
   constructor() {

package/src/graph/DependencyManager.ts

@@ -14,29 +14,29 @@
  * limitations under the License.
  */
 
-import { ROOT_PACKAGE_NAME } from '../MetaModelConst';
+import { ROOT_PACKAGE_NAME } from '../MetaModelConst.js';
 import { type Clazz, guaranteeNonNullable } from '@finos/legend-shared';
-import type { PackageableElement } from '../models/metamodels/pure/packageableElements/PackageableElement';
-import type { Enumeration } from '../models/metamodels/pure/packageableElements/domain/Enumeration';
-import type { Type } from '../models/metamodels/pure/packageableElements/domain/Type';
-import type { Class } from '../models/metamodels/pure/packageableElements/domain/Class';
-import type { Mapping } from '../models/metamodels/pure/packageableElements/mapping/Mapping';
-import type { Profile } from '../models/metamodels/pure/packageableElements/domain/Profile';
-import { Package } from '../models/metamodels/pure/packageableElements/domain/Package';
-import type { ConcreteFunctionDefinition } from '../models/metamodels/pure/packageableElements/domain/ConcreteFunctionDefinition';
-import type { Store } from '../models/metamodels/pure/packageableElements/store/Store';
-import type { Association } from '../models/metamodels/pure/packageableElements/domain/Association';
-import type { Service } from '../models/metamodels/pure/packageableElements/service/Service';
-import { BasicModel } from '../graph/BasicModel';
-import type { PackageableRuntime } from '../models/metamodels/pure/packageableElements/runtime/PackageableRuntime';
-import type { PackageableConnection } from '../models/metamodels/pure/packageableElements/connection/PackageableConnection';
-import type { FileGenerationSpecification } from '../models/metamodels/pure/packageableElements/fileGeneration/FileGenerationSpecification';
-import type { GenerationSpecification } from '../models/metamodels/pure/packageableElements/generationSpecification/GenerationSpecification';
-import type { Measure } from '../models/metamodels/pure/packageableElements/domain/Measure';
-import type { SectionIndex } from '../models/metamodels/pure/packageableElements/section/SectionIndex';
+import type { PackageableElement } from '../models/metamodels/pure/packageableElements/PackageableElement.js';
+import type { Enumeration } from '../models/metamodels/pure/packageableElements/domain/Enumeration.js';
+import type { Type } from '../models/metamodels/pure/packageableElements/domain/Type.js';
+import type { Class } from '../models/metamodels/pure/packageableElements/domain/Class.js';
+import type { Mapping } from '../models/metamodels/pure/packageableElements/mapping/Mapping.js';
+import type { Profile } from '../models/metamodels/pure/packageableElements/domain/Profile.js';
+import { Package } from '../models/metamodels/pure/packageableElements/domain/Package.js';
+import type { ConcreteFunctionDefinition } from '../models/metamodels/pure/packageableElements/domain/ConcreteFunctionDefinition.js';
+import type { Store } from '../models/metamodels/pure/packageableElements/store/Store.js';
+import type { Association } from '../models/metamodels/pure/packageableElements/domain/Association.js';
+import type { Service } from '../models/metamodels/pure/packageableElements/service/Service.js';
+import { BasicModel } from '../graph/BasicModel.js';
+import type { PackageableRuntime } from '../models/metamodels/pure/packageableElements/runtime/PackageableRuntime.js';
+import type { PackageableConnection } from '../models/metamodels/pure/packageableElements/connection/PackageableConnection.js';
+import type { FileGenerationSpecification } from '../models/metamodels/pure/packageableElements/fileGeneration/FileGenerationSpecification.js';
+import type { GenerationSpecification } from '../models/metamodels/pure/packageableElements/generationSpecification/GenerationSpecification.js';
+import type { Measure } from '../models/metamodels/pure/packageableElements/domain/Measure.js';
+import type { SectionIndex } from '../models/metamodels/pure/packageableElements/section/SectionIndex.js';
 import type { Entity } from '@finos/legend-model-storage';
-import type { Database } from '../models/metamodels/pure/packageableElements/store/relational/model/Database';
-import type { DataElement } from '../models/metamodels/pure/packageableElements/data/DataElement';
+import type { Database } from '../models/metamodels/pure/packageableElements/store/relational/model/Database.js';
+import type { DataElement } from '../models/metamodels/pure/packageableElements/data/DataElement.js';
 
 class DependencyModel extends BasicModel {
   constructor(

package/src/graph/PureGraphExtension.ts

@@ -15,7 +15,7 @@
  */
 
 import type { Clazz } from '@finos/legend-shared';
-import type { PackageableElement } from '../models/metamodels/pure/packageableElements/PackageableElement';
+import type { PackageableElement } from '../models/metamodels/pure/packageableElements/PackageableElement.js';
 
 export class PureGraphExtension<T extends PackageableElement> {
   private readonly _class: Clazz<T>;

package/src/graph/PureGraphPlugin.ts

@@ -15,10 +15,10 @@
  */
 
 import { AbstractPlugin, type Clazz } from '@finos/legend-shared';
-import type { PackageableElement } from '../models/metamodels/pure/packageableElements/PackageableElement';
-import type { PureModel } from './PureModel';
-import type { GraphPluginManager } from '../GraphPluginManager';
-import type { Testable } from '../models/metamodels/pure/test/Testable';
+import type { PackageableElement } from '../models/metamodels/pure/packageableElements/PackageableElement.js';
+import type { PureModel } from './PureModel.js';
+import type { GraphPluginManager } from '../GraphPluginManager.js';
+import type { Testable } from '../models/metamodels/pure/test/Testable.js';
 
 export type DeadReferencesCleaner = (graph: PureModel) => void;
 export type TestablesCollector = (graph: PureModel) => Testable[];