Update tab callback

NeogitOrg · Nov 28, 2023 · 185c69b · 185c69b
1 parent f95809a
commit 185c69b
Showing 1 changed file with 40 additions and 30 deletions.
diff --git a/lua/neogit/buffers/refs_view/init.lua b/lua/neogit/buffers/refs_view/init.lua
@@ -137,38 +137,48 @@ function M:open()
         ["<enter>"] = function()
           CommitViewBuffer.new(self.buffer.ui:get_commits_in_selection()[1]):open()
         end,
-        ["{"] = function()
-          pcall(vim.cmd, "normal! zc")
-
-          vim.cmd("normal! k")
-          for _ = vim.fn.line("."), 0, -1 do
-            if vim.fn.foldlevel(".") > 0 then
-              break
-            end
-
-            vim.cmd("normal! k")
-          end
-
-          pcall(vim.cmd, "normal! zo")
-          vim.cmd("normal! zz")
-        end,
-        ["}"] = function()
-          pcall(vim.cmd, "normal! zc")
-
-          vim.cmd("normal! j")
-          for _ = vim.fn.line("."), vim.fn.line("$"), 1 do
-            if vim.fn.foldlevel(".") > 0 then
-              break
+        -- ["{"] = function()
+        --   pcall(vim.cmd, "normal! zc")
+        --
+        --   vim.cmd("normal! k")
+        --   for _ = vim.fn.line("."), 0, -1 do
+        --     if vim.fn.foldlevel(".") > 0 then
+        --       break
+        --     end
+        --
+        --     vim.cmd("normal! k")
+        --   end
+        --
+        --   pcall(vim.cmd, "normal! zo")
+        --   vim.cmd("normal! zz")
+        -- end,
+        -- ["}"] = function()
+        --   pcall(vim.cmd, "normal! zc")
+        --
+        --   vim.cmd("normal! j")
+        --   for _ = vim.fn.line("."), vim.fn.line("$"), 1 do
+        --     if vim.fn.foldlevel(".") > 0 then
+        --       break
+        --     end
+        --
+        --     vim.cmd("normal! j")
+        --   end
+        --
+        --   pcall(vim.cmd, "normal! zo")
+        --   vim.cmd("normal! zz")
+        -- end,
+        ["<tab>"] = function()
+          local fold = self.buffer.ui:get_fold_under_cursor()
+          if fold then
+            if fold.options.on_open then
+              fold.options.on_open(fold, self.buffer.ui)
+            else
+              local ok, _ = pcall(vim.cmd, "normal! za")
+              if ok then
+                fold.options.folded = not fold.options.folded
+              end
             end
-
-            vim.cmd("normal! j")
           end
-
-          pcall(vim.cmd, "normal! zo")
-          vim.cmd("normal! zz")
-        end,
-        ["<tab>"] = function()
-          pcall(vim.cmd, "normal! za")
         end,
         ["d"] = function()
           if not config.check_integration("diffview") then