From 0207e4025e5558fdd0b3951f250689eede5c75b2 Mon Sep 17 00:00:00 2001 From: Munif Tanjim Date: Mon, 5 Jun 2023 16:40:18 +0600 Subject: [PATCH] feat: 'Neotest' command (#254) --- plugin/neotest.lua | 270 +++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 270 insertions(+) create mode 100644 plugin/neotest.lua diff --git a/plugin/neotest.lua b/plugin/neotest.lua new file mode 100644 index 00000000..2463f001 --- /dev/null +++ b/plugin/neotest.lua @@ -0,0 +1,270 @@ +local cmd_completion_store = { + [""] = { + "attach", + "jump", + "output", + "output-panel", + "run", + "stop", + "summary", + }, + jump = { + "next", + "prev", + }, + ["output-panel"] = { + "open", + "close", + "toggle", + }, + run = { + "file", + "last", + }, + summary = { + "close", + "mark", + "open", + "toggle", + }, + ["summary:mark"] = { + "clear", + "run", + "toggle", + }, +} + +local opt_completion_store = { + adapter = function(arg) + local part = arg:gsub("^adapter=", "") + return vim.tbl_filter(function(cmd) + return not not string.find(cmd, "^" .. part) + end, require("neotest").state.adapter_ids()) + end, + strategy = function(arg) + local part = arg:gsub("^strategy=", "") + return vim.tbl_filter(function(cmd) + return not not string.find(cmd, "^" .. part) + end, { "integrated", "dap" }) + end, +} + +local commands +commands = { + attach = function(params) + require("neotest").run.attach(params.opts) + end, + jump = { + next = function(params) + require("neotest").jump.next(params.opts) + end, + prev = function(params) + require("neotest").jump.prev(params.opts) + end, + }, + output = function(params) + require("neotest").output.open(params.opts) + end, + ["output-panel"] = { + function() + commands["output-panel"].toggle() + end, + close = function() + require("neotest").output_panel.close() + end, + open = function() + require("neotest").output_panel.open() + end, + toggle = function() + require("neotest").output_panel.toggle() + end, + }, + run = { + function(params) + require("neotest").run.run(params.opts) + end, + file = function(params) + params.opts[1] = vim.fn.expand("%") + commands.run[1](params.opts) + end, + last = function(params) + require("neotest").run.run_last(params.opts) + end, + }, + stop = function(params) + require("neotest").run.stop(params.opts) + end, + summary = { + function() + commands.summary.toggle() + end, + close = function() + require("neotest").summary.close() + end, + mark = { + function() + commands.summary.mark.toggle() + end, + clear = function(params) + require("neotest").summary.clear_marked(params.opts) + end, + run = function(params) + require("neotest").summary.run_marked(params.opts) + end, + toggle = function() + local key = require("neotest.config").summary.mappings.mark + if type(key) == "table" then + key = key[1] + end + vim.api.nvim_feedkeys(vim.api.nvim_replace_termcodes(key, true, false, true), "m", false) + end, + }, + open = function() + require("neotest").summary.open() + end, + toggle = function() + require("neotest").summary.toggle() + end, + }, +} + +local function eval_luastring(value) + return loadstring("return " .. value, value)() +end + +local function make_params(info, args) + local params = { + bang = info.bang, + opts = {}, + } + + for _, arg in ipairs(args) do + if arg:find("=", 1) then + local parts = vim.split(arg, "=") + local key = table.remove(parts, 1) + local value = table.concat(parts, "=") + params.opts[key] = eval_luastring(value) + else + table.insert(params.opts, eval_luastring(arg)) + end + end + + return params +end + +vim.api.nvim_create_user_command("Neotest", function(info) + local args = info.fargs + + ---@type string|nil + local cmd_name = table.remove(args, 1) + if not cmd_name then + return vim.api.nvim_err_writeln("[Neotest] missing command") + end + + local cmd = commands[cmd_name] + if type(cmd) == "function" then + return cmd(make_params(info, args)) + end + + if type(cmd) ~= "table" then + return vim.api.nvim_err_writeln(string.format("[Neotest] unknown command: %s", cmd_name)) + end + + if cmd[args[1]] then + local subcmd_name = table.remove(args, 1) + local subcmd = cmd[subcmd_name] + + if type(subcmd) == "function" then + return subcmd(make_params(info, args)) + end + + if type(subcmd) ~= "table" then + return vim.api.nvim_err_writeln( + string.format("[Neotest] unknown subcommand: %s", subcmd_name) + ) + end + + if subcmd[args[1]] then + local subsubcmd_name = table.remove(args, 1) + local subsubcmd = subcmd[subsubcmd_name] + + if type(subsubcmd) ~= "function" then + return vim.api.nvim_err_writeln( + string.format("[Neotest] unknown subcommand: %s %s", subcmd_name, subsubcmd_name) + ) + end + + return subsubcmd(make_params(info, args)) + end + + if not subcmd[1] then + return vim.api.nvim_err_writeln("[Neotest] missing subcommand") + end + + return subcmd[1](make_params(info, args)) + end + + if not cmd[1] then + return vim.api.nvim_err_writeln("[Neotest] missing subcommand") + end + + return cmd[1](make_params(info, args)) +end, { + bang = true, + nargs = "*", + range = true, + complete = function(_, cmd_line) + local args = vim.split(cmd_line, "%s+", { trimempty = true }) + local last_idx = #args + local last_arg = args[last_idx] + + local is_partial = not string.match(cmd_line, "%s$") + + local cmd_scope = "" + + -- command + if last_idx == 1 then + return cmd_completion_store[cmd_scope] + elseif last_idx == 2 and is_partial then + return vim.tbl_filter(function(cmd) + return not not string.find(cmd, "^" .. last_arg) + end, cmd_completion_store[cmd_scope]) + end + + -- sub-command + cmd_scope = args[2] + if last_idx == 2 and cmd_completion_store[cmd_scope] then + return cmd_completion_store[cmd_scope] + elseif #args == 3 and is_partial and cmd_completion_store[cmd_scope] then + local items = vim.tbl_filter(function(cmd) + return not not string.find(cmd, "^" .. last_arg) + end, cmd_completion_store[cmd_scope]) + if #items > 0 then + return + end + end + + -- sub-sub-command + cmd_scope = string.format("%s:%s", args[2], args[3]) + if last_idx == 3 and cmd_completion_store[cmd_scope] then + return cmd_completion_store[cmd_scope] + elseif #args == 4 and is_partial and cmd_completion_store[cmd_scope] then + local items = vim.tbl_filter(function(cmd) + return not not string.find(cmd, "^" .. last_arg) + end, cmd_completion_store[cmd_scope]) + if #items > 0 then + return + end + end + + -- adpaters + if string.match(last_arg, "^adapter=") and is_partial then + return opt_completion_store.adapter(last_arg) + end + + -- strategies + if string.match(last_arg, "^strategy=") and is_partial then + return opt_completion_store.strategy(last_arg) + end + end, +})