Grasscutter/src/main/java/emu/grasscutter/game/managers/blossom/BlossomActivity.java

package emu.grasscutter.game.managers.blossom;

import emu.grasscutter.data.GameData;
import emu.grasscutter.game.dungeons.challenge.WorldChallenge;
import emu.grasscutter.game.dungeons.challenge.trigger.*;
import emu.grasscutter.game.entity.*;
import emu.grasscutter.game.props.FightProperty;
import emu.grasscutter.game.world.*;
import emu.grasscutter.scripts.data.*;
import emu.grasscutter.utils.Utils;
import java.util.*;

public final class BlossomActivity {

    private final SceneGroup tempSceneGroup;
    private final WorldChallenge challenge;
    private final EntityGadget gadget;
    private EntityGadget chest;
    private int step;
    private final int goal;
    private int generatedCount;
    private final int worldLevel;
    private boolean pass = false;
    private final List<EntityMonster> activeMonsters = new ArrayList<>();
    private final Queue<Integer> candidateMonsters = new ArrayDeque<>();
    private static final int BLOOMING_GADGET_ID = 70210109;

    public BlossomActivity(
            EntityGadget entityGadget, List<Integer> monsters, int timeout, int worldLevel) {
        this.tempSceneGroup = new SceneGroup();
        this.tempSceneGroup.id = entityGadget.getId();
        this.gadget = entityGadget;
        this.step = 0;
        this.goal = monsters.size();
        this.candidateMonsters.addAll(monsters);
        this.worldLevel = worldLevel;
        ArrayList<ChallengeTrigger> challengeTriggers = new ArrayList<>();
        this.challenge =
                new WorldChallenge(
                        entityGadget.getScene(),
                        tempSceneGroup,
                        1,
                        1,
                        List.of(goal, timeout),
                        timeout,
                        goal,
                        challengeTriggers);
        challengeTriggers.add(new KillMonsterCountTrigger());
        // this.challengeTriggers.add(new InTimeTrigger());
    }

    public WorldChallenge getChallenge() {
        return this.challenge;
    }

    public void setMonsters(List<EntityMonster> monsters) {
        this.activeMonsters.clear();
        this.activeMonsters.addAll(monsters);
        for (EntityMonster monster : monsters) {
            monster.setGroupId(this.tempSceneGroup.id);
        }
    }

    public int getAliveMonstersCount() {
        int count = 0;
        for (EntityMonster monster : activeMonsters) {
            if (monster.isAlive()) {
                count++;
            }
        }
        return count;
    }

    public boolean getPass() {
        return pass;
    }

    public void start() {
        challenge.start();
    }

    public void onTick() {
        Scene scene = gadget.getScene();
        Position pos = gadget.getPosition();
        if (getAliveMonstersCount() <= 2) {
            if (generatedCount < goal) {
                step++;

                var worldLevelData = GameData.getWorldLevelDataMap().get(worldLevel);
                int worldLevelOverride = 0;
                if (worldLevelData != null) {
                    worldLevelOverride = worldLevelData.getMonsterLevel();
                }

                List<EntityMonster> newMonsters = new ArrayList<>();
                int willSpawn = Utils.randomRange(3, 5);
                if (generatedCount + willSpawn > goal) {
                    willSpawn = goal - generatedCount;
                }
                generatedCount += willSpawn;
                for (int i = 0; i < willSpawn; i++) {
                    var entry = candidateMonsters.poll();
                    if (entry == null) continue;

                    var monsterData = GameData.getMonsterDataMap().get((int) entry);
                    var level = scene.getEntityLevel(1, worldLevelOverride);
                    var entity =
                            new EntityMonster(scene, monsterData, pos.nearby2d(4f), Position.ZERO, level);
                    scene.addEntity(entity);
                    newMonsters.add(entity);
                }
                setMonsters(newMonsters);
            } else {
                if (getAliveMonstersCount() == 0) {
                    this.pass = true;
                    this.challenge.done();
                }
            }
        }
    }

    public EntityGadget getGadget() {
        return gadget;
    }

    public EntityGadget getChest() {
        if (chest == null) {
            EntityGadget rewardGadget =
                    new EntityGadget(gadget.getScene(), BLOOMING_GADGET_ID, gadget.getPosition());
            SceneGadget metaGadget = new SceneGadget();
            metaGadget.boss_chest = new SceneBossChest();
            metaGadget.boss_chest.resin = 20;
            rewardGadget.setFightProperty(FightProperty.FIGHT_PROP_BASE_HP, Float.POSITIVE_INFINITY);
            rewardGadget.setFightProperty(FightProperty.FIGHT_PROP_CUR_HP, Float.POSITIVE_INFINITY);
            rewardGadget.setFightProperty(FightProperty.FIGHT_PROP_MAX_HP, Float.POSITIVE_INFINITY);
            rewardGadget.setMetaGadget(metaGadget);
            rewardGadget.buildContent();
            chest = rewardGadget;
        }
        return chest;
    }
}