From 279ee740185b8e01aa498e7cd57c31a4defbf17a Mon Sep 17 00:00:00 2001
From: Arto Simonyan <artosimonyan@protonmail.com>
Date: Sun, 8 Dec 2024 01:36:06 +0200
Subject: [PATCH] Buff and Corpse fixes (#551)

---
 go.mod                  | 2 +-
 go.sum                  | 2 ++
 internal/action/buff.go | 9 +++++++++
 3 files changed, 12 insertions(+), 1 deletion(-)

diff --git a/go.mod b/go.mod
index c60886f79..d5964906f 100644
--- a/go.mod
+++ b/go.mod
@@ -7,7 +7,7 @@ require (
 	github.com/bwmarrin/discordgo v0.28.1
 	github.com/go-telegram-bot-api/telegram-bot-api/v5 v5.5.1
 	github.com/gorilla/websocket v1.5.3
-	github.com/hectorgimenez/d2go v0.0.0-20241207062615-aecd76dcb779
+	github.com/hectorgimenez/d2go v0.0.0-20241207223919-823a0a05b094
 	github.com/inkeliz/gowebview v1.0.1
 	github.com/lxn/win v0.0.0-20210218163916-a377121e959e
 	github.com/otiai10/copy v1.14.0
diff --git a/go.sum b/go.sum
index 0d27ba59e..3e512c1fd 100644
--- a/go.sum
+++ b/go.sum
@@ -19,6 +19,8 @@ github.com/hectorgimenez/d2go v0.0.0-20241206181459-bd7e8656b856 h1:Ly5ITqzXGTV9
 github.com/hectorgimenez/d2go v0.0.0-20241206181459-bd7e8656b856/go.mod h1:EOVayMaK8D13wsZiZ6n8AK3+Qflm1wHZsCqnzlVIci0=
 github.com/hectorgimenez/d2go v0.0.0-20241207062615-aecd76dcb779 h1:g/1WvtIuNeDlz4b0LmS3gWJQh2umeh0yyFju1R4bl/k=
 github.com/hectorgimenez/d2go v0.0.0-20241207062615-aecd76dcb779/go.mod h1:EOVayMaK8D13wsZiZ6n8AK3+Qflm1wHZsCqnzlVIci0=
+github.com/hectorgimenez/d2go v0.0.0-20241207223919-823a0a05b094 h1:PejzsBQctNYzzOWW/rSj566S5XKd6Jtu4nxPuSL0JeY=
+github.com/hectorgimenez/d2go v0.0.0-20241207223919-823a0a05b094/go.mod h1:EOVayMaK8D13wsZiZ6n8AK3+Qflm1wHZsCqnzlVIci0=
 github.com/inkeliz/gowebview v1.0.1 h1:4gpLE2qt4kV3DB+xHkHKUeLLiGPN5Xw3or9A3hVqYyA=
 github.com/inkeliz/gowebview v1.0.1/go.mod h1:4SNjXp/fogE11MwvJD67kMBmSObY2BBqinEgH8+8eM8=
 github.com/inkeliz/w32 v1.0.2 h1:Es8Bmw9ApOY0PVRpGs7wsqIKdK5C3xBkP5TOATfVmtU=
diff --git a/internal/action/buff.go b/internal/action/buff.go
index 6f0b17ecb..2d9e740d0 100644
--- a/internal/action/buff.go
+++ b/internal/action/buff.go
@@ -45,6 +45,15 @@ func Buff() {
 		return
 	}
 
+	// Check if we're in loading screen
+	if ctx.Data.OpenMenus.LoadingScreen {
+		ctx.Logger.Debug("Loading screen detected. Waiting for game to load before buffing...")
+		ctx.WaitForGameToLoad()
+
+		// Give it half a second more
+		utils.Sleep(500)
+	}
+
 	preKeys := make([]data.KeyBinding, 0)
 	for _, buff := range ctx.Char.PreCTABuffSkills() {
 		kb, found := ctx.Data.KeyBindings.KeyBindingForSkill(buff)