Merge pull request arunsetty#1 from ibttf/reset

working on reset

Author: neel-shanmugam

electron/main.ts

@@ -210,6 +210,24 @@ class AppState {
     }
   }
 
+  public clearQueues(): void {
+    // Clear screenshotQueue
+    this.screenshotQueue.forEach((screenshotPath) => {
+      fs.unlink(screenshotPath, (err) => {
+        if (err) console.error(`Error deleting screenshot at ${screenshotPath}:`, err);
+      });
+    });
+    this.screenshotQueue = [];
+
+    // Clear extraScreenshotQueue
+    this.extraScreenshotQueue.forEach((screenshotPath) => {
+      fs.unlink(screenshotPath, (err) => {
+        if (err) console.error(`Error deleting extra screenshot at ${screenshotPath}:`, err);
+      });
+    });
+    this.extraScreenshotQueue = [];
+  }
+
   // Screenshot management methods
   private async takeScreenshot(): Promise<string> {
     if (!this.mainWindow) throw new Error("No main window available")
@@ -510,6 +528,17 @@ class AppState {
     ipcMain.handle("toggle-window", async () => {
       this.toggleMainWindow()
     })
+
+    ipcMain.handle("reset-queues", async () => {
+      try {
+        AppState.getInstance().clearQueues();
+        console.log("Screenshot queues have been cleared.");
+        return { success: true };
+      } catch (error: any) {
+        console.error("Error resetting queues:", error);
+        return { success: false, error: error.message };
+      }
+    });
   }
   // Global shortcuts setup
   public setupGlobalShortcuts(): void {
@@ -533,6 +562,23 @@ class AppState {
       await this.processScreenshots()
     })
 
+    globalShortcut.register("CommandOrControl+R", () => {
+      console.log("Resetting screenshot queues and switching view to 'queue'...");
+  
+      // Clear both screenshot queues
+      this.clearQueues();
+  
+      console.log("Cleared queues.");
+  
+      // **Update the view state to 'queue'**
+      this.view = "queue";
+  
+      // Notify renderer process to switch view to 'queue'
+      if (this.mainWindow && !this.mainWindow.isDestroyed()) {
+        this.mainWindow.webContents.send("reset-view");
+      }
+    });
+
     globalShortcut.register("CommandOrControl+B", () => {
       this.toggleMainWindow()
       // If window exists and we're showing it, bring it to front
@@ -549,6 +595,8 @@ class AppState {
         }
       }
     })
+
+    
   }
 }
 

electron/preload.ts

@@ -4,6 +4,7 @@ import { contextBridge, ipcRenderer } from "electron"
 interface ElectronAPI {
   takeScreenshot: () => Promise<string>
   onSolutionsReady: (callback: (solutions: string) => void) => () => void
+  onResetView: (callback: () => void) => () => void
 }
 
 const PROCESSING_EVENTS = {
@@ -77,5 +78,12 @@ contextBridge.exposeInMainWorld("electronAPI", {
     return () => {
       ipcRenderer.removeListener("solutions-ready", subscription)
     }
-  }
+  },
+  onResetView: (callback: () => void) => { 
+    const subscription = () => callback();
+    ipcRenderer.on("reset-view", subscription);
+    return () => {
+      ipcRenderer.removeListener("reset-view", subscription);
+    };
+  },
 } as ElectronAPI)

src/App.tsx

@@ -6,6 +6,7 @@ import Solutions from "./_pages/Solutions"
 import { QueryClient, QueryClientProvider } from "react-query"
 import axios from "axios"
 
+
 declare global {
   interface Window {
     electronAPI: {
@@ -25,6 +26,7 @@ declare global {
       onProcessingError: (callback: (error: string) => void) => () => void
       onProcessingNoScreenshots: (callback: () => void) => () => void
       updateContentHeight: (height: number) => Promise<void>
+      onResetView: (callback: () => void) => () => void
     }
   }
 }
@@ -36,6 +38,20 @@ const App: React.FC = () => {
   const containerRef = useRef<HTMLDivElement>(null)
 
   // Effect for height monitoring
+
+  useEffect(() => {
+    const cleanup = window.electronAPI.onResetView(() => {
+      console.log("Received 'reset-view' message from main process.")
+      setView("queue")
+      console.log("View reset to 'queue' via Command+R shortcut.")
+      queryClient.invalidateQueries(["screenshots"])
+    })
+
+    return () => {
+      cleanup()
+    }
+  }, [])
+
   useEffect(() => {
     if (!containerRef.current) return