Fix wrong label placement after completion

monkoose · Sep 11, 2024 · 3cec4a2 · 3cec4a2
1 parent 2fb9869
commit 3cec4a2
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/lua/neocodeium/completer.lua b/lua/neocodeium/completer.lua
@@ -363,6 +363,8 @@ function Completer:accept()
          utils.set_lines(lnum, lnum, block)
          utils.set_cursor(pos)
       end
+      -- required to update label position
+      renderer.pos = pos
    end, 0)
 end
 

diff --git a/lua/neocodeium/renderer.lua b/lua/neocodeium/renderer.lua
@@ -319,6 +319,12 @@ function Renderer:update_forward_line()
       else
          self.inline[1].text = self.block.text:sub(col + 1)
          self:clear_block()
+         -- required to update label position
+         nvim_buf_set_extmark(0, ns, self.pos[1], 0, {
+            id = self.label.id,
+            virt_text = { { " 0 ", "NeoCodeiumLabel" } },
+            virt_text_win_col = -4,
+         })
       end
       self.inline[1].id = show_inline(nil, self.inline[1].text, lnum, col)
    end