From 23989366e1ccb98b4c97363f9af8891e5dab6f85 Mon Sep 17 00:00:00 2001
From: Melledy <121644117+Melledy@users.noreply.github.com>
Date: Mon, 2 Oct 2023 06:12:09 -0700
Subject: [PATCH] Prevent the player from adding more than 1 main character

---
 .../java/emu/lunarcore/game/avatar/AvatarStorage.java     | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/main/java/emu/lunarcore/game/avatar/AvatarStorage.java b/src/main/java/emu/lunarcore/game/avatar/AvatarStorage.java
index 2edf639..853a656 100644
--- a/src/main/java/emu/lunarcore/game/avatar/AvatarStorage.java
+++ b/src/main/java/emu/lunarcore/game/avatar/AvatarStorage.java
@@ -3,6 +3,7 @@ package emu.lunarcore.game.avatar;
 import java.util.Iterator;
 import java.util.stream.Stream;
 
+import emu.lunarcore.GameConstants;
 import emu.lunarcore.LunarRail;
 import emu.lunarcore.data.GameData;
 import emu.lunarcore.data.excel.AvatarExcel;
@@ -39,10 +40,15 @@ public class AvatarStorage extends BasePlayerManager implements Iterable<GameAva
     }
 
     public boolean addAvatar(GameAvatar avatar) {
-        // Sanity
+        // Sanity - Dont add avatars we already have OR avatars that dont have excel data
         if (avatar.getExcel() == null || this.hasAvatar(avatar.getAvatarId())) {
             return false;
         }
+        
+        // Dont add more than 1 main character
+        if (avatar.isHero() && this.hasAvatar(GameConstants.TRAILBLAZER_AVATAR_ID)) {
+            return false;
+        }
 
         // Set owner first
         avatar.setOwner(getPlayer());