pake-cli 3.11.4 → 3.11.6

package/src-tauri/src/inject/{component.js → fullscreen.js}

@@ -1,28 +1,10 @@
-document.addEventListener("DOMContentLoaded", () => {
-  // Toast
-  function pakeToast(msg) {
-    const m = document.createElement("div");
-    m.innerHTML = msg;
-    m.style.cssText =
-      "max-width:60%;min-width: 80px;padding:0 12px;height: 32px;color: rgb(255, 255, 255);line-height: 32px;text-align: center;border-radius: 8px;position: fixed; bottom:24px;right: 28px;z-index: 999999;background: rgba(0, 0, 0,.8);font-size: 13px;";
-    document.body.appendChild(m);
-    setTimeout(function () {
-      const d = 0.5;
-      m.style.transition =
-        "transform " + d + "s ease-in, opacity " + d + "s ease-in";
-      m.style.opacity = "0";
-      setTimeout(function () {
-        document.body.removeChild(m);
-      }, d * 1000);
-    }, 3000);
-  }
-
-  window.pakeToast = pakeToast;
-});
-
-// Polyfill for HTML5 Fullscreen API in Tauri webview
-// This bridges the HTML5 Fullscreen API to Tauri's native window fullscreen
-// Works for all video sites (YouTube, Vimeo, Bilibili, etc.)
+// Polyfill for HTML5 Fullscreen API in Tauri webview.
+// Bridges the standard requestFullscreen / exitFullscreen DOM API to Tauri's
+// native window fullscreen so video sites (YouTube, Vimeo, Bilibili, etc.) can
+// go true fullscreen on their player buttons.
+//
+// Split out from component.js so a future CLI flag (or custom.js override)
+// can short-circuit the polyfill for apps that don't need video fullscreen.
 (function () {
   if (window.__PAKE_FULLSCREEN_POLYFILL__) return;
   window.__PAKE_FULLSCREEN_POLYFILL__ = true;
@@ -42,7 +24,6 @@ document.addEventListener("DOMContentLoaded", () => {
     let wasInBody = false;
     let monitorId = null;
 
-    // Inject fullscreen styles
     if (!document.getElementById("pake-fullscreen-style")) {
       const styleEl = document.createElement("style");
       styleEl.id = "pake-fullscreen-style";
@@ -93,7 +74,6 @@ document.addEventListener("DOMContentLoaded", () => {
       monitorId = null;
     }
 
-    // Find the actual video element
     function findMediaElement() {
       const videos = document.querySelectorAll("video");
       if (videos.length > 0) {
@@ -112,11 +92,9 @@ document.addEventListener("DOMContentLoaded", () => {
       return null;
     }
 
-    // Enter fullscreen
     function enterFullscreen(element) {
       fullscreenElement = element;
 
-      // If html/body element, find the video instead
       let targetElement = element;
       if (element === document.documentElement || element === document.body) {
         const mediaElement = findMediaElement();
@@ -130,7 +108,6 @@ document.addEventListener("DOMContentLoaded", () => {
         actualFullscreenElement = element;
       }
 
-      // Save original state
       originalStyles = {
         position: targetElement.style.position,
         top: targetElement.style.top,
@@ -152,7 +129,6 @@ document.addEventListener("DOMContentLoaded", () => {
         originalNextSibling = targetElement.nextSibling;
       }
 
-      // Apply fullscreen
       targetElement.classList.add("pake-fullscreen-element");
       document.body.classList.add("pake-fullscreen-active");
 
@@ -160,7 +136,6 @@ document.addEventListener("DOMContentLoaded", () => {
         document.body.appendChild(targetElement);
       }
 
-      // Fullscreen window
       appWindow.setFullscreen(true).then(() => {
         startFullscreenMonitor();
         const event = new Event("fullscreenchange", { bubbles: true });
@@ -177,7 +152,6 @@ document.addEventListener("DOMContentLoaded", () => {
       return Promise.resolve();
     }
 
-    // Exit fullscreen
     function exitFullscreen() {
       if (!fullscreenElement) {
         return Promise.resolve();
@@ -188,7 +162,6 @@ document.addEventListener("DOMContentLoaded", () => {
       const exitingElement = fullscreenElement;
       const targetElement = actualFullscreenElement;
 
-      // Restore styles and position
       targetElement.classList.remove("pake-fullscreen-element");
       document.body.classList.remove("pake-fullscreen-active");
 
@@ -209,7 +182,6 @@ document.addEventListener("DOMContentLoaded", () => {
         }
       }
 
-      // Reset state
       fullscreenElement = null;
       actualFullscreenElement = null;
       originalStyles = null;
@@ -217,7 +189,6 @@ document.addEventListener("DOMContentLoaded", () => {
       originalNextSibling = null;
       wasInBody = false;
 
-      // Exit window fullscreen
       return appWindow.setFullscreen(false).then(() => {
         const event = new Event("fullscreenchange", { bubbles: true });
         document.dispatchEvent(event);
@@ -231,7 +202,6 @@ document.addEventListener("DOMContentLoaded", () => {
       });
     }
 
-    // Override fullscreenEnabled
     Object.defineProperty(document, "fullscreenEnabled", {
       get: () => true,
       configurable: true,
@@ -241,7 +211,6 @@ document.addEventListener("DOMContentLoaded", () => {
       configurable: true,
     });
 
-    // Override fullscreenElement
     Object.defineProperty(document, "fullscreenElement", {
       get: () => fullscreenElement,
       configurable: true,
@@ -255,7 +224,6 @@ document.addEventListener("DOMContentLoaded", () => {
       configurable: true,
     });
 
-    // Override requestFullscreen
     Element.prototype.requestFullscreen = function () {
       return enterFullscreen(this);
     };
@@ -266,12 +234,10 @@ document.addEventListener("DOMContentLoaded", () => {
       return enterFullscreen(this);
     };
 
-    // Override exitFullscreen
     document.exitFullscreen = exitFullscreen;
     document.webkitExitFullscreen = exitFullscreen;
     document.webkitCancelFullScreen = exitFullscreen;
 
-    // Handle Escape key
     document.addEventListener(
       "keydown",
       (e) => {

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

@@ -0,0 +1,22 @@
+// Lightweight in-page toast used by Rust `show_toast` (download status, etc).
+// Kept tiny and always loaded so the Rust side can rely on `window.pakeToast`.
+document.addEventListener("DOMContentLoaded", () => {
+  function pakeToast(msg) {
+    const m = document.createElement("div");
+    m.innerHTML = msg;
+    m.style.cssText =
+      "max-width:60%;min-width: 80px;padding:0 12px;height: 32px;color: rgb(255, 255, 255);line-height: 32px;text-align: center;border-radius: 8px;position: fixed; bottom:24px;right: 28px;z-index: 999999;background: rgba(0, 0, 0,.8);font-size: 13px;";
+    document.body.appendChild(m);
+    setTimeout(function () {
+      const d = 0.5;
+      m.style.transition =
+        "transform " + d + "s ease-in, opacity " + d + "s ease-in";
+      m.style.opacity = "0";
+      setTimeout(function () {
+        document.body.removeChild(m);
+      }, d * 1000);
+    }, 3000);
+  }
+
+  window.pakeToast = pakeToast;
+});

package/src-tauri/src/lib.rs

@@ -13,7 +13,8 @@ const WINDOW_SHOW_DELAY: u64 = 50;
 
 use app::{
     invoke::{
-        clear_cache_and_restart, download_file, download_file_by_binary, send_notification,
+        clear_cache_and_restart, clear_dock_badge, download_file, download_file_by_binary,
+        increment_dock_badge, send_notification, set_dock_badge, set_dock_badge_label,
         update_theme_mode,
     },
     setup::{set_global_shortcut, set_system_tray},
@@ -42,6 +43,7 @@ pub fn run_app() {
     let start_to_tray = pake_config.windows[0].start_to_tray && show_system_tray; // Only valid when tray is enabled
     let multi_instance = pake_config.multi_instance;
     let multi_window = pake_config.multi_window;
+    let enable_find = pake_config.windows[0].enable_find;
 
     let window_state_plugin = WindowStatePlugin::default()
         .with_state_flags(if init_fullscreen {
@@ -81,6 +83,10 @@ pub fn run_app() {
             download_file,
             download_file_by_binary,
             send_notification,
+            increment_dock_badge,
+            set_dock_badge,
+            set_dock_badge_label,
+            clear_dock_badge,
             update_theme_mode,
             clear_cache_and_restart,
         ])
@@ -93,8 +99,7 @@ pub fn run_app() {
             // --- Menu Construction Start ---
             #[cfg(target_os = "macos")]
             {
-                let menu = app::menu::get_menu(app.app_handle(), multi_window)?;
-                app.set_menu(menu)?;
+                app::menu::set_app_menu(app.app_handle(), multi_window, enable_find)?;
 
                 // Event Handling for Custom Menu Item
                 app.on_menu_event(move |app_handle, event| {
@@ -103,7 +108,7 @@ pub fn run_app() {
             }
             // --- Menu Construction End ---
 
-            let window = set_window(app.app_handle(), &pake_config, &tauri_config);
+            let window = set_window(app.app_handle(), &pake_config, &tauri_config)?;
             set_system_tray(
                 app.app_handle(),
                 show_system_tray,
@@ -174,7 +179,10 @@ pub fn run_app() {
             }
         })
         .build(tauri::generate_context!())
-        .expect("error while building tauri application")
+        .unwrap_or_else(|error| {
+            eprintln!("[Pake] Fatal error while building Tauri application: {error}");
+            std::process::exit(1);
+        })
         .run(|_app, _event| {
             // Handle macOS dock icon click to reopen hidden window
             #[cfg(target_os = "macos")]

package/src-tauri/src/util.rs

@@ -23,25 +23,35 @@ pub fn get_pake_config() -> (PakeConfig, Config) {
     (pake_config, tauri_config)
 }
 
-pub fn get_data_dir(app: &AppHandle, package_name: String) -> PathBuf {
-    {
-        let data_dir = app
-            .path()
-            .config_dir()
-            .expect("Failed to get data dirname")
-            .join(package_name);
-
-        if !data_dir.exists() {
-            std::fs::create_dir(&data_dir)
-                .unwrap_or_else(|_| panic!("Can't create dir {}", data_dir.display()));
-        }
-        data_dir
+pub fn get_data_dir(app: &AppHandle, package_name: String) -> std::io::Result<PathBuf> {
+    let data_dir = app
+        .path()
+        .config_dir()
+        .map_err(|err| {
+            std::io::Error::new(
+                std::io::ErrorKind::NotFound,
+                format!("Failed to resolve config dir: {err}"),
+            )
+        })?
+        .join(package_name);
+
+    if !data_dir.exists() {
+        std::fs::create_dir_all(&data_dir).map_err(|err| {
+            std::io::Error::new(
+                err.kind(),
+                format!("Can't create dir {}: {err}", data_dir.display()),
+            )
+        })?;
     }
+
+    Ok(data_dir)
 }
 
 pub fn show_toast(window: &WebviewWindow, message: &str) {
     let script = format!(r#"pakeToast("{message}");"#);
-    window.eval(&script).unwrap();
+    if let Err(error) = window.eval(&script) {
+        eprintln!("[Pake] Failed to show toast: {error}");
+    }
 }
 
 pub enum MessageType {
@@ -101,10 +111,16 @@ pub fn get_download_message_with_lang(
     .to_string()
 }
 
-// Check if the file exists, if it exists, add a number to file name
+/// Check if the file exists. If it does, append `-N` to the stem until a free
+/// path is found.
+///
+/// Robustness notes:
+/// - Files without an extension are handled (we keep them extensionless).
+/// - If the numeric suffix would overflow `u32::MAX` we fall back to the
+///   original file_path so the caller never enters an infinite loop on
+///   pathologically large filenames (regression guard for #1183).
 pub fn check_file_or_append(file_path: &str) -> String {
     let mut new_path = PathBuf::from(file_path);
-    let mut counter = 0;
 
     while new_path.exists() {
         let file_stem = new_path
@@ -116,16 +132,24 @@ pub fn check_file_or_append(file_path: &str) -> String {
             .map(|e| e.to_string_lossy().to_string());
         let parent_dir = new_path.parent().unwrap_or(Path::new(""));
 
-        let new_file_stem = match file_stem.rfind('-') {
-            Some(index) if file_stem[index + 1..].parse::<u32>().is_ok() => {
+        let parsed_suffix = file_stem.rfind('-').and_then(|index| {
+            file_stem[index + 1..]
+                .parse::<u32>()
+                .ok()
+                .map(|n| (index, n))
+        });
+
+        let new_file_stem = match parsed_suffix {
+            Some((index, current)) => {
+                let Some(next) = current.checked_add(1) else {
+                    // u32::MAX collisions are a sign of something pathological;
+                    // bail with the original path instead of looping forever.
+                    return file_path.to_string();
+                };
                 let base_name = &file_stem[..index];
-                counter = file_stem[index + 1..].parse::<u32>().unwrap() + 1;
-                format!("{base_name}-{counter}")
-            }
-            _ => {
-                counter += 1;
-                format!("{file_stem}-{counter}")
+                format!("{base_name}-{next}")
             }
+            None => format!("{file_stem}-1"),
         };
 
         new_path = match &extension {
@@ -136,3 +160,86 @@ pub fn check_file_or_append(file_path: &str) -> String {
 
     new_path.to_string_lossy().into_owned()
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    use std::env;
+    use std::fs;
+    use std::path::PathBuf;
+
+    fn temp_path(name: &str) -> PathBuf {
+        let mut dir = env::temp_dir();
+        dir.push(format!(
+            "pake-util-test-{}-{}",
+            std::process::id(),
+            std::time::SystemTime::now()
+                .duration_since(std::time::UNIX_EPOCH)
+                .map(|d| d.as_nanos())
+                .unwrap_or(0)
+        ));
+        fs::create_dir_all(&dir).unwrap();
+        dir.push(name);
+        dir
+    }
+
+    #[test]
+    fn check_file_or_append_returns_input_when_missing() {
+        let path = temp_path("ghost.txt");
+        let resolved = check_file_or_append(path.to_str().unwrap());
+        assert_eq!(resolved, path.to_string_lossy());
+        let _ = fs::remove_dir_all(path.parent().unwrap());
+    }
+
+    #[test]
+    fn check_file_or_append_increments_suffix() {
+        let path = temp_path("dup.txt");
+        fs::write(&path, b"existing").unwrap();
+        let resolved = check_file_or_append(path.to_str().unwrap());
+        assert!(resolved.ends_with("dup-1.txt"), "got {resolved}");
+        let _ = fs::remove_dir_all(path.parent().unwrap());
+    }
+
+    #[test]
+    fn check_file_or_append_handles_files_without_extension() {
+        let path = temp_path("README");
+        fs::write(&path, b"existing").unwrap();
+        let resolved = check_file_or_append(path.to_str().unwrap());
+        assert!(resolved.ends_with("README-1"), "got {resolved}");
+        let _ = fs::remove_dir_all(path.parent().unwrap());
+    }
+
+    #[test]
+    fn check_file_or_append_does_not_panic_on_huge_suffix() {
+        let path = temp_path(&format!("huge-{}.txt", u32::MAX));
+        fs::write(&path, b"existing").unwrap();
+        let resolved = check_file_or_append(path.to_str().unwrap());
+        assert!(resolved.contains("huge-"));
+        let _ = fs::remove_dir_all(path.parent().unwrap());
+    }
+
+    #[test]
+    fn download_message_falls_back_to_english_for_unknown_locale() {
+        let msg = get_download_message_with_lang(MessageType::Start, Some("fr-FR".to_string()));
+        assert_eq!(msg, "Start downloading~");
+    }
+
+    #[test]
+    fn download_message_picks_chinese_for_zh_locales() {
+        for tag in ["zh", "zh-CN", "zh-TW", "en-CN", "en-HK"] {
+            let msg = get_download_message_with_lang(MessageType::Success, Some(tag.to_string()));
+            assert_eq!(
+                msg, "下载成功，已保存到下载目录~",
+                "expected Chinese for {tag}"
+            );
+        }
+    }
+
+    #[test]
+    fn download_message_failure_localized() {
+        let en = get_download_message_with_lang(MessageType::Failure, Some("en".into()));
+        let zh = get_download_message_with_lang(MessageType::Failure, Some("zh".into()));
+        assert!(en.contains("Download failed"));
+        assert!(zh.contains("下载失败"));
+    }
+}

package/src-tauri/tauri.conf.json

@@ -1,7 +1,7 @@
 {
   "productName": "Weekly",
   "identifier": "com.pake.weekly",
-  "version": "3.11.4",
+  "version": "3.11.6",
   "app": {
     "withGlobalTauri": true,
     "trayIcon": {