Fix table reformating when added to list item

Author: kristijanhusak

lua/orgmode/treesitter/table.lua

@@ -87,20 +87,10 @@ function TsTable:_get_content()
   return indented
 end
 
-local function get_table_from_node(node)
-  if not node then
-    return false
-  end
-  local table_node = utils.get_closest_parent_of_type(node, 'table', true)
-  if not table_node then
-    return false
-  end
-  return TsTable:new({ node = table_node })
-end
-
 function TsTable:add_row()
   local line = vim.fn.line('.')
-  vim.api.nvim_buf_set_lines(0, line, line, true, { '|' })
+  local indent = config:get_indent(vim.fn.indent(line))
+  vim.api.nvim_buf_set_lines(0, line, line, true, { ('%s|'):format(indent) })
   return TsTable.from_current_node():reformat()
 end
 

tests/plenary/ui/table_spec.lua

@@ -77,4 +77,22 @@ describe('Tables', function()
       '|      |       |',
     }, vim.api.nvim_buf_get_lines(0, 0, -1, false))
   end)
+
+  it('should add new row on enter in list item', function()
+    helpers.load_file_content({
+      '* TODO Test',
+      '  DEADLINE: <2021-07-21 Wed 22:02>',
+      '  - Some list item',
+      '    | test |',
+    })
+    vim.fn.cursor({ 4, 10 })
+    vim.cmd([[exe "norm a\<CR>"]])
+    assert.are.same({
+      '* TODO Test',
+      '  DEADLINE: <2021-07-21 Wed 22:02>',
+      '  - Some list item',
+      '    | test |',
+      '    |      |',
+    }, vim.api.nvim_buf_get_lines(0, 0, -1, false))
+  end)
 end)