fix: call schedule before event dispatch

Author: yioneko

lua/nvim-tree/actions/fs/create-file.lua

@@ -23,7 +23,6 @@ local function create_and_notify(file)
   end
   if M.enable_async then
     async.call(vim.loop.fs_close, fd)
-    async.schedule()
   else
     vim.loop.fs_close(fd)
   end
@@ -104,7 +103,6 @@ local async_fn = async.wrap(function(node)
         is_error = true
         break
       end
-      async.schedule()
       events._dispatch_folder_created(new_file_path)
     end
   end

lua/nvim-tree/api.lua

@@ -129,7 +129,12 @@ Api.marks.navigate.next = require("nvim-tree.marks.navigation").next
 Api.marks.navigate.prev = require("nvim-tree.marks.navigation").prev
 Api.marks.navigate.select = require("nvim-tree.marks.navigation").select
 
-Api.async.call = require("nvim-tree.async").call
+Api.async.call = function(...)
+  if not require("nvim-tree.async").in_async() then
+    error "The async call is not triggered in an async context of nvim-tree, please use `api.in_async` to detect beforehand"
+  end
+  require("nvim-tree.async").call(...)
+end
 Api.async.in_async = require("nvim-tree.async").in_async
 
 return Api

lua/nvim-tree/async.lua

@@ -101,15 +101,6 @@ function M.schedule()
   end)
 end
 
-function M.unwrap_err(...)
-  local args = { ... }
-  local err = table.remove(args, 1)
-  if err then
-    error(err)
-  end
-  return unpack(args)
-end
-
 ---@class Interrupter
 ---@field yield fun()
 ---@field interval number

lua/nvim-tree/events.lua

@@ -56,10 +56,12 @@ local function dispatch(event_name, payload)
   safe_call_handlers(evt_handlers.normal)
 
   if async.in_async() then
+    async.schedule() -- schedule to avoid manually calling `vim.schedule` in event handlers
     safe_call_handlers(evt_handlers.async)
   else
     -- if we are not in async context, we should execute them in a new async context
     async.exec(function()
+      async.schedule()
       safe_call_handlers(evt_handlers.async)
     end)
   end