ModuleFactory.java

package in.twizmwaz.cardinal.module;

import in.twizmwaz.cardinal.Cardinal;
import in.twizmwaz.cardinal.match.Match;
import in.twizmwaz.cardinal.module.modules.appliedRegion.AppliedRegionBuilder;
import in.twizmwaz.cardinal.module.modules.armorKeep.ArmorKeepBuilder;
import in.twizmwaz.cardinal.module.modules.attackSpeed.AttackSpeedBuilder;
import in.twizmwaz.cardinal.module.modules.blitz.BlitzBuilder;
import in.twizmwaz.cardinal.module.modules.blockdrops.BlockdropsBuilder;
import in.twizmwaz.cardinal.module.modules.broadcasts.BroadcastModuleBuilder;
import in.twizmwaz.cardinal.module.modules.buildHeight.BuildHeightBuilder;
import in.twizmwaz.cardinal.module.modules.cardinalNotifications.CardinalNotificationsBuilder;
import in.twizmwaz.cardinal.module.modules.chat.ChatModuleBuilder;
import in.twizmwaz.cardinal.module.modules.chatChannels.ChatChannelModuleBuilder;
import in.twizmwaz.cardinal.module.modules.classModule.ClassModuleBuilder;
import in.twizmwaz.cardinal.module.modules.cores.CoreObjectiveBuilder;
import in.twizmwaz.cardinal.module.modules.craftingModule.CraftingModuleBuilder;
import in.twizmwaz.cardinal.module.modules.ctf.FlagBuilder;
import in.twizmwaz.cardinal.module.modules.ctf.net.NetBuilder;
import in.twizmwaz.cardinal.module.modules.ctf.post.PostBuilder;
import in.twizmwaz.cardinal.module.modules.damageIndicator.DamageIndicatorBuilder;
import in.twizmwaz.cardinal.module.modules.deathMessages.DeathMessagesBuilder;
import in.twizmwaz.cardinal.module.modules.deathTracker.DeathTrackerBuilder;
import in.twizmwaz.cardinal.module.modules.destroyable.DestroyableObjectiveBuilder;
import in.twizmwaz.cardinal.module.modules.difficulty.MapDifficultyBuilder;
import in.twizmwaz.cardinal.module.modules.disableDamage.DisableDamageBuilder;
import in.twizmwaz.cardinal.module.modules.fallingBlocks.FallingBlocksBuilder;
import in.twizmwaz.cardinal.module.modules.feedback.FeedbackModuleBuilder;
import in.twizmwaz.cardinal.module.modules.filter.FilterModuleBuilder;
import in.twizmwaz.cardinal.module.modules.freeze.FreezeModuleBuilder;
import in.twizmwaz.cardinal.module.modules.friendlyFire.FriendlyFireBuilder;
import in.twizmwaz.cardinal.module.modules.gameComplete.GameCompleteBuilder;
import in.twizmwaz.cardinal.module.modules.gamerules.GamerulesBuilder;
import in.twizmwaz.cardinal.module.modules.guiKeep.GuiKeepModuleBuilder;
import in.twizmwaz.cardinal.module.modules.header.HeaderModuleBuilder;
import in.twizmwaz.cardinal.module.modules.hill.HillObjectiveBuilder;
import in.twizmwaz.cardinal.module.modules.hunger.HungerBuilder;
import in.twizmwaz.cardinal.module.modules.invisibleBlock.InvisibleBlockBuilder;
import in.twizmwaz.cardinal.module.modules.itemKeep.ItemKeepBuilder;
import in.twizmwaz.cardinal.module.modules.itemMods.ItemModsBuilder;
import in.twizmwaz.cardinal.module.modules.itemRemove.ItemRemoveBuilder;
import in.twizmwaz.cardinal.module.modules.killReward.KillRewardBuilder;
import in.twizmwaz.cardinal.module.modules.killStreakCount.KillStreakBuilder;
import in.twizmwaz.cardinal.module.modules.kit.KitBuilder;
import in.twizmwaz.cardinal.module.modules.mapNotification.MapNotificationBuilder;
import in.twizmwaz.cardinal.module.modules.match.MatchModuleBuilder;
import in.twizmwaz.cardinal.module.modules.matchTimer.MatchTimerBuilder;
import in.twizmwaz.cardinal.module.modules.mob.MobModuleBuilder;
import in.twizmwaz.cardinal.module.modules.monumentModes.MonumentModesBuilder;
import in.twizmwaz.cardinal.module.modules.motd.MOTDBuilder;
import in.twizmwaz.cardinal.module.modules.multitrade.MultitradeBuilder;
import in.twizmwaz.cardinal.module.modules.observers.ObserverModuleBuilder;
import in.twizmwaz.cardinal.module.modules.permissions.PermissionModuleBuilder;
import in.twizmwaz.cardinal.module.modules.playable.PlayableBuilder;
import in.twizmwaz.cardinal.module.modules.portal.PortalBuilder;
import in.twizmwaz.cardinal.module.modules.potionRemover.PotionRemoverBuilder;
import in.twizmwaz.cardinal.module.modules.projectileParticles.ProjectileParticlesModuleBuilder;
import in.twizmwaz.cardinal.module.modules.projectiles.ProjectilesBuilder;
import in.twizmwaz.cardinal.module.modules.proximityAlarm.ProximityAlarmBuilder;
import in.twizmwaz.cardinal.module.modules.rage.RageBuilder;
import in.twizmwaz.cardinal.module.modules.rank.RankModuleBuilder;
import in.twizmwaz.cardinal.module.modules.regions.RegionModuleBuilder;
import in.twizmwaz.cardinal.module.modules.respawn.RespawnModuleBuilder;
import in.twizmwaz.cardinal.module.modules.score.ScoreModuleBuilder;
import in.twizmwaz.cardinal.module.modules.scoreboard.ScoreboardModuleBuilder;
import in.twizmwaz.cardinal.module.modules.scorebox.ScoreboxBuilder;
import in.twizmwaz.cardinal.module.modules.snowflakes.SnowflakesBuilder;
import in.twizmwaz.cardinal.module.modules.sound.SoundModuleBuilder;
import in.twizmwaz.cardinal.module.modules.spawn.SpawnModuleBuilder;
import in.twizmwaz.cardinal.module.modules.spectatorTools.SpectatorToolsBuilder;
import in.twizmwaz.cardinal.module.modules.stats.StatsBuilder;
import in.twizmwaz.cardinal.module.modules.tasker.TaskerModuleBuilder;
import in.twizmwaz.cardinal.module.modules.team.TeamModuleBuilder;
import in.twizmwaz.cardinal.module.modules.teamManager.TeamManagerModuleBuilder;
import in.twizmwaz.cardinal.module.modules.teamPicker.TeamPickerBuilder;
import in.twizmwaz.cardinal.module.modules.timeLimit.TimeLimitBuilder;
import in.twizmwaz.cardinal.module.modules.timeLock.TimeLockBuilder;
import in.twizmwaz.cardinal.module.modules.timeNotifications.TimeNotificationsBuilder;
import in.twizmwaz.cardinal.module.modules.timers.TimersBuilder;
import in.twizmwaz.cardinal.module.modules.titleRespawn.TitleRespawnBuilder;
import in.twizmwaz.cardinal.module.modules.tnt.TntBuilder;
import in.twizmwaz.cardinal.module.modules.tntDefuse.TntDefuseBuilder;
import in.twizmwaz.cardinal.module.modules.tntTracker.TntTrackerBuilder;
import in.twizmwaz.cardinal.module.modules.toolRepair.ToolRepairBuilder;
import in.twizmwaz.cardinal.module.modules.tracker.TrackerBuilder;
import in.twizmwaz.cardinal.module.modules.tutorial.TutorialBuilder;
import in.twizmwaz.cardinal.module.modules.visibility.VisibilityBuilder;
import in.twizmwaz.cardinal.module.modules.wildcard.WildCardBuilder;
import in.twizmwaz.cardinal.module.modules.wools.WoolObjectiveBuilder;
import in.twizmwaz.cardinal.module.modules.worldFreeze.WorldFreezeBuilder;
import in.twizmwaz.cardinal.util.Config;
import org.bukkit.Bukkit;
import org.bukkit.ChatColor;

