From 813165c4bb6ce397f52c101929d954e629794f8b Mon Sep 17 00:00:00 2001
From: maxym <maksim.parfenov@protonmail.ch>
Date: Sun, 26 Nov 2023 00:26:57 +0100
Subject: [PATCH] feature: add update modal and block system modal

---
 app/components/common/Version.tsx             | 38 ++++++++++-
 .../modal/UpdateApplicationWarningModal.tsx   | 67 +++++++++++++++++++
 desktop/main/autoUpdater.ts                   |  1 -
 desktop/main/createMainWindow.ts              |  6 +-
 desktop/main/promptBeforeClose.ts             |  9 ++-
 desktop/main/reactions/handleCloseApp.ts      |  8 ++-
 desktop/main/sources/autoUpdate.ts            | 12 ++--
 desktop/main/startApp.ts                      |  7 +-
 shared/constants.ts                           |  4 +-
 9 files changed, 134 insertions(+), 18 deletions(-)
 create mode 100644 app/screens/modal/UpdateApplicationWarningModal.tsx

diff --git a/app/components/common/Version.tsx b/app/components/common/Version.tsx
index ec4face74..389c1f852 100644
--- a/app/components/common/Version.tsx
+++ b/app/components/common/Version.tsx
@@ -21,6 +21,9 @@ import { getNetworkInfo } from '../../redux/network/selectors';
 import { checkUpdates as checkUpdatesIco } from '../../assets/images';
 import { AppThDispatch } from '../../types';
 import updaterSlice from '../../redux/updater/slice';
