Correct last input for idle timeout for Wayland handlers

2e3s · May 16, 2024 · f960dd6 · f960dd6
1 parent be59315
commit f960dd6
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 7 deletions.
diff --git a/watchers/src/watchers/wl_ext_idle_notify.rs b/watchers/src/watchers/wl_ext_idle_notify.rs
@@ -19,6 +19,7 @@ struct IdleState {
     last_input_time: DateTime<Utc>,
     is_idle: bool,
     is_changed: bool,
+    idle_timeout: Duration,
 }
 
 impl Drop for IdleState {
@@ -29,18 +30,20 @@ impl Drop for IdleState {
 }
 
 impl IdleState {
-    fn new(idle_notification: ExtIdleNotificationV1) -> Self {
+    fn new(idle_notification: ExtIdleNotificationV1, idle_timeout: Duration) -> Self {
         Self {
             idle_notification,
             last_input_time: Utc::now(),
             is_idle: false,
             is_changed: false,
+            idle_timeout,
         }
     }
 
     fn idle(&mut self) {
         self.is_idle = true;
         self.is_changed = true;
+        self.last_input_time -= self.idle_timeout;
         debug!("Idle");
     }
 
@@ -138,6 +141,7 @@ impl Watcher for IdleWatcher {
             connection
                 .get_ext_idle_notification(timeout.unwrap())
                 .unwrap(),
+            Duration::from_std(client.config.idle_timeout).unwrap(),
         );
         connection.event_queue.roundtrip(&mut idle_state).unwrap();
 

diff --git a/watchers/src/watchers/wl_kwin_idle.rs b/watchers/src/watchers/wl_kwin_idle.rs
@@ -16,32 +16,35 @@ use wayland_protocols_plasma::idle::client::org_kde_kwin_idle_timeout::{
 };
 
 struct IdleState {
-    idle_timeout: OrgKdeKwinIdleTimeout,
+    kwin_idle_timeout: OrgKdeKwinIdleTimeout,
     last_input_time: DateTime<Utc>,
     is_idle: bool,
     is_changed: bool,
+    idle_timeout: Duration,
 }
 
 impl Drop for IdleState {
     fn drop(&mut self) {
         info!("Releasing idle timeout");
-        self.idle_timeout.release();
+        self.kwin_idle_timeout.release();
     }
 }
 
 impl IdleState {
-    fn new(idle_timeout: OrgKdeKwinIdleTimeout) -> Self {
+    fn new(kwin_idle_timeout: OrgKdeKwinIdleTimeout, idle_timeout: Duration) -> Self {
         Self {
-            idle_timeout,
+            kwin_idle_timeout,
             last_input_time: Utc::now(),
             is_idle: false,
             is_changed: false,
+            idle_timeout,
         }
     }
 
     fn idle(&mut self) {
         self.is_idle = true;
         self.is_changed = true;
+        self.last_input_time -= self.idle_timeout;
         debug!("Idle");
     }
 
@@ -135,8 +138,10 @@ impl Watcher for IdleWatcher {
         connection.get_kwin_idle()?;
 
         let timeout = u32::try_from(client.config.idle_timeout.as_secs() * 1000);
-        let mut idle_state =
-            IdleState::new(connection.get_kwin_idle_timeout(timeout.unwrap()).unwrap());
+        let mut idle_state = IdleState::new(
+            connection.get_kwin_idle_timeout(timeout.unwrap()).unwrap(),
+            Duration::from_std(client.config.idle_timeout).unwrap(),
+        );
         connection.event_queue.roundtrip(&mut idle_state).unwrap();
 
         Ok(Self {