Grasscutter/src/main/java/emu/grasscutter/server/packet/send/PacketAvatarDataNotify.java

package emu.grasscutter.server.packet.send;

import java.util.Map.Entry;

import emu.grasscutter.game.GenshinPlayer;
import emu.grasscutter.game.TeamInfo;
import emu.grasscutter.game.avatar.GenshinAvatar;
import emu.grasscutter.net.packet.GenshinPacket;
import emu.grasscutter.net.packet.PacketOpcodes;
import emu.grasscutter.net.proto.AvatarDataNotifyOuterClass.AvatarDataNotify;
import emu.grasscutter.net.proto.AvatarTeamOuterClass.AvatarTeam;

public class PacketAvatarDataNotify extends GenshinPacket {
	
	public PacketAvatarDataNotify(GenshinPlayer player) {
		super(PacketOpcodes.AvatarDataNotify, 2);

		AvatarDataNotify.Builder proto = AvatarDataNotify.newBuilder()
				.setCurAvatarTeamId(player.getTeamManager().getCurrentTeamId())
				.setChooseAvatarGuid(player.getTeamManager().getCurrentCharacterGuid())
				.addAllOwnedFlycloakList(player.getFlyCloakList())
				.addAllOwnedCostumeList(player.getCostumeList());
				
		for (GenshinAvatar avatar : player.getAvatars()) {
			proto.addAvatarList(avatar.toProto());
		}
		
		for (Entry<Integer, TeamInfo> entry : player.getTeamManager().getTeams().entrySet()) {
			TeamInfo teamInfo = entry.getValue();
			AvatarTeam.Builder avatarTeam = AvatarTeam.newBuilder()
					.setTeamName(teamInfo.getName());
			
			for (int i = 0; i < teamInfo.getAvatars().size(); i++) {
				GenshinAvatar avatar = player.getAvatars().getAvatarById(teamInfo.getAvatars().get(i));
				avatarTeam.addAvatarGuidList(avatar.getGuid());
			}
			
			proto.putAvatarTeamMap(entry.getKey(), avatarTeam.build());
		}
		
		this.setData(proto.build());
	}

}