Insert markdown table

[malko42]

If you ever wish to easily insert a markdown table feel free to use the following function:

insert_markdown_table = function()
	local get_size = function()
		local INPUT_CANCELLED = "~~~INPUT-CANCELLED~~~"
		local input = vim.fn.input { prompt = "Size of the table - columns by rows (e.g. 2x3)", cancelreturn = INPUT_CANCELLED }
		if input == INPUT_CANCELLED then
			-- early return if the user cancels the input
			return nil, nil
		end
		local string_column, string_row = input:match("([^x]+)x([^x]+)")
		return tonumber(string_column), tonumber(string_row)
	end
	local columns, rows = get_size();
	if columns == nil or rows == nil or columns < 1 or rows < 1 then
		vim.api.nvim_err_writeln("Invalid input. Please provide a valid size (e.g. 2x3)")
		return
	end

	-- Create a row
	-- @param row number
	-- @param header boolean
	local create_row = function(length, header)
		local row = "|"
		for _ = 1, length do
			row = row .. (header and " ---" or "   ") .. " |"
		end
		return row
	end

	local rows_table = {}

	-- First row with header row
	table.insert(rows_table, create_row(columns, false))
	table.insert(rows_table, create_row(columns, true))
	for _ = 2, rows do
		table.insert(rows_table, create_row(columns, false))
	end

	vim.api.nvim_put(rows_table, "l", true, false)
end

When called, it will prompt you for a size which should be in the form of nb_columnsxnb_rows meaning that if you wish to create a table of 4 columns by 2 rows you should input 4x2