refactor: use the new refactored migrateSituation from @publicodes/tools

src/app/_components/MainLayoutProviders.tsx

@@ -2,7 +2,8 @@
 
 import { IframeOptionsProvider } from '@/app/_components/mainLayoutProviders/IframeOptionsContext'
 import { UserProvider } from '@/publicodes-state'
-import { MigrationType, RegionFromGeolocation } from '@/publicodes-state/types'
+import { RegionFromGeolocation } from '@/publicodes-state/types'
+import { Migration } from '@publicodes/tools/migration'
 import { PropsWithChildren } from 'react'
 import MainHooks from './mainLayoutProviders/MainHooks'
 import { PreventNavigationProvider } from './mainLayoutProviders/PreventNavigationProvider'
@@ -11,7 +12,7 @@ import SimulationSyncProvider from './mainLayoutProviders/SimulationSyncProvider
 
 type Props = {
   region: RegionFromGeolocation
-  migrationInstructions: MigrationType
+  migrationInstructions: Migration
 }
 export default function MainLayoutProviders({
   children,

src/helpers/modelFetching/getMigrationInstructions.ts

@@ -1,5 +1,5 @@
-import { MigrationType } from '@/publicodes-state/types'
 import migration from '@incubateur-ademe/nosgestesclimat/public/migration.json'
+import { Migration } from '@publicodes/tools/migration'
 import { importPreviewFile } from './importPreviewFile'
 
 type Props = {
@@ -10,7 +10,7 @@ type Props = {
  */
 export async function getMigrationInstructions({
   PRNumber,
-}: Props = {}): Promise<MigrationType> {
+}: Props = {}): Promise<Migration> {
   if (PRNumber) {
     const fileName = `migration.json`
     return importPreviewFile({ fileName, PRNumber })

src/helpers/simulation/generateSimulation.ts

@@ -1,5 +1,6 @@
 import { migrateSimulation } from '@/publicodes-state/helpers/migrateSimulation'
-import { MigrationType, Simulation } from '@/publicodes-state/types'
+import { Simulation } from '@/publicodes-state/types'
+import { Migration } from '@publicodes/tools/migration'
 import { captureException } from '@sentry/react'
 import { v4 as uuidv4 } from 'uuid'
 
@@ -18,7 +19,7 @@ export function generateSimulation({
   savedViaEmail,
   migrationInstructions,
 }: Partial<Simulation> & {
-  migrationInstructions?: MigrationType
+  migrationInstructions?: Migration
 } = {}): Simulation {
   let simulation = {
     id,
@@ -37,10 +38,7 @@ export function generateSimulation({
 
   if (migrationInstructions) {
     try {
-      simulation = migrateSimulation({
-        simulation,
-        migrationInstructions,
-      })
+      simulation = migrateSimulation(simulation, migrationInstructions)
     } catch (error) {
       console.warn('Error trying to migrate LocalStorage:', error)
       captureException(error)

src/publicodes-state/helpers/migrateSimulation.ts

@@ -1,28 +1,23 @@
-// eslint-disable-next-line @typescript-eslint/ban-ts-comment
-// @ts-ignore
-import { migrateSituation } from '@publicodes/tools/migration'
-import { MigrationType, Simulation } from '../types'
+import { Migration, migrateSituation } from '@publicodes/tools/migration'
+import { Simulation } from '../types'
 
-type Props = {
-  simulation: Simulation
-  migrationInstructions: MigrationType
-}
-type Return = Simulation
-
-export function migrateSimulation({
-  simulation: oldSimulation,
-  migrationInstructions,
-}: Props): Return {
-  const simulation = JSON.parse(JSON.stringify(oldSimulation)) as Simulation
-
-  const { situationMigrated, foldedStepsMigrated } = migrateSituation({
-    situation: simulation.situation,
-    foldedSteps: simulation.foldedSteps,
-    migrationInstructions,
-  })
+export function migrateSimulation(
+  simulation: Simulation,
+  migrationInstructions: Migration
+): Simulation {
+  simulation.situation = migrateSituation(
+    simulation.situation,
+    migrationInstructions
+  )
 
-  simulation.situation = situationMigrated
-  simulation.foldedSteps = foldedStepsMigrated
+  // NOTE: folded steps (i.e. answered rules) are can be map to a situation,
+  // where the keys are the rule names and the value is null.
+  simulation.foldedSteps = Object.keys(
+    migrateSituation(
+      Object.fromEntries(simulation.foldedSteps.map((step) => [step, null])),
+      migrationInstructions
+    )
+  )
 
   return simulation
 }

src/publicodes-state/hooks/useUser/useSimulations.ts

@@ -1,20 +1,18 @@
 'use client'
 
 import { generateSimulation } from '@/helpers/simulation/generateSimulation'
+import { Migration } from '@publicodes/tools/migration'
 import { Dispatch, SetStateAction, useCallback, useMemo } from 'react'
-import {
-  MigrationType,
-  Simulation,
-  UpdateCurrentSimulationProps,
-} from '../../types'
+import { Simulation, UpdateCurrentSimulationProps } from '../../types'
 
 type Props = {
   simulations: Simulation[]
   setSimulations: Dispatch<SetStateAction<Simulation[]>>
   currentSimulationId: string
   setCurrentSimulationId: Dispatch<SetStateAction<string>>
-  migrationInstructions: MigrationType
+  migrationInstructions: Migration
 }
+
 export default function useSimulations({
   simulations,
   setSimulations,

src/publicodes-state/providers/userProvider/context.ts

@@ -1,8 +1,9 @@
 'use client'
 
+import { Migration } from '@publicodes/tools/migration'
 import { Dispatch, SetStateAction, createContext } from 'react'
 import { v4 as uuid } from 'uuid'
-import { MigrationType, Simulation, Tutorials, User } from '../../types'
+import { Simulation, Tutorials, User } from '../../types'
 
 type UserContextType = {
   user: User
@@ -13,7 +14,7 @@ type UserContextType = {
   setSimulations: Dispatch<SetStateAction<Simulation[]>>
   currentSimulationId: string
   setCurrentSimulationId: Dispatch<SetStateAction<string>>
-  migrationInstructions: MigrationType
+  migrationInstructions: Migration
 }
 
 export default createContext<UserContextType>({

src/publicodes-state/providers/userProvider/index.tsx

@@ -2,7 +2,8 @@
 
 import { PropsWithChildren } from 'react'
 
-import { MigrationType, RegionFromGeolocation } from '@/publicodes-state/types'
+import { RegionFromGeolocation } from '@/publicodes-state/types'
+import { Migration } from '@publicodes/tools/migration'
 import UserContext from './context'
 import useUpdateOldLocalStorage from './useOldLocalStorage'
 import usePersistentSimulations from './usePersistentSimulations'
@@ -21,7 +22,7 @@ type Props = {
   /**
    * The migration instructions for old localstorage
    */
-  migrationInstructions: MigrationType
+  migrationInstructions: Migration
 }
 export default function UserProvider({
   children,

src/publicodes-state/providers/userProvider/usePersistentSimulations.ts

@@ -1,13 +1,14 @@
 import { generateSimulation } from '@/helpers/simulation/generateSimulation'
 import { getIsLocalStorageAvailable } from '@/utils/getIsLocalStorageAvailable'
+import { Migration } from '@publicodes/tools/migration'
 import { useEffect, useState } from 'react'
-import { MigrationType, Simulation } from '../../types'
+import { Simulation } from '../../types'
 
 const isLocalStorageAvailable = getIsLocalStorageAvailable()
 
 type Props = {
   storageKey: string
-  migrationInstructions: MigrationType
+  migrationInstructions: Migration
 }
 export default function usePersistentSimulations({
   storageKey,

src/publicodes-state/types.d.ts

@@ -134,8 +134,3 @@ type MosaicInfos = {
 }
 
 type Formule = any
-
-export type MigrationType = {
-  keysToMigrate: Record<DottedName, DottedName>
-  valuesToMigrate: Record<DottedName, Record<string, NodeValue>>
-}