fix(table): align cells in wrapped tables

Both the wrapped and regular cells.

Author: okuuva

lua/render-markdown/render/markdown/table.lua

@@ -712,9 +712,22 @@ function Render:row_wrapped_lines(row, row_index)
             local col_width = self.wrap_layout.col_widths[i]
             line:text(border_icon, highlight)
             line:pad(padding, filler)
-            local chunk = cell_lines[i][visual_line + 1] or Line.new(filler)
-            line:extend(chunk)
-            line:pad(col_width - chunk:width(), filler)
+            local chunks = cell_lines[i]
+            local chunk = chunks[visual_line + 1] or Line.new(filler)
+            local fill = col_width - chunk:width()
+            local alignment = self.data.cols[i].alignment
+            if alignment == Alignment.center then
+                local left = math.floor(fill / 2)
+                line:pad(left, filler)
+                line:extend(chunk)
+                line:pad(fill - left, filler)
+            elseif alignment == Alignment.right then
+                line:pad(fill, filler)
+                line:extend(chunk)
+            else
+                line:extend(chunk)
+                line:pad(fill, filler)
+            end
             line:pad(padding, filler)
         end
         line:text(border_icon, highlight)

tests/table_spec.lua

@@ -377,6 +377,51 @@ describe('table', function()
         end)
     end)
 
+    it('wrapped aligns unwrapped cells', function()
+        util.setup.text({
+            '',
+            '|Left|Right|Center|Long|',
+            '|:---|----:|:----:|----|',
+            '|a|2|mid|one two three four five six seven|',
+        }, {
+            pipe_table = { max_table_width = 40 },
+            win_options = { wrap = { default = false, rendered = true } },
+        })
+
+        util.assert_screen({
+            '┌──────┬───────┬────────┬──────────────┐',
+            '│ Left │ Right │ Center │ Long         │',
+            '├━─────┼──────━┼━──────━┼──────────────┤',
+            '│ a    │     2 │  mid   │ one two      │',
+            '│      │       │        │ three four   │',
+            '│      │       │        │ five six     │',
+            '│      │       │        │ seven        │',
+            '└──────┴───────┴────────┴──────────────┘',
+        })
+    end)
+
+    it('wrapped aligns wrapped cells', function()
+        util.setup.text({
+            '',
+            '| L | R | C |',
+            '|:--|--:|:-:|',
+            '| a | one two three four five six | one two three four five six |',
+        }, {
+            pipe_table = { max_table_width = 40 },
+            win_options = { wrap = { default = false, rendered = true } },
+        })
+
+        util.assert_screen({
+            '┌─────┬───────────────┬───────────────┐',
+            '│ L   │             R │       C       │',
+            '├━────┼──────────────━┼━─────────────━┤',
+            '│ a   │       one two │    one two    │',
+            '│     │    three four │  three four   │',
+            '│     │      five six │   five six    │',
+            '└─────┴───────────────┴───────────────┘',
+        })
+    end)
+
     it('wrapped long delimiter', function()
         util.setup.text({
             '',