fix: terminate MTP process on shutdown (#59)

* fix: terminate MTP process on shutdown

* fix: terminate vstest process on shutdown

Author: Nsidorenco

lua/neotest-vstest/mtp/client.lua

@@ -98,6 +98,16 @@ end
 ---@return nio.control.Future<vim.lsp.Client> client_future, integer mtp_process_pid
 function M.create_client(dll_path, on_update, on_log, mtp_env)
   local server_future, mtp_process = start_server(dll_path, mtp_env)
+
+  nio.scheduler()
+  local cleanup_autocmd_id = vim.api.nvim_create_autocmd("QuitPre", {
+    group = vim.api.nvim_create_augroup("neotest_vstest_mtp_shutdown", { clear = false }),
+    desc = "Shutdown dotnet MTP client process on Neovim exit",
+    callback = function()
+      mtp_process:kill(vim.uv.constants.SIGKILL)
+    end,
+  })
+
   local client_future = nio.control.future()
 
   nio.run(function()
@@ -112,6 +122,7 @@ function M.create_client(dll_path, on_update, on_log, mtp_env)
         )
         server:shutdown()
         mtp_process:kill(vim.uv.constants.SIGKILL)
+        vim.api.nvim_del_autocmd(cleanup_autocmd_id)
       end,
       before_init = function(params)
         params.processId = vim.fn.getpid()

lua/neotest-vstest/vstest/cli_wrapper.lua

@@ -12,6 +12,10 @@ local function get_vstest_path()
       args = { "--info" },
     })
 
+    logger.debug(
+      "neotest-vstest: starting process to detect dotnet sdk path " .. tostring(process.pid)
+    )
+
     local default_sdk_path
     if vim.fn.has("win32") then
       default_sdk_path = "C:/Program Files/dotnet/sdk/"
@@ -104,14 +108,24 @@ function M.create_test_runner(project)
     logger.warn(obj.stderr)
   end)
 
+  nio.scheduler()
+  local cleanup_autocmd_id = vim.api.nvim_create_autocmd("QuitPre", {
+    group = vim.api.nvim_create_augroup("neotest_vstest_server_shutdown", { clear = false }),
+    desc = "Shutdown dotnet VSTest client process on Neovim exit",
+    callback = function()
+      process:kill(vim.uv.constants.SIGKILL)
+    end,
+  })
+
   logger.info(string.format("neotest-vstest: spawned vstest process with pid: %s", process.pid))
 
   return {
     execute = function(content)
       process:write(content .. "\n")
     end,
     stop = function()
-      process:kill(9)
+      process:kill(vim.uv.constants.SIGKILL)
+      vim.api.nvim_del_autocmd(cleanup_autocmd_id)
     end,
   }
 end