X-Git-Url: http://git.scottworley.com/vopamoi/blobdiff_plain/b56a37d31873fa82e6196cde9479b689a14e4629..ad72cd51833c7202a3af3bd403c06616e4050b88:/vopamoi.ts

diff --git a/vopamoi.ts b/vopamoi.ts
index a3b7824..7bdf4b8 100644
--- a/vopamoi.ts
+++ b/vopamoi.ts
@@ -149,7 +149,6 @@ function Log(prefix: string = "vp-") {
 }
 const log = Log();
 
-
 function UI() {
   const undoLog: string[] = [];
   return {
@@ -179,14 +178,25 @@ function UI() {
 }
 const ui = UI();
 
+enum CommitOrAbort {
+  Commit,
+  Abort,
+}
+
 function BrowserUI() {
   var currentViewState = "todo";
+  var taskFocusedBeforeJumpingToInput: HTMLElement | null = null;
   return {
     addTask: function (event: KeyboardEvent) {
       const input = <HTMLInputElement>document.getElementById("taskName");
       if (input.value) {
         const task = ui.addTask(input.value);
-        if (task && task instanceof HTMLElement) task.focus();
+        if (currentViewState === "todo") {
+          task instanceof HTMLElement && task.focus();
+        } else if (this.returnFocusAfterInput()) {
+        } else {
+          this.firstVisibleTask()?.focus();
+        }
         input.value = "";
         if (event.getModifierState("Control")) {
           this.setPriority(task, null, document.getElementsByClassName("task")[0]);
@@ -209,7 +219,7 @@ function BrowserUI() {
       event.preventDefault();
     },
 
-    completeEdit: function (event: Event) {
+    completeEdit: function (event: Event, resolution: CommitOrAbort = CommitOrAbort.Commit) {
       const input = event.target as HTMLInputElement;
       const task = input.parentElement!;
       const oldDescription = task.getAttribute("data-description")!;
@@ -218,7 +228,7 @@ function BrowserUI() {
       task.removeChild(task.children[0]);
       task.removeAttribute("data-description");
       task.focus();
-      if (newDescription === oldDescription) {
+      if (newDescription === oldDescription || resolution === CommitOrAbort.Abort) {
         task.textContent = oldDescription;
       } else {
         ui.edit(task.getAttribute("data-created")!, newDescription, oldDescription);
@@ -234,6 +244,9 @@ function BrowserUI() {
     },
 
     focusTaskNameInput: function (event: Event) {
+      if (document.activeElement instanceof HTMLElement) {
+        taskFocusedBeforeJumpingToInput = document.activeElement;
+      }
       document.getElementById("taskName")!.focus();
       event.preventDefault();
     },
@@ -279,6 +292,14 @@ function BrowserUI() {
       }
     },
 
+    returnFocusAfterInput: function (): boolean {
+      if (taskFocusedBeforeJumpingToInput) {
+        taskFocusedBeforeJumpingToInput.focus();
+        return true;
+      }
+      return false;
+    },
+
     // Change task's priority to be between other tasks a and b.
     setPriority: function (task: Element, a: Element | null, b: Element | null) {
       const aPriority = a === null ? 0 : Model.getPriority(a);
@@ -331,8 +352,10 @@ function handleKey(event: any) {
   if (event.target.tagName === "INPUT") {
     if (event.target.id === "taskName") {
       if (event.key == "Enter") return browserUI.addTask(event);
+      if (event.key == "Escape") return browserUI.returnFocusAfterInput();
     } else {
       if (event.key == "Enter") return browserUI.completeEdit(event);
+      if (event.key == "Escape") return browserUI.completeEdit(event, CommitOrAbort.Abort);
     }
   } else {
     if (inputState === InputState.Command) {