import java.lang.reflect.InvocationTargetException;
import java.util.Comparator;
import java.util.List;
import java.util.Objects;
import java.util.logging.Level;
import java.util.stream.Collectors;
import java.util.stream.Stream;

public class ModuleFactory {

    private final List<? extends ModuleBuilder> builders;

    private Stream<Class<? extends ModuleBuilder>> getBuilders() {
        return Stream.of(
                BuildHeightBuilder.class,
                WoolObjectiveBuilder.class,
                CoreObjectiveBuilder.class,
                DestroyableObjectiveBuilder.class,
                ItemRemoveBuilder.class,
                ToolRepairBuilder.class,
                DisableDamageBuilder.class,
                GamerulesBuilder.class,
                KitBuilder.class,
                TimeLockBuilder.class,
                FriendlyFireBuilder.class,
                HungerBuilder.class,
                MapDifficultyBuilder.class,
                HungerBuilder.class,
                ProjectilesBuilder.class,
                TntTrackerBuilder.class,
                VisibilityBuilder.class,
                MOTDBuilder.class,
                WorldFreezeBuilder.class,
                TeamManagerModuleBuilder.class,
                RespawnModuleBuilder.class,
                ObserverModuleBuilder.class,
                KillStreakBuilder.class,
                TeamPickerBuilder.class,
                ScoreboardModuleBuilder.class,
                TeamModuleBuilder.class,
                SpawnModuleBuilder.class,
                DeathMessagesBuilder.class,
                TntDefuseBuilder.class,
                ScoreModuleBuilder.class,
                GameCompleteBuilder.class,
                RegionModuleBuilder.class,
                TaskerModuleBuilder.class,
                MatchTimerBuilder.class,
                ItemKeepBuilder.class,
                ArmorKeepBuilder.class,
                BroadcastModuleBuilder.class,
                MatchModuleBuilder.class,
                TimeNotificationsBuilder.class,
                HillObjectiveBuilder.class,
                ChatModuleBuilder.class,
                MonumentModesBuilder.class,
                RageBuilder.class,
                BlitzBuilder.class,
                MapNotificationBuilder.class,
                FilterModuleBuilder.class,
                AppliedRegionBuilder.class,
                KillRewardBuilder.class,
                PortalBuilder.class,
                ClassModuleBuilder.class,
                TrackerBuilder.class,
                ScoreboxBuilder.class,
                BlockdropsBuilder.class,
                ProximityAlarmBuilder.class,
                /* BloodBuilder.class, */
                PermissionModuleBuilder.class,
                ChatChannelModuleBuilder.class,
                TntBuilder.class,
                MobModuleBuilder.class,
                DeathTrackerBuilder.class,
                SnowflakesBuilder.class,
                SoundModuleBuilder.class,
                TimersBuilder.class,
                HeaderModuleBuilder.class,
                TimeLimitBuilder.class,
                PlayableBuilder.class,
                TutorialBuilder.class,
                WildCardBuilder.class,
                CardinalNotificationsBuilder.class,
                PotionRemoverBuilder.class,
                InvisibleBlockBuilder.class,
                TitleRespawnBuilder.class,
                GuiKeepModuleBuilder.class,
                RankModuleBuilder.class,
                MultitradeBuilder.class,
                AttackSpeedBuilder.class,
                DamageIndicatorBuilder.class,
                ItemModsBuilder.class,
                FreezeModuleBuilder.class,
                CraftingModuleBuilder.class,
                PostBuilder.class,
                FlagBuilder.class,
                NetBuilder.class,
                StatsBuilder.class,
                FallingBlocksBuilder.class,
                SpectatorToolsBuilder.class,
                ProjectileParticlesModuleBuilder.class,
                FeedbackModuleBuilder.class
        );
    }

