feat: hooks

README.md

@@ -92,6 +92,14 @@ hunk.setup({
     folder_open = "",
     folder_closed = "",
   },
+
+  -- Called right after each window and buffer are created.
+  hooks = {
+    ---@param _context { buf: number, tree: NuiTree, opts: table }
+    on_tree_mount = function(_context) end,
+    ---@param _context { buf: number, win: number }
+    on_diff_mount = function(_context) end,
+  },
 })
 ```
 

lua/hunk/config.lua

@@ -41,8 +41,10 @@ local M = {
   },
 
   hooks = {
-    on_tree_mount = function() end,
-    on_diff_mount = function() end,
+    ---@param _context { buf: number, tree: NuiTree, opts: table }
+    on_tree_mount = function(_context) end,
+    ---@param _context { buf: number, win: number }
+    on_diff_mount = function(_context) end,
   },
 }
 

lua/hunk/ui/file.lua

@@ -65,6 +65,8 @@ function M.create(window, params)
     end, { buffer = buf })
   end
 
+  config.hooks.on_diff_mount({ buf = buf, win = window })
+
   local function apply_signs()
     api.signs.clear_signs(buf)
 

lua/hunk/ui/tree.lua

@@ -263,6 +263,8 @@ function M.create(opts)
     end, { buffer = buf })
   end
 
+  config.hooks.on_tree_mount({ buf = buf, tree = tree, opts = opts })
+
   local file_tree
   if config.ui.tree.mode == "nested" then
     file_tree = build_file_tree(opts.changeset)