Fix wrong label placement after completion

Author: monkoose

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
 

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