From bef8612665829b5b00cc049f3466057e89b52f1b Mon Sep 17 00:00:00 2001
From: toasted-nutbread <toasted-nutbread@users.noreply.github.com>
Date: Sat, 19 Dec 2020 16:00:47 -0500
Subject: [PATCH] Focus the current entry earlier in the generation process
 (#1137)

---
 ext/mixed/js/display.js | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/ext/mixed/js/display.js b/ext/mixed/js/display.js
index 9e2cd931..f600da63 100644
--- a/ext/mixed/js/display.js
+++ b/ext/mixed/js/display.js
@@ -879,6 +879,7 @@ class Display extends EventDispatcher {
         }
 
         let {sentence=null, optionsContext=null, focusEntry=null, scrollX=null, scrollY=null} = state;
+        if (typeof focusEntry !== 'number') { focusEntry = 0; }
         if (!(typeof optionsContext === 'object' && optionsContext !== null)) {
             optionsContext = this.getOptionsContext();
             state.optionsContext = optionsContext;
@@ -943,9 +944,11 @@ class Display extends EventDispatcher {
             entry.dataset.index = `${i}`;
             this._addEntryEventListeners(entry);
             container.appendChild(entry);
+            if (focusEntry === i) {
+                this._focusEntry(i, false);
+            }
         }
 
-        this._focusEntry(typeof focusEntry === 'number' ? focusEntry : 0, false);
         if (typeof scrollX === 'number' || typeof scrollY === 'number') {
             let {x, y} = this._windowScroll;
             if (typeof scrollX === 'number') { x = scrollX; }