export stable id of cost surface and map this to unique id of copy of…

… each cost surface on import [MRXN23-606]
Vizzuality · Mar 10, 2024 · 2e1137e · 2e1137e
1 parent 8c4e665
commit 2e1137e
Show file tree

Hide file tree

Showing 2 changed files with 37 additions and 2 deletions.
diff --git a/api/apps/geoprocessing/src/export/pieces-exporters/scenario-metadata.piece-exporter.ts b/api/apps/geoprocessing/src/export/pieces-exporters/scenario-metadata.piece-exporter.ts
@@ -71,6 +71,16 @@ export class ScenarioMetadataPieceExporter implements ExportPieceProcessor {
       .where('s.id = :scenarioId', { scenarioId })
       .execute();
 
+    const scenarioCostSurfaceStableId: string = await this.entityManager
+      .createQueryBuilder()
+      .select(['stable_id'])
+      .from('cost_surfaces', 'cs')
+      .where('cs.id = :costSurfaceId', {
+        costSurfaceId: scenario.cost_surface_id,
+      })
+      .execute()
+      .then((result: { stable_id: string }[]) => result[0]?.stable_id);
+
     if (!scenario) {
       const errorMessage = `${ScenarioMetadataPieceExporter.name} - Scenario ${scenarioId} does not exist.`;
       this.logger.error(errorMessage);
@@ -101,7 +111,7 @@ export class ScenarioMetadataPieceExporter implements ExportPieceProcessor {
       solutionsAreLocked: scenario.solutions_are_locked,
       type: scenario.type,
       status: scenario.status ?? undefined,
-      cost_surface_id: scenario.cost_surface_id,
+      cost_surface_id: scenarioCostSurfaceStableId,
     };
 
     const relativePath = ClonePieceRelativePathResolver.resolveFor(

diff --git a/api/apps/geoprocessing/src/import/pieces-importers/scenario-metadata.piece-importer.ts b/api/apps/geoprocessing/src/import/pieces-importers/scenario-metadata.piece-importer.ts
@@ -84,6 +84,21 @@ export class ScenarioMetadataPieceImporter implements ImportPieceProcessor {
       .execute();
   }
 
+  private async mapCostSurfaceStableIdToIdOfClonedCostSurface(
+    em: EntityManager,
+    costSurfaceId: string,
+    projectId: string,
+  ): Promise<string> {
+    return await em
+      .createQueryBuilder()
+      .select('id')
+      .from('cost_surfaces', 'cs')
+      .where('cs.stable_id = :costSurfaceId', { costSurfaceId })
+      .andWhere('cs.project_id = :projectId', { projectId })
+      .execute()
+      .then((result) => result[0]?.id);
+  }
+
   async run(input: ImportJobInput): Promise<ImportJobOutput> {
     const {
       pieceResourceId: scenarioId,
@@ -121,6 +136,13 @@ export class ScenarioMetadataPieceImporter implements ImportPieceProcessor {
       const scenarioCloning = resourceKind === ResourceKind.Scenario;
 
       await this.entityManager.transaction(async (em) => {
+        const idOfClonedCostSurfaceLinkedToScenario =
+          await this.mapCostSurfaceStableIdToIdOfClonedCostSurface(
+            em,
+            metadata.cost_surface_id,
+            projectId,
+          );
+
         if (scenarioCloning) {
           await this.updateScenario(em, scenarioId, metadata, input.ownerId);
         } else {
@@ -135,7 +157,10 @@ export class ScenarioMetadataPieceImporter implements ImportPieceProcessor {
             em,
             scenarioId,
             projectId,
-            metadata,
+            {
+              ...metadata,
+              cost_surface_id: idOfClonedCostSurfaceLinkedToScenario,
+            },
             input.ownerId,
           );
         }