neovim: add ai module that I don't use

neovim/.config/nvim/lua/ai.lua

@@ -0,0 +1,153 @@
+--[
+-- i don't use this module but i didn't want to delete either. i'll just leave it here until the end of times.
+--]
+
+local curl = require("plenary.curl")
+local async = require("plenary.async")
+
+---@param messages {role: string; content: string}[]
+---@param model string
+---@param ollama_url string
+---@param callback fun(data:any)
+local send_prompt_request = function(messages, model, ollama_url, callback)
+  curl.request({
+    method = "POST",
+    url = ollama_url .. "/api/chat",
+    body = vim.json.encode({
+      model = model,
+      messages = messages,
+      stream = true,
+    }),
+    headers = {
+      content_type = "application/json",
+    },
+    stream = function(_, chunk)
+      if chunk == nil then
+        return
+      end
+      callback(chunk)
+    end,
+  })
+end
+
+local M = {
+  sessions = {},
+}
+
+---@param opts {
+--- ollama_url: string;
+--- model: string;
+---}
+function M.setup(opts)
+  M.ollama_url = opts.ollama_url
+  M.model = opts.model
+end
+
+---@param buf_name string
+---@return integer
+local function create_buffer(buf_name)
+  local buf = vim.api.nvim_create_buf(false, false)
+  vim.api.nvim_buf_set_name(buf, buf_name)
+  vim.bo[buf].buftype = "acwrite"
+  vim.bo[buf].bufhidden = "hide"
+  vim.bo[buf].swapfile = false
+  vim.bo[buf].modifiable = true
+
+  return buf
+end
+
+---@param buf integer
+---@return integer
+local function create_window(buf)
+  local term_width = tonumber(vim.api.nvim_get_option_value("columns", { scope = "global" }))
+  local term_height = tonumber(vim.api.nvim_get_option_value("lines", { scope = "global" }))
+
+  local row, col = unpack(vim.api.nvim_win_get_cursor(0))
+
+  local win_width = math.min(term_width - col - 5, 150)
+  local win_height = math.min(term_height - row - 5, 30)
+
+  local win = vim.api.nvim_open_win(
+    buf,
+    true,
+    { relative = "cursor", row = 1, col = 1, width = win_width, height = win_height, anchor = "NW", border = "single" }
+  )
+
+  vim.wo[win].signcolumn = "no"
+  vim.wo[win].number = false
+  vim.wo[win].relativenumber = false
+
+  return win
+end
+
+---@param buf integer
+function M._close_buffer(buf)
+  vim.api.nvim_win_close(vim.api.nvim_get_current_win(), true)
+  vim.api.nvim_buf_delete(buf, { unload = true, force = true })
+  M.sessions = {}
+end
+
+function M.prompt()
+  local id = #M.sessions + 1
+  M.sessions[id] = { messages = {} }
+
+  local buf = create_buffer("ai_prompt_" .. id)
+
+  vim.api.nvim_create_autocmd("BufWriteCmd", {
+    buffer = buf,
+    callback = function()
+      vim.bo[buf].modified = false
+
+      local lines = vim.api.nvim_buf_get_lines(buf, 0, -1, false)
+      local content = table.concat(lines, "\n")
+      table.insert(lines, "")
+      table.insert(lines, "")
+
+      vim.api.nvim_buf_set_lines(buf, 0, -1, false, lines)
+
+      table.insert(M.sessions[id].messages, { role = "user", content = content })
+
+      async.void(send_prompt_request)(M.sessions[id].messages, M.model, M.ollama_url, function(chunk)
+        if chunk == nil then
+          return
+        end
+
+        vim.schedule(function()
+          local data = vim.json.decode(chunk)
+          if data.done then
+            return
+          end
+
+          local lastMessage = M.sessions[id].messages[#M.sessions[id].messages]
+          if lastMessage.role == "user" then
+            table.insert(M.sessions[id].messages, data.message)
+            lastMessage = data.message
+          else
+            M.sessions[id].messages[#M.sessions[id].messages] = {
+              role = lastMessage.role,
+              content = lastMessage.content .. data.message.content,
+            }
+          end
+
+          local all_lines = vim.api.nvim_buf_get_lines(buf, 0, -1, false)
+          all_lines[#all_lines] = all_lines[#all_lines] .. data.message.content
+          local s = table.concat(all_lines, "\n")
+
+          vim.api.nvim_buf_set_lines(buf, 0, -1, false, vim.split(s, "\n"))
+        end)
+      end)
+    end,
+  })
+
+  vim.keymap.set("n", "q", function()
+    M._close_buffer(buf)
+  end, { buf = buf })
+
+  vim.keymap.set("n", "c", function()
+    vim.api.nvim_buf_set_lines(buf, 0, -1, false, {})
+  end, { buf = buf })
+
+  create_window(buf)
+end
+
+return M