mirror of
https://github.com/Myzel394/easytables.nvim.git
synced 2025-06-18 23:05:27 +02:00
113 lines
3.5 KiB
Lua
113 lines
3.5 KiB
Lua
local table = require("easytables.table")
|
|
local window = require("easytables.window")
|
|
local inputHelper = require("easytables.input")
|
|
local o = require("easytables.options")
|
|
local export = require("easytables.export")
|
|
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
|
|
|
|
-- tuples 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,
|
|
{
|
|
on_export = function()
|
|
local new_table = export:export_table(markdown_table)
|
|
|
|
vim.schedule(function()
|
|
local cursor = vim.api.nvim_win_get_cursor(0)
|
|
|
|
vim.api.nvim_buf_set_text(
|
|
0,
|
|
cursor[1] - 1,
|
|
cursor[2],
|
|
cursor[1] - 1,
|
|
cursor[2],
|
|
new_table
|
|
)
|
|
end)
|
|
end
|
|
}
|
|
)
|
|
|
|
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
|
|
|
|
local raw_table = import.extract_table(buffer, start_row, end_row)
|
|
|
|
local markdown_table = table:import(raw_table)
|
|
|
|
local win = window:create(
|
|
markdown_table,
|
|
{
|
|
on_export = function()
|
|
local new_table = export:export_table(markdown_table)
|
|
|
|
vim.schedule(function()
|
|
-- Remove old table
|
|
vim.api.nvim_buf_set_lines(buffer, start_row, end_row, false, new_table)
|
|
end)
|
|
end
|
|
}
|
|
)
|
|
|
|
win:show()
|
|
win:register_listeners()
|
|
win:draw_table()
|
|
end,
|
|
{
|
|
desc = "Import the current markdown table at the cursor's position into EasyTables"
|
|
}
|
|
)
|
|
end
|
|
|
|
return {
|
|
setup = setup,
|
|
}
|