controller updates

Author: kmcfaul

packages/module/src/QuickStartController.tsx

@@ -1,7 +1,11 @@
 import * as React from 'react';
 import QuickStartContent from './controller/QuickStartContent';
 import QuickStartFooter from './controller/QuickStartFooter';
-import { QuickStartContext, QuickStartContextValues } from './utils/quick-start-context';
+import {
+  getDefaultQuickStartState,
+  QuickStartContext,
+  QuickStartContextValues,
+} from './utils/quick-start-context';
 import { QuickStart, QuickStartStatus, QuickStartTaskStatus } from './utils/quick-start-types';
 
 interface QuickStartControllerProps {
@@ -35,13 +39,26 @@ export const QuickStartController: React.FC<QuickStartControllerProps> = ({
   } = quickStart;
   const totalTasks = tasks?.length;
   const {
+    activeQuickStartID,
+    allQuickStartStates,
+    setAllQuickStartStates,
     activeQuickStartState,
     setActiveQuickStart,
     setQuickStartTaskNumber,
     setQuickStartTaskStatus,
     nextStep,
     previousStep,
   } = React.useContext<QuickStartContextValues>(QuickStartContext);
+  React.useEffect(() => {
+    // If activeQuickStartID was changed through prop from QuickStartContainer, need to init the state if it does not exist yet
+    if (activeQuickStartID && !allQuickStartStates[activeQuickStartID]) {
+      setAllQuickStartStates({
+        ...allQuickStartStates,
+        [activeQuickStartID]: getDefaultQuickStartState(),
+      });
+    }
+  }, [activeQuickStartID, allQuickStartStates, setAllQuickStartStates]);
+
   const status = activeQuickStartState?.status as QuickStartStatus;
   const taskNumber = activeQuickStartState?.taskNumber as number;
   const allTaskStatuses = tasks.map(