Merge pull request #209 from enaboapps/iss207

Fix to escape row
enaboapps · Mar 9, 2024 · 4ab3729 · 4ab3729
2 parents 07038bf + d0e87af
commit 4ab3729
Showing 1 changed file with 17 additions and 3 deletions.
diff --git a/app/src/main/java/com/enaboapps/switchify/service/scanning/tree/ScanTree.kt b/app/src/main/java/com/enaboapps/switchify/service/scanning/tree/ScanTree.kt
@@ -195,22 +195,36 @@ class ScanTree(
     }
 
     /**
-     * This function checks if escaping the current row is necessary
-     * @return True if escaping the current row is necessary, false otherwise
+     * This function checks if the current row should be escaped
+     * @return Whether the current row should be escaped
      */
     private fun shouldEscapeCurrentRow(): Boolean {
         // If at the last node, activate the escape row
         if (currentColumn == tree[currentRow].nodes.size - 1 && !shouldEscapeRow && scanDirection == ScanDirection.RIGHT) {
             shouldEscapeRow = true
             highlightCurrentRow()
+
+            return true
         } else if (currentColumn == 0 && !shouldEscapeRow && scanDirection == ScanDirection.LEFT) {
             shouldEscapeRow = true
             highlightCurrentRow()
+
+            return true
         } else if (shouldEscapeRow) {
             shouldEscapeRow = false
             unhighlightCurrentRow()
+
+            // Ensure that the index is correct
+            currentColumn = if (scanDirection == ScanDirection.RIGHT) {
+                0
+            } else {
+                tree[currentRow].nodes.size - 1
+            }
+            highlightCurrentNode()
+
+            return true
         }
-        return shouldEscapeRow
+        return false
     }
 
     /**