From 9cf745939d792204a18d7ad10a54d22386ececf3 Mon Sep 17 00:00:00 2001
From: Folke Lemaitre <folke.lemaitre@gmail.com>
Date: Fri, 28 Jun 2024 20:17:34 +0200
Subject: [PATCH] feat(task): build procs can now yield a LazyMsg for more
 control

---
 lua/lazy/manage/task/init.lua | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/lua/lazy/manage/task/init.lua b/lua/lazy/manage/task/init.lua
index 2b21001..277a592 100644
--- a/lua/lazy/manage/task/init.lua
+++ b/lua/lazy/manage/task/init.lua
@@ -112,9 +112,13 @@ function Task:_run(task)
   task(self, self._opts)
 end
 
----@param msg string|string[]
+---@param msg string|string[]|LazyMsg
 ---@param level? number
 function Task:log(msg, level)
+  if type(msg) == "table" and msg.msg then
+    level = msg.level or level
+    msg = msg.msg
+  end
   level = level or vim.log.levels.DEBUG
   self._level = math.max(self._level or 0, level or 0)
   msg = type(msg) == "table" and table.concat(msg, "\n") or msg
@@ -170,8 +174,8 @@ function Task:_done()
   if self._opts.on_done then
     self._opts.on_done(self)
   end
+  self:render()
   vim.schedule(function()
-    self:render()
     vim.api.nvim_exec_autocmds("User", {
       pattern = "LazyPlugin" .. self.name:sub(1, 1):upper() .. self.name:sub(2),
       data = { plugin = self.plugin.name },