refactor: when you cancel editing a new task in the new homework, we return to the remaining task, if there is one, otherwise to the homework. For an existing task, we keep the previous transition to the initial state

Author: Andrw-404

hwproj.front/src/store/storeHooks/taskEditorHooks.ts

@@ -240,9 +240,15 @@ export const useTaskEditing = () => {
     }, [commitTaskRemoval, dispatch]);
 
     const discardTaskDraft = useCallback((task: HomeworkTaskViewModel, homework: HomeworkViewModel, isNewTask: boolean) => {
+        const draftHomework = draftHomeworks.find(dh => dh.id === task.homeworkId)
+        const remainingTasks = (draftHomework?.tasks || []).filter(t => t.id !== task.id);
         discardTaskDraftByIds(task.id!, task.homeworkId!);
         if (isNewTask) {
-            dispatch(setSelectedItem({ isHomework: true, id: homework.id }));
+            if (homework.id! < 0 && remainingTasks.length > 0) {
+                dispatch(setSelectedItem({ isHomework: false, id: remainingTasks[0].id }));
+            } else {
+                dispatch(setSelectedItem({ isHomework: true, id: homework.id }));
+            }
         } else {
             dispatch(setSelectedItem({ isHomework: false, id: task.id }));
         }