From 11489152d8e18d8fbcef4c7f1d46d901a19781f6 Mon Sep 17 00:00:00 2001
From: Dan Williams <dan@ioncontrol.co>
Date: Sun, 6 Jul 2025 18:15:19 -0500
Subject: [PATCH] sleep-context: add ability to increase timeout

Track when the timeout was started, so that if it needs to be increased
we can calculate how long the new timeout should be. Then cancel the
original timeout and start a new one with.

Signed-off-by: Dan Williams <dan@ioncontrol.co>
---
 src/mm-sleep-context.c | 33 ++++++++++++++++++++++++++++++---
 src/mm-sleep-context.h |  3 +++
 2 files changed, 33 insertions(+), 3 deletions(-)

diff --git a/src/mm-sleep-context.c b/src/mm-sleep-context.c
index 69a9da22..d6c01198 100644
--- a/src/mm-sleep-context.c
+++ b/src/mm-sleep-context.c
@@ -27,6 +27,7 @@ G_DEFINE_TYPE_EXTENDED (MMSleepContext, mm_sleep_context, G_TYPE_OBJECT, 0,
 struct _MMSleepContextPrivate {
     GTask *task;
     guint  timeout_id;
+    gint64 timeout_start;
 };
 
 enum {
@@ -52,6 +53,7 @@ timeout_cleanup (MMSleepContext *self)
     if (self->priv->timeout_id)
         g_source_remove (self->priv->timeout_id);
     self->priv->timeout_id = 0;
+    self->priv->timeout_start = 0;
 }
 
 void
@@ -95,6 +97,33 @@ operation_ready (MMSleepContext *self,
     g_signal_emit (self, signals[DONE], 0, error);
 }
 
+static void
+timeout_add (MMSleepContext *self,
+             guint           interval)
+{
+    timeout_cleanup (self);
+    self->priv->timeout_start = g_get_monotonic_time ();
+    self->priv->timeout_id = g_timeout_add_seconds (interval,
+                                                    (GSourceFunc)operation_timeout,
+                                                    self);
+}
+
+void
+mm_sleep_context_timeout_backoff (MMSleepContext *self,
+                                  guint           more_seconds)
+{
+    gint64 cur_time;
+    gint64 new_interval;
+
+    new_interval = more_seconds * G_USEC_PER_SEC;
+    if (self->priv->timeout_start) {
+        cur_time = g_get_monotonic_time ();
+        new_interval += (cur_time - self->priv->timeout_start);
+    }
+
+    timeout_add (self, new_interval / G_USEC_PER_SEC);
+}
+
 /*****************************************************************************/
 
 MMSleepContext *
@@ -107,9 +136,7 @@ mm_sleep_context_new (guint timeout_seconds)
                                    NULL,
                                    (GAsyncReadyCallback)operation_ready,
                                    NULL);
-    self->priv->timeout_id = g_timeout_add_seconds (timeout_seconds,
-                                                    (GSourceFunc)operation_timeout,
-                                                    self);
+    timeout_add (self, timeout_seconds);
     mm_obj_dbg (self, "new context with %d second timeout", timeout_seconds);
 
     return self;
diff --git a/src/mm-sleep-context.h b/src/mm-sleep-context.h
index cd45c1f2..44b3c28a 100644
--- a/src/mm-sleep-context.h
+++ b/src/mm-sleep-context.h
@@ -51,6 +51,9 @@ G_DEFINE_AUTOPTR_CLEANUP_FUNC (MMSleepContext, g_object_unref)
 
 MMSleepContext *mm_sleep_context_new      (guint           timeout_seconds);
 
+void            mm_sleep_context_timeout_backoff (MMSleepContext *self,
+                                                  guint           more_seconds);
+
 /* The MMSleepContext assumes ownership of @error */
 void            mm_sleep_context_complete (MMSleepContext *self,
                                            GError         *error);