Add component append

NeogitOrg · Nov 28, 2023 · c1ff05e · c1ff05e
1 parent ec64f37
commit c1ff05e
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/lua/neogit/lib/ui/component.lua b/lua/neogit/lib/ui/component.lua
@@ -108,6 +108,11 @@ function Component:get_highlight()
   return self.options.highlight or (self.parent and self.parent:get_highlight() or nil)
 end
 
+function Component:append(c)
+  table.insert(self.children, c)
+  return self
+end
+
 function Component.new(f)
   local x = {}
   setmetatable(x, {

diff --git a/lua/neogit/lib/ui/init.lua b/lua/neogit/lib/ui/init.lua
@@ -129,6 +129,13 @@ function Ui:get_component_stack_under_cursor()
   end)
 end
 
+function Ui:get_fold_under_cursor()
+  local cursor = vim.api.nvim_win_get_cursor(0)
+  return self:find_component(function(c)
+    return c.options.foldable and c:is_under_cursor(cursor)
+  end)
+end
+
 function Ui:get_component_stack_in_linewise_selection()
   local range = { vim.fn.getpos("v")[2], vim.fn.getpos(".")[2] }
   table.sort(range)
@@ -313,7 +320,8 @@ function Ui:_render(first_line, first_col, parent, components, flags)
       if c.options.foldable then
         self.buf:buffered_create_fold(
           #self.buf.line_buffer - (c.position.row_end - c.position.row_start),
-          #self.buf.line_buffer
+          #self.buf.line_buffer,
+          not c.options.folded
         )
       end
     end