bug(core) Ignore invisible headings

Previously, `next-visible-heading`, moved to the next heading, regardless of its visibility.
nvim-orgmode · Jul 15, 2024 · cc0e081 · cc0e081
1 parent 85c1b44
commit cc0e081
Show file tree

Hide file tree

Showing 2 changed files with 116 additions and 4 deletions.
diff --git a/lua/orgmode/org/mappings.lua b/lua/orgmode/org/mappings.lua
@@ -924,12 +924,26 @@ function OrgMappings:export()
   return require('orgmode.export').prompt()
 end
 
+---Find and move cursor to next visible heading.
+---@return integer
 function OrgMappings:next_visible_heading()
-  return vim.fn.search([[^\*\+]], 'W')
+  return vim.fn.search([[^\*\+]], 'W', 0, 0, self._skip_invisible_heading)
 end
 
+---Find and move cursor to previous visible heading.
+---@return integer
 function OrgMappings:previous_visible_heading()
-  return vim.fn.search([[^\*\+]], 'bW')
+  return vim.fn.search([[^\*\+]], 'bW', 0, 0, self._skip_invisible_heading)
+end
+
+---Check if heading is visible. If not, skip it.
+---@return integer
+function OrgMappings:_skip_invisible_heading()
+  local fold = vim.fn.foldclosed('.')
+  if fold == -1 or vim.fn.line('.') == fold then
+    return 0
+  end
+  return 1
 end
 
 function OrgMappings:forward_heading_same_level()

diff --git a/tests/plenary/ui/mappings/headline_spec.lua b/tests/plenary/ui/mappings/headline_spec.lua
@@ -498,7 +498,7 @@ describe('Heading mappings', function()
       '   2. Second item',
     })
 
-    vim.cmd([[norm gg]])
+    vim.cmd([[norm ggzR]])
     assert.are.same(1, vim.fn.line('.'))
     vim.cmd([[norm }]])
     assert.are.same(3, vim.fn.line('.'))
@@ -514,6 +514,53 @@ describe('Heading mappings', function()
     assert.are.same(16, vim.fn.line('.'))
   end)
 
+  it('should jump to next visible heading on any level (org_next_visible_heading)', function()
+    helpers.create_file({
+      '#TITLE: Test',
+      '',
+      '* TODO Test orgmode',
+      '  DEADLINE: <2021-07-21 Wed 22:02>',
+      '** TODO [#A] Test orgmode level 2 :PRIVATE:',
+      'Some content for level 2',
+      '*** NEXT [#1] Level 3',
+      'Content Level 3',
+      '* DONE top level todo :WORK:',
+      'content for top level todo',
+      '* TODO top level todo with multiple tags :OFFICE:PROJECT:',
+      '  - [ ] The checkbox',
+      '  - [X] The checkbox 2',
+      '    - [ ] Nested checkbox',
+      'multiple tags content, tags not read from content :FROMCONTENT:',
+      '** NEXT Working on this now :OFFICE:NESTED:',
+      '   1. First item',
+      '   2. Second item',
+    })
+
+    -- Default collapsed
+    vim.cmd([[norm gg]])
+    assert.are.same(1, vim.fn.line('.'))
+    vim.cmd([[norm }]])
+    assert.are.same(3, vim.fn.line('.'))
+    vim.cmd([[norm }]])
+    assert.are.same(9, vim.fn.line('.'))
+    vim.cmd([[norm }]])
+    assert.are.same(11, vim.fn.line('.'))
+
+    -- Open first level
+    vim.cmd([[norm ggzr]])
+    assert.are.same(1, vim.fn.line('.'))
+    vim.cmd([[norm }]])
+    assert.are.same(3, vim.fn.line('.'))
+    vim.cmd([[norm }]])
+    assert.are.same(5, vim.fn.line('.'))
+    vim.cmd([[norm }]])
+    assert.are.same(9, vim.fn.line('.'))
+    vim.cmd([[norm }]])
+    assert.are.same(11, vim.fn.line('.'))
+    vim.cmd([[norm }]])
+    assert.are.same(16, vim.fn.line('.'))
+  end)
+
   it('should jump to previous heading on any level (org_previous_visible_heading)', function()
     helpers.create_file({
       '#TITLE: Test',
@@ -536,7 +583,7 @@ describe('Heading mappings', function()
       '   2. Second item',
     })
 
-    vim.cmd([[norm G]])
+    vim.cmd([[norm GzR]])
     assert.are.same(18, vim.fn.line('.'))
     vim.cmd([[norm {]])
     assert.are.same(16, vim.fn.line('.'))
@@ -554,6 +601,57 @@ describe('Heading mappings', function()
     assert.are.same(3, vim.fn.line('.'))
   end)
 
+  it('should jump to previous visible heading on any level (org_previous_visible_heading)', function()
+    helpers.create_file({
+      '#TITLE: Test',
+      '',
+      '* TODO Test orgmode',
+      '  DEADLINE: <2021-07-21 Wed 22:02>',
+      '** TODO [#A] Test orgmode level 2 :PRIVATE:',
+      'Some content for level 2',
+      '*** NEXT [#1] Level 3',
+      'Content Level 3',
+      '* DONE top level todo :WORK:',
+      'content for top level todo',
+      '* TODO top level todo with multiple tags :OFFICE:PROJECT:',
+      '  - [ ] The checkbox',
+      '  - [X] The checkbox 2',
+      '    - [ ] Nested checkbox',
+      'multiple tags content, tags not read from content :FROMCONTENT:',
+      '** NEXT Working on this now :OFFICE:NESTED:',
+      '   1. First item',
+      '   2. Second item',
+    })
+
+    -- Default to only top level
+    vim.cmd([[norm G]])
+    assert.are.same(18, vim.fn.line('.'))
+    vim.cmd([[norm {]])
+    assert.are.same(11, vim.fn.line('.'))
+    vim.cmd([[norm {]])
+    assert.are.same(9, vim.fn.line('.'))
+    vim.cmd([[norm {]])
+    assert.are.same(3, vim.fn.line('.'))
+    vim.cmd([[norm {]])
+    assert.are.same(3, vim.fn.line('.'))
+
+    -- open one dir
+    vim.cmd([[exec "norm Gzr"]])
+    assert.are.same(18, vim.fn.line('.'))
+    vim.cmd([[norm {]])
+    assert.are.same(16, vim.fn.line('.'))
+    vim.cmd([[norm {]])
+    assert.are.same(11, vim.fn.line('.'))
+    vim.cmd([[norm {]])
+    assert.are.same(9, vim.fn.line('.'))
+    vim.cmd([[norm {]])
+    assert.are.same(5, vim.fn.line('.'))
+    vim.cmd([[norm {]])
+    assert.are.same(3, vim.fn.line('.'))
+    vim.cmd([[norm {]])
+    assert.are.same(3, vim.fn.line('.'))
+  end)
+
   it('should jump to next heading on same level (org_backward_heading_same_level)', function()
     helpers.create_file({
       '#TITLE: Test',