pake-cli 3.0.3 → 3.1.1

package/src-tauri/src/app/setup.rs

@@ -0,0 +1,95 @@
+use std::str::FromStr;
+use std::sync::{Arc, Mutex};
+use std::time::{Duration, Instant};
+use tauri::{
+    menu::{MenuBuilder, MenuItemBuilder},
+    tray::TrayIconBuilder,
+    AppHandle, Manager,
+};
+use tauri_plugin_global_shortcut::{GlobalShortcutExt, Shortcut};
+use tauri_plugin_window_state::{AppHandleExt, StateFlags};
+
+pub fn set_system_tray(app: &AppHandle, show_system_tray: bool) -> tauri::Result<()> {
+    if !show_system_tray {
+        app.remove_tray_by_id("pake-tray");
+        return Ok(());
+    }
+
+    let hide_app = MenuItemBuilder::with_id("hide_app", "Hide").build(app)?;
+    let show_app = MenuItemBuilder::with_id("show_app", "Show").build(app)?;
+    let quit = MenuItemBuilder::with_id("quit", "Quit").build(app)?;
+
+    let menu = MenuBuilder::new(app)
+        .items(&[&hide_app, &show_app, &quit])
+        .build()?;
+
+    app.app_handle().remove_tray_by_id("pake-tray");
+
+    let tray = TrayIconBuilder::new()
+        .menu(&menu)
+        .on_menu_event(move |app, event| match event.id().as_ref() {
+            "hide_app" => {
+                if let Some(window) = app.get_webview_window("pake") {
+                    window.minimize().unwrap();
+                }
+            }
+            "show_app" => {
+                if let Some(window) = app.get_webview_window("pake") {
+                    window.show().unwrap();
+                }
+            }
+            "quit" => {
+                app.save_window_state(StateFlags::all()).unwrap();
+                std::process::exit(0);
+            }
+            _ => (),
+        })
+        .icon(app.default_window_icon().unwrap().clone())
+        .build(app)?;
+
+    tray.set_icon_as_template(false)?;
+    Ok(())
+}
+
+pub fn set_global_shortcut(app: &AppHandle, shortcut: String) -> tauri::Result<()> {
+    if shortcut.is_empty() {
+        return Ok(());
+    }
+
+    let app_handle = app.clone();
+    let shortcut_hotkey = Shortcut::from_str(&shortcut).unwrap();
+    let last_triggered = Arc::new(Mutex::new(Instant::now()));
+
+    app_handle
+        .plugin(
+            tauri_plugin_global_shortcut::Builder::new()
+                .with_handler({
+                    let last_triggered = Arc::clone(&last_triggered);
+                    move |app, event, _shortcut| {
+                        let mut last_triggered = last_triggered.lock().unwrap();
+                        if Instant::now().duration_since(*last_triggered) < Duration::from_millis(300) {
+                            return;
+                        }
+                        *last_triggered = Instant::now();
+
+                        if shortcut_hotkey.eq(event) {
+                            if let Some(window) = app.get_webview_window("pake") {
+                                let is_visible = window.is_visible().unwrap();
+                                if is_visible {
+                                    window.hide().unwrap();
+                                } else {
+                                    window.show().unwrap();
+                                    window.set_focus().unwrap();
+                                }
+                            }
+                        }
+                    }
+                })
+                .build(),
+        )
+        .expect("Failed to set global shortcut");
+
+    app.global_shortcut().register(shortcut_hotkey).unwrap();
+
+    Ok(())
+}

package/src-tauri/src/app/window.rs

@@ -1,11 +1,15 @@
 use crate::app::config::PakeConfig;
+use crate::util::get_data_dir;
 use std::{path::PathBuf, str::FromStr};
-use tauri::{App, Url, WebviewUrl, WebviewWindow, WebviewWindowBuilder};
+use tauri::{App, Config, Url, WebviewUrl, WebviewWindow, WebviewWindowBuilder};
 
 #[cfg(target_os = "macos")]
 use tauri::{Theme, TitleBarStyle};
 
