From 03dbf362840eca332184069954cfa778be2b513a Mon Sep 17 00:00:00 2001
From: Tyler Miller <tmillr@proton.me>
Date: Fri, 13 Sep 2024 15:37:23 -0700
Subject: [PATCH] feat: restart timer on config change

---
 TODO.md              |  2 --
 lua/sos/observer.lua | 25 +++++++++++++++++--------
 2 files changed, 17 insertions(+), 10 deletions(-)

diff --git a/TODO.md b/TODO.md
index 8896d86..883b7d0 100644
--- a/TODO.md
+++ b/TODO.md
@@ -106,5 +106,3 @@
       file/buf the err msg is for).
 
 - [ ] feat: opt (and maybe cmd/func) to disable timer
-
-- [ ] Calling `setup()` should reset timer if it is active?
diff --git a/lua/sos/observer.lua b/lua/sos/observer.lua
index 03211c3..f52c55d 100644
--- a/lua/sos/observer.lua
+++ b/lua/sos/observer.lua
@@ -19,6 +19,13 @@ function MultiBufObserver:new()
     pending_detach = {},
   }
 
+  function instance:debounce()
+    local result, err, _ = timer:stop()
+    assert(result == 0, err)
+    result, err, _ = timer:start(self.timeout, 0, self.on_timer)
+    assert(result == 0, err)
+  end
+
   ---Called whenever a buffer incurs a savable change (i.e. writing the buffer
   ---would change the file's contents on the filesystem). All this does is
   ---debounce the timer.
@@ -28,10 +35,7 @@ function MultiBufObserver:new()
   ---@return true | nil
   function instance:on_change(buf)
     if not running or self.pending_detach[buf] then return true end -- detach
-    local result, err, _ = timer:stop()
-    assert(result == 0, err)
-    result, err, _ = timer:start(self.timeout, 0, self.on_timer)
-    assert(result == 0, err)
+    self:debounce()
   end
 
   ---Attach buffer callbacks if not already attached
@@ -133,20 +137,25 @@ function MultiBufObserver:new()
     self.autocmds = {}
   end
 
-  ---@class sos.MultiBufObserver.start.opts
-  ---@field [string] any
+  ---@class sos.MultiBufObserver.opts
   ---@field timeout integer timeout in milliseconds
   ---@field on_timer function
   ---@field should_observe_buf fun(buf: integer): boolean
 
   ---Begin observing buffers with this observer. Ok to call when already
   ---running.
-  ---@param opts sos.MultiBufObserver.start.opts
+  ---@param opts sos.MultiBufObserver.opts
   function instance:start(opts)
     self.timeout = opts.timeout
     self.on_timer = vim.schedule_wrap(opts.on_timer)
     self.should_observe_buf_cb = opts.should_observe_buf
-    if running then return end
+
+    if running then
+      -- Timeout may have changed
+      if self:due_in() > 0 then self:debounce() end
+      return
+    end
+
     running = true
 
     vim.list_extend(self.autocmds, {