fix: windows impl.

Author: Kokoro2336

plugins/recent-doc/src/commands.rs

@@ -7,17 +7,18 @@ use tauri::{command, AppHandle, Runtime};
 
 #[cfg(target_os = "windows")]
 use windows::{
-    core::{Interface, HSTRING, PCWSTR, PWSTR},
+    core::{HSTRING, PCWSTR},
     // CRITICAL FIX: Changed `Interface` to `ComInterface` so `.cast()` works
     Win32::{
-        Foundation::MAX_PATH,
         System::Com::{
-            CoCreateInstance, CoInitializeEx, CoTaskMemFree, CoUninitialize, IPersistFile,
-            CLSCTX_INPROC_SERVER, COINIT_APARTMENTTHREADED, STGM_READ,
+            CoCreateInstance, CoInitializeEx, CoTaskMemFree, CoUninitialize, CLSCTX_INPROC_SERVER,
+            COINIT_APARTMENTTHREADED,
         },
         UI::Shell::{
-            FOLDERID_Recent, IShellLinkW, SHAddToRecentDocs, SHCreateItemFromParsingName,
-            SHGetKnownFolderPath, ShellLink, KNOWN_FOLDER_FLAG, SHARDAPPIDINFO, SHARD_APPIDINFO,
+            ApplicationDestinations, ApplicationDocumentLists, Common::IObjectArray,
+            IApplicationDestinations, IApplicationDocumentLists, IShellItem, SHAddToRecentDocs,
+            SHCreateItemFromParsingName, ADLT_RECENT, SHARDAPPIDINFO, SHARD_APPIDINFO,
+            SIGDN_FILESYSPATH,
         },
     },
 };
@@ -56,18 +57,14 @@ impl Drop for ComGuard {
 pub(crate) fn add_recent_document<R: Runtime>(app: AppHandle<R>, _path: &str) -> Result<()> {
     #[cfg(target_os = "windows")]
     unsafe {
-        // Set guard first
         let _com_guard = ComGuard::new();
 
-        // Get app id
         let app_id = app.config().identifier.clone();
         let app_id_hstring = HSTRING::from(app_id);
 
-        // Convert path to HSTRING
         let path_hstring = HSTRING::from(_path);
         let item = SHCreateItemFromParsingName(&path_hstring, None)?;
 
-        // construct info
         let info = SHARDAPPIDINFO {
             pszAppID: PCWSTR::from_raw(app_id_hstring.as_ptr()),
             psi: std::mem::ManuallyDrop::new(Some(item)),
@@ -102,8 +99,19 @@ pub(crate) fn add_recent_document<R: Runtime>(app: AppHandle<R>, _path: &str) ->
 pub(crate) fn clear_recent_documents<R: Runtime>(_app: AppHandle<R>) -> Result<()> {
     #[cfg(target_os = "windows")]
     unsafe {
-        let _com_guard = ComGuard::new();
-        SHAddToRecentDocs(SHARD_APPIDINFO.0 as u32, None);
+        let _guard = ComGuard::new();
+
+        let app_id = _app.config().identifier.clone();
+        let app_id_hstring = HSTRING::from(app_id);
+
+        if let Ok(dests) = CoCreateInstance::<_, IApplicationDestinations>(
+            &ApplicationDestinations as *const _,
+            None,
+            CLSCTX_INPROC_SERVER,
+        ) {
+            let _ = dests.SetAppID(PCWSTR::from_raw(app_id_hstring.as_ptr()));
+            let _ = dests.RemoveAllDestinations();
+        }
     }
 
     #[cfg(target_os = "macos")]
@@ -130,26 +138,31 @@ pub(crate) fn get_recent_documents<R: Runtime>(_app: AppHandle<R>) -> Result<Vec
 
     #[cfg(target_os = "windows")]
     unsafe {
-        // Ensure COM is initialized for this worker thread
         let _guard = ComGuard::new();
 
-        // Retrieve the absolute path to "%APPDATA%\Microsoft\Windows\Recent"
-        let recent_path_pwstr = SHGetKnownFolderPath(&FOLDERID_Recent, KNOWN_FOLDER_FLAG(0), None)?;
-
-        if !recent_path_pwstr.as_ptr().is_null() {
-            let recent_path = recent_path_pwstr.to_string()?;
-            // Free the string memory allocated by the Windows Shell
-            CoTaskMemFree(Some(recent_path_pwstr.as_ptr() as *mut _));
-
-            // Iterate through the hidden directory searching for .lnk files
-            if let Ok(entries) = std::fs::read_dir(recent_path) {
-                for entry in entries.flatten() {
-                    let path = entry.path();
-
-                    if path.extension().and_then(|s| s.to_str()) == Some("lnk") {
-                        // Resolve the shortcut binary to extract its actual target path
-                        if let Ok(resolved) = resolve_shortcut(&path) {
-                            recent_docs.push(resolved);
+        let app_id = _app.config().identifier.clone();
+        let app_id_hstring = HSTRING::from(app_id);
+        let app_id_pcwstr = PCWSTR::from_raw(app_id_hstring.as_ptr());
+
+        if let Ok(doc_lists) = CoCreateInstance::<_, IApplicationDocumentLists>(
+            &ApplicationDocumentLists,
+            None,
+            CLSCTX_INPROC_SERVER,
+        ) {
+            let _ = doc_lists.SetAppID(app_id_pcwstr);
+            if let Ok(obj_array) = doc_lists.GetList::<IObjectArray>(ADLT_RECENT, 30) {
+                let count = obj_array.GetCount().unwrap_or(0);
+
+                for i in 0..count {
+                    if let Ok(shell_item) = obj_array.GetAt::<IShellItem>(i) {
+                        if let Ok(name_pwstr) = shell_item.GetDisplayName(SIGDN_FILESYSPATH) {
+                            if let Ok(path) = name_pwstr.to_string() {
+                                if !path.is_empty() {
+                                    recent_docs.push(path);
+                                }
+                            }
+
+                            CoTaskMemFree(Some(name_pwstr.as_ptr() as *const core::ffi::c_void));
                         }
                     }
                 }
@@ -179,29 +192,3 @@ pub(crate) fn get_recent_documents<R: Runtime>(_app: AppHandle<R>) -> Result<Vec
 
     Ok(recent_docs)
 }
-
-/// Helper function to extract the real file system path from a Windows .lnk shortcut file.
-#[cfg(target_os = "windows")]
-fn resolve_shortcut(lnk_path: &std::path::Path) -> Result<String> {
-    unsafe {
-        // Instantiate the ShellLink COM Object
-        let shell_link: IShellLinkW =
-            CoCreateInstance(&ShellLink as *const _, None, CLSCTX_INPROC_SERVER)?;
-
-        // Query the IPersistFile interface to load the shortcut binary from disk
-        // This .cast() now compiles perfectly because ComInterface is imported
-        let persist_file: IPersistFile = shell_link.cast()?;
-
-        let path_hstring = HSTRING::from(lnk_path.as_os_str());
-        persist_file.Load(&path_hstring, STGM_READ)?;
-
-        // Allocate a buffer to hold the resolved target path
-        let mut target_path = [0u16; MAX_PATH as usize];
-        shell_link.GetPath(&mut target_path, std::ptr::null_mut(), 0)?;
-
-        // Convert the UTF-16 C-string back to a standard Rust String
-        let path_string = PWSTR::from_raw(target_path.as_mut_ptr()).to_string()?;
-
-        Ok(path_string)
-    }
-}