feat: Connect to synchronized kernel

src/jupynium/events_control.py

@@ -522,6 +522,13 @@ def process_request_event(nvim_info: NvimInfo, driver, event):
  event[3].send(ret_obj)
  return True, None
 
+ elif event[1] == "kernel_connect_info":
+ driver.switch_to.window(nvim_info.window_handles[bufnr])
+ code = "return Jupyter.notebook.kernel.id"
+ ret_obj = driver.execute_script(code)
+ event[3].send(ret_obj)
+ return True, None
+
  if event[3] is not None:
  event[3].send("OK")
 

src/jupynium/lua/commands.lua

@@ -29,3 +29,4 @@ vim.api.nvim_create_user_command("JupyniumKernelRestart", "lua Jupynium_kernel_r
 vim.api.nvim_create_user_command("JupyniumKernelInterrupt", "lua Jupynium_kernel_interrupt()", {})
 vim.api.nvim_create_user_command("JupyniumKernelSelect", "lua Jupynium_kernel_select()", {})
 vim.api.nvim_create_user_command("JupyniumKernelHover", "lua Jupynium_kernel_hover()", {})
+vim.api.nvim_create_user_command("JupyniumConnectKernel", Jupynium_connect_kernel_cmd, { nargs = "?" })

src/jupynium/lua/helpers.lua

@@ -688,3 +688,36 @@ function Jupynium_kernel_complete_async(bufnr, code_line, col, callback)
 
  Jupynium_rpcnotify("kernel_complete_async", bufnr, true, code_line, col, callback_id)
 end
+
+function Jupynium_connect_kernel(bufnr, hostname)
+ if bufnr == nil or bufnr == 0 then
+ bufnr = vim.api.nvim_get_current_buf()
+ end
+ if Jupynium_syncing_bufs[bufnr] == nil then
+ Jupynium_notify.error { [[Cannot get kernel list without synchronising.]], [[Run `:JupyniumStartSync`]] }
+ return
+ end
+ local kernel_id = Jupynium_rpcrequest("kernel_connect_info", bufnr, true)
+ local cmd = "jupyter console --existing " .. kernel_id
+ if hostname ~= "" then
+ cmd = "ssh " .. hostname .. " -t " .. cmd
+ end
+ local _ipython = nil
+ if not _ipython then
+ _ipython = require("toggleterm.terminal").Terminal:new {
+ cmd = cmd,
+ direction = "horizontal",
+ close_on_exit = true,
+ hidden = true,
+ }
+ end
+ _ipython:toggle()
+ Jupynium_notify.info { "Connect to kernel " .. kernel_id }
+ print(kernel_id)
+end
+
+function Jupynium_connect_kernel_cmd(args)
+ local hostname = args.args
+ local buf = vim.api.nvim_get_current_buf()
+ Jupynium_connect_kernel(buf, hostname)
+end