tray menus with always-on-top toggle

Author: yujonglee

Cargo.lock

@@ -1,7 +1,7 @@
 {
   "$schema": "https://schema.tauri.app/config/2",
   "version": "0.0.2",
-  "productName": "Hyprnote",
+  "productName": "Hyprnote Dev",
   "mainBinaryName": "Hyprnote Dev",
   "identifier": "com.hyprnote.dev",
   "build": {

apps/desktop/src-tauri/tauri.conf.json

@@ -16,7 +16,11 @@ specta-typescript = { workspace = true }
 
 [dependencies]
 tauri = { workspace = true, features = ["tray-icon", "image-png"] }
+tauri-plugin-store2 = { workspace = true }
 tauri-plugin-windows = { workspace = true }
 
 specta = { workspace = true }
+strum = { workspace = true, features = ["derive"] }
 tauri-specta = { workspace = true, features = ["derive", "typescript"] }
+
+webbrowser = "1.0.4"

plugins/tray/Cargo.toml

@@ -1,20 +1,29 @@
+use crate::PLUGIN_NAME;
 use tauri::{
     image::Image,
-    menu::{Menu, MenuId, MenuItem, PredefinedMenuItem},
+    menu::{CheckMenuItem, Menu, MenuId, MenuItem, PredefinedMenuItem},
     tray::TrayIconBuilder,
-    AppHandle, Manager, Result,
+    AppHandle, Result,
 };
 
 pub enum TrayItem {
-    Open,
+    Info,
+    Github,
+    Twitter,
+    Discord,
     Quit,
+    AlwaysOnTop,
 }
 
 impl From<TrayItem> for MenuId {
     fn from(value: TrayItem) -> Self {
         match value {
-            TrayItem::Open => "open_hypr",
+            TrayItem::Info => "info_hypr",
+            TrayItem::Github => "github_hypr",
+            TrayItem::Twitter => "twitter_hypr",
+            TrayItem::Discord => "discord_hypr",
             TrayItem::Quit => "quit_hypr",
+            TrayItem::AlwaysOnTop => "always_on_top_hypr",
         }
         .into()
     }
@@ -24,8 +33,12 @@ impl From<MenuId> for TrayItem {
     fn from(id: MenuId) -> Self {
         let id = id.0.as_str();
         match id {
-            "open_hypr" => TrayItem::Open,
+            "info_hypr" => TrayItem::Info,
+            "github_hypr" => TrayItem::Github,
+            "twitter_hypr" => TrayItem::Twitter,
+            "discord_hypr" => TrayItem::Discord,
             "quit_hypr" => TrayItem::Quit,
+            "always_on_top_hypr" => TrayItem::AlwaysOnTop,
             _ => unreachable!(),
         }
     }
@@ -35,16 +48,35 @@ pub trait TrayPluginExt<R: tauri::Runtime> {
     fn create_tray(&self) -> Result<()>;
 }
 
+#[derive(Debug, PartialEq, Eq, Hash, strum::Display)]
+enum StoreKey {
+    MainWindowAlwaysOnTop,
+}
+
+impl tauri_plugin_store2::ScopedStoreKey for StoreKey {}
+
 impl<T: tauri::Manager<tauri::Wry>> TrayPluginExt<tauri::Wry> for T {
     fn create_tray(&self) -> Result<()> {
         let app = self.app_handle();
 
+        let store = {
+            use tauri_plugin_store2::StorePluginExt;
+            app.scoped_store::<StoreKey>(PLUGIN_NAME).unwrap()
+        };
+
+        let always_on_top = always_on_top_menu(app, always_on_top_state(&store))?;
+
         let menu = Menu::with_items(
             app,
             &[
-                &MenuItem::with_id(app, TrayItem::Open, "Open", true, None::<&str>)?,
+                &info_menu(app)?,
+                &github_menu(app)?,
+                &twitter_menu(app)?,
+                &discord_menu(app)?,
+                &PredefinedMenuItem::separator(app)?,
+                &always_on_top,
                 &PredefinedMenuItem::separator(app)?,
-                &MenuItem::with_id(app, TrayItem::Quit, "Quit", true, None::<&str>)?,
+                &quit_menu(app)?,
             ],
         )?;
 
@@ -57,10 +89,34 @@ impl<T: tauri::Manager<tauri::Wry>> TrayPluginExt<tauri::Wry> for T {
             .show_menu_on_left_click(true)
             .on_menu_event({
                 move |app: &AppHandle, event| match TrayItem::from(event.id.clone()) {
-                    TrayItem::Open => {
-                        if let Some(window) = app.get_webview_window("main") {
-                            window.show().unwrap();
-                            window.set_focus().unwrap();
+                    TrayItem::Info => {}
+                    TrayItem::Github => {
+                        let _ = webbrowser::open("https://github.com/fastrepl/hyprnote");
+                    }
+                    TrayItem::Twitter => {
+                        let _ = webbrowser::open("https://hyprnote.com/x");
+                    }
+                    TrayItem::Discord => {
+                        let _ = webbrowser::open("https://hyprnote.com/discord");
+                    }
+                    TrayItem::AlwaysOnTop => {
+                        let next_always_on_top = !always_on_top_state(&store);
+
+                        let toggled = {
+                            use tauri_plugin_windows::HyprWindowId;
+                            if let Some(window) = HyprWindowId::Main.get(app) {
+                                window.set_always_on_top(next_always_on_top).is_ok()
+                            } else {
+                                false
+                            }
+                        };
+
+                        if toggled {
+                            store
+                                .set(StoreKey::MainWindowAlwaysOnTop, next_always_on_top)
+                                .unwrap();
+
+                            always_on_top.set_checked(next_always_on_top).unwrap();
                         }
                     }
                     TrayItem::Quit => {
@@ -77,3 +133,55 @@ impl<T: tauri::Manager<tauri::Wry>> TrayPluginExt<tauri::Wry> for T {
         Ok(())
     }
 }
+
+fn info_menu<R: tauri::Runtime>(app: &AppHandle<R>) -> Result<MenuItem<R>> {
+    let info = app.package_info();
+
+    let display_name = match info.name.as_str() {
+        "Hyprnote" => format!("Hyprnote v{}", &info.version),
+        "Hyprnote Dev" => format!("Hyprnote v{} (dev)", &info.version),
+        "Hyprnote Nightly" => format!("Hyprnote v{} (nightly)", &info.version),
+        _ => unreachable!(),
+    };
+
+    MenuItem::with_id(app, TrayItem::Info, &display_name, false, None::<&str>)
+}
+
+fn github_menu<R: tauri::Runtime>(app: &AppHandle<R>) -> Result<MenuItem<R>> {
+    MenuItem::with_id(app, TrayItem::Github, "GitHub", true, None::<&str>)
+}
+
+fn twitter_menu<R: tauri::Runtime>(app: &AppHandle<R>) -> Result<MenuItem<R>> {
+    MenuItem::with_id(app, TrayItem::Twitter, "Twitter", true, None::<&str>)
+}
+
+fn discord_menu<R: tauri::Runtime>(app: &AppHandle<R>) -> Result<MenuItem<R>> {
+    MenuItem::with_id(app, TrayItem::Discord, "Discord", true, None::<&str>)
+}
+
+fn quit_menu<R: tauri::Runtime>(app: &AppHandle<R>) -> Result<MenuItem<R>> {
+    MenuItem::with_id(app, TrayItem::Quit, "Quit Completely", true, Some("cmd+q"))
+}
+
+fn always_on_top_menu<R: tauri::Runtime>(
+    app: &AppHandle<R>,
+    initial_state: bool,
+) -> Result<CheckMenuItem<R>> {
+    CheckMenuItem::with_id(
+        app,
+        TrayItem::AlwaysOnTop,
+        "Always on top",
+        true,
+        initial_state,
+        Some("cmd+shift+t"),
+    )
+}
+
+fn always_on_top_state<R: tauri::Runtime>(
+    store: &tauri_plugin_store2::ScopedStore<R, StoreKey>,
+) -> bool {
+    store
+        .get::<bool>(StoreKey::MainWindowAlwaysOnTop)
+        .unwrap_or(Some(false))
+        .unwrap_or(false)
+}