feat: cache method (#5)

Author: yetone

Cargo.toml

@@ -13,7 +13,6 @@ description = "A tiny Rust library that allows you to easily obtain selected tex
 [target.'cfg(not(target_os = "macos"))'.dependencies]
 arboard = "3.2.0"
 enigo = { version = "0.2.0", features = [ "xdo" ] }
-parking_lot = "0.12.1"
 
 [target.'cfg(target_os = "macos")'.dependencies]
 cocoa = "0.24"
@@ -24,3 +23,9 @@ core-graphics = "0.22.3"
 accessibility-ng = "0.1.6"
 accessibility-sys-ng = "0.1.3"
 
+[dependencies]
+active-win-pos-rs = "0.8.3"
+debug_print = "1.0.0"
+lru = "0.12.3"
+parking_lot = "0.12.1"
+

src/lib.rs

@@ -33,6 +33,13 @@ mod tests {
 
     #[test]
     fn test_get_selected_text() {
+        println!("--- get_selected_text ---");
+        let text = get_selected_text().unwrap();
+        println!("selected text: {:#?}", text);
+        println!("--- get_selected_text ---");
+        let text = get_selected_text().unwrap();
+        println!("selected text: {:#?}", text);
+        println!("--- get_selected_text ---");
         let text = get_selected_text().unwrap();
         println!("selected text: {:#?}", text);
     }

src/macos.rs

@@ -1,15 +1,52 @@
+use std::num::NonZeroUsize;
+
 use accessibility_ng::{AXAttribute, AXUIElement};
 use accessibility_sys_ng::{kAXFocusedUIElementAttribute, kAXSelectedTextAttribute};
+use active_win_pos_rs::get_active_window;
 use core_foundation::string::CFString;
+use debug_print::debug_println;
+use lru::LruCache;
+use parking_lot::Mutex;
+
+static GET_SELECTED_TEXT_METHOD: Mutex<Option<LruCache<String, u8>>> = Mutex::new(None);
 
 pub fn get_selected_text() -> Result<String, Box<dyn std::error::Error>> {
+    if GET_SELECTED_TEXT_METHOD.lock().is_none() {
+        let cache = LruCache::new(NonZeroUsize::new(100).unwrap());
+        *GET_SELECTED_TEXT_METHOD.lock() = Some(cache);
+    }
+    let mut cache = GET_SELECTED_TEXT_METHOD.lock();
+    let cache = cache.as_mut().unwrap();
+    let app_name = match get_active_window() {
+        Ok(window) => window.app_name,
+        Err(_) => return Err("No active window found".into()),
+    };
+    // debug_println!("app_name: {}", app_name);
+    if let Some(text) = cache.get(&app_name) {
+        if *text == 0 {
+            return get_selected_text_by_ax();
+        }
+        return get_selected_text_by_clipboard_using_applescript();
+    }
     match get_selected_text_by_ax() {
-        Ok(text) => Ok(text),
-        Err(_) => get_selected_text_by_clipboard_using_applescript(),
+        Ok(text) => {
+            cache.put(app_name, 0);
+            Ok(text)
+        }
+        Err(_) => match get_selected_text_by_clipboard_using_applescript() {
+            Ok(text) => {
+                if !text.is_empty() {
+                    cache.put(app_name, 1);
+                }
+                Ok(text)
+            }
+            Err(e) => Err(e),
+        },
     }
 }
 
 fn get_selected_text_by_ax() -> Result<String, Box<dyn std::error::Error>> {
+    // debug_println!("get_selected_text_by_ax");
     let system_element = AXUIElement::system_wide();
     let Some(selected_element) = system_element
         .attribute(&AXAttribute::new(&CFString::from_static_string(
@@ -79,6 +116,7 @@ theSelectedText
 
 fn get_selected_text_by_clipboard_using_applescript() -> Result<String, Box<dyn std::error::Error>>
 {
+    // debug_println!("get_selected_text_by_clipboard_using_applescript");
     let output = std::process::Command::new("osascript")
         .arg("-e")
         .arg(APPLE_SCRIPT)