import uuid from fabric.notifications.service import Notification, NotificationAction, Notifications from fabric.widgets.box import Box from fabric.widgets.button import Button from fabric.widgets.centerbox import CenterBox from fabric.widgets.image import Image from fabric.widgets.label import Label from fabric.widgets.revealer import Revealer from gi.repository import GLib, Gtk from loguru import logger import config.data as data import utils.icons as icons from utils.custom_image import CustomImage from utils.wayland import WaylandWindow as Window from utils.notification_utils import ( cache_notification_pixbuf, load_scaled_pixbuf, get_shared_notification_history, NotificationHistory, delete_cached_image, ) def get_limited_apps_history(): config = data.load_config() return config.get("limited_apps_history", ["Spotify"]) def get_history_ignored_apps(): config = data.load_config() return config.get("history_ignored_apps", ["Modus"]) class ActionButton(Button): def __init__( self, action: NotificationAction, index: int, total: int, notification_box ): super().__init__( name="action-button", h_expand=True, on_clicked=self.on_clicked, child=Label( name="button-label", h_expand=True, h_align="fill", ellipsization="end", max_chars_width=1, label=action.label, ), ) self.action = action self.notification_box = notification_box style_class = ( "start-action" if index == 0 else "end-action" if index == total - 1 else "middle-action" ) self.add_style_class(style_class) self.connect( "enter-notify-event", lambda *_: notification_box.hover_button(self) ) self.connect( "leave-notify-event", lambda *_: notification_box.unhover_button(self) ) def on_clicked(self, *_): self.action.invoke() self.action.parent.close("dismissed-by-user") class NotificationBox(Box): def __init__(self, notification: Notification, timeout_ms=5000, **kwargs): super().__init__( name="notification-box", orientation="v", h_align="fill", h_expand=True, children=[], ) self.notification = notification self.uuid = str(uuid.uuid4()) if timeout_ms == 0: self.timeout_ms = 0 else: live_timeout = getattr(self.notification, "timeout", -1) self.timeout_ms = live_timeout if live_timeout != -1 else timeout_ms self._timeout_id = None self._container = None self.cached_image_path = None if self.timeout_ms > 0: self.start_timeout() if self.notification.image_pixbuf: cache_path = cache_notification_pixbuf(self) if cache_path: self.cached_image_path = cache_path else: logger.warning( f"NotificationBox { self.uuid }: Caching failed, cached_image_path not set." ) else: logger.debug(f"NotificationBox {self.uuid}: No image to cache.") content = self.create_content() action_buttons = self.create_action_buttons() self.add(content) if action_buttons: self.add(action_buttons) self.connect("enter-notify-event", self.on_hover_enter) self.connect("leave-notify-event", self.on_hover_leave) self._destroyed = False self._is_history = False def set_is_history(self, is_history): # Ensure the object is properly initialized if not hasattr(self, "_is_history"): self._is_history = False self._is_history = is_history def set_container(self, container): # Ensure the object is properly initialized if not hasattr(self, "_container"): self._container = None self._container = container def get_container(self): return getattr(self, "_container", None) def create_header(self): notification = self.notification self.app_icon_image = ( Image( name="notification-icon", image_file=notification.app_icon[7:], size=24, ) if "file://" in notification.app_icon else Image( name="notification-icon", icon_name="dialog-information-symbolic" or notification.app_icon, icon_size=24, ) ) self.app_name_label_header = Label( notification.app_name, name="notification-app-name", h_align="start" ) self.header_close_button = self.create_close_button() return CenterBox( name="notification-title", start_children=[ Box( spacing=4, children=[ self.app_icon_image, self.app_name_label_header, ], ) ], end_children=[self.header_close_button], ) def create_content(self): notification = self.notification pixbuf = load_scaled_pixbuf(self, 48, 48) self.notification_image_box = Box( name="notification-image", orientation="v", children=[CustomImage(pixbuf=pixbuf), Box(v_expand=True)], ) self.notification_summary_label = Label( name="notification-summary", markup=notification.summary, h_align="start", max_chars_width=16, ellipsization="end", ) self.notification_app_name_label_content = Label( name="notification-app-name", markup=notification.app_name, h_align="start", max_chars_width=16, ellipsization="end", ) self.notification_body_label = ( Label( markup=notification.body, h_align="start", max_chars_width=34, ellipsization="end", ) if notification.body else Box() ) self.notification_body_label.set_single_line_mode( True ) if notification.body else None self.notification_text_box = Box( name="notification-text", orientation="v", v_align="center", h_expand=True, h_align="start", children=[ Box( name="notification-summary-box", orientation="h", children=[ self.notification_summary_label, Box( name="notif-sep", h_expand=False, v_expand=False, h_align="center", v_align="center", ), self.notification_app_name_label_content, ], ), self.notification_body_label, ], ) self.content_close_button = self.create_close_button() self.content_close_button_box = Box( orientation="v", children=[ self.content_close_button, ], ) return Box( name="notification-content", spacing=8, children=[ self.notification_image_box, self.notification_text_box, self.content_close_button_box, ], ) def create_action_buttons(self): notification = self.notification if not notification.actions: return None grid = Gtk.Grid() grid.set_column_homogeneous(True) grid.set_column_spacing(4) for i, action in enumerate(notification.actions): action_button = ActionButton(action, i, len(notification.actions), self) grid.attach(action_button, i, 0, 1, 1) return grid def create_close_button(self): self.close_button = Button( name="notif-close-button", child=Label(name="notif-close-label", markup=icons.cancel), on_clicked=lambda *_: self.notification.close("dismissed-by-user"), ) self.close_button.connect( "enter-notify-event", lambda *_: self.hover_button(self.close_button) ) self.close_button.connect( "leave-notify-event", lambda *_: self.unhover_button(self.close_button) ) return self.close_button def on_hover_enter(self, *args): container = getattr(self, "_container", None) if container: container.pause_and_reset_all_timeouts() def on_hover_leave(self, *args): container = getattr(self, "_container", None) if container: container.resume_all_timeouts() def start_timeout(self): self.stop_timeout() self._timeout_id = GLib.timeout_add(self.timeout_ms, self.close_notification) def stop_timeout(self): if hasattr(self, "_timeout_id") and self._timeout_id is not None: GLib.source_remove(self._timeout_id) self._timeout_id = None def close_notification(self): # Safety check: ensure _destroyed attribute exists if not hasattr(self, "_destroyed"): logger.warning( f"NotificationBox { getattr(self, 'uuid', 'unknown') } missing _destroyed attribute, skipping close" ) return False if not self._destroyed: try: logger.debug( f"Notification { self.notification.id } timeout expired, closing notification." ) self.notification.close("expired") self.stop_timeout() except Exception as e: logger.error( f"Error in close_notification for notification { self.notification.id }: {e}" ) return False def destroy(self, from_history_delete=False): if ( hasattr(self, "cached_image_path") and self.cached_image_path and (not getattr(self, "_is_history", False) or from_history_delete) ): delete_cached_image(self.cached_image_path) # Ensure _destroyed attribute exists before setting it if hasattr(self, "_destroyed"): self._destroyed = True else: # If _destroyed doesn't exist, create it self._destroyed = True self.stop_timeout() super().destroy() def hover_button(self, button): container = getattr(self, "_container", None) if container: container.pause_and_reset_all_timeouts() def unhover_button(self, button): container = getattr(self, "_container", None) if container: container.resume_all_timeouts() class NotificationContainer(Box): def __init__( self, notification_history_instance: NotificationHistory, revealer_transition_type: str = "slide-down", ): super().__init__(name="notification-container-main", orientation="v", spacing=4) self.notification_history = notification_history_instance self._server = Notifications() self._server.connect("notification-added", self.on_new_notification) self._pending_removal = False self._is_destroying = False self.stack = Gtk.Stack( name="notification-stack", transition_type=Gtk.StackTransitionType.SLIDE_LEFT_RIGHT, transition_duration=200, visible=True, ) self.navigation = Box( name="notification-navigation", spacing=4, h_align="center" ) self.stack_box = Box( name="notification-stack-box", h_align="center", h_expand=False, children=[self.stack], ) self.prev_button = Button( name="nav-button", child=Label(name="nav-button-label", markup=icons.chevron_left), on_clicked=self.show_previous, ) self.close_all_button = Button( name="nav-button", child=Label(name="nav-button-label", markup=icons.cancel), on_clicked=self.close_all_notifications, ) self.close_all_button_label = self.close_all_button.get_child() self.close_all_button_label.add_style_class("close") self.next_button = Button( name="nav-button", child=Label(name="nav-button-label", markup=icons.chevron_right), on_clicked=self.show_next, ) for button in [self.prev_button, self.close_all_button, self.next_button]: button.connect( "enter-notify-event", lambda *_: self.pause_and_reset_all_timeouts() ) button.connect("leave-notify-event", lambda *_: self.resume_all_timeouts()) self.navigation.add(self.prev_button) self.navigation.add(self.close_all_button) self.navigation.add(self.next_button) self.navigation_revealer = Revealer( transition_type="slide-down", transition_duration=200, child=self.navigation, reveal_child=False, ) self.notification_box_container = Box( name="notification-box-internal-container", orientation="v", children=[self.stack_box, self.navigation_revealer], ) self.main_revealer = Revealer( name="notification-main-revealer", transition_type=revealer_transition_type, transition_duration=250, child_revealed=False, child=self.notification_box_container, ) self.add(self.main_revealer) self.notifications = [] self.current_index = 0 self.update_navigation_buttons() self._destroyed_notifications = set() def on_new_notification(self, fabric_notif, id): notification_history_instance = self.notification_history if notification_history_instance.do_not_disturb_enabled: notification = fabric_notif.get_notification_from_id(id) new_box = NotificationBox(notification) if notification.image_pixbuf: cache_notification_pixbuf(new_box) new_box.set_is_history(True) notification_history_instance.add_notification(new_box) return notification = fabric_notif.get_notification_from_id(id) new_box = NotificationBox(notification) new_box.set_container(self) notification.connect("closed", self.on_notification_closed) app_name = notification.app_name if app_name in get_history_ignored_apps(): logger.info( f"Ignoring notification from {app_name} as it is in the ignored list." ) # Don't add to history, just show the notification and let it timeout normally new_box.set_is_history(False) # Continue with normal notification display but skip history logic self.stack.add_named(new_box, str(id)) self.notifications.append(new_box) self.current_index = len(self.notifications) - 1 self.stack.set_visible_child(new_box) for notification_box in self.notifications: notification_box.start_timeout() self.main_revealer.show_all() self.main_revealer.set_reveal_child(True) self.update_navigation_buttons() return if app_name in get_limited_apps_history(): notification_history_instance.clear_history_for_app(app_name) existing_notification_index = -1 for index, existing_box in enumerate(self.notifications): if existing_box.notification.app_name == app_name: existing_notification_index = index break if existing_notification_index != -1: old_notification_box = self.notifications.pop( existing_notification_index ) self.stack.remove(old_notification_box) old_notification_box.set_is_history(True) notification_history_instance.add_notification(old_notification_box) old_notification_box.destroy() self.stack.add_named(new_box, str(id)) self.notifications.append(new_box) self.current_index = len(self.notifications) - 1 self.stack.set_visible_child(new_box) else: while len(self.notifications) >= 5: oldest_notification = self.notifications[0] oldest_notification.set_is_history(True) notification_history_instance.add_notification(oldest_notification) self.stack.remove(oldest_notification) self.notifications.pop(0) if self.current_index > 0: self.current_index -= 1 self.stack.add_named(new_box, str(id)) self.notifications.append(new_box) self.current_index = len(self.notifications) - 1 self.stack.set_visible_child(new_box) else: while len(self.notifications) >= 5: oldest_notification = self.notifications[0] oldest_notification.set_is_history(True) notification_history_instance.add_notification(oldest_notification) self.stack.remove(oldest_notification) self.notifications.pop(0) if self.current_index > 0: self.current_index -= 1 self.stack.add_named(new_box, str(id)) self.notifications.append(new_box) self.current_index = len(self.notifications) - 1 self.stack.set_visible_child(new_box) for notification_box in self.notifications: notification_box.start_timeout() self.main_revealer.show_all() self.main_revealer.set_reveal_child(True) self.update_navigation_buttons() def show_previous(self, *args): if self.current_index > 0: self.current_index -= 1 self.stack.set_visible_child(self.notifications[self.current_index]) self.update_navigation_buttons() def show_next(self, *args): if self.current_index < len(self.notifications) - 1: self.current_index += 1 self.stack.set_visible_child(self.notifications[self.current_index]) self.update_navigation_buttons() def update_navigation_buttons(self): self.prev_button.set_sensitive(self.current_index > 0) self.next_button.set_sensitive(self.current_index < len(self.notifications) - 1) should_reveal = len(self.notifications) > 1 self.navigation_revealer.set_reveal_child(should_reveal) def on_notification_closed(self, notification, reason): if self._is_destroying: return if notification.id in self._destroyed_notifications: return self._destroyed_notifications.add(notification.id) try: notif_to_remove = None for i, notif_box in enumerate(self.notifications): if notif_box.notification.id == notification.id: notif_to_remove = (i, notif_box) break if not notif_to_remove: return i, notif_box = notif_to_remove reason_str = str(reason) notification_history_instance = self.notification_history # Check if this notification should be ignored from history app_name = notification.app_name if app_name not in get_history_ignored_apps(): notif_box.set_is_history(True) notification_history_instance.add_notification(notif_box) if reason_str == "NotificationCloseReason.DISMISSED_BY_USER": logger.info( f"User dismissed notification { notification.id }, added to history" ) elif ( reason_str == "NotificationCloseReason.EXPIRED" or reason_str == "NotificationCloseReason.CLOSED" or reason_str == "NotificationCloseReason.UNDEFINED" ): logger.info( f"Notification { notification.id } closed automatically, added to history" ) else: logger.warning( f"Unknown close reason: {reason_str} for notification { notification.id }. Still added to history." ) else: logger.info( f"Notification from {app_name} closed but not added to history (ignored app)" ) notif_box.stop_timeout() if len(self.notifications) == 1: self._is_destroying = True self.main_revealer.set_reveal_child(False) GLib.timeout_add( self.main_revealer.get_transition_duration(), self._destroy_container, ) return new_index = i if i == self.current_index: new_index = max(0, i - 1) elif i < self.current_index: new_index = self.current_index - 1 if notif_box.get_parent() == self.stack: self.stack.remove(notif_box) self.notifications.pop(i) if new_index >= len(self.notifications) and len(self.notifications) > 0: new_index = len(self.notifications) - 1 self.current_index = new_index if len(self.notifications) > 0: self.stack.set_visible_child(self.notifications[self.current_index]) self.update_navigation_buttons() except Exception as e: logger.error(f"Error closing notification: {e}") def _destroy_container(self): try: self.notifications.clear() self._destroyed_notifications.clear() for child in self.stack.get_children(): self.stack.remove(child) child.destroy() self.current_index = 0 except Exception as e: logger.error(f"Error cleaning up the container: {e}") finally: self._is_destroying = False return False def pause_and_reset_all_timeouts(self): if self._is_destroying: return for notification in self.notifications[:]: try: if hasattr(notification, "pause_timeout"): notification.pause_timeout() except Exception as e: logger.error(f"Error pausing timeout: {e}") def resume_all_timeouts(self): if self._is_destroying: return for notification in self.notifications[:]: try: if hasattr(notification, "resume_timeout"): notification.resume_timeout() except Exception as e: logger.error(f"Error resuming timeout: {e}") def close_all_notifications(self, *args): notifications_to_close = self.notifications.copy() for notification_box in notifications_to_close: notification_box.notification.close("dismissed-by-user") class NotificationPopup(Window): def __init__(self, **kwargs): y_pos = data.NOTIF_POS.lower() if hasattr(data, "NOTIF_POS") else "top" x_pos = "right" if hasattr(data, "DOCK_POSITION") and data.DOCK_POSITION in ["Top", "Bottom"]: x_pos = "right" super().__init__( name="notification-popup", anchor=f"{x_pos} {y_pos}", layer="top", keyboard_mode="none", exclusivity="none", visible=True, all_visible=True, ) self.widgets = kwargs.get("widgets", None) self.notification_history = ( self.widgets.notification_history if self.widgets else get_shared_notification_history() ) self.notification_container = NotificationContainer( notification_history_instance=self.notification_history, revealer_transition_type="slide-down" if y_pos == "top" else "slide-up", ) self.show_box = Box() self.show_box.set_size_request(1, 1) self.add( Box( name="notification-popup-box", orientation="v", children=[self.notification_container, self.show_box], ) )