feat: generic test runner

Author: fcoury

lua/vial/init.lua

@@ -1,136 +1,157 @@
 local M = {}
-local last_test = nil
-local vial_path = "vial"
-local test_cmd = "cargo test -- %s --nocapture --color=always"
-local toggle_state = {
-	enabled = false,
-	extensions = {},
-}
+local vial_path = nil
+local file_types = nil
+local enabled = false
 
 function M.setup(opts)
 	opts = opts or {}
 	vial_path = opts.vial_path or vial_path
-	test_cmd = opts.command or test_cmd
-	toggle_state.extensions = opts.extensions or {}
+	file_types = opts.file_types
+	enabled = true
+
+	vim.cmd([[autocmd BufWritePost * lua require'vial'.on_buf_write_post()]])
+end
+
+-- Function to get the current context and run command
+local function run_command(command)
+	local handle = io.popen(string.format("%s send '%s' > /dev/null 2>&1", vial_path, command))
+
+	-- we don't handle responses yet
+	-- local result = handle:read("*a")
+	-- handle:close()
 end
 
-local function current_test_name()
-	-- Get the current buffer and cursor position
-	local bufnr = vim.api.nvim_get_current_buf()
-	---@diagnostic disable-next-line: deprecated
-	local row, _ = unpack(vim.api.nvim_win_get_cursor(0))
-
-	local function_declaration_line = 0
-	local function_name = ""
-
-	-- Check lines above the cursor for the #[test] attribute and function definition
-	for i = row, 1, -1 do
-		local line = vim.api.nvim_buf_get_lines(bufnr, i - 1, i, false)[1]
-
-		-- Check for function start and capture the function name
-		if line:match("^%s*fn%s") then
-			function_declaration_line = i
-			function_name = line:match("^%s*fn%s([%w_]+)%s*%(")
-			break
-		end
+local function current_settings()
+	if file_types == nil then
+		print("No file types defined")
+		return
 	end
 
-	for i = function_declaration_line, 1, -1 do
-		local line = vim.api.nvim_buf_get_lines(bufnr, i - 1, i, false)[1]
-		if line:match("^%s*#%[test%]") then
-			return function_name
-		end
+	local filetype = vim.bo.filetype
+	return file_types[filetype]
+end
+
+local function get(setting)
+	local settings = current_settings()
+	if settings == nil then
+		return nil
 	end
 
-	return nil
+	return settings[setting]
 end
 
--- Function to get the current context and run command
-local function run_command(command)
-	-- local file = vim.api.nvim_buf_get_name(0)
-	-- local line = vim.fn.line(".")
-	-- local context_command = string.format("%s -file=%s -line=%s", command, file, line)
+local function set(setting, value)
+	local settings = current_settings()
+	if settings == nil then
+		return
+	end
 
-	-- Execute the Rust binary and pass the command
-	local handle = io.popen(string.format("%s send '%s' > /dev/null 2>&1", vial_path, command))
-	-- local result = handle:read("*a")
-	-- handle:close()
+	settings[setting] = value
 end
 
 local function run_test(test_name)
-	local cmd = string.format(test_cmd, test_name)
+	local raw_cmd = get("command")
+	if raw_cmd == nil then
+		print("No command defined")
+		return
+	end
+
+	local cmd = string.format(raw_cmd, test_name)
 	run_command(cmd)
 end
 
 function M.run_test()
-	local test_name = current_test_name()
+	local test_name = get("extract")()
 
 	if test_name == nil then
 		print("No test found")
 		return
 	end
 
-	last_test = test_name
+	set("last_test", test_name)
 	run_test(test_name)
 end
 
 function M.run_last_test()
+	local last_test = get("last_test")
+
 	if last_test == nil then
-		print("No last test found")
 		return
 	end
 
 	run_test(last_test)
 end
 
-local function is_valid_extension(extension)
-	for _, ext in ipairs(toggle_state.extensions) do
-		if ext == extension then
-			return true
-		end
-	end
-	return false
-end
-
-local function get_extension(filename)
-	return filename:match("^.+(%..+)$")
-end
+local function is_inside_test()
+	local extract_fn = get("extract")
 
-local function on_buf_write_post()
-	if not toggle_state.enabled then
-		return
+	if extract_fn == nil then
+		return false
 	end
 
-	local filename = vim.api.nvim_buf_get_name(0)
-	local extension = get_extension(filename)
+	return get("extract")() ~= nil
+end
 
-	if is_valid_extension(extension) then
+local function run_last_or_current_test()
+	if is_inside_test() then
 		M.run_test()
+	else
+		M.run_last_test()
 	end
 end
 
-function M.toggle_automated_tests()
-	toggle_state.enabled = not toggle_state.enabled
-	if toggle_state.enabled then
+local function on_buf_write_post()
+	run_last_or_current_test()
+end
+
+function M.toggle_active()
+	if enabled then
 		vim.cmd([[autocmd BufWritePost * lua require'vial'.on_buf_write_post()]])
-		print("Automated tests enabled")
+		enabled = false
 	else
-		vim.cmd([[autocmd! BufWritePost * lua require'vial'.on_buf_write_post()]])
-		print("Automated tests disabled")
+		vim.cmd([[autocmd BufWritePost * lua require'vial'.on_buf_write_post()]])
+		enabled = true
 	end
 end
 
+function M.clear_last_test()
+	set("last_test", nil)
+end
+
+function M.show_last_test()
+	local last_test = get("last_test")
+
+	if last_test == nil then
+		print("No last test found")
+		return
+	end
+
+	print(last_test)
+end
+
 vim.api.nvim_set_keymap("n", "<leader>tt", "<cmd>lua require'vial'.run_test()<cr>", { noremap = true, silent = true })
 vim.api.nvim_set_keymap(
 	"n",
 	"<leader>tl",
 	"<cmd>lua require'vial'.run_last_test()<cr>",
 	{ noremap = true, silent = true }
 )
+vim.api.nvim_set_keymap(
+	"n",
+	"<leader>tc",
+	"<cmd>lua require'vial'.clear_last_test()<cr>",
+	{ noremap = true, silent = true }
+)
 vim.api.nvim_set_keymap(
 	"n",
 	"<leader>ta",
-	"<cmd>lua require'vial'.toggle_automated_tests()<cr>",
+	"<cmd>lua require'vial'.toggle_active()<cr>",
+	{ noremap = true, silent = true }
+)
+vim.api.nvim_set_keymap(
+	"n",
+	"<leader>ts",
+	"<cmd>lua require'vial'.show_last_test()<cr>",
 	{ noremap = true, silent = true }
 )