package com.cursesmp.curse; import com.cursesmp.curse.curses.combat.*; import com.cursesmp.curse.curses.environmental.*; import com.cursesmp.curse.curses.mining.*; import com.cursesmp.curse.curses.movement.*; import com.cursesmp.curse.curses.mythic.*; import com.cursesmp.curse.curses.rng.*; import com.cursesmp.curse.curses.social.*; import java.util.*; public class CurseRegistry { private final Map curses = new HashMap<>(); private final Map> cursesByTier = new EnumMap<>(CurseTier.class); public CurseRegistry() { for (CurseTier tier : CurseTier.values()) { this.cursesByTier.put(tier, new ArrayList<>()); } registerDefaultCurses(); } private void registerDefaultCurses() { registerCurse(new WindwalkerCurse()); registerCurse(new PhantomStepCurse()); registerCurse(new GravityShiftCurse()); registerCurse(new ShadowStepCurse()); registerCurse(new BerserkerCurse()); registerCurse(new GlassCannonCurse()); registerCurse(new VampiricCurse()); registerCurse(new ThornsAuraCurse()); registerCurse(new AssassinCurse()); registerCurse(new BloodPactCurse()); registerCurse(new IronwillCurse()); registerCurse(new VeinMinerCurse()); registerCurse(new FortuneGambleCurse()); registerCurse(new MidasTouchCurse()); registerCurse(new DwarvenMightCurse()); registerCurse(new AquaticCurse()); registerCurse(new PyromancerCurse()); registerCurse(new FrostbornCurse()); registerCurse(new StormcallerCurse()); registerCurse(new WildMagicCurse()); registerCurse(new ChaosCurse()); registerCurse(new LuckManipulatorCurse()); registerCurse(new PackLeaderCurse()); registerCurse(new SymbioteCurse()); registerCurse(new CurseWeaverCurse()); registerCurse(new TimeDialatorCurse()); registerCurse(new SoulReaperCurse()); registerCurse(new VoidTouchedCurse()); } public void registerCurse(Curse curse) { this.curses.put(curse.getId(), curse); this.cursesByTier.get(curse.getTier()).add(curse); } public Curse getCurse(String id) { return this.curses.get(id); } public Collection getAllCurses() { return this.curses.values(); } public List getCursesByTier(CurseTier tier) { return new ArrayList<>(this.cursesByTier.get(tier)); } public Curse getRandomCurse(Map probabilities) { int totalWeight = probabilities.values().stream().mapToInt(Integer::intValue).sum(); int random = new Random().nextInt(totalWeight); int currentWeight = 0; for (Map.Entry entry : probabilities.entrySet()) { currentWeight += entry.getValue(); if (random < currentWeight) { List tierCurses = getCursesByTier(entry.getKey()); if (!tierCurses.isEmpty()) { return tierCurses.get(new Random().nextInt(tierCurses.size())); } } } return getAllCurses().iterator().next(); } }