From 2e63463f1971d88d02e4b6301fedc48093ac49ba Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=C5=81ukasz=20Maria=C5=84ski?=
 <13919176+m00nwtchr@users.noreply.github.com>
Date: Wed, 18 Dec 2024 16:29:44 +0000
Subject: [PATCH] Remove dependency on X11 (#672)

* Remove dependency on X11 from break_screen
---
 safeeyes/ui/break_screen.py | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

diff --git a/safeeyes/ui/break_screen.py b/safeeyes/ui/break_screen.py
index 7546fd83..bc826221 100644
--- a/safeeyes/ui/break_screen.py
+++ b/safeeyes/ui/break_screen.py
@@ -44,7 +44,7 @@ class BreakScreen:
     def __init__(self, context, on_skipped, on_postponed, style_sheet_path):
         self.context = context
         self.count_labels = []
-        self.display = Display()
+        self.display = None
         self.enable_postpone = False
         self.enable_shortcut = False
         self.is_pretified = False
@@ -56,6 +56,9 @@ def __init__(self, context, on_skipped, on_postponed, style_sheet_path):
         self.strict_break = False
         self.windows = []
 
+        if not self.context['is_wayland']:
+            self.display = Display()
+
         # Initialize the theme
         css_provider = Gtk.CssProvider()
         css_provider.load_from_path(style_sheet_path)
@@ -131,7 +134,8 @@ def close(self):
         Hide the break screen from active window and destroy all other windows
         """
         logging.info("Close the break screen(s)")
-        self.__release_keyboard()
+        if not self.context['is_wayland']:
+            self.__release_keyboard()
 
         # Destroy other windows if exists
         GLib.idle_add(lambda: self.__destroy_all_screens())
@@ -149,7 +153,11 @@ def __show_break_screen(self, message, image_path, widget, tray_actions):
         Show an empty break screen on all screens.
         """
         # Lock the keyboard
-        utility.start_thread(self.__lock_keyboard)
+        if not self.context['is_wayland']:
+            utility.start_thread(self.__lock_keyboard)
+        else:
+            # TODO: Wayland keyboard locking
+            logging.warning("Keyboard locking not yet implemented for Wayland.")
 
         display = Gdk.Display.get_default()
         screen = display.get_default_screen()
@@ -243,7 +251,7 @@ def __update_count_down(self, count):
 
     def __lock_keyboard(self):
         """
-        Lock the keyboard to prevent the user from using keyboard shortcuts
+        Lock the keyboard to prevent the user from using keyboard shortcuts (X11 only)
         """
         logging.info("Lock the keyboard")
         self.lock_keyboard = True