Grasscutter/src/main/java/emu/grasscutter/data/excels/AvatarSkillDepotData.java

package emu.grasscutter.data.excels;

import java.util.List;
import java.util.Optional;
import java.util.stream.IntStream;

import emu.grasscutter.data.GameData;
import emu.grasscutter.data.GameDepot;
import emu.grasscutter.data.GameResource;
import emu.grasscutter.data.ResourceLoader.AvatarConfig;
import emu.grasscutter.data.ResourceType;
import emu.grasscutter.data.ResourceType.LoadPriority;
import emu.grasscutter.data.binout.AbilityEmbryoEntry;
import emu.grasscutter.game.props.ElementType;
import emu.grasscutter.utils.Utils;
import it.unimi.dsi.fastutil.ints.IntArrayList;
import it.unimi.dsi.fastutil.ints.IntList;
import lombok.Getter;

@ResourceType(name = "AvatarSkillDepotExcelConfigData.json", loadPriority = LoadPriority.HIGH)
public class AvatarSkillDepotData extends GameResource {

    private int id;
    @Getter private int energySkill;
    @Getter private int attackModeSkill;

    @Getter private List<Integer> skills;
    @Getter private List<Integer> subSkills;
    @Getter private List<String> extraAbilities;
    @Getter private List<Integer> talents;
    @Getter private List<InherentProudSkillOpens> inherentProudSkillOpens;

    @Getter private String talentStarName;
    @Getter private String skillDepotAbilityGroup;

    // Transient
    @Getter private AvatarSkillData energySkillData;
    @Getter private ElementType elementType;
    @Getter private IntList abilities;
    @Getter private int talentCostItemId;

    @Override
    public int getId() {
        return this.id;
    }

    public void setAbilities(AbilityEmbryoEntry info) {
        this.abilities = new IntArrayList(info.getAbilities().length);
        for (String ability : info.getAbilities()) {
            this.abilities.add(Utils.abilityHash(ability));
        }
    }

    @Override
    public void onLoad() {
        // Set energy skill data
        this.energySkillData = GameData.getAvatarSkillDataMap().get(this.energySkill);
        if (this.energySkillData != null) {
            this.elementType = this.energySkillData.getCostElemType();
        } else {
            this.elementType = ElementType.None;
        }
        // Set embryo abilities (if player skill depot)
        if (getSkillDepotAbilityGroup() != null && getSkillDepotAbilityGroup().length() > 0) {
            AvatarConfig config = GameDepot.getPlayerAbilities().get(getSkillDepotAbilityGroup());

            if (config != null) {
                this.setAbilities(new AbilityEmbryoEntry(getSkillDepotAbilityGroup(), config.abilities.stream().map(Object::toString).toArray(String[]::new)));
            }
        }

        // Get constellation item from GameData
        Optional.ofNullable(this.talents)
            .map(talents -> talents.get(0))
            .map(i -> GameData.getAvatarTalentDataMap().get((int) i))
            .map(talentData -> talentData.getMainCostItemId())
            .ifPresent(itemId -> this.talentCostItemId = itemId);
    }

    public static class InherentProudSkillOpens {
        @Getter private int proudSkillGroupId;
        @Getter private int needAvatarPromoteLevel;
    }

    public IntStream getSkillsAndEnergySkill() {
        return IntStream.concat(this.skills.stream().mapToInt(i -> i), IntStream.of(this.energySkill))
                        .filter(skillId -> skillId > 0);
    }
}