feat(add_note): include heading title on add_note editor (#1100)

* feat(add_note): include heading title when adding note

* test(add_note): includes heading title when adding note

---------

Co-authored-by: Kristijan Husak <husakkristijan@gmail.com>

Author: celsobenedetti

lua/orgmode/org/mappings.lua

@@ -862,12 +862,14 @@ function OrgMappings:add_note()
   local headline = self.files:get_closest_headline()
   local indent = headline:get_indent()
   local text = ('%s- Note taken on %s \\\\'):format(indent, Date.now():to_wrapped_string(false))
-  return self:_get_note(text, indent, 'Insert note for entry.'):next(function(note)
-    if not note then
-      return false
-    end
-    return headline:add_note(note)
-  end)
+  return self
+    :_get_note(text, indent, string.format('Insert note for %s.', headline:get_title() or 'entry'))
+    :next(function(note)
+      if not note then
+        return false
+      end
+      return headline:add_note(note)
+    end)
 end
 
 function OrgMappings:open_at_point()

tests/plenary/ui/mappings/add_note_spec.lua

@@ -0,0 +1,35 @@
+local helpers = require('tests.plenary.helpers')
+local orgmode = require('orgmode')
+local Promise = require('orgmode.utils.promise')
+
+describe('Add Note', function()
+  after_each(function()
+    vim.cmd([[silent! %bw!]])
+  end)
+
+  it('includes heading title in note prompt', function()
+    helpers.create_file({
+      '* TODO Example Title',
+      'Some content',
+    })
+
+    -- Place cursor on the headline
+    vim.fn.cursor(1, 1)
+
+    local captured_title = nil
+
+    helpers.with_var(orgmode.capture, 'build_note_capture', function(_, title)
+      captured_title = title
+      -- Return a stub with an open() that resolves immediately
+      return {
+        open = function()
+          return Promise.resolve({ 'Test note content' })
+        end,
+      }
+    end, function()
+      orgmode.action('org_mappings.add_note')
+    end)
+
+    assert.are.same('Insert note for Example Title.', captured_title)
+  end)
+end)