refactor: rewrite the plugin to be lua-based (#83)

* Refactor the plugin to be lua-based

Signed-off-by: Tomas Slusny <slusnucky@gmail.com>

* Add open/close/toggle commands and function

* Add support for prompt.mapping to map prompts to keys

* Fix issue with system_prompt replace not using correct value and fix naming of USER_ prompts

* Move some of chat buffer logic to separate file, allow changing window layout properly

* Disable python part of the plugin for now

* Fix check if message is copilot message

Signed-off-by: Tomas Slusny <slusnucky@gmail.com>

---------

Signed-off-by: Tomas Slusny <slusnucky@gmail.com>

Author: deathbeam

lua/CopilotChat/chat.lua

@@ -0,0 +1,53 @@
+local Spinner = require('CopilotChat.spinner')
+local class = require('CopilotChat.utils').class
+
+local function create_buf()
+  local bufnr = vim.api.nvim_create_buf(false, true)
+  vim.api.nvim_buf_set_name(bufnr, 'copilot-chat')
+  vim.bo[bufnr].filetype = 'markdown'
+  vim.treesitter.start(bufnr, 'markdown')
+  return bufnr
+end
+
+local Chat = class(function(self, name)
+  self.bufnr = create_buf()
+  self.spinner = Spinner(self.bufnr, name)
+end)
+
+function Chat:validate()
+  if not vim.api.nvim_buf_is_valid(self.bufnr) then
+    self.bufnr = create_buf()
+    self.spinner.bufnr = self.bufnr
+  end
+end
+
+function Chat:append(str)
+  self:validate()
+
+  local last_line, last_column = self:last()
+  vim.api.nvim_buf_set_text(
+    self.bufnr,
+    last_line,
+    last_column,
+    last_line,
+    last_column,
+    vim.split(str, '\n')
+  )
+
+  return self:last()
+end
+
+function Chat:last()
+  self:validate()
+
+  local last_line = vim.api.nvim_buf_line_count(self.bufnr) - 1
+  local last_line_content = vim.api.nvim_buf_get_lines(self.bufnr, -2, -1, false)
+  local last_column = #last_line_content[1]
+  return last_line, last_column
+end
+
+function Chat:clear()
+  vim.api.nvim_buf_set_lines(self.bufnr, 0, -1, false, {})
+end
+
+return Chat

lua/CopilotChat/code_actions.lua

@@ -4,106 +4,73 @@ local telescope_pickers = require('telescope.pickers')
 local finders = require('telescope.finders')
 local themes = require('telescope.themes')
 local conf = require('telescope.config').values
-local utils = require('CopilotChat.utils')
+local select = require('CopilotChat.select')
+local chat = require('CopilotChat')
 
 local help_actions = {}
 local user_prompt_actions = {}
 
 local function generate_fix_diagnostic_prompt()
-  local diagnostic = utils.get_diagnostics()
-  if diagnostic == 'No diagnostics available' then
-    return diagnostic
+  local diagnostic = select.diagnostics()
+  if not diagnostic then
+    return 'No diagnostics available'
   end
 
-  local file_name = vim.fn.expand('%:t')
-  local line_number = vim.fn.line('.')
   return 'Please assist with fixing the following diagnostic issue in file: "'
-    .. file_name
-    .. ':'
-    .. line_number
-    .. '". '
-    .. diagnostic
+    .. diagnostic.prompt_extra
 end
 
 local function generate_explain_diagnostic_prompt()
-  local diagnostic = utils.get_diagnostics()
-  if diagnostic == 'No diagnostics available' then
-    return diagnostic
+  local diagnostic = select.diagnostics()
+  if not diagnostic then
+    return 'No diagnostics available'
   end
 
-  local file_name = vim.fn.expand('%:t')
-  local line_number = vim.fn.line('.')
-  return 'Please explain the following diagnostic issue in file: "'
-    .. file_name
-    .. ':'
-    .. line_number
-    .. '". '
-    .. diagnostic
+  return 'Please explain the following diagnostic issue in file: "' .. diagnostic.prompt_extra
 end
 
 --- Help command for telescope picker
---- This will copy all the lines in the buffer to the unnamed register
+--- This will send whole buffer to copilot
 --- Then will send the diagnostic to copilot chat
----@param prefix string
-local function diagnostic_help_command(prefix)
-  if prefix == nil then
-    prefix = ''
-  else
-    prefix = prefix .. ' '
-  end
-
-  return function(prompt_bufnr, _)
-    actions.select_default:replace(function()
-      actions.close(prompt_bufnr)
-      local selection = action_state.get_selected_entry()
-
-      -- Select all the lines in the buffer to uname register
-      vim.cmd('normal! ggVG"*y')
-
-      -- Get value from the help_actions and execute the command
-      local value = ''
-      for _, action in pairs(help_actions) do
-        if action.name == selection[1] then
-          value = action.label
-          break
-        end
+local function diagnostic_help_command(prompt_bufnr, _)
+  actions.select_default:replace(function()
+    actions.close(prompt_bufnr)
+    local selection = action_state.get_selected_entry()
+
+    -- Get value from the help_actions and execute the command
+    local value = ''
+    for _, action in pairs(help_actions) do
+      if action.name == selection[1] then
+        value = action.label
+        break
       end
+    end
 
-      vim.cmd(prefix .. value)
-    end)
-    return true
-  end
+    chat.ask(value, { selection = select.buffer })
+  end)
+  return true
 end
 
 --- Prompt command for telescope picker
 --- This will show all the user prompts in the telescope picker
 --- Then will execute the command selected by the user
----@param prefix string
-local function generate_prompt_command(prefix)
-  if prefix == nil then
-    prefix = ''
-  else
-    prefix = prefix .. ' '
-  end
-
-  return function(prompt_bufnr, _)
-    actions.select_default:replace(function()
-      actions.close(prompt_bufnr)
-      local selection = action_state.get_selected_entry()
-
-      -- Get value from the prompt_actions and execute the command
-      local value = ''
-      for _, action in pairs(user_prompt_actions) do
-        if action.name == selection[1] then
-          value = action.label
-          break
-        end
+local function generate_prompt_command(prompt_bufnr, _)
+  actions.select_default:replace(function()
+    actions.close(prompt_bufnr)
+    local selection = action_state.get_selected_entry()
+
+    -- Get value from the prompt_actions and execute the command
+    local value = ''
+    for _, action in pairs(user_prompt_actions) do
+      if action.name == selection[1] then
+        value = action.label
+        break
       end
+    end
 
-      vim.cmd(prefix .. value)
-    end)
-    return true
-  end
+    chat.ask(value)
+  end)
+  return true
 end
 
 local function show_help_actions()
@@ -136,25 +103,18 @@ local function show_help_actions()
         results = action_names,
       }),
       sorter = conf.generic_sorter(opts),
-      attach_mappings = diagnostic_help_command('CopilotChat'),
+      attach_mappings = diagnostic_help_command,
     })
     :find()
 end
 
 --- Show prompt actions
----@param is_in_visual_mode boolean?
-local function show_prompt_actions(is_in_visual_mode)
+local function show_prompt_actions()
   -- Convert user prompts to a table of actions
   user_prompt_actions = {}
 
-  for key, prompt in pairs(vim.g.copilot_chat_user_prompts) do
-    table.insert(user_prompt_actions, { name = key, label = prompt })
-  end
-
-  local cmd = 'CopilotChat'
-
-  if is_in_visual_mode then
-    cmd = "'<,'>CopilotChatVisual"
+  for key, prompt in pairs(chat.get_prompts(true)) do
+    table.insert(user_prompt_actions, { name = key, label = prompt.prompt })
   end
 
   -- Show the menu with telescope pickers
@@ -170,7 +130,7 @@ local function show_prompt_actions(is_in_visual_mode)
         results = action_names,
       }),
       sorter = conf.generic_sorter(opts),
-      attach_mappings = generate_prompt_command(cmd),
+      attach_mappings = generate_prompt_command,
     })
     :find()
 end