From 53024c2c42c44aa4fe3aa612c194a738446c7b58 Mon Sep 17 00:00:00 2001
From: Oj <oj@oojmed.com>
Date: Sat, 11 Dec 2021 13:30:27 +0000
Subject: [PATCH] [AsarUpdate] Add buttons to modal for restarting

---
 src/asarUpdate.js | 17 ++++++++++++++---
 1 file changed, 14 insertions(+), 3 deletions(-)

diff --git a/src/asarUpdate.js b/src/asarUpdate.js
index cbd67df..07ac5db 100644
--- a/src/asarUpdate.js
+++ b/src/asarUpdate.js
@@ -32,7 +32,7 @@ module.exports = () => { // (Try) update asar
     res.pipe(file);
   });
 
-  file.on('finish', () => {
+  file.on('finish', async () => {
     file.close();
     log('AsarUpdate', 'Completed download');
 
@@ -45,10 +45,21 @@ New Hash: ${newHash}
 Changed: ${changed}`);
 
     if (changed) {
-      electron.dialog.showMessageBox(null, {
+      const { response } = await electron.dialog.showMessageBox(null, {
         message: 'Updated OpenAsar',
-        detail: `New version will be used next restart.`
+        detail: `Restart required to use new version.`,
+        buttons: ['Restart Now', 'Later'],
+        defaultId: 0
       });
+
+      log('AsarUpdate', 'Modal response', response);
+
+      if (response === 0) {
+        log('AsarUpdate', 'Restarting');
+
+        electron.app.relaunch();
+        electron.app.exit();
+      }
     }
   });
 };
\ No newline at end of file