Merge pull request #14020 from chentau/float_resize

Update lines after shrinking floating window

1 files changed, 14 insertions, 3 deletions

diff --git a/src/nvim/ui_compositor.c b/src/nvim/ui_compositor.c
index 06efc9fa99..946215d957 100644
--- a/src/nvim/ui_compositor.c
+++ b/src/nvim/ui_compositor.c
@@ -127,6 +127,9 @@ bool ui_comp_put_grid(ScreenGrid *grid, int row, int col, int height, int width,
                       bool valid, bool on_top)
 {
   bool moved;
+
+  grid->comp_height = height;
+  grid->comp_width = width;
   if (grid->comp_index != 0) {
     moved = (row != grid->comp_row) || (col != grid->comp_col);
     if (ui_comp_should_draw()) {
@@ -334,17 +337,25 @@ static void compose_line(Integer row, Integer startcol, Integer endcol,
   sattr_T *bg_attrs = &default_grid.attrs[default_grid.line_offset[row]
                                           +(size_t)startcol];
 
+  int grid_width, grid_height;
   while (col < endcol) {
     int until = 0;
     for (size_t i = 0; i < kv_size(layers); i++) {
       ScreenGrid *g = kv_A(layers, i);
-      if (g->comp_row > row || row >= g->comp_row + g->Rows
+      // compose_line may have been called after a shrinking operation but
+      // before the resize has actually been applied. Therefore, we need to
+      // first check to see if any grids have pending updates to width/height,
+      // to ensure that we don't accidentally put any characters into `linebuf`
+      // that have been invalidated.
+      grid_width = MIN(g->Columns, g->comp_width);
+      grid_height = MIN(g->Rows, g->comp_height);
+      if (g->comp_row > row || row >= g->comp_row + grid_height
           || g->comp_disabled) {
         continue;
       }
-      if (g->comp_col <= col && col < g->comp_col+g->Columns) {
+      if (g->comp_col <= col && col < g->comp_col + grid_width) {
         grid = g;
-        until = g->comp_col+g->Columns;
+        until = g->comp_col + grid_width;
       } else if (g->comp_col > col) {
         until = MIN(until, g->comp_col);
       }