timetables: add customisedModules prop when migrating to V2 schema

nusmodifications · Aug 4, 2023 · 3d05c4f · 3d05c4f
1 parent 4ce23c1
commit 3d05c4f
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/website/src/reducers/timetables.ts b/website/src/reducers/timetables.ts
@@ -5,7 +5,7 @@ import { createMigrate, PersistedState } from 'redux-persist';
 import { PersistConfig } from 'storage/persistReducer';
 import { ModuleCode } from 'types/modules';
 import { ModuleLessonConfig, SemTimetableConfig, TimetableConfig } from 'types/timetables';
-import { ColorMapping, TimetablesState } from 'types/reducers';
+import { ColorMapping, CustomisedModulesMap, TimetablesState } from 'types/reducers';
 
 import config from 'config';
 import {
@@ -35,8 +35,11 @@ export function migrateV1toV2(
 ): TimetablesState & PersistedState {
   const newLessons: TimetableConfig = {};
   const oldLessons = oldState.lessons;
+  const newCustomisedModules: CustomisedModulesMap = {};
 
   Object.entries(oldLessons).forEach(([semester, modules]) => {
+    newCustomisedModules[semester] = [];
+
     Object.entries(modules).forEach(([moduleCode, lessons]) => {
       const newSemester: { [moduleCode: string]: { [lessonType: string]: string[] } } = {
         [moduleCode]: {},
@@ -56,6 +59,7 @@ export function migrateV1toV2(
   return {
     ...oldState,
     lessons: newLessons,
+    customisedModules: newCustomisedModules,
   };
 }