graceful shutdown protocol implemented for ai highlighter (#5268)

stream-labs · Dec 18, 2024 · bc3c6a7 · bc3c6a7
1 parent efc54a8
commit bc3c6a7
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/app/services/highlighter/ai-highlighter/ai-highlighter.ts b/app/services/highlighter/ai-highlighter/ai-highlighter.ts
@@ -4,6 +4,7 @@ import { AiHighlighterUpdater } from './updater';
 import { duration } from 'moment';
 import { ICoordinates } from '..';
 import kill from 'tree-kill';
+import { getOS, OS } from 'util/operating-systems';
 
 export enum EHighlighterInputTypes {
   KILL = 'kill',
@@ -130,7 +131,13 @@ export function getHighlightClips(
       cancelSignal.addEventListener('abort', () => {
         console.log('ending highlighter process');
         messageBuffer.clear();
-        kill(childProcess.pid!, 'SIGINT');
+
+        // windows doesn't support signals and we have to use the custom graceful shutdown
+        if (getOS() === OS.Windows) {
+          childProcess.stdin?.write('quit\n');
+        } else {
+          kill(childProcess.pid!, 'SIGINT');
+        }
         reject(new Error('Highlight generation canceled'));
       });
     }