X-Git-Url: http://git.scottworley.com/vopamoi/blobdiff_plain/997d8d55032aba314fa674c19ce25463af30c972..f36b20d6c8aff57f2582ffc800d20f03ab0bb44c:/vopamoi.ts

diff --git a/vopamoi.ts b/vopamoi.ts
index 5f759d0..8e72809 100644
--- a/vopamoi.ts
+++ b/vopamoi.ts
@@ -378,8 +378,9 @@ function BrowserUI() {
       return target as HTMLElement;
     },
 
-    firstVisibleTask: function () {
-      for (const task of document.getElementsByClassName("task")) {
+    firstVisibleTask: function (root: Element | null = null) {
+      if (root === null) root = document.body;
+      for (const task of root.getElementsByClassName("task")) {
         const state = task.getAttribute("data-state");
         if (
           task instanceof HTMLElement &&
@@ -447,7 +448,9 @@ function BrowserUI() {
     },
 
     moveCursorRight: function () {
-      (this.currentTask()?.getElementsByClassName("task")[0] as HTMLElement)?.focus();
+      const active = this.currentTask();
+      if (!active) return false;
+      (this.firstVisibleTask(active) as HTMLElement | null)?.focus();
     },
 
     moveCursorVertically: function (offset: number): boolean {