ENCOA-311

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

@@ -1,10 +1,10 @@
 import AutoExpandingTextArea from "@/components/Low/AutoExpandingTextarea";
 import { Card, CardContent } from "@/components/ui/card";
-import { WriteBlanksExercise, ReadingPart } from "@/interfaces/exam";
+import { WriteBlanksExercise, ReadingPart, Difficulty } from "@/interfaces/exam";
 import useExamEditorStore from "@/stores/examEditor";
 import { DragEndEvent } from "@dnd-kit/core";
 import { arrayMove } from "@dnd-kit/sortable";
-import { useState, useEffect } from "react";
+import { useState, useEffect, useCallback } from "react";
 import { MdEditOff, MdEdit, MdDelete, MdAdd } from "react-icons/md";
 import { toast } from "react-toastify";
 import useSectionEdit from "../../Hooks/useSectionEdit";
@@ -21,6 +21,7 @@ import PromptEdit from "../Shared/PromptEdit";
 
 const WriteBlanksForm: React.FC<{ sectionId: number; exercise: WriteBlanksExercise }> = ({ sectionId, exercise }) => {
     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)!
     );
@@ -208,12 +209,24 @@ const WriteBlanksForm: React.FC<{ sectionId: number; exercise: WriteBlanksExerci
         updateLocal({ ...local, text: newText });
     };
 
+    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="Write Blanks: Form Exercise"
                 description="Edit questions and their solutions"
                 editing={editing}
+                difficulty={exercise.difficulty}
+                saveDifficulty={saveDifficulty}
                 handleSave={handleSave}
                 handleDiscard={handleDiscard}
                 handleDelete={handleDelete}