From 4c58f97981be1c1a23c6df59f0d02a7ecb393793 Mon Sep 17 00:00:00 2001
From: Paulus Schoutsen <balloob@gmail.com>
Date: Wed, 10 Nov 2021 21:36:49 -0800
Subject: [PATCH] Give ESP time to restart after write and disable button until
 initialized

---
 src/install-dialog.ts | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/install-dialog.ts b/src/install-dialog.ts
index e197509..329c0c8 100644
--- a/src/install-dialog.ts
+++ b/src/install-dialog.ts
@@ -466,6 +466,7 @@ class EwtInstallDialog extends LitElement {
         <ewt-button
           slot="primaryAction"
           .label=${supportsImprov ? "Next" : "Close"}
+          .disabled=${this._client === undefined}
           dialogAction=${ifDefined(supportsImprov ? undefined : "close")}
           @click=${!supportsImprov
             ? undefined
@@ -636,7 +637,9 @@ class EwtInstallDialog extends LitElement {
         this._installState = state;
 
         if (state.state === FlashStateType.FINISHED) {
-          this._initialize().then(() => this.requestUpdate());
+          sleep(100)
+            .then(() => this._initialize())
+            .then(() => this.requestUpdate());
         }
       },
       this.port,