Fix adding closed date to a headline plan that is not of a specific type. Closes #309

Author: kristijanhusak

lua/orgmode/treesitter/headline.lua

@@ -185,7 +185,8 @@ function Headline:add_closed_date()
       string.format('%s%s', indent, closed_text),
     })
   end
-  local last_child = dates['DEADLINE'] or dates['SCHEDULED']
+  local keys = vim.tbl_keys(dates)
+  local last_child = dates['DEADLINE'] or dates['SCHEDULED'] or dates[keys[#keys]]
   local ptext = query.get_node_text(last_child, 0)
   local text = ptext .. ' ' .. closed_text
   tree_utils.set_node_text(last_child, text)

tests/plenary/ui/mappings/todo_spec.lua

@@ -114,4 +114,19 @@ describe('Todo mappings', function()
     vim.cmd('norm ciT')
     assert.are.same('* test TODO', vim.fn.getline(1))
   end)
+
+  it('Should properly add closed date when plan date is not of specific type', function()
+    helpers.load_file_content({
+      '#TITLE: Test',
+      '',
+      '* TODO Test orgmode',
+      '  <2021-07-21 Wed 22:02>',
+    })
+    vim.fn.cursor({ 3, 1 })
+    vim.cmd([[norm cit]])
+    assert.are.same({
+      '* DONE Test orgmode',
+      '  <2021-07-21 Wed 22:02> CLOSED: [' .. Date.now():to_string() .. ']',
+    }, vim.api.nvim_buf_get_lines(0, 2, 4, false))
+  end)
 end)