    public ModuleFactory() {
        this.builders = getBuilders().map(clazz -> {
            try {
                return clazz.getConstructor().newInstance();
            } catch (Exception e) {
                Bukkit.getLogger().log(Level.SEVERE, "Error enabling " + clazz.getSimpleName() + " as a ModuleBuilder.");
                e.printStackTrace();
            }
            return null;
        }).filter(Objects::nonNull)
                .sorted(Comparator.comparing(this::getLoadTime))
                .collect(Collectors.toList());
    }

    private ModuleLoadTime getLoadTime(ModuleBuilder builder) {
        LoadTime time = builder.getClass().getAnnotation(LoadTime.class);
        return time == null ? ModuleLoadTime.NORMAL : time.value();
    }

    public void buildModules(Match match) {
        builders.stream().flatMap(builder -> build(builder, match).stream())
                .peek(this::registerEvents).forEach(match.getModules()::add);
    }

    private ModuleCollection<Module> build(ModuleBuilder builder, Match match) {
        ModuleCollection<Module> results = new ModuleCollection<>();
        try {
            results.addAll(builder.load(match));
            if (Config.debug_displayModuleLoadMessages)
                Bukkit.getConsoleSender().sendMessage(ChatColor.GREEN + builder.getClass().getSimpleName() +
                        " was successfully built for " + match.getLoadedMap().toShortMessage(ChatColor.DARK_AQUA, true, true));
        } catch (Exception e) {
            Bukkit.getConsoleSender().sendMessage(ChatColor.RED + builder.getClass().getSimpleName() +
                    " failed to load for " + match.getLoadedMap().toShortMessage(ChatColor.DARK_AQUA, true, true));
            e.printStackTrace();
        }
        return results;
    }

    private void registerEvents(Module module) {
        try {
            Bukkit.getServer().getPluginManager().registerEvents(module, Cardinal.getInstance());
        } catch (IllegalStateException e) { // Happens if you try to register a listener more than once
        } catch (Exception e) {
            Bukkit.getConsoleSender().sendMessage(ChatColor.RED + this.getClass().getSimpleName() +
                    " failed to get registered as Listener");
            e.printStackTrace();
        }
    }

}