ENCOA-311

src/components/ExamEditor/Exercises/TrueFalse/index.tsx

@@ -1,9 +1,9 @@
-import React, { useEffect, useState } from 'react';
+import React, { useCallback, useEffect, useState } from 'react';
 import {
     MdAdd,
 } from 'react-icons/md';
 import Alert, { AlertItem } from '../Shared/Alert';
-import { ReadingPart, TrueFalseExercise } from '@/interfaces/exam';
+import { Difficulty, ReadingPart, TrueFalseExercise } from '@/interfaces/exam';
 import QuestionsList from '../Shared/QuestionsList';
 import Header from '../../Shared/Header';
 import SortableQuestion from '../Shared/SortableQuestion';
@@ -19,6 +19,7 @@ import PromptEdit from '../Shared/PromptEdit';
 
 const TrueFalse: React.FC<{ exercise: TrueFalseExercise, sectionId: number }> = ({ exercise, sectionId }) => {
     const { currentModule, dispatch } = useExamEditorStore();
+    const difficulty = useExamEditorStore((state) => state.modules[currentModule].difficulty);
     const { state } = useExamEditorStore(
         (state) => state.modules[currentModule].sections.find((section) => section.sectionId === sectionId)!
     );
@@ -131,12 +132,24 @@ const TrueFalse: React.FC<{ exercise: TrueFalseExercise, sectionId: number }> =
         setLocal(handleTrueFalseReorder(event, local));
     }
 
+    const saveDifficulty = useCallback((diff: Difficulty) => {
+        if (!difficulty.includes(diff)) {
+            dispatch({ type: 'UPDATE_MODULE', payload: { updates: { difficulty: [...difficulty, diff]} } });
+        }
+        const updatedExercise = { ...exercise, difficulty: diff };
+        const newState = { ...section };
+        newState.exercises = newState.exercises.map((ex) => ex.id === exercise.id ? updatedExercise : ex );
+        dispatch({ type: 'UPDATE_SECTION_STATE', payload: { sectionId, update: newState, module: currentModule } });
+    }, [currentModule, difficulty, dispatch, exercise, section, sectionId]);
+
     return (
         <div className="p-4">
             <Header
                 title='True/False/Not Given Exercise'
                 description='Edit questions and their solutions'
                 editing={editing}
+                difficulty={exercise.difficulty}
+                saveDifficulty={saveDifficulty}
                 handleSave={handleSave}
                 handleDelete={handleDelete}
                 handleDiscard={handleDiscard}