From d4e732a7f068c50e1916f005bd033ab9159a774c Mon Sep 17 00:00:00 2001
From: Filip Kilibarda <filipcsc@gmail.com>
Date: Sat, 13 Jul 2019 16:08:41 -0700
Subject: [PATCH] Added idle time support for gnome/wayland (#332)

* Added idle time support for gnome/wayland

* Removed is_gnome, using Utility.DESKTOP_ENVIRONMENT instead
---
 safeeyes/plugins/smartpause/plugin.py | 44 +++++++++++++++++++++++++++
 1 file changed, 44 insertions(+)

diff --git a/safeeyes/plugins/smartpause/plugin.py b/safeeyes/plugins/smartpause/plugin.py
index 9f03605..7c131c9 100644
--- a/safeeyes/plugins/smartpause/plugin.py
+++ b/safeeyes/plugins/smartpause/plugin.py
@@ -20,6 +20,8 @@ import datetime
 import logging
 import subprocess
 import threading
+import re
+import os
 
 from safeeyes import Utility
 from safeeyes.model import State
@@ -44,12 +46,54 @@ waiting_time = 2
 interpret_idle_as_break = False
 
 
+def is_wayland():
+    """
+    Determine if Wayland is running
+    https://unix.stackexchange.com/a/325972/222290
+    """
+    try:
+        output = subprocess.check_output(['loginctl'])
+        second_line = output.split(b'\n')[1]
+        session_id = re.search(rb'^ *?(\d+)', second_line).group(1)
+        output = subprocess.check_output(
+            ['loginctl', 'show-session', session_id, '-p', 'Type']
+        )
+    except BaseException:
+        logging.info('Unable to determine if wayland is running. Assuming no.')
+        return False
+    else:
+        return bool(re.search(b'wayland', output, re.IGNORECASE))
+
+
+def __gnome_wayland_idle_time():
+    """
+    Determine system idle time in seconds, specifically for gnome with wayland.
+    If there's a failure, return 0.
+    https://unix.stackexchange.com/a/492328/222290
+    """
+    try:
+        output = subprocess.check_output([
+            'dbus-send',
+            '--print-reply',
+            '--dest=org.gnome.Mutter.IdleMonitor',
+            '/org/gnome/Mutter/IdleMonitor/Core',
+            'org.gnome.Mutter.IdleMonitor.GetIdletime'
+        ])
+        return int(re.search(rb'\d+$', output).group(0)) / 1000
+    except BaseException as e:
+        logging.warning("Failed to get system idle time for gnome/wayland.")
+        logging.warning(str(e))
+        return 0
+
+
 def __system_idle_time():
     """
     Get system idle time in minutes.
     Return the idle time if xprintidle is available, otherwise return 0.
     """
     try:
+        if is_wayland() and Utility.DESKTOP_ENVIRONMENT == 'gnome':
+            return __gnome_wayland_idle_time()
         # Convert to seconds
         return int(subprocess.check_output(['xprintidle']).decode('utf-8')) / 1000
     except BaseException: