feat: 支持获取窗口是否 focused (#184)

* feat: 优化代码

* feat: windows 支持获取窗口是否 focused

* feat: linux 实现获取窗口是否 focused

* feat: macos 实现获取窗口是否 focused

* chore: 修改版本号

* fix: 修复命名

---------

Co-authored-by: nashaofu <[email protected]>

Author: nashaofu

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "xcap"
-version = "0.2.1"
+version = "0.2.2"
 edition = "2021"
 description = "XCap is a cross-platform screen capture library written in Rust. It supports Linux (X11, Wayland), MacOS, and Windows. XCap supports screenshot and video recording (WIP)."
 license = "Apache-2.0"
@@ -23,6 +23,11 @@ thiserror = "2.0"
 [target.'cfg(target_os = "macos")'.dependencies]
 core-foundation = "0.10"
 core-graphics = "0.24"
+objc2-app-kit = { version = "0.2.2", features = [
+    "libc",
+    "NSWorkspace",
+    "NSRunningApplication",
+] }
 
 [target.'cfg(target_os = "windows")'.dependencies]
 windows = { version = "0.58", features = [

examples/window.rs

@@ -16,7 +16,7 @@ fn main() {
             window.current_monitor().name(),
             (window.x(), window.y(), window.z()),
             (window.width(), window.height()),
-            (window.is_minimized(), window.is_maximized())
+            (window.is_minimized(), window.is_maximized(), window.is_focused())
         );
     }
 }

src/linux/impl_window.rs

@@ -28,6 +28,7 @@ pub(crate) struct ImplWindow {
     pub height: u32,
     pub is_minimized: bool,
     pub is_maximized: bool,
+    pub is_focused: bool,
 }
 
 fn get_atom(conn: &Connection, name: &str) -> XCapResult<Atom> {
@@ -38,7 +39,7 @@ fn get_atom(conn: &Connection, name: &str) -> XCapResult<Atom> {
     let atom_reply = conn.wait_for_reply(atom_cookie)?;
     let atom = atom_reply.atom();
 
-    if atom == ATOM_NONE {
+    if atom.is_none() {
         return Err(XCapError::new(format!("{} not supported", name)));
     }
 
@@ -79,12 +80,29 @@ pub fn get_window_pid(conn: &Connection, window: &Window) -> XCapResult<u32> {
         .copied()
 }
 
+fn get_active_window_id(conn: &Connection) -> Option<u32> {
+    let active_window_atom = get_atom(conn, "_NET_ACTIVE_WINDOW").ok()?;
+    let setup = conn.get_setup();
+
+    for screen in setup.roots() {
+        let root_window = screen.root();
+        let active_window_id =
+            get_window_property(conn, root_window, active_window_atom, ATOM_NONE, 0, 4).ok()?;
+        if let Some(&active_window_id) = active_window_id.value::<u32>().first() {
+            return Some(active_window_id);
+        }
+    }
+
+    None
+}
+
 impl ImplWindow {
     fn new(
         conn: &Connection,
         window: &Window,
         pid: u32,
         z: i32,
+        is_focused: bool,
         impl_monitors: &Vec<ImplMonitor>,
     ) -> XCapResult<ImplWindow> {
         let title = {
@@ -197,6 +215,7 @@ impl ImplWindow {
             height,
             is_minimized,
             is_maximized,
+            is_focused,
         })
     }
 
@@ -208,6 +227,7 @@ impl ImplWindow {
         // https://specifications.freedesktop.org/wm-spec/1.5/ar01s03.html#id-1.4.4
         // list all windows by stacking order
         let client_list_atom = get_atom(&conn, "_NET_CLIENT_LIST_STACKING")?;
+        let active_window_id = get_active_window_id(&conn);
 
         let mut impl_windows = Vec::new();
         let impl_monitors = ImplMonitor::all()?;
@@ -247,7 +267,10 @@ impl ImplWindow {
                         }
                     };
 
-                    if let Ok(impl_window) = ImplWindow::new(&conn, client, pid, z, &impl_monitors)
+                    let is_focused = active_window_id.eq(&Some(client.resource_id()));
+
+                    if let Ok(impl_window) =
+                        ImplWindow::new(&conn, client, pid, z, is_focused, &impl_monitors)
                     {
                         impl_windows.push(impl_window);
                     } else {

src/macos/impl_window.rs

@@ -16,6 +16,7 @@ use core_graphics::{
     window::{kCGNullWindowID, kCGWindowSharingNone},
 };
 use image::RgbaImage;
+use objc2_app_kit::NSWorkspace;
 
 use crate::{error::XCapResult, XCapError};
 
@@ -35,6 +36,7 @@ pub(crate) struct ImplWindow {
     pub height: u32,
     pub is_minimized: bool,
     pub is_maximized: bool,
+    pub is_focused: bool,
 }
 
 unsafe impl Send for ImplWindow {}
@@ -129,9 +131,10 @@ impl ImplWindow {
         window_name: String,
         window_owner_name: String,
         z: i32,
+        focused_app_pid: Option<i32>,
     ) -> XCapResult<ImplWindow> {
         let id = get_cf_number_i32_value(window_cf_dictionary_ref, "kCGWindowNumber")? as u32;
-        let pid = get_cf_number_i32_value(window_cf_dictionary_ref, "kCGWindowOwnerPID")? as u32;
+        let pid = get_cf_number_i32_value(window_cf_dictionary_ref, "kCGWindowOwnerPID")?;
 
         let cg_rect = get_window_cg_rect(window_cf_dictionary_ref)?;
 
@@ -164,11 +167,13 @@ impl ImplWindow {
         let is_minimized =
             !get_cf_bool_value(window_cf_dictionary_ref, "kCGWindowIsOnscreen")? && !is_maximized;
 
+        let is_focused = focused_app_pid.eq(&Some(pid));
+
         Ok(ImplWindow {
             id,
             title: window_name,
             app_name: window_owner_name,
-            pid,
+            pid: pid as u32,
             current_monitor: current_monitor.clone(),
             x: cg_rect.origin.x as i32,
             y: cg_rect.origin.y as i32,
@@ -177,12 +182,18 @@ impl ImplWindow {
             height: cg_rect.size.height as u32,
             is_minimized,
             is_maximized,
+            is_focused,
         })
     }
 
     pub fn all() -> XCapResult<Vec<ImplWindow>> {
         unsafe {
             let impl_monitors = ImplMonitor::all()?;
+            let workspace = NSWorkspace::sharedWorkspace();
+            let focused_app_pid = workspace
+                .frontmostApplication()
+                .map(|focused_app| focused_app.processIdentifier());
+
             let mut impl_windows = Vec::new();
 
             // CGWindowListCopyWindowInfo 返回窗口顺序为从顶层到最底层
@@ -240,6 +251,7 @@ impl ImplWindow {
                     window_name.clone(),
                     window_owner_name.clone(),
                     num_windows as i32 - i as i32 - 1,
+                    focused_app_pid,
                 ) {
                     impl_windows.push(impl_window);
                 } else {

src/window.rs

@@ -74,6 +74,10 @@ impl Window {
     pub fn is_maximized(&self) -> bool {
         self.impl_window.is_maximized
     }
+    /// The window is focused.
+    pub fn is_focused(&self) -> bool {
+        self.impl_window.is_focused
+    }
 }
 
 impl Window {

src/windows/capture.rs

@@ -10,8 +10,8 @@ use windows::Win32::{
             OBJ_BITMAP, SRCCOPY,
         },
     },
-    Storage::Xps::{PrintWindow, PRINT_WINDOW_FLAGS},
-    UI::WindowsAndMessaging::{GetDesktopWindow, WINDOWINFO},
+    Storage::Xps::{PrintWindow, PRINT_WINDOW_FLAGS, PW_CLIENTONLY},
+    UI::WindowsAndMessaging::{GetDesktopWindow, PW_RENDERFULLCONTENT, WINDOWINFO},
 };
 
 use crate::error::{XCapError, XCapResult};

src/windows/impl_window.rs

@@ -17,9 +17,9 @@ use windows::{
             },
         },
         UI::WindowsAndMessaging::{
-            EnumWindows, GetClassNameW, GetWindowInfo, GetWindowLongPtrW, GetWindowTextLengthW,
-            GetWindowTextW, GetWindowThreadProcessId, IsIconic, IsWindow, IsWindowVisible,
-            IsZoomed, GWL_EXSTYLE, WINDOWINFO, WINDOW_EX_STYLE, WS_EX_TOOLWINDOW,
+            EnumWindows, GetClassNameW, GetForegroundWindow, GetWindowInfo, GetWindowLongPtrW,
+            GetWindowTextLengthW, GetWindowTextW, GetWindowThreadProcessId, IsIconic, IsWindow,
+            IsWindowVisible, IsZoomed, GWL_EXSTYLE, WINDOWINFO, WINDOW_EX_STYLE, WS_EX_TOOLWINDOW,
         },
     },
 };
@@ -52,6 +52,7 @@ pub(crate) struct ImplWindow {
     pub height: u32,
     pub is_minimized: bool,
     pub is_maximized: bool,
+    pub is_focused: bool,
 }
 
 fn is_window_cloaked(hwnd: HWND) -> bool {
@@ -326,6 +327,7 @@ impl ImplWindow {
             let rc_client = window_info.rcClient;
             let is_minimized = IsIconic(hwnd).as_bool();
             let is_maximized = IsZoomed(hwnd).as_bool();
+            let is_focused = GetForegroundWindow() == hwnd;
 
             Ok(ImplWindow {
                 hwnd,
@@ -342,6 +344,7 @@ impl ImplWindow {
                 height: (rc_client.bottom - rc_client.top) as u32,
                 is_minimized,
                 is_maximized,
+                is_focused,
             })
         }
     }