From 4c33d24a7b856488df73407ae418c5df9701c27d Mon Sep 17 00:00:00 2001
From: Red Adaya <red.adaya@gmail.com>
Date: Tue, 12 Mar 2024 09:31:35 +0800
Subject: [PATCH] close modals on ECS

---
 src/app/common/modals/alert.tsx  |  2 +-
 src/models/bookmarks.ts          |  2 +-
 src/models/clientsettingsview.ts | 10 ++++++++++
 src/models/connectionsview.ts    | 10 ++++++++++
 src/models/historyview.ts        |  2 +-
 src/models/modals.ts             |  3 ++-
 src/models/model.ts              | 11 +++++++----
 7 files changed, 32 insertions(+), 8 deletions(-)

diff --git a/src/app/common/modals/alert.tsx b/src/app/common/modals/alert.tsx
index f128f26d8..714869aec 100644
--- a/src/app/common/modals/alert.tsx
+++ b/src/app/common/modals/alert.tsx
@@ -44,7 +44,7 @@ class AlertModal extends React.Component<{}, {}> {
                         <Markdown text={message?.message ?? ""} extraClassName="bottom-margin" />
                     </If>
                     <If condition={!message?.markdown}>{message?.message}</If>
-                    <If condition={message.confirmflag}>
+                    <If condition={message?.confirmflag}>
                         <Checkbox
                             onChange={this.handleDontShowAgain}
                             label={"Don't show me this again"}
diff --git a/src/models/bookmarks.ts b/src/models/bookmarks.ts
index ea69c6a45..a5e58f55b 100644
--- a/src/models/bookmarks.ts
+++ b/src/models/bookmarks.ts
@@ -206,7 +206,7 @@ class BookmarksModel {
     }
 
     handleDocKeyDown(e: any): void {
-        let waveEvent = adaptFromReactOrNativeKeyEvent(e);
+        const waveEvent = adaptFromReactOrNativeKeyEvent(e);
         if (checkKeyPressed(waveEvent, "Escape")) {
             e.preventDefault();
             if (this.editingBookmark.get() != null) {
diff --git a/src/models/clientsettingsview.ts b/src/models/clientsettingsview.ts
index ad177e7ab..e74a02a53 100644
--- a/src/models/clientsettingsview.ts
+++ b/src/models/clientsettingsview.ts
@@ -3,6 +3,7 @@
 
 import * as mobx from "mobx";
 import { Model } from "./model";
+import { checkKeyPressed, adaptFromReactOrNativeKeyEvent } from "@/util/keyutil";
 
 class ClientSettingsViewModel {
     globalModel: Model;
@@ -21,6 +22,15 @@ class ClientSettingsViewModel {
             this.globalModel.activeMainView.set("clientsettings");
         })();
     }
+
+    handleDocKeyDown(e: any): void {
+        const waveEvent = adaptFromReactOrNativeKeyEvent(e);
+        if (checkKeyPressed(waveEvent, "Escape")) {
+            e.preventDefault();
+            this.closeView();
+            return;
+        }
+    }
 }
 
 export { ClientSettingsViewModel };
diff --git a/src/models/connectionsview.ts b/src/models/connectionsview.ts
index 6d819b88d..297beda1c 100644
--- a/src/models/connectionsview.ts
+++ b/src/models/connectionsview.ts
@@ -3,6 +3,7 @@
 
 import * as mobx from "mobx";
 import { Model } from "./model";
+import { checkKeyPressed, adaptFromReactOrNativeKeyEvent } from "@/util/keyutil";
 
 class ConnectionsViewModel {
     globalModel: Model;
@@ -21,6 +22,15 @@ class ConnectionsViewModel {
             this.globalModel.activeMainView.set("connections");
         })();
     }
+
+    handleDocKeyDown(e: any): void {
+        const waveEvent = adaptFromReactOrNativeKeyEvent(e);
+        if (checkKeyPressed(waveEvent, "Escape")) {
+            e.preventDefault();
+            this.closeView();
+            return;
+        }
+    }
 }
 
 export { ConnectionsViewModel };
diff --git a/src/models/historyview.ts b/src/models/historyview.ts
index 6892b854a..63b934d20 100644
--- a/src/models/historyview.ts
+++ b/src/models/historyview.ts
@@ -291,7 +291,7 @@ class HistoryViewModel {
     }
 
     handleDocKeyDown(e: any): void {
-        let waveEvent = adaptFromReactOrNativeKeyEvent(e);
+        const waveEvent = adaptFromReactOrNativeKeyEvent(e);
         if (checkKeyPressed(waveEvent, "Escape")) {
             e.preventDefault();
             this.closeView();
diff --git a/src/models/modals.ts b/src/models/modals.ts
index 528c4992a..4c4dcfd99 100644
--- a/src/models/modals.ts
+++ b/src/models/modals.ts
@@ -18,10 +18,11 @@ class ModalsModel {
         }
     }
 
-    popModal() {
+    popModal(callback?: () => void) {
         mobx.action(() => {
             this.store.pop();
         })();
+        callback && callback();
     }
 }
 
diff --git a/src/models/model.ts b/src/models/model.ts
index 233049a83..423ad80aa 100644
--- a/src/models/model.ts
+++ b/src/models/model.ts
@@ -457,7 +457,7 @@ class Model {
         if (this.alertMessage.get() != null) {
             if (checkKeyPressed(waveEvent, "Escape")) {
                 e.preventDefault();
-                this.cancelAlert();
+                this.modalsModel.popModal(() => this.cancelAlert());
                 return;
             }
             if (checkKeyPressed(waveEvent, "Enter")) {
@@ -467,6 +467,10 @@ class Model {
             }
             return;
         }
+        if (checkKeyPressed(waveEvent, "Escape") && this.modalsModel.store.length > 0) {
+            this.modalsModel.popModal();
+            return;
+        }
         if (this.activeMainView.get() == "bookmarks") {
             this.bookmarksModel.handleDocKeyDown(e);
             return;
@@ -476,14 +480,13 @@ class Model {
             return;
         }
         if (this.activeMainView.get() == "connections") {
-            this.historyViewModel.handleDocKeyDown(e);
+            this.connectionViewModel.handleDocKeyDown(e);
             return;
         }
         if (this.activeMainView.get() == "clientsettings") {
-            this.historyViewModel.handleDocKeyDown(e);
+            this.clientSettingsViewModel.handleDocKeyDown(e);
             return;
         }
-
         if (checkKeyPressed(waveEvent, "Escape")) {
             e.preventDefault();
             if (this.activeMainView.get() == "webshare") {