From 0e61c9ba1ead56cd3b0b45b67ccddd13892fdc19 Mon Sep 17 00:00:00 2001
From: keifufu <github@keifufu.dev>
Date: Wed, 12 Jun 2024 09:06:42 +0200
Subject: [PATCH] add hide_tray_icon option. closes #7

---
 src/config.rs        |   1 +
 src/data/config.yaml |   1 +
 src/main.rs          | 128 ++++++++++++++++++++++---------------------
 3 files changed, 68 insertions(+), 62 deletions(-)

diff --git a/src/config.rs b/src/config.rs
index 18e0fea..a032f8a 100644
--- a/src/config.rs
+++ b/src/config.rs
@@ -34,6 +34,7 @@ pub struct WindowRule {
 #[derive(Debug, Serialize, Deserialize, Clone)]
 pub struct Config {
   pub run_at_startup: bool,
+  pub hide_tray_icon: Option<bool>, // option because i cant be bothered handling config upgrades
   pub window_rules: Vec<WindowRule>,
 }
 
diff --git a/src/data/config.yaml b/src/data/config.yaml
index 2bcc8a3..11c8023 100644
--- a/src/data/config.yaml
+++ b/src/data/config.yaml
@@ -1,4 +1,5 @@
 run_at_startup: false
+hide_tray_icon: false
 window_rules:
   - match: "Global"
     active_border_color: "#c6a0f6"
diff --git a/src/main.rs b/src/main.rs
index 7ae72ca..012551f 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,4 +1,5 @@
 #![windows_subsystem = "windows"]
+#![allow(unused_assignments)]
 
 use config::Config;
 use config::RuleMatch;
@@ -66,69 +67,72 @@ fn main() {
       std::process::exit(1);
     }
 
-    let tray_menu_builder = Menu::with_items(&[
-      &MenuItemBuilder::new()
-        .text("Open config")
-        .enabled(true)
-        .id(MenuId::new("0"))
-        .build(),
-      &MenuItemBuilder::new()
-        .text("Reload config")
-        .enabled(true)
-        .id(MenuId::new("1"))
-        .build(),
-      &MenuItemBuilder::new()
-        .text("Exit")
-        .enabled(true)
-        .id(MenuId::new("2"))
-        .build(),
-    ]);
-
-    let tray_menu = match tray_menu_builder {
-      Ok(tray_menu) => tray_menu,
-      Err(err) => {
-        Logger::log("[ERROR] Failed to build tray icon");
-        Logger::log(&format!("[DEBUG] {:?}", err));
-        std::process::exit(1);
-      }
-    };
-
-    let icon = match Icon::from_resource(1, Some((64, 64))) {
-      Ok(icon) => icon,
-      Err(err) => {
-        Logger::log("[ERROR] Failed to create icon");
-        Logger::log(&format!("[DEBUG] {:?}", err));
-        std::process::exit(1);
-      }
-    };
-
-    let tray_icon_builder = TrayIconBuilder::new()
-      .with_menu(Box::new(tray_menu))
-      .with_menu_on_left_click(true)
-      .with_icon(icon)
-      .with_tooltip(format!("cute-borders v{}", env!("CARGO_PKG_VERSION")));
-
     #[allow(unused_variables)]
-    let tray_icon = match tray_icon_builder.build() {
-      Ok(tray_icon) => tray_icon,
-      Err(err) => {
-        Logger::log("[ERROR] Failed to build tray icon");
-        Logger::log(&format!("[DEBUG] {:?}", err));
-        std::process::exit(1);
-      }
-    };
-
-    MenuEvent::set_event_handler(Some(|event: MenuEvent| {
-      if event.id == MenuId::new("0") {
-        let _ = open::that(get_file_path("config.yaml"));
-      } else if event.id == MenuId::new("1") {
-        Config::reload();
-        apply_colors(false);
-      } else if event.id == MenuId::new("2") {
-        apply_colors(true);
-        std::process::exit(0);
-      }
-    }));
+    let tray_icon; // needs to be in the main scope
+    if !Config::get().hide_tray_icon.unwrap_or(false) {
+      let tray_menu_builder = Menu::with_items(&[
+        &MenuItemBuilder::new()
+          .text("Open config")
+          .enabled(true)
+          .id(MenuId::new("0"))
+          .build(),
+        &MenuItemBuilder::new()
+          .text("Reload config")
+          .enabled(true)
+          .id(MenuId::new("1"))
+          .build(),
+        &MenuItemBuilder::new()
+          .text("Exit")
+          .enabled(true)
+          .id(MenuId::new("2"))
+          .build(),
+      ]);
+  
+      let tray_menu = match tray_menu_builder {
+        Ok(tray_menu) => tray_menu,
+        Err(err) => {
+          Logger::log("[ERROR] Failed to build tray icon");
+          Logger::log(&format!("[DEBUG] {:?}", err));
+          std::process::exit(1);
+        }
+      };
+  
+      let icon = match Icon::from_resource(1, Some((64, 64))) {
+        Ok(icon) => icon,
+        Err(err) => {
+          Logger::log("[ERROR] Failed to create icon");
+          Logger::log(&format!("[DEBUG] {:?}", err));
+          std::process::exit(1);
+        }
+      };
+  
+      let tray_icon_builder = TrayIconBuilder::new()
+        .with_menu(Box::new(tray_menu))
+        .with_menu_on_left_click(true)
+        .with_icon(icon)
+        .with_tooltip(format!("cute-borders v{}", env!("CARGO_PKG_VERSION")));
+  
+      tray_icon = match tray_icon_builder.build() {
+        Ok(tray_icon) => tray_icon,
+        Err(err) => {
+          Logger::log("[ERROR] Failed to build tray icon");
+          Logger::log(&format!("[DEBUG] {:?}", err));
+          std::process::exit(1);
+        }
+      };
+  
+      MenuEvent::set_event_handler(Some(|event: MenuEvent| {
+        if event.id == MenuId::new("0") {
+          let _ = open::that(get_file_path("config.yaml"));
+        } else if event.id == MenuId::new("1") {
+          Config::reload();
+          apply_colors(false);
+        } else if event.id == MenuId::new("2") {
+          apply_colors(true);
+          std::process::exit(0);
+        }
+      }));
+    }
 
     let mut msg = std::mem::zeroed();
     while GetMessageW(&mut msg, std::ptr::null_mut(), 0, 0) != 0 {