feat(toc): add close_after_use configuration option (#785)

nvim-neorg · Apr 19, 2023 · e5d7fbb · e5d7fbb
1 parent 1fecaab
commit e5d7fbb
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/lua/neorg/modules/core/norg/qol/toc/module.lua b/lua/neorg/modules/core/norg/qol/toc/module.lua
@@ -39,6 +39,12 @@ module.load = function()
     end)
 end
 
+module.config.public = {
+    -- If `true`, will close the Table of Contents after an entry in the table
+    -- is picked.
+    close_after_use = false,
+}
+
 module.public = {
     parse_toc_macro = function(buffer)
         local toc, toc_name = false, nil
@@ -205,6 +211,11 @@ module.public = {
                 vim.api.nvim_set_current_win(original_window)
                 vim.api.nvim_set_current_buf(original_buffer)
                 vim.api.nvim_win_set_cursor(original_window, { location[1] + 1, location[2] })
+
+                if module.config.public.close_after_use then
+
+                    vim.api.nvim_buf_delete(ui_buffer, { force = true })
+                end
             end,
         })
     end,