Move between headlines with curly braces

[andreadev-it]

I'm coming from orgmode.nvim (still using it as my daily driver), and I found this functionality to be extremly useful, expecially if I wanted to open some folds very fast. When I use curly braces in normal mode, instead of moving me between paragraphs, it moves by headings.

I was able to do it with some code that I've inserted into my personal config, but maybe it could be useful to make it part of the default experience? What do you think?

Here is the code I've used:

local function goto_headline(which)
  local ts_utils = require('nvim-treesitter.ts_utils')

  local tsparser = vim.treesitter.get_parser()
  local tstree = tsparser:parse()
  local root = tstree[1]:root()

  local cursor = vim.api.nvim_win_get_cursor(0)
  local cursor_range = { cursor[1] - 1, cursor[2] }

  -- Query all headings (from 1 to 6)
  local query = vim.treesitter.query.parse_query('norg', [[
    (heading1) @h1
    (heading2) @h2
    (heading3) @h3
    (heading4) @h4
    (heading5) @h5
    (heading6) @h6
  ]])

  local previous_headline = nil
  local next_headline = nil

  -- Find the previous and next heading from the captures
  for _, captures, metadata in query:iter_matches(root) do
    for i, node in pairs(captures) do
      local row = node:start()
      if row < cursor_range[1] then
        previous_headline = node
      elseif row > cursor_range[1] and next_headline == nil then
        next_headline = node
        break
      end
    end
  end

  if which == 'previous' then
    ts_utils.goto_node(previous_headline)
  elseif which == 'next' then
    ts_utils.goto_node(next_headline)
  end
end

-- Go to previous headline
local function goto_previous_headline()
  goto_headline('previous')
end

-- Go to next headline
local function goto_next_headline()
  goto_headline('next')
end

vim.api.nvim_create_augroup('NeorgKeymaps', { clear = true })
vim.api.nvim_create_autocmd("FileType", {
  pattern = { 'norg' },
  callback = function ()
    vim.keymap.set('n', '{', goto_previous_headline, { desc = 'Neorg: Go to previous headline', buffer=true })
    vim.keymap.set('n', '}', goto_next_headline, { desc = 'Neorg: Go to next headline', buffer=true })
  end
})