refactor: refactor child block completion

openedx-unsupported · Jul 26, 2023 · bea2f85 · bea2f85
1 parent 0404094
commit bea2f85
Showing 1 changed file with 13 additions and 10 deletions.
diff --git a/Source/CourseOutlineTableSource.swift b/Source/CourseOutlineTableSource.swift
@@ -769,19 +769,22 @@ extension CourseOutlineTableController: BlockCompletionDelegate {
 
         if let index = groups.firstIndex(where: { $0.block.blockID == blockGroup.block.blockID }) {
             groups[index] = blockGroup
+            collapsedSections.removeAll()
+            hasAddedToCollapsedSections = false
+            setGroups(groups)
         } else {
-            if let index = groups.firstIndex(where: { $0.block.blockID == blockGroup.block.blockID }),
-               let child = courseQuerier.childrenOfBlockWithID(blockID: groups[index].block.blockID, forMode: .full).value,
-               let indexOfBlock = child.children.firstIndex(where: { $0.blockID == blockGroup.block.blockID }) {
-                var updatedChild = child
-                updatedChild.children[indexOfBlock] = blockGroup.block
-                groups[index] = updatedChild
+            for (index, group) in groups.enumerated() {
+                guard var child = courseQuerier.childrenOfBlockWithID(blockID: group.block.blockID, forMode: .full).value else { continue }
+                if let indexOfBlock = child.children.firstIndex(where: { $0.blockID == blockGroup.block.blockID }) {
+                    child.children[indexOfBlock] = blockGroup.block
+                    groups[index] = child
+                    collapsedSections.removeAll()
+                    hasAddedToCollapsedSections = false
+                    setGroups(groups)
+                    break
+                }
             }
         }
-
-        collapsedSections.removeAll()
-        hasAddedToCollapsedSections = false
-        setGroups(groups)
     }
 }