feat(app): terminate any running AI processes before closing or resta…

…rting #22
KindaBrazy · Nov 2, 2024 · b683de6 · b683de6
1 parent 454fe3a
commit b683de6
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 5 deletions.
diff --git a/src/renderer/src/App/Components/MainContents/StatusBar.tsx b/src/renderer/src/App/Components/MainContents/StatusBar.tsx
@@ -8,8 +8,6 @@ const classNames = 'flex size-full items-center overflow-x-scroll scrollbar-hide
 const StatusBar = () => {
   const [statusBar] = useState(extensionsData.statusBar);
 
-  console.log(statusBar);
-
   const isEmptyAdd = useMemo(() => {
     return isEmpty(statusBar.addStart) && isEmpty(statusBar.addCenter) && isEmpty(statusBar.addEnd);
   }, [statusBar]);

diff --git a/src/renderer/src/App/Components/TitleBar/WindowButtons.tsx b/src/renderer/src/App/Components/TitleBar/WindowButtons.tsx
@@ -33,12 +33,24 @@ export default function WindowButtons() {
     [dispatch],
   );
 
+  const close = useCallback(() => {
+    rendererIpc.pty.process('stop', '');
+    // Close the app with a delay so ensure pty terminated
+    setTimeout(() => rendererIpc.win.changeWinState('close'), 200);
+  }, []);
+
+  const restart = useCallback(() => {
+    rendererIpc.pty.process('stop', '');
+    // Close the app with a delay so ensure pty terminated
+    setTimeout(() => rendererIpc.win.changeWinState('restart'), 200);
+  }, []);
+
   const changeWindowState = useCallback(
     (operation: WindowOperation) => {
       if (operation === 'close') {
         rendererIpc.storage.update('app', {lastPage: currentPage});
         if (isHotkeyPressed('control') || !showCloseConfirm) {
-          rendererIpc.win.changeWinState('close');
+          close();
         } else if (!isConfigVisible) {
           setIsConfigVisible(true);
           Modal.error({
@@ -67,10 +79,10 @@ export default function WindowButtons() {
                   Stay
                 </Button>
                 <div className="space-x-2">
-                  <Button size="sm" color="warning" onPress={() => rendererIpc.win.changeWinState('restart')}>
+                  <Button size="sm" color="warning" onPress={restart}>
                     Restart
                   </Button>
-                  <Button size="sm" color="danger" onPress={() => rendererIpc.win.changeWinState('close')}>
+                  <Button size="sm" color="danger" onPress={close}>
                     Exit
                   </Button>
                 </div>