Add support for percentage cookies

Author: TravonteD

lua/orgmode/treesitter/headline.lua

@@ -121,7 +121,11 @@ function Headline:remove_closed_date()
 end
 
 function Headline:cookie()
-  return self:parse('%[%d?/%d?%]')
+  local cookie = self:parse('%[%d?/%d?%]')
+  if cookie then
+    return cookie
+  end
+  return self:parse('%[%d?%d?%d?%%%]')
 end
 
 -- @return tsnode, string

lua/orgmode/treesitter/list.lua

@@ -23,7 +23,9 @@ function List:parent_cookie()
     local content = top_item:field('contents')[1]
     -- The cookie should be the last thing on the line
     local cookie_node = content:named_child(content:named_child_count() - 1)
-    if query.get_node_text(cookie_node, 0):match('%[%d?/%d?%]') then
+
+    local text = query.get_node_text(cookie_node, 0)
+    if text:match('%[%d?/%d?%]') or text:match('%[%d?%d?%d?%%%]') then
       return cookie_node
     end
   end
@@ -42,7 +44,12 @@ function List:update_parent_cookie()
   local checked_boxes = vim.tbl_filter(function(box)
     return box:match('%[%w%]')
   end, checkboxes)
-  local new_status = ('[%d/%d]'):format(#checked_boxes, #checkboxes)
+  local new_status
+  if query.get_node_text(parent_cookie, 0):find('%%') then
+    new_status = ('[%d%%]'):format((#checked_boxes/#checkboxes) * 100)
+  else
+    new_status = ('[%d/%d]'):format(#checked_boxes, #checkboxes)
+  end
   tree_utils.set_node_text(parent_cookie, new_status)
 end
 

lua/orgmode/utils/treesitter.lua

@@ -1,5 +1,6 @@
 local ts_utils = require('nvim-treesitter.ts_utils')
 local config = require('orgmode.config')
+local query = vim.treesitter.query
 local M = {}
 
 function M.current_node()

tests/plenary/ui/mappings_spec.lua

@@ -1618,4 +1618,27 @@ describe('Mappings', function()
     vim.cmd([[exe "norm \<C-space>"]])
     assert.are.same('- Test orgmode [1/2]', vim.fn.getline(1))
   end)
+
+  it('should update the checklist cookies with a percentage within a headline', function()
+    helpers.load_file_content({
+      '* Test orgmode [%]',
+      '- [ ] checkbox item',
+      '- [ ] checkbox item',
+    })
+    vim.fn.cursor(2, 1)
+    vim.cmd([[exe "norm \<C-space>"]])
+    assert.are.same('* Test orgmode [50%]', vim.fn.getline(1))
+  end)
+
+  it('should update the checklist cookies with a percentage within a nested list', function()
+    helpers.load_file_content({
+      '- Test orgmode [%]',
+      '  - [ ] checkbox item',
+      '  - [ ] checkbox item',
+      '  - [ ] checkbox item',
+    })
+    vim.fn.cursor(2, 1)
+    vim.cmd([[exe "norm \<C-space>"]])
+    assert.are.same('- Test orgmode [33%]', vim.fn.getline(1))
+  end)
 end)