fixed dynamic popup window sizing

lua/ogpt/common/popup.lua

@@ -6,6 +6,7 @@ function Popup:init(options, edgy)
   options = options or {}
   self.edgy = false
   self.init_update = false
+  self.parent_winid = nil
   if options.edgy and options.border or edgy then
     self.edgy = true
     options.border = nil

lua/ogpt/config.lua

@@ -169,7 +169,7 @@ function M.defaults()
       relative = "editor",
       size = {
         width = "40%",
-        height = 10,
+        height = 5,
       },
       padding = { 1, 1, 1, 1 },
       enter = true,

lua/ogpt/flows/actions/popup/init.lua

@@ -31,6 +31,7 @@ function PopupAction:init(name, opts)
   self.ui = opts.ui or {}
   self.cur_win = vim.api.nvim_get_current_win()
   self.output_panel = PopupWindow(popup_options)
+  self.output_panel.parent_winid = self.cur_win
   self.spinner = Spinner:new(function(state) end)
 
   self:update_variables()
@@ -47,7 +48,6 @@ end
 
 function PopupAction:run()
   -- self.stop = false
-  local response = Response(self.provider)
   local params = self:get_params()
   local _, start_row, start_col, end_row, end_col = self:get_visual_selection()
   local opts = {
@@ -68,6 +68,11 @@ function PopupAction:run()
       -- self.stop = true
     end,
   }
+  local response = Response(self.provider, {
+    on_post_render = function(response)
+      self.output_panel:update_popup_size(opts)
+    end,
+  })
 
   if self.strategy == STRATEGY_DISPLAY then
     self:set_loading(true)

lua/ogpt/flows/actions/popup/window.lua

@@ -19,8 +19,10 @@ function PopupWindow:update_popup_size(opts)
   end
   opts = vim.tbl_extend("force", self.options, opts or {})
   opts.lines = vim.api.nvim_buf_get_lines(self.bufnr, 0, -1, false)
-  local ui_opts = self:calculate_size(opts)
-  self:update_layout(ui_opts)
+  if self.options.dynamic_resize then
+    opts = self:calculate_size(opts)
+  end
+  self:update_layout(opts)
 end
 
 function PopupWindow:calculate_size(opts)
@@ -49,10 +51,6 @@ function PopupWindow:calculate_size(opts)
   ui_h = math.min(ui_h, max_h)
   ui_h = math.max(ui_h, 1)
 
-  -- use opts
-  ui_w = opts.ui_w or ui_w
-  ui_h = opts.ui_h or ui_h
-
   -- build ui opts
   local ui_opts = vim.tbl_deep_extend("keep", {
     size = {
@@ -95,9 +93,9 @@ function PopupWindow:mount(opts)
   if self.options.dynamic_resize then
     -- set the event
     -- https://github.com/MunifTanjim/nui.nvim/blob/main/lua/nui/popup/README.md#popupupdate_layout
-    self:on(event.CursorMoved, function()
-      self:update_popup_size(opts)
-    end)
+    -- self:on(event.CursorMoved, function()
+    --   self:update_popup_size(opts)
+    -- end)
   else
     opts = {
       cur_win = opts.cur_win,

lua/ogpt/flows/chat/base.lua

@@ -304,13 +304,12 @@ function Chat:addAnswerPartial(response)
     for i, line in ipairs(lines) do
       local currentLine = vim.api.nvim_buf_get_lines(buffer, -2, -1, false)[1]
       vim.api.nvim_buf_set_lines(buffer, -2, -1, false, { currentLine .. line })
-
       local last_line_num = vim.api.nvim_buf_line_count(buffer)
       Signs.set_for_lines(self.chat_window.bufnr, start_line, last_line_num - 1, "chat")
       if i == length and i > 1 then
         vim.api.nvim_buf_set_lines(buffer, -1, -1, false, { "" })
       end
-      if self:is_buf_visiable() then
+      if self:is_buf_visible() then
         vim.api.nvim_win_set_cursor(win, { last_line_num, 0 })
       end
     end
@@ -555,7 +554,7 @@ function Chat:is_buf_exists()
   return vim.fn.bufexists(self.chat_window.bufnr) == 1
 end
 
-function Chat:is_buf_visiable()
+function Chat:is_buf_visible()
   -- Get all windows in the current tab
   local wins = vim.api.nvim_tabpage_list_wins(0)
   -- Traverse the window list to determine whether the buffer of chat_window is visible in the window
@@ -585,7 +584,7 @@ function Chat:add_highlight(hl_group, line, col_start, col_end)
 end
 
 function Chat:set_cursor(pos)
-  if self:is_buf_visiable() then
+  if self:is_buf_visible() then
     pcall(vim.api.nvim_win_set_cursor, self.chat_window.winid, pos)
   end
 end

lua/ogpt/response.lua

@@ -127,11 +127,17 @@ function Response:pop_content()
   if content[2] == "END" and (content[1] and content[1] == "") then
     content[1] = self:get_processed_text()
   end
+  if self.events.on_pop_content then
+    self.events.on_pop_content()
+  end
   return content
 end
 
 function Response:render()
   self.partial_result_cb(self)
+  if self.events.on_post_render then
+    self.events.on_post_render(self)
+  end
 end
 
 function Response:pop_chunk()