From e5beebfdc4f082888c447364b12091d2ee8ecfa5 Mon Sep 17 00:00:00 2001
From: Jos de Jong <wjosdejong@gmail.com>
Date: Tue, 19 Sep 2023 14:43:57 +0200
Subject: [PATCH] fix: select active element on undo only when existing

---
 src/lib/components/modes/treemode/TreeMode.svelte | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/lib/components/modes/treemode/TreeMode.svelte b/src/lib/components/modes/treemode/TreeMode.svelte
index 10563d47..fe83cb77 100644
--- a/src/lib/components/modes/treemode/TreeMode.svelte
+++ b/src/lib/components/modes/treemode/TreeMode.svelte
@@ -1783,11 +1783,11 @@
       // TODO: find a better way to restore focus
       // TODO: implement a proper TypeScript solution to check whether this is an element with blur, focus, select
       const activeElement = document.activeElement as HTMLInputElement
-      if (activeElement && activeElement.blur && activeElement.focus) {
+      if (activeElement && activeElement.blur && activeElement.select) {
         activeElement.blur()
         setTimeout(() => {
           handleUndo()
-          setTimeout(() => activeElement.select())
+          setTimeout(() => activeElement?.select())
         })
       } else {
         handleUndo()
@@ -1800,11 +1800,11 @@
       // TODO: find a better way to restore focus
       // TODO: implement a proper TypeScript solution to check whether this is an element with blur, focus, select
       const activeElement = document.activeElement as HTMLInputElement
-      if (activeElement && activeElement.blur && activeElement.focus) {
+      if (activeElement && activeElement.blur && activeElement.select) {
         activeElement.blur()
         setTimeout(() => {
           handleRedo()
-          setTimeout(() => activeElement.select())
+          setTimeout(() => activeElement?.select())
         })
       } else {
         handleRedo()