Register colorscheme highlight refresh

1 files changed, 15 insertions, 1 deletions

diff --git a/lua/clever_f/composition_root.lua b/lua/clever_f/composition_root.lua
index 8d75359..527fec6 100644
--- a/lua/clever_f/composition_root.lua
+++ b/lua/clever_f/composition_root.lua
@@ -71,6 +71,7 @@ function CompositionRoot.new(options)
     coordinator = coordinator,
     facade = facade,
     activation = nil,
+    last_highlight_refresh = nil,
   }
   return root
 end
@@ -82,14 +83,27 @@ end
 function CompositionRoot:activate()
   local record = root_records[self]
   if record.activation == nil then
+    local highlights = record.feedback:evaluate_highlights()
+    local colorscheme_registration = record.host:register_events(
+      "ColorScheme",
+      function()
+        record.last_highlight_refresh = record.feedback:evaluate_highlights()
+      end,
+      { owner = "clever_f", lifecycle = "colorscheme" }
+    )
     record.activation = {
       state = record.state,
-      highlights = record.feedback:evaluate_highlights(),
+      highlights = highlights,
+      colorscheme_registration = colorscheme_registration,
     }
   end
   return record.activation
 end
 
+function CompositionRoot:last_highlight_refresh()
+  return root_records[self].last_highlight_refresh
+end
+
 function CompositionRoot:host()
   return root_records[self].host
 end