fix(archive): Set correct archive category on archived headline

Fixes #882

Author: kristijanhusak

lua/orgmode/capture/init.lua

@@ -297,6 +297,8 @@ function Capture:refile_file_headline_to_archive(headline)
 
   local destination_file = self.files:get(archive_location)
   local todo_state = headline:get_todo()
+  local headline_category = headline:get_category()
+  local outline_path = headline:get_outline_path()
 
   return self
     :_refile_from_org_file({
@@ -310,11 +312,10 @@ function Capture:refile_file_headline_to_archive(headline)
         local archived_headline = archive_file:get_closest_headline({ target_line, 0 })
         archived_headline:set_property('ARCHIVE_TIME', Date.now():to_string())
         archived_headline:set_property('ARCHIVE_FILE', file.filename)
-        local outline_path = headline:get_outline_path()
         if outline_path ~= '' then
           archived_headline:set_property('ARCHIVE_OLPATH', outline_path)
         end
-        archived_headline:set_property('ARCHIVE_CATEGORY', headline:get_category())
+        archived_headline:set_property('ARCHIVE_CATEGORY', headline_category)
         archived_headline:set_property('ARCHIVE_TODO', todo_state or '')
       end)
     end)

tests/plenary/ui/mappings/archive_spec.lua

@@ -93,4 +93,34 @@ describe('Archive', function()
       '  Body text baz',
     }, vim.api.nvim_buf_get_lines(0, 0, -1, false))
   end)
+
+  it('sets the correct archive category on the archived headline', function()
+    local file = helpers.create_agenda_file({
+      '* foobar',
+      '  :PROPERTIES:',
+      '  :CATEGORY: custom_category',
+      '  :END:',
+      '* foo',
+    })
+
+    local now = Date.now()
+    vim.cmd([[exe "norm ,o$"]])
+    -- Pause to finish the archiving
+    vim.wait(50)
+    assert.are.same({
+      '* foo',
+    }, vim.api.nvim_buf_get_lines(0, 0, -1, false))
+
+    vim.cmd(('edit %s'):format(file.filename .. '_archive'))
+    assert.are.same({
+      '* foobar',
+      '  :PROPERTIES:',
+      '  :CATEGORY: custom_category',
+      '  :ARCHIVE_TIME: ' .. now:to_string(),
+      '  :ARCHIVE_FILE: ' .. file.filename,
+      '  :ARCHIVE_CATEGORY: custom_category',
+      '  :ARCHIVE_TODO: ',
+      '  :END:',
+    }, vim.api.nvim_buf_get_lines(0, 0, -1, false))
+  end)
 end)