enhance: skip throtting last process if found full screen app in foreground

Author: mokurin000

Cargo.toml

@@ -29,6 +29,8 @@ windows = { version = "0.61", features = [
     "Win32_UI_WindowsAndMessaging",
     # enable SeDebugPrivilege for SYSTEM processes
     "Win32_Security",
+    # SHQueryUserNotificationStat
+    "Win32_UI_Shell",
 ] }
 windows-version = "0.1.4"
 

src/main.rs

@@ -2,7 +2,7 @@ use std::error::Error;
 use std::ffi::OsString;
 
 use ahash::AHashSet;
-use spdlog::{Level, LevelFilter, debug, error, info, warn};
+use spdlog::{Level, LevelFilter, debug, error, info, trace, warn};
 use win32_ecoqos::process::toggle_efficiency_mode;
 
 use rustystar::bypass::whitelisted;
@@ -12,6 +12,9 @@ use rustystar::logging::log_error;
 use rustystar::privilege::try_enable_se_debug_privilege;
 use rustystar::utils::{process_child_process, toggle_all};
 use rustystar::{PID_SENDER, WHITELIST};
+use windows::Win32::UI::Shell::{
+    QUNS_BUSY, QUNS_RUNNING_D3D_FULL_SCREEN, SHQueryUserNotificationState,
+};
 
 #[compio::main]
 async fn main() -> Result<(), Box<dyn Error>> {
@@ -97,19 +100,25 @@ async fn main() -> Result<(), Box<dyn Error>> {
             let mut last_pid = None;
 
             while let Ok(pid) = rx.recv().await {
-                debug!("received: {pid}");
+                trace!("received: {pid}");
 
                 match last_pid {
                     // skip boosting
                     Some(last) if last == pid => {
                         continue;
                     }
-                    Some(last_pid) => {
-                        _ = compio::runtime::spawn_blocking(move || {
-                            process_child_process(Some(true), last_pid)
-                        })
-                        .await;
-                    }
+                    Some(last_pid) => match unsafe { SHQueryUserNotificationState() } {
+                        Ok(QUNS_BUSY) | Ok(QUNS_RUNNING_D3D_FULL_SCREEN) => {
+                            debug!("detected full screen app! skip throttling");
+                        }
+                        _ => {
+                            _ = compio::runtime::spawn_blocking(move || {
+                                process_child_process(Some(true), last_pid)
+                            })
+                            .await;
+                        }
+                    },
+
                     None => {}
                 }
 

src/utils/mod.rs

@@ -1,7 +1,7 @@
 use std::collections::BTreeMap;
 
 use rustc_hash::FxHashSet;
-use spdlog::{info, warn};
+use spdlog::{debug, info, warn};
 use win32_ecoqos::{
     process::toggle_efficiency_mode,
     utils::{Process, Processes},
@@ -13,7 +13,7 @@ use crate::bypass::whitelisted;
 pub fn process_child_process(enable: Option<bool>, main_pid: u32) -> windows_result::Result<()> {
     let action = match enable {
         Some(true) => "throtting",
-        Some(false) => "boosting ",
+        Some(false) => "boosting",
         None => "recovering",
     };
 
@@ -23,13 +23,13 @@ pub fn process_child_process(enable: Option<bool>, main_pid: u32) -> windows_res
         .find(|Process { process_id, .. }| process_id == &main_pid)
     {
         if whitelisted(process_name) {
-            info!("skipping whitelisted process: {process_name:?}");
+            debug!("[{action:^10}] skipping {process_name:?}");
             return Ok(());
         }
 
-        info!("{action} process {main_pid:6}: {process_name:?}");
+        info!("[{action:^10}] process {main_pid:6}: {process_name:?}");
     } else {
-        info!("{action} process {main_pid:6}");
+        info!("[{action:^10}] process {main_pid:6}");
     }
 
     let relations = BTreeMap::from_iter(procs.iter().map(