feat: share lsp diagnostic errors with goose (#53)

Author: azorng

README.md

@@ -162,6 +162,7 @@ The following editor context is automatically captured and included in your conv
 | Current file | Path to the focused file before entering goose |
 | Selected text | Text and lines currently selected in visual mode |
 | Mentioned files | File info added through [mentions](#file-mentions) |
+| Diagnostics | Error diagnostics from the current file (if any) |
 
 <a id="file-mentions"></a>
 ### Adding more files to context through file mentions

lua/goose/context.lua

@@ -12,12 +12,14 @@ M.context = {
 
   -- attachments
   mentioned_files = nil,
-  selections = nil
+  selections = nil,
+  linter_errors = nil
 }
 
 function M.unload_attachments()
   M.context.mentioned_files = nil
   M.context.selections = nil
+  M.context.linter_errors = nil
 end
 
 function M.load()
@@ -27,6 +29,7 @@ function M.load()
 
     M.context.current_file = current_file
     M.context.cursor_data = cursor_data
+    M.context.linter_errors = M.check_linter_errors()
   end
 
   local current_selection = M.get_current_selection()
@@ -40,6 +43,23 @@ function M.load()
   end
 end
 
+function M.check_linter_errors()
+  local diagnostics = vim.diagnostic.get(0, { severity = vim.diagnostic.severity.ERROR })
+  if #diagnostics == 0 then
+    return nil
+  end
+
+  local message = "Found " .. #diagnostics .. " error" .. (#diagnostics > 1 and "s" or "") .. ":"
+
+  for i, diagnostic in ipairs(diagnostics) do
+    local line_number = diagnostic.lnum + 1 -- Convert to 1-based line numbers
+    local short_message = diagnostic.message:gsub("%s+", " "):gsub("^%s", ""):gsub("%s$", "")
+    message = message .. "\n Line " .. line_number .. ": " .. short_message
+  end
+
+  return message
+end
+
 function M.new_selection(file, content, lines)
   return {
     file = file,

template/prompt.tpl

@@ -1,12 +1,12 @@
-<? if current_file or mentioned_files or selections then ?>
+<? if current_file or mentioned_files or selections or linter_errors then ?>
   <additional-data>
     Below is context that may help answer the user query. Ignore if not relevant
     <? if current_file then ?>
       <current-file>
         Path: <%= current_file.path %>
       </current-file>
     <? end ?>
-    <? if selections or mentioned_files then ?>
+    <? if selections or mentioned_files or linter_errors then ?>
       <attached-files>
         <? if selections then ?>
           <? for x, selection in ipairs(selections) do ?>
@@ -30,6 +30,11 @@
             </mentioned-file>
           <? end ?>
         <? end ?>
+        <? if linter_errors then ?>
+          <linter-errors>
+            <%= linter_errors %>
+          </linter-errors>
+        <? end ?>
       </attached-files>
     <? end ?>
   </additional-data>
@@ -38,4 +43,4 @@
   </user-query>
 <? else ?>
   <%= prompt %>
-<? end ?>
+<? end ?>