nit

Author: tadghh

.cargo/Config.toml

@@ -1,9 +1,10 @@
-[unstable]
-build-std         =["alloc", "core", "panic_abort", "proc_macro", "std"]
-build-std-features=["panic_immediate_abort"]
+# [unstable]
+# build-std         =["alloc", "core", "panic_abort", "proc_macro", "std"]
+# build-std-features=["panic_immediate_abort"]
 
-[build]
-rustflags=["--cfg=has_std", "-C", "panic=abort", "target-cpu=native"]
+# [build]
+# rustflags=["--cfg=has_std", "-C", "panic=abort", "target-cpu=native"]
 
-[target.x86_64-pc-windows-msvc]
-rustflags=["-C", "link-arg=/LIMIT:10240"]
+# [target.x86_64-pc-windows-msvc]
+# rustflags=["-C", "link-arg=/LIMIT:10240"]
+# #

src/app_state.rs

@@ -3,6 +3,7 @@ use std::path::PathBuf;
 use std::sync::Arc;
 use tokio::sync::{broadcast, RwLock};
 
+use crate::monitor::{refresh_config, reset_config};
 use crate::transparency::create_rules_window;
 use crate::util::Config;
 use crate::win_utils::{self, create_percentage_window};
@@ -99,6 +100,7 @@ impl AppState {
                 Ok(class) => {
                     if let Some(info) = class {
                         window_config.set_window_class(info.1);
+                        refresh_config(window_config.clone());
                         config.get_windows().remove(&window_config.get_key());
                         window_config.set_old_classname(Some(old_class));
                     }
@@ -108,17 +110,32 @@ impl AppState {
         }
 
         if !window_config.is_wide() {
+            println!("d");
             match find_parent_from_child_class(window_config.get_window_class()) {
                 Ok(class) => {
                     if let Some(info) = class {
-                        config.get_windows().remove(&format!(
-                            "{}|{}",
-                            window_config.get_name(),
-                            info.1
-                        ));
+                        let key = &format!("{}|{}", window_config.get_name(), info.1);
+                        println!("hrere {:?}", info);
+                        if config.get_windows().contains_key(key) {
+                            println!("key yo {:?}", key);
+                            let clone_config = window_config.clone();
+                            let real_class = clone_config.get_window_class();
+                            window_config.set_window_class(info.1);
+                            reset_config(window_config.clone());
+                            // let handles = window_config.get_window_hwnds();
+                            window_config.set_window_class(real_class.to_string());
+                            // println!("handles yo {:?}", handles);
+
+                            config.get_windows().remove(key);
+                        }
+                    } else {
+                        println!("2{:?}", class);
                     }
                 }
-                Err(_) => (),
+                Err(_) => {
+                    println!("2");
+                    ()
+                }
             }
         }
 

src/main.rs

@@ -1,4 +1,4 @@
-#![windows_subsystem = "windows"]
+// #![windows_subsystem = "windows"]
 #![feature(let_chains)]
 use anyhow::Result;
 use std::sync::Arc;

src/monitor.rs

@@ -1,3 +1,4 @@
+use crate::window_config::WindowConfig;
 use crate::{app_state::AppState, util::Config, win_utils::make_window_transparent};
 
 use core::time::Duration;
@@ -53,6 +54,7 @@ pub async fn monitor_windows(app_state: Arc<AppState>) {
     let mut config = app_state.get_config().await;
     let mut is_enabled = app_state.is_enabled().await;
     let mut config_rx = app_state.subscribe_config_updates();
+
     let mut enabled_rx = app_state.subscribe_enabled_updates();
 
     loop {
@@ -63,6 +65,7 @@ pub async fn monitor_windows(app_state: Arc<AppState>) {
             Ok(new_config) = config_rx.recv() => {
                 config = new_config;
             }
+
             Ok(state) = enabled_rx.recv() => {
                 if state != is_enabled && is_enabled {
                     reset_windows(&mut window_cache);
@@ -140,3 +143,21 @@ fn reset_windows(window_cache: &mut HashMap<String, Vec<WindowHandleState>>) {
             }
         });
 }
+
+pub fn reset_config(window_config: WindowConfig) {
+    let handles = window_config.get_window_hwnds();
+    for handle in handles {
+        if make_window_transparent(HWND(handle as *mut c_void), 255).is_ok() {}
+    }
+}
+pub fn refresh_config(window_config: WindowConfig) {
+    let handles = window_config.get_window_hwnds();
+    for handle in handles {
+        if make_window_transparent(
+            HWND(handle as *mut c_void),
+            window_config.get_transparency(),
+        )
+        .is_ok()
+        {}
+    }
+}

src/window_config.rs

@@ -216,8 +216,12 @@ fn get_window_class_name(hwnd: HWND) -> Option<String> {
 pub fn find_parent_from_child_class(
     child_class: &str,
 ) -> windows::core::Result<Option<(HWND, String)>> {
+    println!("{:?}", child_class);
     let child_hwnd = match find_window_by_class(child_class)? {
-        Some(hwnd) => hwnd,
+        Some(hwnd) => {
+            println!("s3");
+            hwnd
+        }
         None => {
             return Ok(None);
         }