more theme checking

Signed-off-by: FTCHD <144691102+FTCHD@users.noreply.github.com>

lib/store.ts

@@ -111,6 +111,8 @@ interface PersistedState {
     lastSkippedVersion: string | undefined
 
     hideStartup: boolean
+
+    theme: 'light' | 'dark' | undefined
 }
 
 const getStorage = (store: LazyStore): StateStorage => ({
@@ -253,6 +255,8 @@ export const usePersistedStore = create<PersistedState>()(
             lastSkippedVersion: undefined,
 
             hideStartup: false,
+
+            theme: undefined,
         }),
         {
             name: 'store',

lib/tray.ts

@@ -9,6 +9,7 @@ import { getAllWindows, getCurrentWindow } from '@tauri-apps/api/window'
 import { ask } from '@tauri-apps/plugin-dialog'
 import { platform } from '@tauri-apps/plugin-os'
 import { buildMenu } from './menu'
+import { usePersistedStore } from './store'
 import { resetMainWindow } from './window'
 
 export async function triggerTrayRebuild() {
@@ -24,19 +25,41 @@ async function getTray() {
 }
 
 async function resolveTrayIconForTheme() {
+    const existingTheme = usePersistedStore.getState().theme
+    if (existingTheme) {
+        return existingTheme === 'dark' ? 'icons/favicon/icon.png' : 'icons/favicon/icon-light.png'
+    }
+
     let theme: 'light' | 'dark' = 'dark'
     try {
-        const oneTheme = (await invoke<string>('get_system_theme')) || 'dark'
+        const oneTheme = await invoke<string>('get_system_theme')
         const currentWindow = getCurrentWindow()
         const twoTheme = await currentWindow.theme()
 
-        if (oneTheme === twoTheme) {
+        if (oneTheme !== twoTheme) {
+            const answer = await ask(
+                'Could not determine your system theme, please select the correct one below',
+                {
+                    title: 'Hmm...',
+                    okLabel: 'Dark',
+                    cancelLabel: 'Light',
+                }
+            )
+
+            if (answer) {
+                theme = 'dark'
+            } else {
+                theme = 'light'
+            }
+        } else {
             theme = oneTheme
         }
     } catch {}
 
     console.log('[resolveTrayIconForTheme] theme', theme)
 
+    usePersistedStore.setState({ theme })
+
     const pickedPath = theme === 'dark' ? 'icons/favicon/icon.png' : 'icons/favicon/icon-light.png'
     console.log('[resolveTrayIconForTheme] pickedPath', pickedPath)