From 8de4999881db17b0824120bace91bf99db32d192 Mon Sep 17 00:00:00 2001
From: Ade Attwood <adrian.attwood@marketdojo.com>
Date: Tue, 26 Dec 2023 10:38:59 +0000
Subject: [PATCH] refactor(vim): update ollama plugin to use older APIs

The ollama plugin was using some APIs that are not yet available in nvim
stable. Now we are using the older APIs we can use this on other
versions of nvim.

The `vim.api.nvim_buf_set_text` does not take some parameters as `-1` to
add text to the end of the line. Now we are getting the text on the last
line adding the new text to it, then replacing the last line.
---
 site-modules/core/files/vim/plugin/ollama.lua | 22 ++++++++++++++-----
 1 file changed, 16 insertions(+), 6 deletions(-)

diff --git a/site-modules/core/files/vim/plugin/ollama.lua b/site-modules/core/files/vim/plugin/ollama.lua
index fdbaf0f..d8fcb4f 100644
--- a/site-modules/core/files/vim/plugin/ollama.lua
+++ b/site-modules/core/files/vim/plugin/ollama.lua
@@ -32,6 +32,16 @@ local function find_buffer_by_name(name)
   return -1
 end
 
+local function append_text(buffer, text)
+  -- Get the total number of lines in the buffer
+  local total_lines = vim.api.nvim_buf_line_count(buffer)
+  local last_line = vim.api.nvim_buf_get_lines(buffer, total_lines - 1, -1, false)[1]
+  local col = #last_line
+
+  -- Append the text to the end of the buffer
+  vim.api.nvim_buf_set_text(buffer, total_lines - 1, col, total_lines - 1, col, text)
+end
+
 local function process_line(results_buffer, line)
   local ok, chunk = pcall(vim.json.decode, line)
   if not ok then
@@ -39,9 +49,9 @@ local function process_line(results_buffer, line)
   end
 
   if chunk.response and chunk.response == "\n" then
-    vim.api.nvim_buf_set_text(results_buffer, -1, -1, -1, -1, { "", "" })
+    append_text(results_buffer, { "", "" })
   elseif chunk.response then
-    vim.api.nvim_buf_set_text(results_buffer, -1, -1, -1, -1, { chunk.response })
+    append_text(results_buffer, { chunk.response })
   end
 
   if chunk.done then
@@ -61,9 +71,9 @@ local function ollama_send()
   local prompt_lines = vim.api.nvim_buf_get_lines(prompt_buffer, 0, -1, false)
 
   -- Update the results buffer with your prompt and then start off the response.
-  vim.api.nvim_buf_set_text(results_buffer, -1, -1, -1, -1, prompt_lines)
+  append_text(results_buffer, prompt_lines)
   local ollama_prompt = { "", "", string.format("<<< Ollama %s", ollama_model), "" }
-  vim.api.nvim_buf_set_text(results_buffer, -1, -1, -1, -1, ollama_prompt)
+  append_text(results_buffer, ollama_prompt)
 
   -- Clear the prompt buffer so it's ready for the next prompt
   vim.api.nvim_buf_set_lines(prompt_buffer, 0, -1, false, {})
@@ -84,7 +94,7 @@ local function ollama_send()
       end
     end,
     on_exit = function()
-      vim.api.nvim_buf_set_text(results_buffer, -1, -1, -1, -1, { "", "", ">>> You", "", "" })
+      append_text(results_buffer, { "", "", ">>> You", "", "" })
     end,
   })
 
@@ -112,7 +122,7 @@ local function ollama_init()
 
   -- Open the response buffer and add the first part of the response
   vim.cmd [[tab new /tmp/ollama-response.md]]
-  vim.api.nvim_buf_set_text(0, -1, -1, -1, -1, { ">>> You", "", "" })
+  append_text(0, { ">>> You", "", "" })
 
   -- Set up the propt buffer read for the user to start chatting
   vim.cmd [[botright split /tmp/ollama-prompt.md | resize 14]]