easytables.nvim/lua/easytables/tablebuilder.lua

local M = {};

DEFAULT_DRAW_REPRESENTATION_OPTIONS = {
    min_width = 3,
    filler = " ",
    top_left = "┌",
    top_right = "┐",
    bottom_left = "└",
    bottom_right = "┘",
    horizontal = "─",
    vertical = "│",
    left_t = "├",
    right_t = "┤",
    top_t = "┬",
    bottom_t = "┴",
    cross = "┼"
}

function create_horizontal_line(cell_widths, left, middle, right, middle_t)
    local string = ""

    string = string .. left

    for i, width in ipairs(cell_widths) do
        string = string .. string.rep(middle, width)

        if i ~= #cell_widths then
            string = string .. middle_t
        end
    end

    string = string .. right

    return string
end

-- Creates a horizontal divider like this:
-- `create_horizontal_divider(5, 5, {variant = "top"})`:
-- ┌─────┬─────┬─────┬─────┬─────┐
-- `create_horizontal_divider(5, 5, {variant = "between"})`:
-- ├─────┼─────┼─────┼─────┼─────┤
-- `create_horizontal_divider(5, 5, {variant = "bottom"})`:
-- └─────┴─────┴─────┴─────┴─────┘
function create_horizontal_divider(
    table,
    options -- [[ table ]] -- optional
)
    local options = options or {}
    local top_left = options.top_left or DEFAULT_DRAW_REPRESENTATION_OPTIONS.top_left
    local top_right = options.top_right or DEFAULT_DRAW_REPRESENTATION_OPTIONS.top_right
    local bottom_left = options.bottom_left or DEFAULT_DRAW_REPRESENTATION_OPTIONS.bottom_left
    local bottom_right = options.bottom_right or DEFAULT_DRAW_REPRESENTATION_OPTIONS.bottom_right
    local horizontal = options.horizontal or DEFAULT_DRAW_REPRESENTATION_OPTIONS.horizontal
    local left_t = options.left_t or DEFAULT_DRAW_REPRESENTATION_OPTIONS.left_t
    local right_t = options.right_t or DEFAULT_DRAW_REPRESENTATION_OPTIONS.right_t
    local top_t = options.top_t or DEFAULT_DRAW_REPRESENTATION_OPTIONS.top_t
    local bottom_t = options.bottom_t or DEFAULT_DRAW_REPRESENTATION_OPTIONS.bottom_t
    local cross = options.cross or DEFAULT_DRAW_REPRESENTATION_OPTIONS.cross
    local min_width = options.min_width or DEFAULT_DRAW_REPRESENTATION_OPTIONS.min_width
    local variant = options.variant or "between"

    local widths = table:get_widths_for_columns(min_width)

    if variant == "top" then
        return create_horizontal_line(widths, top_left, horizontal, top_right, top_t)
    elseif variant == "between" then
        return create_horizontal_line(widths, left_t, horizontal, right_t, cross)
    elseif variant == "bottom" then
        return create_horizontal_line(widths, bottom_left, horizontal, bottom_right, bottom_t)
    end
end

function table.draw_representation(table, options)
    local options = options or {}
    local min_width = options.min_width or DEFAULT_DRAW_REPRESENTATION_OPTIONS.min_width
    local filler = options.filler or DEFAULT_DRAW_REPRESENTATION_OPTIONS.filler
    local vertical = options.vertical or DEFAULT_DRAW_REPRESENTATION_OPTIONS.vertical

    local representation = {}

    local horizontal_divider = create_horizontal_divider(table, options)

    representation[#representation + 1] = create_horizontal_divider(table, { variant = "top" })

    local column_widths = table:get_widths_for_columns(min_width)

    for i = 1, table:rows_amount() do
        local line = ""
        for j = 1, table:cols_amount() do
            local length = column_widths[j]
            local cell = table:value_at(i, j)
            local cell_width = vim.api.nvim_strwidth(cell)

            if cell_width < length then
                cell = cell .. string.rep(filler, length - cell_width)
            end

            cell = vertical .. cell

            -- Add most right vertical divider
            if j == table:cols_amount() then
                cell = cell .. vertical
            end

            line = line .. cell
        end

        representation[#representation + 1] = line

        if i ~= table:rows_amount() then
            representation[#representation + 1] = horizontal_divider
        end
    end

    representation[#representation + 1] = create_horizontal_divider(table, { variant = "bottom" })

    return representation
end

function table.from_representation(representation, options)
    local opts = options or {}

    local table = {}

    for i = 1, #representation do
        local character = representation[i]
    end
end

return table