timetables: add customisedModules prop when migrating to V2 schema

Author: zwliew

website/src/reducers/timetables.test.ts

@@ -260,7 +260,6 @@ describe('redux schema migration', () => {
     hidden: {},
     academicYear: '2022/2023',
     archive: {},
-    customisedModules: {},
     _persist: {
       version: 1,
       rehydrated: false,
@@ -285,7 +284,10 @@ describe('redux schema migration', () => {
     hidden: {},
     academicYear: '2022/2023',
     archive: {},
-    customisedModules: {},
+    customisedModules: {
+      [1]: [],
+      [2]: [],
+    },
     _persist: {
       version: 1, // version kept the same because the framework does not support it in unit tests
       rehydrated: false,

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 {
@@ -27,16 +27,19 @@ import { SET_EXPORTED_DATA } from 'actions/constants';
 import { Actions } from '../types/actions';
 
 // Migration from state V1 -> V2
-type TimetableStateV1 = Omit<TimetablesState, 'lessons'> & {
+type TimetableStateV1 = Omit<TimetablesState, 'lessons' | 'customisedModules'> & {
   lessons: { [semester: string]: { [moduleCode: string]: { [lessonType: string]: string } } };
 };
 export function migrateV1toV2(
   oldState: TimetableStateV1 & PersistedState,
 ): 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,
   };
 }