From 17580e5f9ae34c0bdd9ce93cce5ba23828bb72aa Mon Sep 17 00:00:00 2001
From: monkoose <pythonproof@gmail.com>
Date: Tue, 5 Nov 2024 09:42:48 +0200
Subject: [PATCH] Attempt to fix missing label in some situations

Seems like doesn't help much, but not bad to control scheduling a little
bit better
---
 lua/neocodeium/completer.lua | 41 ++++++++++++++++++++++--------------
 lua/neocodeium/init.lua      |  1 -
 2 files changed, 25 insertions(+), 17 deletions(-)

diff --git a/lua/neocodeium/completer.lua b/lua/neocodeium/completer.lua
index b804bb2..22d7b8e 100644
--- a/lua/neocodeium/completer.lua
+++ b/lua/neocodeium/completer.lua
@@ -210,11 +210,11 @@ function Completer:enabled()
 end
 
 function Completer:update_label()
-   if utils.is_insert() and utils.is_empty(self.inline) and not self.block.text then
-      vim.schedule(function()
+   vim.schedule(function()
+      if utils.is_insert() and utils.is_empty(self.inline) and not self.block.text then
          self:display_label()
-      end)
-   end
+      end
+   end)
 end
 
 ---@private
@@ -352,13 +352,9 @@ end
 ---@private
 function Completer:start_clear_timer()
    if not self.clear_timer:is_active() then
-      self.clear_timer:start(
-         350,
-         0,
-         vim.schedule_wrap(function()
-            self:clear_all()
-         end)
-      )
+      self.clear_timer:start(350, 0, function()
+         self:clear_all(false, true)
+      end)
    end
 end
 
@@ -498,20 +494,33 @@ end
 
 ---Clears plugin's namespace and resets cache
 ---@param with_reset? boolean
-function Completer:clear_all(with_reset)
+---@param scheduled? boolean
+function Completer:clear_all(with_reset, scheduled)
    if with_reset then
       self.clear_timer:stop()
-      nvim_buf_clear_namespace(0, ns, 0, -1)
       self.label.id = nil
       self.inline = {}
       self.block.id = nil
       self.block.text = nil
       self.fulltext = ""
+      if scheduled then
+         vim.schedule(function()
+            nvim_buf_clear_namespace(0, ns, 0, -1)
+         end)
+      else
+         nvim_buf_clear_namespace(0, ns, 0, -1)
+      end
       events.emit("NeoCodeiumCompletionCleared", nil, true)
    else
-      -- self:clear_label()
-      self:clear_inline()
-      self:clear_block()
+      if scheduled then
+         vim.schedule(function()
+            self:clear_inline()
+            self:clear_block()
+         end)
+      else
+         self:clear_inline()
+         self:clear_block()
+      end
    end
 end
 
diff --git a/lua/neocodeium/init.lua b/lua/neocodeium/init.lua
index 56806c0..3e1f57b 100644
--- a/lua/neocodeium/init.lua
+++ b/lua/neocodeium/init.lua
@@ -125,7 +125,6 @@ local function enable_autocmds()
 
    create_autocmd("InsertEnter", {
       callback = function()
-         completer:update_label()
          completer:initiate()
       end,
    })