feat: add set_badge_count for Linux, iOS; set_badge_label for Macos; set_overlay_icon for Windows

.changes/badge_count.md

@@ -0,0 +1,5 @@
+---
+"tao": patch
+---
+
+Add **set_badge_count** for Linux, iOS; **set_badge_label** for Macos; **set_overlay_icon** for Windows

examples/overlay.rs

@@ -0,0 +1,79 @@
+// Copyright 2014-2021 The winit contributors
+// Copyright 2021-2023 Tauri Programme within The Commons Conservancy
+// SPDX-License-Identifier: Apache-2.0
+
+use std::env::current_dir;
+use tao::{
+  event::{ElementState, Event, KeyEvent, WindowEvent}, event_loop::{ControlFlow, EventLoop}, keyboard::{Key, ModifiersState}, window::WindowBuilder
+};
+
+#[allow(clippy::single_match)]
+fn main() {
+  env_logger::init();
+  let event_loop = EventLoop::new();
+
+  let window = WindowBuilder::new().build(&event_loop).unwrap();
+
+  let mut modifiers = ModifiersState::default();
+
+  eprintln!("Key mappings:");
+  #[cfg(windows)]
+  eprintln!("  [any key]: Show the Overlay Icon");
+  #[cfg(not(windows))]
+  eprintln!("  [1-5]: Show a Badge count");
+  eprintln!("  Ctrl+1: Clear");
+
+  event_loop.run(move |event, _, control_flow| {
+    *control_flow = ControlFlow::Wait;
+
+    match event {
+      Event::WindowEvent {
+        event: WindowEvent::CloseRequested,
+        ..
+      } => *control_flow = ControlFlow::Exit,
+      Event::WindowEvent { event, .. } => match event {
+        WindowEvent::ModifiersChanged(new_state) => {
+          modifiers = new_state;
+        }
+        WindowEvent::KeyboardInput {
+          event:
+            KeyEvent {
+              logical_key: Key::Character(key_str),
+              state: ElementState::Released,
+              ..
+            },
+          ..
+        } => {
+          #[cfg(not(windows))]
+          if modifiers.is_empty() {
+            window.set_badge_count(Some(match key_str {
+              "1" => 1,
+              "2" => 2,
+              "3" => 3,
+              "4" => 4,
+              "5" => 5,
+              _ => 20
+            }), None);
+          } else if modifiers.control_key() && key_str == "1" {
+            window.set_badge_count(None, None);
+          }
+
+          #[cfg(windows)]
+          {
+            use tao::{window::Icon, platform::windows::WindowExtWindows, platform::windows::IconExtWindows};
+            if modifiers.is_empty() {
+              let mut path = current_dir().unwrap();
+              path.push("./examples/icon.ico");
+
+              window.set_overlay_icon(Some(Icon::from_path(path, None).unwrap()));
+            } else if modifiers.control_key() && key_str == "1" {
+              window.set_overlay_icon(None);
+            }
+          }
+        }
+        _ => {}
+      },
+      _ => {}
+    }
+  });
+}

src/platform/macos.rs

