From 42ecdb179139e3187ee0115242a7d0c14bbca142 Mon Sep 17 00:00:00 2001
From: Oleh Polisan <polisanoleg@gmail.com>
Date: Wed, 19 Jun 2024 15:58:54 +0300
Subject: [PATCH] fix: force regenerate icons if can't open

---
 src/main/mainWindow.ts | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/src/main/mainWindow.ts b/src/main/mainWindow.ts
index d752f89..2c18968 100755
--- a/src/main/mainWindow.ts
+++ b/src/main/mainWindow.ts
@@ -506,7 +506,12 @@ export async function createWindows() {
 export async function setTrayIcon(iconName: string) {
     if (!tray || tray.isDestroyed()) return;
     if (iconName !== "icon") {
-        tray.setImage(join(DATA_DIR, "TrayIcons", iconName + ".png"));
+        try {
+            tray.setImage(join(DATA_DIR, "TrayIcons", iconName + ".png"));
+        } catch (error) {
+            console.log("Error: ", error, "Regenerating tray icons.");
+            generateTrayIcons(true);
+        }
         return;
     }
     tray.setImage(join(STATIC_DIR, "icon.png"));
@@ -528,10 +533,10 @@ export async function createTrayIcon(iconName: string, iconDataURL: string) {
     mainWin.webContents.send(IpcEvents.SET_CURRENT_VOICE_TRAY_ICON);
 }
 
-export async function generateTrayIcons() {
+export async function generateTrayIcons(force = false) {
     // this function generates tray icons as .png's in Vesktop cache for future use
     mkdirSync(join(DATA_DIR, "TrayIcons"), { recursive: true });
-    if (!Settings.store.trayCustom) {
+    if (force || !Settings.store.trayCustom) {
         const Icons = ["speaking", "muted", "deafened", "idle"];
         for (const icon of Icons) {
             mainWin.webContents.send(IpcEvents.CREATE_TRAY_ICON_REQUEST, icon);