feat(cookie): implement virtual cookies

Closes #745

Author: PriceHiller

docs/configuration.org

@@ -3012,3 +3012,30 @@ require('orgmode').setup({
 #+end_src
 
 📝 NOTE: If you are using a plugin for =vim.ui.input=, make sure it supports autocompletion for better experience. [[https://github.com/folke/snacks.nvim][snacks.nvim]] input module supports autocompletion.
+
+*** cookies_use_extmarks
+:PROPERTIES:
+:CUSTOM_ID: cookies_use_extmarks
+:END:
+- Type: =boolean=
+- Default: =false=
+Possible values:
+- =true= - Uses /Virtual/ cookies to show live progress for cookies overlayed on top of the actual cookie text.
+- =false= - Do not add any /Virtual/ cookies.
+
+You can toggle Virtual cookies on the fly by executing command =:Org cookie_mode= when in a org buffer.
+This additionally sets the buffer variable =vim.b.org_cookie_mode= to =true= or =false=, depending on the current state.
+
+Currently this only applies Virtual cookies to headlines.
+
+Uses the following highlights:
+- ~@org.cookie.delimiter~: The highlight to use for the delimiters (brackets: ~[~ & ~]~)
+  - ~@org.cookie.delimiter.left~: A more granular highlight for just the left ~[~ bracket
+  - ~@org.cookie.delimiter.right~: A more granular highlight for just the right ~]~ bracket
+- ~@org.cookie.number~: The numbers used in the cookie (~50~)
+  - ~org.cookie.number.complete~: A more granular highlight for the left side of ~[1/5]~ (in this case the ~1~)
+  - ~org.cookie.number.total~: A more granular highlight for the right side of ~[1/5]~ (in this case the ~5~)
+- ~@org.cookie.sign~: The sign used in the cookie (e.g. ~/~, ~%~, or ~???~)
+  - ~@org.cookie.sign.unknown~: More granular if there wasn't items to calculate on (~???~) and you want a different highlight for the unknown sign
+  - ~@org.cookie.sign.div~: More granular for the div sign, the ~/~ in ~[1/5]~
+  - ~@org.cookie.sign.percent~: More granular for the percent sign, the ~%~ in ~[100%]~

ftplugin/org.lua

@@ -18,6 +18,10 @@ if config.org_startup_indented then
   require('orgmode.ui.virtual_indent'):new(bufnr):attach()
 end
 
+if config.ui.cookies_use_extmarks then
+  require('orgmode.ui.virtcookie').new(bufnr):attach()
+end
+
 vim.bo.modeline = false
 vim.opt_local.fillchars:append('fold: ')
 vim.opt_local.foldmethod = 'expr'

lua/orgmode/colors/highlights.lua

@@ -71,6 +71,11 @@ function M.link_highlights()
     ['@org.table.delimiter'] = '@punctuation.special',
     ['@org.table.heading'] = '@markup.heading',
     ['@org.edit_src'] = 'Visual',
+
+    -- For cookie extmarks (applicable if enabled)
+    ['@org.cookie.delimiter'] = 'Delimiter',
+    ['@org.cookie.sign'] = 'Special',
+    ['@org.cookie.number'] = 'Number',
   }
 
   for src, def in pairs(links) do

lua/orgmode/config/_meta.lua

@@ -174,6 +174,7 @@
 ---@field folds? { colored: boolean } Should folds be colored or use the default folding highlight. Default: { colored: true }
 ---@field menu? { handler: fun() | nil } Menu configuration
 ---@field input? { use_vim_ui: boolean } Input configuration
+---@field cookies_use_extmarks? boolean Should virtual text/extmarks be used for headline cookie progress
 
 ---@class OrgMappingsConfig
 ---@field disable_all? boolean Disable all mappings. Default: false

lua/orgmode/config/defaults.lua

@@ -221,6 +221,7 @@ local DefaultConfig = {
     input = {
       use_vim_ui = false,
     },
+    cookies_use_extmarks = false,
   },
 }
 

lua/orgmode/files/headline.lua

@@ -903,22 +903,16 @@ function Headline:_set_cookie(cookie, num, denum)
   return self:_set_node_text(cookie, new_cookie_val)
 end
 
-function Headline:update_cookie()
-  -- Update cookie state from a check box state change
-
-  -- Return early if the headline doesn't have a cookie
-  local cookie = self:get_cookie()
-  if not cookie then
-    return self
-  end
-
+---@return { [1]: integer, [2]: integer }? checked_unchecked The first integer is the number of checked boxes, the second integer is the total boxes
+function Headline:_get_checkbox_progress()
+  -- Count done checkboxes  and total checkboxes for the headline
   local section = self:node():parent()
+  local num_checked_boxes, num_boxes = 0, 0
   if not section then
-    return self
+    return
   end
 
   -- Count checked boxes from all lists
-  local num_checked_boxes, num_boxes = 0, 0
   local body = section:field('body')[1]
   if body then
     for node in body:iter_children() do
@@ -933,6 +927,53 @@ function Headline:update_cookie()
     end
   end
 
+  if num_boxes == 0 then
+    return
+  end
+
+  return { num_checked_boxes, num_boxes }
+end
+
+---@return { [1]: integer, [2]: integer }? done_total The first integer is the number of "DONE" todos, the second integer is the total todos
+function Headline:_get_todo_progress()
+  -- Count done children headlines and total children with TODO keywords
+  local children = self:get_child_headlines()
+  local headlines_with_todo = vim.tbl_filter(function(h)
+    local todo, _, _ = h:get_todo()
+    return todo ~= nil
+  end, children)
+
+  local dones = vim.tbl_filter(function(h)
+    return h:is_done()
+  end, headlines_with_todo)
+
+  if #headlines_with_todo == 0 then
+    return
+  end
+
+  return { #dones, #headlines_with_todo }
+end
+
+function Headline:update_cookie()
+  -- Update cookie state from a check box state change
+
+  -- Return early if the headline doesn't have a cookie
+  local cookie = self:get_cookie()
+  if not cookie then
+    return self
+  end
+
+  local section = self:node():parent()
+  if not section then
+    return self
+  end
+
+  local progress = self:_get_checkbox_progress()
+  if not progress then
+    return self
+  end
+  local num_checked_boxes, num_boxes = unpack(progress)
+
   -- Set the cookie
   return self:_set_cookie(cookie, num_checked_boxes, num_boxes)
 end
@@ -946,19 +987,15 @@ function Headline:update_todo_cookie()
     return self
   end
 
-  -- Count done children headlines and total children with TODO keywords
-  local children = self:get_child_headlines()
-  local headlines_with_todo = vim.tbl_filter(function(h)
-    local todo, _, _ = h:get_todo()
-    return todo ~= nil
-  end, children)
+  local progress = self:_get_todo_progress()
+  if not progress then
+    return self
+  end
 
-  local dones = vim.tbl_filter(function(h)
-    return h:is_done()
-  end, headlines_with_todo)
+  local dones, headlines_with_todo = unpack(progress)
 
   -- Set the cookie
-  return self:_set_cookie(cookie, #dones, #headlines_with_todo)
+  return self:_set_cookie(cookie, dones, headlines_with_todo)
 end
 
 function Headline:update_parent_cookie()

lua/orgmode/org/global.lua

@@ -98,6 +98,12 @@ local build = function(orgmode)
     indent_mode = function()
       require('orgmode.ui.virtual_indent').toggle_buffer_indent_mode()
     end,
+    cookie_mode = function()
+      local virtcookie = require('orgmode.ui.virtcookie').get()
+      if virtcookie then
+        virtcookie:toggle()
+      end
+    end,
   }
 
   _G.Org = OrgGlobal