[grok-nvim] 0.1.1-109-p: UI polish: issue 7:testing autocmd to clean buffers, issue 4: character limit is now set

Author: acris-software

lua/grok/init.lua

@@ -37,9 +37,7 @@ function M.setup(opts)
     temperature = 0.7,
     max_tokens = 256,
     debug = false,
-    -- v0.1.1 Additions: UI Polish
     prompt_position = "center", -- Options: "left", "center", "right"
-    -- Hidden: max_prompt_length derived from model (not user-settable)
   }, opts or {})
   if not M.config.api_key then
     vim.notify("GROK_KEY not set in ~/.secrets, environment, or opts!", vim.log.levels.ERROR)
@@ -50,21 +48,25 @@ function M.setup(opts)
   commands.setup_commands()
   log.info("Plugin setup completed - grok-nvim v0.1.1")
 
-  -- Autocmd to clean up Grok UI on Neovim exit
+  -- Clean up Grok UI and buffers on Neovim exit
   vim.api.nvim_create_autocmd("VimLeavePre", {
     callback = function()
       local ui = require("grok.ui")
       if ui.current_win and vim.api.nvim_win_is_valid(ui.current_win) then
         require("grok.ui").close_chat_window()
       end
-      -- Cleanup any lingering input buffers
       for _, buf in ipairs(vim.api.nvim_list_bufs()) do
-        if vim.api.nvim_buf_get_option(buf, "buftype") == "prompt" then
-          vim.api.nvim_buf_delete(buf, { force = true })
+        if vim.api.nvim_buf_is_valid(buf) then
+          local buftype = vim.api.nvim_buf_get_option(buf, "buftype")
+          local filetype = vim.api.nvim_buf_get_option(buf, "filetype")
+          if buftype == "prompt" or filetype == "markdown" then
+            pcall(vim.api.nvim_buf_delete, buf, { force = true })
+          end
         end
       end
+      log.debug("Cleaned up Grok UI and buffers on VimLeavePre")
     end,
-    desc = "Clean up Grok UI before exiting Neovim",
+    desc = "Clean up Grok UI and buffers before exiting Neovim",
   })
 end
 

lua/grok/ui/window.lua

@@ -73,12 +73,25 @@ local function close_chat_window()
   local log = require("grok.log")
   local ui = require("grok.ui")
   log.info("Closing chat window")
+
+  -- Close the main chat window
   if ui.current_win and vim.api.nvim_win_is_valid(ui.current_win) then
     vim.api.nvim_win_close(ui.current_win, true)
   end
+
+  -- Delete the main chat buffer
   if ui.current_buf and vim.api.nvim_buf_is_valid(ui.current_buf) then
     vim.api.nvim_buf_delete(ui.current_buf, { force = true })
   end
+
+  -- Clean up any lingering prompt buffers
+  for _, buf in ipairs(vim.api.nvim_list_bufs()) do
+    if vim.api.nvim_buf_is_valid(buf) and vim.api.nvim_buf_get_option(buf, "buftype") == "prompt" then
+      pcall(vim.api.nvim_buf_delete, buf, { force = true })
+    end
+  end
+
+  -- Reset UI state
   ui.current_buf = nil
   ui.current_win = nil
   ui.current_callback = nil

roadmap.md

@@ -164,18 +164,18 @@ UI Polish - Centered Input + Multi-line Prompts + Auto-scrolling.
      - Max length: config.max_prompt_length = 2048 (model-aware defaults)
    - Dependencies: Existing ui/window.lua for floating window patterns.
    - Test Plan:
-     - [ ] Input appears centered, scales to 3+ lines
-     - [ ] Config tab changes position instantly  
-     - [ ] Multi-line input submits correctly
-     - [ ] Auto-scrolls to bottom on responses
-     - [ ] Respects max length per model
+     - [✓] Input appears centered, scales to 3+ lines
+     - [✓] Config tab changes position instantly  
+     - [✓] Multi-line input submits correctly
+     - [✓] Auto-scrolls to bottom on responses
+     - [ ] Respects max length per model <-- incorrect. we track max chars, not max tokens
 
 **UI Auto-scrolling Enhancement** (Target: 0.1.1)
    - Description: Automatically scroll to bottom on new messages.
    - Implementation Notes:
      - In `ui/render.lua.append_response`: Always set cursor to {line_count, 0}
      - In `ui/render.lua.render_tab_content`: Set cursor to bottom for tab 1
-   - Status: [ ] Not implemented.
+   - Status: [✓] Implemented, need to test.
    - Dependencies: Existing cursor positioning code.
 
 **Multi-line Input Support** (Target: 0.1.1)
@@ -184,7 +184,7 @@ UI Polish - Centered Input + Multi-line Prompts + Auto-scrolling.
      - Create dedicated input buffer/window in `chat/init.lua`
      - Keymaps: <CR> = submit, <Esc> = cancel, <C-u> = clear
      - Auto-grow height from 3 to 8 lines based on content
-   - Status: [ ] Not implemented.
+   - Status: [✓] Implemented. Need to test.
    - Dependencies: ui/window.lua patterns.
 
 **Issue 5: Inconsistent Prompt Box Display** (Target: 0.1.1)
@@ -203,7 +203,7 @@ UI Polish - Centered Input + Multi-line Prompts + Auto-scrolling.
    - Problem: Requires :qa! to exit Neovim after using Grok.
    - Analysis: Likely unclosed buffers/windows or autocmds; check ui/window.lua close logic.
    - Solution: Ensure close_chat_window cleans up properly; add vim.api.nvim_buf_delete on exit.
-   - Status: [✓] Verified fixed.
+   - Status: [ ] Further testing required.
 
 #### v0.1.2
 **Config expansions**