@@ -84,6 +84,9 @@ pub trait WindowExtMacOS {
   ///
   /// <https://developer.apple.com/documentation/appkit/nswindow/1419167-titlebarappearstransparent>
   fn set_titlebar_transparent(&self, transparent: bool);
+
+  /// Sets the badge label on the taskbar
+  fn set_badge_label(&self, label: Option<String>);
 }
 
 impl WindowExtMacOS for Window {
@@ -161,6 +164,11 @@ impl WindowExtMacOS for Window {
   fn set_titlebar_transparent(&self, transparent: bool) {
     self.window.set_titlebar_transparent(transparent);
   }
+
+  #[inline]
+  fn set_badge_label(&self, label: Option<String>) {
+    self.window.set_badge_label(label);
+  }
 }
 
 /// Corresponds to `NSApplicationActivationPolicy`.
@@ -388,6 +396,9 @@ pub trait EventLoopWindowTargetExtMacOS {
   /// To set the activation policy before the app starts running, see
   /// [`EventLoopExtMacOS::set_activation_policy`](crate::platform::macos::EventLoopExtMacOS::set_activation_policy).
   fn set_activation_policy_at_runtime(&self, activation_policy: ActivationPolicy);
+
+  /// Sets the badge label on macos dock
+  fn set_badge_label(&self, label: Option<String>);
 }
 
 impl<T> EventLoopWindowTargetExtMacOS for EventLoopWindowTarget<T> {
@@ -415,4 +426,8 @@ impl<T> EventLoopWindowTargetExtMacOS for EventLoopWindowTarget<T> {
     let ns_activation_policy: NSApplicationActivationPolicy = activation_policy.into();
     unsafe { msg_send![app, setActivationPolicy: ns_activation_policy] }
   }
+
+  fn set_badge_label(&self, label: Option<String>) {
+    set_badge_label(label);
+  }
 }

src/platform/unix.rs

@@ -78,6 +78,8 @@ pub trait WindowExtUnix {
 
   /// Whether to show the window icon in the taskbar or not.
   fn set_skip_taskbar(&self, skip: bool) -> Result<(), ExternalError>;
+
+  fn set_badge_count(&self, count: Option<i64>, desktop_filename: Option<String>);
 }
 
 impl WindowExtUnix for Window {
@@ -100,6 +102,10 @@ impl WindowExtUnix for Window {
     let window = UnixWindow::new_from_gtk_window(&event_loop_window_target.p, window)?;
     Ok(Window { window: window })
   }
+
+  fn set_badge_count(&self, count: Option<i64>, desktop_filename: Option<String>) {
+    self.window.set_badge_count(count, desktop_filename);
+  }
 }
 
 pub trait WindowBuilderExtUnix {
@@ -208,6 +214,9 @@ pub trait EventLoopWindowTargetExtUnix {
 
   /// Returns the gtk application for this event loop.
   fn gtk_app(&self) -> &gtk::Application;
+
+  /// Sets the badge count on the taskbar
+  fn set_badge_count(&self, count: Option<i64>, desktop_filename: Option<String>);
 }
 
 impl<T> EventLoopWindowTargetExtUnix for EventLoopWindowTarget<T> {
@@ -249,6 +258,11 @@ impl<T> EventLoopWindowTargetExtUnix for EventLoopWindowTarget<T> {
   fn gtk_app(&self) -> &gtk::Application {
     &self.p.app
   }
+
+  #[inline]
+  fn set_badge_count(&self, count: Option<i64>, desktop_filename: Option<String>) {
+    self.p.set_badge_count(count, desktop_filename);
+  }
 }
 
 unsafe extern "C" fn x_error_callback(

src/platform/windows.rs

@@ -159,6 +159,9 @@ pub trait WindowExtWindows {
   /// This sets `ICON_BIG`. A good ceiling here is 256x256.
   fn set_taskbar_icon(&self, taskbar_icon: Option<Icon>);
 
+  /// This sets the overlay icon
+  fn set_overlay_icon(&self, icon: Option<Icon>);
+
   /// Returns the current window theme.
   fn theme(&self) -> Theme;
 
@@ -238,6 +241,11 @@ impl WindowExtWindows for Window {
   fn set_rtl(&self, rtl: bool) {
     self.window.set_rtl(rtl)
   }
+
+  #[inline]
+  fn set_overlay_icon(&self, icon: Option<Icon>) {
+    self.window.set_overlay_icon(icon);
+  }
 }
 
 /// Additional methods on `WindowBuilder` that are specific to Windows.

src/platform_impl/ios/badge.rs

@@ -0,0 +1,10 @@
+use objc::runtime::{Class, Object};
+use objc::{msg_send, sel, sel_impl};
+
+pub fn set_badge_count(count: i32) {
+  unsafe {
+    let ui_application = Class::get("UIApplication").expect("Failed to get UIApplication class");
+    let app: *mut Object = msg_send![ui_application, sharedApplication];
+    let _: () = msg_send![app, setApplicationIconBadgeNumber:count];
+  }
+}

src/platform_impl/ios/event_loop.rs

@@ -22,6 +22,7 @@ use crate::{
 };
 
 use crate::platform_impl::platform::{
+  set_badge_count,
   app_state,
   ffi::{
     id, kCFRunLoopAfterWaiting, kCFRunLoopBeforeWaiting, kCFRunLoopCommonModes,
@@ -94,6 +95,11 @@ impl<T: 'static> EventLoopWindowTarget<T> {
     debug!("`EventLoopWindowTarget::cursor_position` is ignored on iOS");
     Ok((0, 0).into())
   }
+
+  /// Sets badge count on iOS launcher. 0 hides the count
+  pub fn set_badge_count(&self, count: i32) {
+    set_badge_count(count);
+  }
 }
 
 #[derive(Default, Debug, Copy, Clone, PartialEq, Eq, Hash)]

src/platform_impl/ios/mod.rs

@@ -80,6 +80,7 @@ mod keycode;
 mod monitor;
 mod view;
 mod window;
+mod badge;
 
 use std::fmt;
 
@@ -90,7 +91,7 @@ pub use self::{
   monitor::{MonitorHandle, VideoMode},
   window::{PlatformSpecificWindowBuilderAttributes, Window, WindowId},
 };
-
+pub(crate) use badge::set_badge_count;
 pub(crate) use crate::icon::NoIcon as PlatformIcon;
 
 // todo: implement iOS keyboard event

src/platform_impl/ios/window.rs

@@ -17,6 +17,7 @@ use crate::{
   monitor::MonitorHandle as RootMonitorHandle,
   platform::ios::{MonitorHandleExtIOS, ScreenEdge, ValidOrientations},
   platform_impl::platform::{
+    set_badge_count,
     app_state,
     event_loop::{self, EventProxy, EventWrapper},
     ffi::{
@@ -440,6 +441,11 @@ impl Inner {
   pub fn theme(&self) -> Theme {
     Theme::Light
   }
+
+  /// Sets badge count on iOS launcher. 0 hides the count
+  pub fn set_badge_count(&self, count: i32) {
+    set_badge_count(count);
+  }
 }
 
 pub struct Window {

src/platform_impl/linux/event_loop.rs

@@ -160,6 +160,16 @@ impl<T> EventLoopWindowTarget<T> {
     }
   }
 
+  #[inline]
+  pub fn set_badge_count(&self, count: Option<i64>, desktop_filename: Option<String>) {
+    if let Err(e) = self
+      .window_requests_tx
+      .send((WindowId::dummy(), WindowRequest::BadgeCount(count, desktop_filename)))
+    {
+      log::warn!("Fail to send update progress bar request: {e}");
+    }
+  }
+
   #[inline]
   pub fn set_theme(&self, theme: Option<Theme>) {
     if let Err(e) = self
@@ -432,6 +442,7 @@ impl<T: 'static> EventLoop<T> {
             };
           }
           WindowRequest::ProgressBarState(_) => unreachable!(),
+          WindowRequest::BadgeCount(_, _) => unreachable!(),
           WindowRequest::SetTheme(_) => unreachable!(),
           WindowRequest::WireUpEvents {
             transparent,
@@ -921,6 +932,9 @@ impl<T: 'static> EventLoop<T> {
           WindowRequest::ProgressBarState(state) => {
             taskbar.update(state);
           }
+          WindowRequest::BadgeCount(count, desktop_filename) => {
+            taskbar.update_count(count, desktop_filename);
+          }
           WindowRequest::SetTheme(theme) => {
             if let Some(settings) = Settings::default() {
               match theme {

src/platform_impl/linux/taskbar.rs

@@ -12,6 +12,9 @@ struct UnityLib {
   unity_launcher_entry_set_progress: unsafe extern "C" fn(entry: *const isize, value: f64) -> i32,
   unity_launcher_entry_set_progress_visible:
     unsafe extern "C" fn(entry: *const isize, value: i32) -> i32,
+  unity_launcher_entry_set_count: unsafe extern "C" fn(entry: *const isize, value: i64) -> i32,
+  unity_launcher_entry_set_count_visible:
+    unsafe extern "C" fn(entry: *const isize, value: bool) -> bool,
 }
 
 pub struct TaskbarIndicator {
@@ -120,4 +123,39 @@ impl TaskbarIndicator {
       }
     }
   }
+
+  pub fn update_count(&mut self, count: Option<i64>, desktop_filename: Option<String>) {
+    if let Some(uri) = desktop_filename {
+      self.desktop_filename = Some(uri);
+    }
+
+    self.ensure_lib_load();
+
+    if !self.is_unity_running() {
+      return;
+    }
+
+    if let Some(uri) = &self.desktop_filename {
+      self.desktop_filename_c_str = Some(CString::new(uri.as_str()).unwrap_or_default());
+    }
+
+    if self.unity_entry.is_none() {
+      self.ensure_entry_load();
+    }
+
+    if let Some(unity_lib) = &self.unity_lib {
+      if let Some(unity_entry) = &self.unity_entry {
+        // Sets count
+        if let Some(count) = count {
+          unsafe { (unity_lib.unity_launcher_entry_set_count)(*unity_entry, count) };
+          unsafe { (unity_lib.unity_launcher_entry_set_count_visible)(*unity_entry, true) };
+        }
+        // removes the count
+        else {
+          unsafe { (unity_lib.unity_launcher_entry_set_count)(*unity_entry, 0) };
+          unsafe { (unity_lib.unity_launcher_entry_set_count_visible)(*unity_entry, false) };
+        }
+      }
+    }
+  }
 }

src/platform_impl/linux/window.rs

@@ -1006,6 +1006,15 @@ impl Window {
     }
   }
 
+  pub fn set_badge_count(&self, count: Option<i64>, desktop_filename: Option<String>) {
+    if let Err(e) = self
+      .window_requests_tx
+      .send((WindowId::dummy(), WindowRequest::BadgeCount(count, desktop_filename)))
+    {
+      log::warn!("Fail to send update badge count request: {}", e);
+    }
+  }
+
   pub fn theme(&self) -> Theme {
     if let Some(theme) = *self.preferred_theme.borrow() {
       return theme;
@@ -1068,6 +1077,7 @@ pub enum WindowRequest {
   },
   SetVisibleOnAllWorkspaces(bool),
   ProgressBarState(ProgressBarState),
+  BadgeCount(Option<i64>, Option<String>),
   SetTheme(Option<Theme>),
   BackgroundColor(CssProvider, Option<RGBA>),
 }

src/platform_impl/macos/badge.rs

@@ -0,0 +1,12 @@
+use cocoa::{appkit::NSApp, base::nil, foundation::NSString};
+
+pub fn set_badge_label(label: Option<String>) {
+  unsafe {
+    let label = match label {
+      None => nil,
+      Some(label) => NSString::alloc(nil).init_str(&label)
+    };
+    let dock_tile: cocoa::base::id = msg_send![NSApp(), dockTile];
+    let _: cocoa::base::id = msg_send![dock_tile, setBadgeLabel: label];
+  }
+}