fix: async consideration for stage with hunk

chrisgrieser · Sep 29, 2023 · c3be8c3 · c3be8c3
1 parent 1374d4e
commit c3be8c3
Showing 1 changed file with 10 additions and 5 deletions.
diff --git a/lua/tinygit/staging.lua b/lua/tinygit/staging.lua
@@ -2,6 +2,8 @@ local M = {}
 local u = require("tinygit.utils")
 --------------------------------------------------------------------------------
 
+---@deprecated
+---TODO make PR to gitsigns.nvim instead?
 function M.stageHunkWithInfo()
 	-- stage
 	local ok, gitsigns = pcall(require, "gitsigns")
@@ -11,11 +13,14 @@ function M.stageHunkWithInfo()
 	end
 	gitsigns.stage_hunk()
 
-	-- display total stage info
-	local info = vim.fn.system { "git", "diff", "--staged", "--stat" }
-	if vim.v.shell_error ~= 0 then return "" end
-	local infoNoLastLine = info:gsub("\n.-$", "")
-	u.notify(infoNoLastLine, "info", "Staged Changes")
+	-- HACK defer since stage_hunk is async
+	vim.defer_fn(function()
+		-- display total stage info
+		local info = vim.fn.system { "git", "diff", "--staged", "--stat" }
+		if vim.v.shell_error ~= 0 then return "" end
+		local infoNoLastLine = info:gsub("\n.-$", "")
+		u.notify(infoNoLastLine, "info", "Staged Changes")
+	end, 100)
 end
 
 --------------------------------------------------------------------------------