-pub fn get_window(app: &mut App, config: &PakeConfig, _data_dir: PathBuf) -> WebviewWindow {
+pub fn set_window(app: &mut App, config: &PakeConfig, tauri_config: &Config) -> WebviewWindow {
+    let package_name = tauri_config.clone().product_name.unwrap();
+    let _data_dir = get_data_dir(app.handle(), package_name);
+
     let window_config = config
         .windows
         .first()

package/src-tauri/src/inject/event.js

@@ -187,7 +187,6 @@ document.addEventListener('DOMContentLoaded', () => {
   };
 
   const detectAnchorElementClick = e => {
-
     const anchorElement = e.target.closest('a');
 
     if (anchorElement && anchorElement.href) {
@@ -256,6 +255,38 @@ document.addEventListener('DOMContentLoaded', () => {
   );
 });
 
+document.addEventListener('DOMContentLoaded', function () {
+  let permVal = 'granted';
+  window.Notification = function (title, options) {
+    const { invoke } = window.__TAURI__.core;
+    const body = options?.body || '';
+    let icon = options?.icon || '';
+
+    // If the icon is a relative path, convert to full path using URI
+    if (icon.startsWith('/')) {
+      icon = window.location.origin + icon;
+    }
+
+    invoke('send_notification', {
+      params: {
+        title,
+        body,
+        icon,
+      },
+    });
+  };
+
+  window.Notification.requestPermission = async () => 'granted';
+
+  Object.defineProperty(window.Notification, 'permission', {
+    enumerable: true,
+    get: () => permVal,
+    set: v => {
+      permVal = v;
+    },
+  });
+});
+
 function setDefaultZoom() {
   const htmlZoom = window.localStorage.getItem('htmlZoom');
   if (htmlZoom) {

package/src-tauri/src/inject/style.js

@@ -295,10 +295,14 @@ window.addEventListener('DOMContentLoaded', _event => {
       padding-top: 12px;
     }
 
-    #__next header.HeaderBar_header__jn5ju{
+    #__next header.HeaderBar_header__jn5ju {
       padding-top: 16px;
     }
 
+    #root > .excalidraw-app> .excalidraw-container .App-menu.App-menu_top{
+       margin-top: 15px;
+    }
+
     .geist-page nav.dashboard_nav__PRmJv,
     #app > div.layout > div.header-container.showSearchBoxOrHeaderFixed > header > a {
       padding-top:10px;
@@ -334,11 +338,11 @@ window.addEventListener('DOMContentLoaded', _event => {
     #__next>div>div>.flex.h-screen.w-full.flex-col.items-center {
       padding-top: 20px;
     }
-    
+
     body > div.relative.flex.h-full.w-full.overflow-hidden.transition-colors.z-0 > div.z-\\[21\\].flex-shrink-0.overflow-x-hidden.bg-token-sidebar-surface-primary.max-md\\:\\!w-0 > div > div > div > nav > div.flex.justify-between.h-\\[60px\\].items-center.md\\:h-header-height {
        padding-top: 25px;
     }
-    
+
     body > div.relative.flex.h-full.w-full.overflow-hidden.transition-colors.z-0 > div.relative.flex.h-full.max-w-full.flex-1.flex-col.overflow-hidden > main > div.composer-parent.flex.h-full.flex-col.focus-visible\\:outline-0 > div.flex-1.overflow-hidden.\\@container\\/thread > div > div.absolute.left-0.right-0 > div{
       padding-top: 35px;
     }
@@ -384,10 +388,14 @@ window.addEventListener('DOMContentLoaded', _event => {
       padding-top:15px;
     }
 
+    #app-root .mat-mdc-tooltip-trigger.main-menu-button.mdc-icon-button {
+      margin-top: 15px;
+    }
+
     .lark > .main-wrapper [data-testid="aside"] {
       top: 15px;
     }
-    
+
     #global > div.header-container > .mask-paper {
       padding-top: 20px;
     }

package/src-tauri/src/lib.rs

@@ -2,120 +2,70 @@
 mod app;
 mod util;
 
-use app::{invoke, menu::set_system_tray, window};
-use invoke::{download_file, download_file_by_binary};
-use std::str::FromStr;
-use std::sync::{Arc, Mutex};
-use std::time::{Duration, Instant};
-
 use tauri::Manager;
-use tauri_plugin_global_shortcut::{GlobalShortcutExt, Shortcut};
-use tauri_plugin_window_state::Builder as windowStatePlugin;
-use util::{get_data_dir, get_pake_config};
-use window::get_window;
+use tauri_plugin_window_state::Builder as WindowStatePlugin;
+use tauri_plugin_window_state::StateFlags;
+
+#[cfg(target_os = "macos")]
+use std::time::Duration;
+
+use app::{
+    invoke::{download_file, download_file_by_binary, send_notification},
+    setup::{set_global_shortcut, set_system_tray},
+    window::set_window,
+};
+use util::get_pake_config;
 
 pub fn run_app() {
     let (pake_config, tauri_config) = get_pake_config();
-
     let tauri_app = tauri::Builder::default();
 
     let show_system_tray = pake_config.show_system_tray();
-
-    // Save the value of toggle_app_shortcut before pake_config is moved
     let activation_shortcut = pake_config.windows[0].activation_shortcut.clone();
     let init_fullscreen = pake_config.windows[0].fullscreen;
 
-    let window_state_plugin = if init_fullscreen {
-        windowStatePlugin::default()
-            .with_state_flags(tauri_plugin_window_state::StateFlags::FULLSCREEN)
-            .build()
-    } else {
-        windowStatePlugin::default().build()
-    };
+    let window_state_plugin = WindowStatePlugin::default()
+        .with_state_flags(if init_fullscreen {
+            StateFlags::FULLSCREEN
+        } else {
+            // Prevent flickering on the first open.
+            StateFlags::all() & !StateFlags::VISIBLE
+        })
+        .build();
 
+    #[allow(deprecated)]
     tauri_app
         .plugin(window_state_plugin)
         .plugin(tauri_plugin_oauth::init())
         .plugin(tauri_plugin_http::init())
         .plugin(tauri_plugin_shell::init())
+        .plugin(tauri_plugin_notification::init())
         .plugin(tauri_plugin_single_instance::init(|_, _, _| ()))
         .invoke_handler(tauri::generate_handler![
             download_file,
-            download_file_by_binary
+            download_file_by_binary,
+            send_notification,
         ])
         .setup(move |app| {
-            let data_dir = get_data_dir(app.app_handle(), tauri_config.clone());
-
-            let _window = get_window(app, &pake_config, data_dir);
-
-            // Prevent initial shaking
-            _window.show().unwrap();
-
-            if show_system_tray {
-                let _ = set_system_tray(app.app_handle());
-            } else {
-                app.app_handle().remove_tray_by_id("pake-tray");
-            }
-
-            if !activation_shortcut.is_empty() {
-                let app_handle = app.app_handle().clone();
-                let shortcut_hotkey = Shortcut::from_str(activation_shortcut.as_str()).unwrap();
-                let last_triggered = Arc::new(Mutex::new(Instant::now()));
-
-                app_handle
-                    .plugin(
-                        tauri_plugin_global_shortcut::Builder::new()
-                            .with_handler({
-                                let last_triggered = Arc::clone(&last_triggered);
-                                move |app, event, _shortcut| {
-                                    // Fixed the bug of tauri's hidden call, which caused repeated execution
-                                    let now = Instant::now();
-                                    let mut last = last_triggered.lock().unwrap();
-                                    if now.duration_since(*last) < Duration::from_millis(500) {
-                                        return;
-                                    }
-                                    *last = now;
-
-                                    if shortcut_hotkey.eq(event) {
-                                        let window = app.get_webview_window("pake").unwrap();
-                                        let is_visible = window.is_visible().unwrap();
-
-                                        match is_visible {
-                                            true => {
-                                                window.minimize().unwrap();
-                                            }
-                                            false => {
-                                                window.unminimize().unwrap();
-                                                window.set_focus().unwrap();
-                                            }
-                                        }
-                                    }
-                                }
-                            })
-                            .build(),
-                    )
-                    .expect("Error registering global evoke shortcuts!");
-
-                app.global_shortcut().register(shortcut_hotkey)?;
-            }
-
+            let window = set_window(app, &pake_config, &tauri_config);
+            set_system_tray(app.app_handle(), show_system_tray).unwrap();
+            set_global_shortcut(app.app_handle(), activation_shortcut).unwrap();
+            // Prevent flickering on the first open.
+            window.show().unwrap();
             Ok(())
         })
         .on_window_event(|_window, _event| {
             #[cfg(target_os = "macos")]
             if let tauri::WindowEvent::CloseRequested { api, .. } = _event {
                 let window = _window.clone();
-                {
-                    tauri::async_runtime::spawn(async move {
-                        if window.is_fullscreen().unwrap_or(false) {
-                            window.set_fullscreen(false).unwrap();
-                            // Give a small delay to ensure the full-screen exit operation is completed.
-                            tokio::time::sleep(Duration::from_millis(900)).await;
-                        }
-                        window.minimize().unwrap();
-                        window.hide().unwrap();
-                    });
-                }
+                tauri::async_runtime::spawn(async move {
+                    if window.is_fullscreen().unwrap_or(false) {
+                        window.set_fullscreen(false).unwrap();
+                        tokio::time::sleep(Duration::from_millis(900)).await;
+                    }
+                    window.minimize().unwrap();
+                    window.hide().unwrap();
+                });
                 api.prevent_close();
             }
         })

package/src-tauri/src/util.rs

@@ -23,9 +23,8 @@ pub fn get_pake_config() -> (PakeConfig, Config) {
     (pake_config, tauri_config)
 }
 
-pub fn get_data_dir(app: &AppHandle, _tauri_config: Config) -> PathBuf {
+pub fn get_data_dir(app: &AppHandle, package_name: String) -> PathBuf {
     {
-        let package_name = _tauri_config.product_name.unwrap();
         let data_dir = app
             .path()
             .config_dir()

package/src-tauri/src/app/menu.rs

@@ -1,36 +0,0 @@
-use tauri::{
-    menu::{MenuBuilder, MenuItemBuilder},
-    tray::TrayIconBuilder,
-    AppHandle, Manager,
-};
-use tauri_plugin_window_state::{AppHandleExt, StateFlags};
-
-pub fn set_system_tray(app: &AppHandle) -> tauri::Result<()> {
-    let hide_app = MenuItemBuilder::with_id("hide_app", "Hide").build(app)?;
-    let show_app = MenuItemBuilder::with_id("show_app", "Show").build(app)?;
-    let quit = MenuItemBuilder::with_id("quit", "Quit").build(app)?;
-    let menu = MenuBuilder::new(app)
-        .items(&[&hide_app, &show_app, &quit])
-        .build()?;
-    app.app_handle().remove_tray_by_id("pake-tray");
-    let tray = TrayIconBuilder::new()
-        .menu(&menu)
-        .on_menu_event(move |app, event| match event.id().as_ref() {
-            "hide_app" => {
-                app.get_webview_window("pake").unwrap().minimize().unwrap();
-            }
-            "show_app" => {
-                app.get_webview_window("pake").unwrap().show().unwrap();
-            }
-            "quit" => {
-                let _res = app.save_window_state(StateFlags::all());
-                std::process::exit(0);
-            }
-            _ => (),
-        })
-        .icon(app.default_window_icon().unwrap().clone())
-        .build(app)?;
-
-    tray.set_icon_as_template(false)?;
-    Ok(())
-}