fix(note): Fix capturing note

Closes #846

Author: kristijanhusak

lua/orgmode/capture/init.lua

@@ -489,6 +489,19 @@ function Capture:build_note_capture(title)
         end)
         :attach(kill_map.default_map, kill_map.user_map, kill_map.opts)
     end,
+    on_close = function(capture_window)
+      local is_modified = vim.bo.modified
+
+      if is_modified then
+        local choice = vim.fn.confirm('Do you want to capture this note?', '&Yes\n&No')
+        vim.cmd([[redraw!]])
+        if choice ~= 1 then
+          return utils.echo_info('Canceled.')
+        end
+      end
+
+      capture_window:finish()
+    end,
   })
 end
 

lua/orgmode/capture/window.lua

@@ -6,7 +6,7 @@ local Promise = require('orgmode.utils.promise')
 ---@field template OrgCaptureTemplate
 ---@field on_open? fun(self: OrgCaptureWindow)
 ---@field on_finish? fun(lines: string[]): string[] | nil
----@field on_close? fun()
+---@field on_close? fun(self: OrgCaptureWindow)
 
 ---@class OrgCaptureWindow :OrgCaptureWindowOpts
 ---@field private _window fun() | nil
@@ -34,7 +34,7 @@ function CaptureWindow:open()
     if not content then
       return utils.echo_info('Canceled.')
     end
-    self._window = utils.open_tmp_org_window(16, config.win_split_mode, config.win_border, self.on_close)
+    self._window = utils.open_tmp_org_window(16, config.win_split_mode, config.win_border, self:_on_close())
     vim.api.nvim_buf_set_lines(0, 0, -1, true, content)
     self.template:setup()
     vim.b.org_capture = true
@@ -50,6 +50,15 @@ function CaptureWindow:open()
   end)
 end
 
+function CaptureWindow:_on_close()
+  if not self.on_close then
+    return nil
+  end
+  return function()
+    self.on_close(self)
+  end
+end
+
 function CaptureWindow:finish()
   local result = vim.api.nvim_buf_get_lines(0, 0, -1, false)
   if self.on_finish then