From 827d086c5455f02efd3035c3ec32014459b24d4e Mon Sep 17 00:00:00 2001
From: Kovid Goyal <kovid@kovidgoyal.net>
Date: Sat, 9 Sep 2017 14:26:16 +0530
Subject: [PATCH] Allow re-adding of timers during timers_call()

---
 kitty/timers.c | 15 ++++++++++++++-
 1 file changed, 14 insertions(+), 1 deletion(-)

diff --git a/kitty/timers.c b/kitty/timers.c
index 0e788224d..924dee61a 100644
--- a/kitty/timers.c
+++ b/kitty/timers.c
@@ -217,15 +217,27 @@ timers_call(Timers *self) {
     double now = monotonic_();
     size_t i, j;
     self->in_call = true;
+    bool needs_sort = false;
     for (i = 0, j = 0; i < self->count; i++) {
+        bool remove = false;
         if (self->events[i].at <= now) {  // expired, call it
             /* PyObject_Print(self->events[i].callback, stdout, 1); */
             /* printf("\n"); */
+            remove = true;
             if (self->events[i].callback != Py_None) {
                 PyObject *ret = self->events[i].args ? PyObject_CallObject(self->events[i].callback, self->events[i].args) : PyObject_CallFunctionObjArgs(self->events[i].callback, NULL);
                 if (ret == NULL) PyErr_Print();
-                else Py_DECREF(ret);
+                else {
+                    if (ret != Py_None && PyFloat_Check(ret)) {
+                        self->events[i].at = monotonic_() + PyFloat_AS_DOUBLE(ret);
+                        remove = false;
+                        needs_sort = true;
+                    }
+                    Py_DECREF(ret);
+                }
             }
+        }
+        if (remove) {
             Py_CLEAR(self->events[i].callback); Py_CLEAR(self->events[i].args);
         } else {
             other[j].callback = self->events[i].callback; other[j].at = self->events[i].at; other[j].args = self->events[i].args;
@@ -235,6 +247,7 @@ timers_call(Timers *self) {
     self->events = other;
     self->count = j;
     self->in_call = false;
+    if (needs_sort) qsort(self->events, self->count, sizeof(TimerEvent), compare_events);
 }
 
 static PyObject *