+import { SECOND } from '../../../shared/constants';
+import { Loader } from '../../basicComponents';
+import UpdateApplicationWarningModal from '../../screens/modal/UpdateApplicationWarningModal';
 import FeedbackButton from './Feedback';
 
 const Container = styled.div`
@@ -194,6 +197,29 @@ const UpdateStatus = () => {
   const isDownloading = useSelector(isUpdateDownloading);
   const isDownloaded = useSelector(isUpdateDownloaded);
   const error = useSelector(getError);
+  const [
+    isOpenUpdateApplicationWarningModal,
+    setIsOpenUpdateApplicationWarningModal,
+  ] = useState(false);
+  const [
+    showUpdateApplicationLoader,
+    setShowUpdateApplicationLoader,
+  ] = useState(false);
+
+  const handleRestartNow = () => {
+    setIsOpenUpdateApplicationWarningModal(false);
+    setShowUpdateApplicationLoader(true);
+    eventsService.installUpdate();
+
+    setTimeout(() => {
+      setShowUpdateApplicationLoader(false);
+    }, 10 * SECOND);
+  };
+
+  const handlePostpone = () => {
+    setIsOpenUpdateApplicationWarningModal(false);
+  };
+
   if (!isDownloading && !isDownloaded) return null;
 
   if (progress !== null && !isDownloaded) {
@@ -207,9 +233,19 @@ const UpdateStatus = () => {
     return (
       <>
         <ProgressChunk>Update is ready to install</ProgressChunk>
-        <PrimaryAction onClick={() => eventsService.installUpdate()}>
+        <PrimaryAction
+          onClick={() => setIsOpenUpdateApplicationWarningModal(true)}
+        >
           Restart Smapp
         </PrimaryAction>
+        <UpdateApplicationWarningModal
+          isOpen={isOpenUpdateApplicationWarningModal}
+          onApprove={handleRestartNow}
+          onCancel={handlePostpone}
+        />
+        {showUpdateApplicationLoader && (
+          <Loader size={Loader.sizes.BIG} note="UPDATE IN PROGESS..." />
+        )}
       </>
     );
   }
diff --git a/app/screens/modal/UpdateApplicationWarningModal.tsx b/app/screens/modal/UpdateApplicationWarningModal.tsx
new file mode 100644
index 000000000..541eccd44
--- /dev/null
+++ b/app/screens/modal/UpdateApplicationWarningModal.tsx
@@ -0,0 +1,67 @@
+import React from 'react';
+import styled from 'styled-components';
+import Modal from '../../components/common/Modal';
+import { Button } from '../../basicComponents';
+import { smColors } from '../../vars';
+
+const ButtonsWrapper = styled.div`
+  display: flex;
+  flex-direction: row;
+  justify-content: space-between;
+  margin: auto 0 15px 0;
+  padding-top: 30px;
+`;
+
+const Message = styled.pre`
+  font-size: 14px;
+  line-height: 1.33em;
+  word-wrap: break-word;
+  white-space: pre-wrap;
+  overflow-y: auto;
+  margin-top: 15px;
+
+  ul {
+    list-style: none;
+    margin-left: 10px;
+  }
+  li {
+    margin: 10px 0;
+  }
+  li:before {
+    content: '• ';
+    padding: 5px;
+  }
+`;
+
+const UpdateApplicationWarningModal = ({ isOpen, onApprove, onCancel }) => {
+  if (!isOpen) return null;
+
+  return (
+    <Modal header="Update SMAPP" height={380}>
+      <Message>
+        <p>
+          Restarting now is <b>CRITICAL</b> and may impact your node’s
+          performance and rewards.
+        </p>
+        <ul>
+          <li>
+            Click <b style={{ color: smColors.green }}>RESTART NOW</b> to apply
+            the update immediately. Delaying the update could result in
+            potential loss of rewards.
+          </li>
+          <li>
+            Click <b style={{ color: smColors.purple }}>POSTPONE</b> to delay
+            the update. Be aware that postponing may slow down your node’s
+            performance and future rewards.
+          </li>
+        </ul>
+      </Message>
+      <ButtonsWrapper>
+        <Button onClick={onCancel} isPrimary={false} text="POSTPONE" />
+        <Button onClick={onApprove} text="RESTART NOW" />
+      </ButtonsWrapper>
+    </Modal>
+  );
+};
+
+export default UpdateApplicationWarningModal;
diff --git a/desktop/main/autoUpdater.ts b/desktop/main/autoUpdater.ts
index 3edceab79..625412db1 100644
--- a/desktop/main/autoUpdater.ts
+++ b/desktop/main/autoUpdater.ts
@@ -92,7 +92,6 @@ export const checkUpdates = async (
   autoUpdater.setFeedURL(feedUrl);
   try {
     const result = await autoUpdater.checkForUpdates();
-
     if (result) {
       return {
         status: UpdateInfoStatus.UpdateAvailable,
diff --git a/desktop/main/createMainWindow.ts b/desktop/main/createMainWindow.ts
index 7a5b75b98..54c28c31a 100644
--- a/desktop/main/createMainWindow.ts
+++ b/desktop/main/createMainWindow.ts
@@ -1,5 +1,4 @@
-import { app, BrowserWindow, ipcMain } from 'electron';
-import { autoUpdater } from 'electron-updater';
+import { app, BrowserWindow, ipcMain, autoUpdater } from 'electron';
 import {
   BehaviorSubject,
   combineLatest,
@@ -35,6 +34,7 @@ export default () => {
   const $quit = new Subject<Electron.IpcMainEvent>();
   const $activate = fromAppEvent('activate');
   const $secondInstance = fromAppEvent('second-instance');
+  const $isUpdateInProgress = new BehaviorSubject<boolean>(false);
 
   //
   // Subscriptions & Reactions
@@ -42,6 +42,7 @@ export default () => {
 
   // Push to $quit event stream
   const handleQuit = async (event) => {
+    logger.log('handleQuit', $isAppClosing.value);
     if (!$isAppClosing.value) {
       $quit.next(event);
     }
@@ -84,5 +85,6 @@ export default () => {
     $showWindowOnLoad,
     $isWindowReady,
     $isSmappActivated: $activate,
+    $isUpdateInProgress,
   };
 };
diff --git a/desktop/main/promptBeforeClose.ts b/desktop/main/promptBeforeClose.ts
index a4633caff..19d1ddd66 100644
--- a/desktop/main/promptBeforeClose.ts
+++ b/desktop/main/promptBeforeClose.ts
@@ -19,7 +19,8 @@ const promptBeforeClose = (
   mainWindow: BrowserWindow,
   managers: Partial<Managers>,
   $isAppClosing: BehaviorSubject<boolean>,
-  $showWindowOnLoad: Subject<boolean>
+  $showWindowOnLoad: Subject<boolean>,
+  $isUpdateInProgress: BehaviorSubject<boolean>
 ) => {
   const showPrompt = async () => {
     if (!mainWindow) return CloseAppPromptResult.KEEP_SMESHING;
@@ -64,11 +65,13 @@ const promptBeforeClose = (
   };
 
   const handleClosingApp = async (event: Electron.Event) => {
-    event.preventDefault();
-    if (!mainWindow) {
+    event?.preventDefault();
+
+    if ($isUpdateInProgress.value || !mainWindow) {
       await quit();
       return;
     }
+
     if (isCloseTriggered) {
       mainWindow.hide();
       return;
diff --git a/desktop/main/reactions/handleCloseApp.ts b/desktop/main/reactions/handleCloseApp.ts
index 851fa2bd8..ad88d7e4d 100644
--- a/desktop/main/reactions/handleCloseApp.ts
+++ b/desktop/main/reactions/handleCloseApp.ts
@@ -12,16 +12,18 @@ export default (
   $managers: Subject<Managers>,
   $mainWindow: Subject<BrowserWindow>,
   $isAppClosing: BehaviorSubject<boolean>,
-  $showWindowOnLoad: Subject<boolean>
+  $showWindowOnLoad: Subject<boolean>,
+  $isUpdateInProgress: BehaviorSubject<boolean>
 ) =>
   makeSubscription(
-    $quit.pipe(withLatestFrom($mainWindow, $managers)),
+    $quit.pipe(withLatestFrom($mainWindow, $managers, $isUpdateInProgress)),
     ([event, mw, managers]) =>
       mw &&
       promptBeforeClose(
         mw,
         managers || {},
         $isAppClosing,
-        $showWindowOnLoad
+        $showWindowOnLoad,
+        $isUpdateInProgress
       )(event).catch((err) => logger.error('promptBeforeClose', err))
   );
diff --git a/desktop/main/sources/autoUpdate.ts b/desktop/main/sources/autoUpdate.ts
index caaf2383c..6a5db5814 100644
--- a/desktop/main/sources/autoUpdate.ts
+++ b/desktop/main/sources/autoUpdate.ts
@@ -1,6 +1,7 @@
 import { BrowserWindow } from 'electron';
 import { UpdateInfo } from 'electron-updater';
 import {
+  BehaviorSubject,
   combineLatest,
   filter,
   first,
@@ -40,7 +41,8 @@ const handleAutoUpdates = (
   everyMs: number,
   $mainWindow: Observable<BrowserWindow>,
   $managers: Observable<Managers>,
-  $currentNetwork: Observable<Network | null>
+  $currentNetwork: Observable<Network | null>,
+  $isUpdateInProgress: BehaviorSubject<boolean>
 ) => {
   type DoDownload = boolean;
   type Data = [BrowserWindow, Network, DoDownload];
@@ -107,6 +109,7 @@ const handleAutoUpdates = (
                   StoreService.set('startNodeOnNextLaunch', true);
                   await managers.node.stopNode();
                 }
+
                 installUpdate();
               } catch (err) {
                 logger.error('forceUpdate', err);
@@ -136,9 +139,10 @@ const handleAutoUpdates = (
       $request.next(true);
     }),
     // Trigger installation
-    fromIPC<void>(ipcConsts.AU_REQUEST_INSTALL).subscribe(() =>
-      installUpdate()
-    ),
+    fromIPC<void>(ipcConsts.AU_REQUEST_INSTALL).subscribe(() => {
+      $isUpdateInProgress.next(true);
+      return installUpdate();
+    }),
   ];
 
   return () => subs.forEach((sub) => sub.unsubscribe());
diff --git a/desktop/main/startApp.ts b/desktop/main/startApp.ts
index 024b77b28..c44f23c3b 100644
--- a/desktop/main/startApp.ts
+++ b/desktop/main/startApp.ts
@@ -133,6 +133,7 @@ const startApp = (): AppStore => {
     $isAppClosing,
     $showWindowOnLoad,
     $isWindowReady,
+    $isUpdateInProgress,
   } = createMainWindow();
   // Store
   const $storeService = observeStoreService();
@@ -218,7 +219,8 @@ const startApp = (): AppStore => {
       $managers,
       $mainWindow,
       $isAppClosing,
-      $showWindowOnLoad
+      $showWindowOnLoad,
+      $isUpdateInProgress
     ),
     // Unlock / Create wallet
     // Switch network
@@ -264,7 +266,8 @@ const startApp = (): AppStore => {
       CHECK_UPDATES_INTERVAL,
       $mainWindow,
       $managers,
-      $currentNetwork
+      $currentNetwork,
+      $isUpdateInProgress
     ),
     handleOpenDashboard($mainWindow, $currentNetwork),
     sendWarningsToRenderer($warnings, $mainWindow, $isWindowReady),
diff --git a/shared/constants.ts b/shared/constants.ts
index 0f002637f..7425d97af 100644
--- a/shared/constants.ts
+++ b/shared/constants.ts
@@ -47,8 +47,8 @@ export enum ExternalLinks {
 }
 
 export const BITS_PER_LABEL = 128;
-
-export const MINUTE = 60 * 1000;
+export const SECOND = 1000;
+export const MINUTE = 60 * SECOND;
 
 export const HOUR = MINUTE * 60;