@finos/legend-graph 32.1.33 → 32.1.34

package/src/graph/BasicModel.ts

@@ -76,6 +76,7 @@ import { INTERNAL__UnknownStore } from './metamodel/pure/packageableElements/sto
 import type { PureGraphPlugin } from './PureGraphPlugin.js';
 import { INTERNAL__UnknownElement } from './metamodel/pure/packageableElements/INTERNAL__UnknownElement.js';
 import { DataProduct } from './metamodel/pure/dataProduct/DataProduct.js';
+import { IngestDefinition } from './metamodel/pure/packageableElements/ingest/IngestDefinition.js';
 
 const FORBIDDEN_EXTENSION_ELEMENT_CLASS = new Set([
   PackageableElement,
@@ -255,6 +256,12 @@ export abstract class BasicModel {
   get ownExecutionEnvironments(): ExecutionEnvironmentInstance[] {
     return Array.from(this.executionEnvironmentsIndex.values());
   }
+  get ownIngests(): IngestDefinition[] {
+    return Array.from(this.INTERNAL__unknownElementsIndex.values()).filter(
+      filterByType(IngestDefinition),
+    );
+  }
+
   get ownGenerationSpecifications(): GenerationSpecification[] {
     return Array.from(this.generationSpecificationsIndex.values());
   }

package/src/graph/DependencyManager.ts

@@ -46,6 +46,7 @@ import { LegendSDLC, type GraphDataOrigin } from './GraphDataOrigin.js';
 import type { FunctionActivator } from './metamodel/pure/packageableElements/function/FunctionActivator.js';
 import type { PureGraphPlugin } from './PureGraphPlugin.js';
 import type { Testable } from './metamodel/pure/test/Testable.js';
+import type { IngestDefinition } from './metamodel/pure/packageableElements/ingest/IngestDefinition.js';
 
 export const DEPENDENCY_ROOT_PACKAGE_PREFIX = '@dependency__';
 export const generateDependencyRootPackageName = (
@@ -306,6 +307,9 @@ export class DependencyManager {
   get executionEnvironments(): ExecutionEnvironmentInstance[] {
     return this.dependencyGraphs.flatMap((dep) => dep.ownExecutionEnvironments);
   }
+  get ingests(): IngestDefinition[] {
+    return this.dependencyGraphs.flatMap((dep) => dep.ownIngests);
+  }
 
   getExtensionElements<T extends PackageableElement>(
     extensionElementClass: Clazz<T>,

package/src/graph/PureModel.ts

@@ -61,6 +61,7 @@ import type { SectionIndex } from '../graph/metamodel/pure/packageableElements/s
 import type { PropertyOwner } from './metamodel/pure/packageableElements/domain/AbstractProperty.js';
 import type { ExecutionEnvironmentInstance } from './metamodel/pure/packageableElements/service/ExecutionEnvironmentInstance.js';
 import { FunctionActivator } from './metamodel/pure/packageableElements/function/FunctionActivator.js';
+import type { IngestDefinition } from './metamodel/pure/packageableElements/ingest/IngestDefinition.js';
 
 export interface GraphTextInputOption {
   graphGrammar: string | undefined;
@@ -353,6 +354,16 @@ export class PureModel extends BasicModel {
     ];
   }
 
+  get ingests(): IngestDefinition[] {
+    return [
+      ...this.coreModel.ownIngests,
+      ...this.systemModel.ownIngests,
+      ...this.dependencyManager.ingests,
+      ...this.ownIngests,
+      ...this.generationModel.ownIngests,
+    ];
+  }
+
   get allElements(): PackageableElement[] {
     return [
       ...this.coreModel.allOwnElements,

package/src/graph-manager/AbstractPureGraphManager.ts

@@ -794,6 +794,13 @@ export abstract class AbstractPureGraphManager {
     },
   ): Entity;
 
+  abstract elementsToPureCode(
+    elements: PackageableElement[],
+    options?: {
+      pruneSourceInformation?: boolean;
+    },
+  ): Promise<string>;
+
   async createBasicGraph(options?: {
     initializeSystem?: boolean;
   }): Promise<PureModel> {

package/src/graph-manager/protocol/pure/v1/V1_PureGraphManager.ts

@@ -4224,6 +4224,23 @@ export class V1_PureGraphManager extends AbstractPureGraphManager {
     return entity;
   };
 
+  override async elementsToPureCode(
+    elements: PackageableElement[],
+    options?: { pruneSourceInformation?: boolean; pretty?: boolean },
+  ): Promise<string> {
+    const graphData = new V1_PureModelContextData();
+    graphData.elements = elements.map((element) =>
+      this.elementToProtocol(element, {
+        keepSourceInformation: !options?.pruneSourceInformation,
+      }),
+    );
+    const jsonToGrammar = await this.engine.transformPureModelContextDataToCode(
+      graphData,
+      Boolean(options?.pretty),
+    );
+    return jsonToGrammar;
+  }
+
   private prunePureModelContextData = (
     data: V1_PureModelContextData,
     elementFilter?: (val: V1_PackageableElement) => boolean,