support close action from studio

outerbase · Dec 4, 2024 · ec206c5 · ec206c5
1 parent 9efac5f
commit ec206c5
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/electron/main.ts b/electron/main.ts
@@ -62,7 +62,7 @@ function createDatabaseWindow(conn: ConnectionStoreItem) {
     show: false,
     width: 1024,
     height: 768,
-    // autoHideMenuBar: true,
+    autoHideMenuBar: true,
     webPreferences: {
       devTools: true,
       additionalArguments: ["--database=" + conn.id],
@@ -74,7 +74,7 @@ function createDatabaseWindow(conn: ConnectionStoreItem) {
 
   const queryString = new URLSearchParams({ name: conn.name }).toString();
 
-  dbWindow.on("close", () => {
+  dbWindow.on("closed", () => {
     win?.show();
     ConnectionPool.close(conn.id);
   });
@@ -200,6 +200,12 @@ ipcMain.handle(
   },
 );
 
+ipcMain.handle("close", async (sender) => {
+  sender.sender.close({
+    waitForBeforeUnload: true,
+  });
+});
+
 ipcMain.handle("connect", (_, conn: ConnectionStoreItem) => {
   createDatabaseWindow(conn);
   if (win) win.hide();

diff --git a/electron/preload.ts b/electron/preload.ts
@@ -46,6 +46,10 @@ const outerbaseIpc = {
     return ipcRenderer.invoke("transaction", connectionId, statements);
   },
 
+  close() {
+    return ipcRenderer.invoke("close");
+  },
+
   connect(conn: ConnectionStoreItem) {
     return ipcRenderer.invoke("connect", conn);
   },