From 0bdc1ebfe878c85482828ca38fa9afab1cade155 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Cl=C3=A9ment?= <clem.auger@hotmail.fr>
Date: Tue, 11 Jun 2024 09:04:22 +0200
Subject: [PATCH] fix: also convert simulation poll and group attr

---
 src/publicodes-state/helpers/migrateSimulation.ts | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/src/publicodes-state/helpers/migrateSimulation.ts b/src/publicodes-state/helpers/migrateSimulation.ts
index db1d5c238..ba43fd2c4 100644
--- a/src/publicodes-state/helpers/migrateSimulation.ts
+++ b/src/publicodes-state/helpers/migrateSimulation.ts
@@ -2,7 +2,7 @@ import { Migration, migrateSituation } from '@publicodes/tools/migration'
 import { Simulation } from '../types'
 
 export function migrateSimulation(
-  simulation: Simulation,
+  simulation: Simulation & { group?: string; poll?: string },
   migrationInstructions: Migration
 ): Simulation {
   simulation.situation = migrateSituation(
@@ -21,5 +21,15 @@ export function migrateSimulation(
     )
   )
 
+  // If group or poll is defined, we convert it to groups or polls and delete it
+  if (simulation.group) {
+    simulation.groups = [simulation.group]
+    delete simulation.group
+  }
+  if (simulation.poll) {
+    simulation.polls = [simulation.poll]
+    delete simulation.poll
+  }
+
   return simulation
 }