1 files changed, 18 insertions, 1 deletions
diff --git a/Pages/Component/Dialog.razor b/Pages/Component/Dialog.razor
index dbfec49..673ec2f 100644
--- a/Pages/Component/Dialog.razor
+++ b/Pages/Component/Dialog.razor
@@ -48,6 +48,23 @@
     public void Show() => Visible = true;
     public void Hide() => Visible = false;
 
+    protected override async void OnAfterRender(bool firstRender) {
+        if(firstRender) {
+            await jsRuntime.InvokeVoidAsync("dialogAddObjectReference", new object[] {
+                dialogDiv,
+                DotNetObjectReference.Create(this)
+            });
+        }
+    }
+
+    [JSInvokable("KeyHandler")]
+    public void KeyHandler(string key) {
+        if(key == "Escape") {
+            Hide();
+            return;
+        }
+    }
+
     private string heightStyle =>
-        $"{(height is null ? "" : $"max-height:{height};")}";
+    $"{(height is null ? "" : $"max-height:{height};")}";
 }