up

lua/custom/plugins/term-helper.lua

@@ -0,0 +1,141 @@
+-- Terminal helper
+-- from: https://github.com/jmbuhr/nvim-config
+--
+
+---@module 'utils.term'
+local M = {
+  ---Map of buffers to a terminals
+  ---@type table<integer, Terminal>
+  buf_term_map = {},
+}
+
+---A terminal has a buffer number, a type and a channel id
+---@class Terminal
+---@field bufnr integer
+---@field type string?
+---@field channel_id integer
+
+---Get the channel id of a terminal buffer
+---@param bufnr integer
+---@return integer|nil
+local function get_channel_id(bufnr)
+  return vim.api.nvim_get_option_value('channel', { buf = bufnr })
+end
+
+---Get the type of the terminal buffer
+---like ipython, R, bash, etc.
+---based on the buffer name
+---@param buf integer
+---@return string|nil
+local function get_term_type(buf)
+  local name = vim.api.nvim_buf_get_name(buf)
+  local term_type = name:match 'term://.*/%d+:(.*)'
+  if not term_type then
+    return nil
+  end
+  -- drop flags like --no-confirm-exit
+  term_type = term_type:match '([^%s]+)'
+  -- drop the path to the executable
+  term_type = term_type:match '([^/]+)$'
+  return term_type
+end
+
+---Find nvim terminal buffers
+local function list_terminals()
+  local terms = {}
+  for _, buf in ipairs(vim.api.nvim_list_bufs()) do
+    if vim.api.nvim_buf_is_valid(buf) and vim.api.nvim_get_option_value('buftype', { buf = buf }) == 'terminal' then
+      local chan_id = get_channel_id(buf)
+      if not chan_id then
+        goto continue
+      end
+      ---@type Terminal
+      local term = {
+        bufnr = buf,
+        type = get_term_type(buf),
+        channel_id = chan_id,
+      }
+      table.insert(terms, term)
+    end
+    ::continue::
+  end
+  return terms
+end
+
+---For ipython sending
+local cpaste_start = '%cpaste -q\n'
+local cpaste_end = '--\n'
+local cpaste_pause = 10
+
+---Send lines to a terminal
+---@param lines string[]
+---@param term Terminal
+---@return nil
+local send_lines = function(lines, term)
+  local chan_id = term.channel_id
+  local text = table.concat(lines, '\n') .. '\n'
+  if term.type == 'ipython' then
+    vim.fn.chansend(chan_id, cpaste_start)
+    vim.uv.sleep(cpaste_pause)
+  end
+  vim.fn.chansend(chan_id, text)
+  if term.type == 'ipython' then
+    vim.fn.chansend(chan_id, cpaste_end)
+  end
+end
+
+local s = [[
+print("hello world!")
+def hello():
+  print("hello")
+hello()
+]]
+
+local ls = { s }
+
+---Connect current buffer to a terminal
+local connect = function()
+  local bufnr = vim.api.nvim_get_current_buf()
+  local terms = list_terminals()
+  vim.ui.select(terms, {
+    prompt = 'Select terminal',
+    format_item = function(item)
+      return item.type .. ' ' .. item.bufnr
+    end,
+  }, function(choice)
+    if choice == nil then
+      return
+    end
+    M.buf_term_map[bufnr] = choice
+  end)
+end
+
+M.send = function(lines)
+  local bufnr = vim.api.nvim_get_current_buf()
+  local term = M.buf_term_map[bufnr]
+  if not term then
+    connect()
+    return
+  end
+  send_lines(lines, term)
+  -- scroll to the bottom
+  local term_win = vim.fn.bufwinid(term.bufnr)
+  local n = vim.api.nvim_buf_line_count(term.bufnr)
+  vim.api.nvim_win_set_cursor(term_win, { n, 0 })
+end
+
+M.send_visual = function()
+  -- get the selected text
+  vim.cmd.normal { '"zy', bang = true }
+  local selection = vim.fn.getreg 'z'
+  M.send { selection }
+end
+
+vim.keymap.set('v', '<CR>', function()
+  M.send_visual()
+end, { noremap = true, silent = true })
+
+-- connect
+vim.keymap.set('n', '<leader>ct', function()
+  connect()
+end, { noremap = true, silent = true })