From c5db68fa26d7e74ca4047f345f8c3abb48625e11 Mon Sep 17 00:00:00 2001
From: cv <34649812+cpvalente@users.noreply.github.com>
Date: Fri, 4 Jun 2021 12:30:36 +0200
Subject: [PATCH] graceful shutdown

- confirmation dialog
- shutdown node service
---
 .../src/features/menu/buttons/QuitIconBtn.jsx | 73 +++++++++++++++----
 server/main.js                                | 10 +++
 server/src/app.js                             | 12 ++-
 server/src/controllers/OscController.js       |  8 +-
 4 files changed, 88 insertions(+), 15 deletions(-)

diff --git a/client/src/features/menu/buttons/QuitIconBtn.jsx b/client/src/features/menu/buttons/QuitIconBtn.jsx
index e22b9cd05..bf5c1364d 100644
--- a/client/src/features/menu/buttons/QuitIconBtn.jsx
+++ b/client/src/features/menu/buttons/QuitIconBtn.jsx
@@ -1,21 +1,68 @@
-import { IconButton } from '@chakra-ui/button';
+import { Button, IconButton } from '@chakra-ui/button';
+import {
+  AlertDialog,
+  AlertDialogBody,
+  AlertDialogContent,
+  AlertDialogFooter,
+  AlertDialogHeader,
+  AlertDialogOverlay,
+} from '@chakra-ui/modal';
 import { Tooltip } from '@chakra-ui/tooltip';
+import { useRef, useState } from 'react';
 import { FiPower } from 'react-icons/fi';
 
 export default function QuitIconBtn(props) {
   const { clickhandler, ...rest } = props;
+  const [isOpen, setIsOpen] = useState(false);
+  const onClose = () => setIsOpen(false);
+  const cancelRef = useRef();
+
+  const handleShutdown = () => {
+    onClose();
+    clickhandler();
+  };
+
   return (
-    <Tooltip label='Quit Application'>
-      <IconButton
-        size={props.size || 'xs'}
-        icon={<FiPower />}
-        colorScheme='red'
-        variant='outline'
-        isRound
-        onClick={clickhandler}
-        _focus={{ boxShadow: 'none' }}
-        {...rest}
-      />
-    </Tooltip>
+    <>
+      <Tooltip label='Quit Application'>
+        <IconButton
+          size={props.size || 'xs'}
+          icon={<FiPower />}
+          colorScheme='red'
+          variant='outline'
+          isRound
+          onClick={() => setIsOpen(true)}
+          _focus={{ boxShadow: 'none' }}
+          {...rest}
+        />
+      </Tooltip>
+      <AlertDialog
+        isOpen={isOpen}
+        leastDestructiveRef={cancelRef}
+        onClose={onClose}
+      >
+        <AlertDialogOverlay>
+          <AlertDialogContent>
+            <AlertDialogHeader fontSize='lg' fontWeight='bold'>
+              Server Shutdown
+            </AlertDialogHeader>
+
+            <AlertDialogBody>
+              This will shutdown the program and all running servers. Are you
+              sure?
+            </AlertDialogBody>
+
+            <AlertDialogFooter>
+              <Button ref={cancelRef} onClick={onClose}>
+                Cancel
+              </Button>
+              <Button colorScheme='red' onClick={handleShutdown} ml={3}>
+                Delete
+              </Button>
+            </AlertDialogFooter>
+          </AlertDialogContent>
+        </AlertDialogOverlay>
+      </AlertDialog>
+    </>
   );
 }
diff --git a/server/main.js b/server/main.js
index 32d7920cc..e09c3ee1d 100644
--- a/server/main.js
+++ b/server/main.js
@@ -195,6 +195,16 @@ ipcMain.on('test-message', (event, arg) => {
 // Terminate
 ipcMain.on('shutdown', (event, arg) => {
   console.log('Got IPC shutdown');
+
+  // terminate node service
+  (async () => {
+    const { shutdown } = await import(
+      path.join('file:///', __dirname, env == 'prod' ? '../' : '', 'src/app.js')
+    );
+    // Start express server
+    await shutdown();
+  })();
+
   win.destroy();
   app.quit();
 });
diff --git a/server/src/app.js b/server/src/app.js
index 6c604b2d2..fd7c26f80 100644
--- a/server/src/app.js
+++ b/server/src/app.js
@@ -106,7 +106,7 @@ export const startServer = (overrideConfig = null) => {
 };
 
 // Start OSC server
-import { initiateOSC } from './controllers/OscController.js';
+import { initiateOSC, shutdownOSCServer } from './controllers/OscController.js';
 
 export const startOSCServer = (overrideConfig = null) => {
   // Setup default port
@@ -119,3 +119,13 @@ export const startOSCClient = (overrideConfig = null) => {
   const oscOutPort = overrideConfig?.port || config.osc.portOut;
   console.log('initialise OSC Client at port: ', oscOutPort);
 };
+
+export const shutdown = () => {
+  console.log('Node service shutdown');
+
+  // shutdown express server
+  server.close();
+  // shutdown OSC Server
+  shutdownOSCServer();
+  // shutdown OSC Client
+};
diff --git a/server/src/controllers/OscController.js b/server/src/controllers/OscController.js
index 939ee829b..727bbe0a1 100644
--- a/server/src/controllers/OscController.js
+++ b/server/src/controllers/OscController.js
@@ -1,7 +1,13 @@
 import { Server } from 'node-osc';
 
+let oscServer = null;
+
+export const shutdownOSCServer = () => {
+  if (oscServer != null) oscServer.close();
+};
+
 export const initiateOSC = (config) => {
-  const oscServer = new Server(config.port, '0.0.0.0', () => {
+  oscServer = new Server(config.port, '0.0.0.0', () => {
     console.log(`OSC Server is listening on port ${config.port}`);
   });