enh: Ask user: Capture error in Sentry when app crashes

src/config/comfySettings.ts

@@ -34,6 +34,22 @@ export class ComfySettings {
     }
   }
 
+  public saveSettings() {
+    try {
+      const dirname = path.dirname(this.filePath);
+      if (!fs.existsSync(dirname)) {
+        log.info(`Settings directory ${dirname} does not exist, creating ...`);
+        fs.mkdirSync(dirname, {
+          recursive: true,
+        });
+      }
+
+      fs.writeFileSync(this.filePath, JSON.stringify(this.settings), 'utf-8');
+    } catch (error) {
+      log.error(`Failed to save settings to ${this.filePath}:`, error);
+    }
+  }
+
   get autoUpdate(): boolean {
     return this.settings['Comfy-Desktop.AutoUpdate'] ?? true;
   }
@@ -42,6 +58,10 @@ export class ComfySettings {
     return this.settings['Comfy-Desktop.SendCrashStatistics'] ?? true;
   }
 
+  set sendCrashStatistics(value: boolean) {
+    this.settings['Comfy-Desktop.SendCrashStatistics'] = value;
+  }
+
   public reload(): void {
     this.settings = this.loadSettings();
   }

src/constants.ts

@@ -17,6 +17,7 @@ export const IPC_CHANNELS = {
   CANCEL_DOWNLOAD: 'cancel-download',
   DELETE_MODEL: 'delete-model',
   GET_ALL_DOWNLOADS: 'get-all-downloads',
+  SET_SEND_CRASH_REPORTS: 'set-send-crash-reports',
   GET_ELECTRON_VERSION: 'get-electron-version',
   SEND_ERROR_TO_SENTRY: 'send-error-to-sentry',
   GET_BASE_PATH: 'get-base-path',

src/main.ts

@@ -90,6 +90,7 @@ const gotTheLock = app.requestSingleInstanceLock();
 
 if (!gotTheLock) {
   log.info('App already running. Exiting...');
+
   app.quit();
 } else {
   store = new Store<StoreType>();
@@ -106,13 +107,24 @@ if (!gotTheLock) {
   Sentry.init({
     dsn: SENTRY_URL_ENDPOINT,
     autoSessionTracking: false,
-    beforeSend(event, hint) {
+    async beforeSend(event, hint) {
       if (event.extra?.comfyUIExecutionError) {
         return event;
       }
 
-      //TODO (use default pop up behavior).
-      return event;
+      let sendCrashReport = comfySettings.sendCrashStatistics;
+
+      if (!sendCrashReport) {
+        const { response } = await dialog.showMessageBox({
+          title: 'Send Crash Statistics',
+          message: `Would you like to send crash statistics to the team?`,
+          buttons: ['Always send crash reports', 'Do not send crash report'],
+        });
+
+        sendCrashReport = response === 0;
+      }
+
+      return sendCrashReport ? event : null;
     },
     integrations: [
       Sentry.childProcessIntegration({
@@ -198,6 +210,9 @@ if (!gotTheLock) {
       ipcMain.on(IPC_CHANNELS.OPEN_DEV_TOOLS, () => {
         mainWindow?.webContents.openDevTools();
       });
+      ipcMain.on(IPC_CHANNELS.SET_SEND_CRASH_REPORTS, (_event, value) => {
+        comfySettings.sendCrashStatistics = value;
+      });
       ipcMain.handle(IPC_CHANNELS.IS_PACKAGED, () => {
         return app.isPackaged;
       });

src/preload.ts

@@ -50,6 +50,9 @@ const electronAPI = {
   selectSetupDirectory: (directory: string) => {
     ipcRenderer.send(IPC_CHANNELS.SELECTED_DIRECTORY, directory);
   },
+  setSendCrashReports: (sendCrashReports: boolean) => {
+    ipcRenderer.send(IPC_CHANNELS.SET_SEND_CRASH_REPORTS, sendCrashReports);
+  },
   openDialog: (options: Electron.OpenDialogOptions) => {
     return ipcRenderer.invoke(IPC_CHANNELS.OPEN_DIALOG, options);
   },