Add fat_tables feature, adds row above and below pipe tables to complete enclosed look

Author: MeanderingProgrammer

README.md

@@ -69,6 +69,7 @@ require('render-markdown').setup({
         (block_quote (block_quote_marker) @quote_marker)
         (block_quote (paragraph (inline (block_continuation) @quote_marker)))
 
+        (pipe_table) @table
         (pipe_table_header) @table_head
         (pipe_table_delimiter_row) @table_delim
         (pipe_table_row) @table_row
@@ -95,6 +96,8 @@ require('render-markdown').setup({
         -- conceallevel used for buffer when being rendered
         rendered = 3,
     },
+    -- Add a line above and below tables to complete look, ends up like a window
+    fat_tables = true,
     -- Define the highlight groups to use when rendering various components
     highlights = {
         heading = {

demo/demo.gif

@@ -1,4 +1,4 @@
-default_zoom := '10'
+default_zoom := '3'
 
 demo zoom=default_zoom:
     rm -f demo/demo.gif

justfile

@@ -32,6 +32,7 @@ local M = {}
 ---@field public bullet? string
 ---@field public quote? string
 ---@field public conceal? UserConceal
+---@field public fat_tables? boolean
 ---@field public highlights? UserHighlights
 
 ---@param opts UserConfig|nil
@@ -59,6 +60,7 @@ function M.setup(opts)
             (block_quote (block_quote_marker) @quote_marker)
             (block_quote (paragraph (inline (block_continuation) @quote_marker)))
 
+            (pipe_table) @table
             (pipe_table_header) @table_head
             (pipe_table_delimiter_row) @table_delim
             (pipe_table_row) @table_row
@@ -75,6 +77,7 @@ function M.setup(opts)
             default = vim.opt.conceallevel:get(),
             rendered = 3,
         },
+        fat_tables = true,
         highlights = {
             heading = {
                 backgrounds = { 'DiffAdd', 'DiffChange', 'DiffDelete' },

lua/render-markdown/init.lua

@@ -14,4 +14,16 @@ function M.clamp_last(values, index)
     return values[math.min(index, #values)]
 end
 
+---@param values string[]
+---@return string
+function M.first(values)
+    return values[1]
+end
+
+---@param values string[]
+---@return string
+function M.last(values)
+    return values[#values]
+end
+
 return M

lua/render-markdown/list.lua

@@ -27,6 +27,7 @@
 ---@field public bullet string
 ---@field public quote string
 ---@field public conceal Conceal
+---@field public fat_tables boolean
 ---@field public highlights Highlights
 
 ---@class State

lua/render-markdown/state.lua

@@ -88,6 +88,30 @@ M.markdown = function(root)
                 virt_text = { virt_text },
                 virt_text_pos = 'overlay',
             })
+        elseif capture == 'table' then
+            if state.config.fat_tables then
+                local lines = vim.api.nvim_buf_get_lines(0, start_row, end_row, false)
+                local table_head = list.first(lines)
+                local table_tail = list.last(lines)
+                if #table_head == #table_tail then
+                    local headings = vim.split(table_head, '|', { plain = true, trimempty = true })
+                    local sections = vim.tbl_map(function(part)
+                        return string.rep('─', #part)
+                    end, headings)
+
+                    local line_above = { { '┌' .. table.concat(sections, '┬') .. '┐', highlights.table.head } }
+                    vim.api.nvim_buf_set_extmark(0, M.namespace, start_row, start_col, {
+                        virt_lines_above = true,
+                        virt_lines = { line_above },
+                    })
+
+                    local line_below = { { '└' .. table.concat(sections, '┴') .. '┘', highlights.table.row } }
+                    vim.api.nvim_buf_set_extmark(0, M.namespace, end_row, start_col, {
+                        virt_lines_above = true,
+                        virt_lines = { line_below },
+                    })
+                end
+            end
         elseif vim.tbl_contains({ 'table_head', 'table_delim', 'table_row' }, capture) then
             local row = value:gsub('|', '│')
             if capture == 'table_delim' then