easytables.nvim/lua/easytables/init.lua

local table = require("easytables.table")
local window = require("easytables.window")
local inputHelper = require("easytables.input")
local o = require("easytables.options")
local import = require("easytables.import")

---Initialize `easytables` with the given options. This function **must** be called.
---@param options table See options.lua for available options
local function setup(options)
    options = options or {}
    o.merge_options(options)

    vim.api.nvim_create_user_command(
        "EasyTablesCreateNew",
        function(opt)
            local input = opt.args

            local success, result = pcall(function() return inputHelper.extract_column_info(input) end)

            if not success then
                error("Don't know how to interpret this message. Please use a format like 3x4 or 3x or 4 or x5")
                return
            end

            -- tuple do not seem to be working with pcall
            local cols = result[1]
            local rows = result[2]

            local markdown_table = table:create(cols, rows)

            local win = window:create(markdown_table)

            win:show()
            win:register_listeners()
            win:draw_table()
        end,
        {
            nargs = 1,
            desc = "Create a new markdown table using EasyTables"
        }
    )

    vim.api.nvim_create_user_command(
        "EasyTablesImportThisTable",
        function()
            local buffer = vim.api.nvim_get_current_buf()
            local start_row = import.find_row_start(buffer)

            if not start_row then
                error("No table found (failed to find start row)")
                return
            end

            local end_row = import.find_row_end(buffer, start_row)

            if not end_row then
                error("No table found (failed to find end row)")
                return
            end

            print(vim.inspect(start_row), vim.inspect(end_row))

            local raw_table = import.extract_table(buffer, start_row, end_row)
            print(vim.inspect(raw_table))

            local markdown_table = table:import(raw_table)

            local win = window:create(markdown_table)

            win:show()
            win:register_listeners()
            win:draw_table()


            -- Remove old table
            vim.api.nvim_buf_set_lines(buffer, start_row - 1, end_row, false, {})
        end,
        {
            desc = "Import the current markdown table at the cursor's position into EasyTables"
        }
    )
end

return {
    setup = setup,
}