dev.mayuna.modularbot.ModularBot Maven / Gradle / Ivy

Go to download
package dev.mayuna.modularbot;

import dev.mayuna.consoleparallax.ConsoleParallax;
import dev.mayuna.mayusjdautils.MayusJDAUtilities;
import dev.mayuna.mayuslibrary.exceptionreporting.UncaughtExceptionReporter;
import dev.mayuna.modularbot.base.ModuleManager;
import dev.mayuna.modularbot.config.ModularBotConfig;
import dev.mayuna.modularbot.console.ModularBotOutputHandler;
import dev.mayuna.modularbot.console.ModularConsoleCommand;
import dev.mayuna.modularbot.console.StopConsoleCommand;
import dev.mayuna.modularbot.managers.DefaultModuleManager;
import dev.mayuna.modularbot.managers.ModularBotDataManager;
import dev.mayuna.modularbot.util.logging.ModularBotLogger;
import lombok.Getter;
import lombok.Setter;

import java.util.concurrent.Executors;

public final class ModularBot {

    private static final ModularBotLogger LOGGER = ModularBotLogger.create(ModularBot.class);

    private static @Getter ModularBotConfig config;
    private static @Getter ConsoleParallax consoleParallax;
    private static @Getter MayusJDAUtilities baseMayusJDAUtilities;
    private static @Getter ModuleManager moduleManager;
    private static @Getter ModularBotDataManager modularBotDataManager;
    private static @Getter ModularBotShardManager modularBotShardManager;

    private static @Getter @Setter boolean stopping;

    private ModularBot() {
    }

    static void start(String[] args) {
        LOGGER.info("Starting ModularDiscordBot @ {}", ModularBotConstants.getVersion());
        LOGGER.info("Made by Mayuna");

        // TODO: Print java info

        LOGGER.info("""
                            \s
                            \033[0;35m  __  __         _      _            ___  _                   _   ___      _  \s
                            \033[0;35m |  \\/  |___  __| |_  _| |__ _ _ _  |   \\(_)___ __ ___ _ _ __| | | _ ) ___| |_\s
                            \033[0;35m | |\\/| / _ \\/ _` | || | / _` | '_| | |) | (_- {
            if (!stopping) {
                LOGGER.warn("Modular Discord Bot has not stopped gracefully! Please, use command 'stop' to stop the application. There is a chance that the modules won't be unloaded fully before JVM termination.");

                LOGGER.info("Shutting down Modular Discord Bot...");
                shutdown();
            }
        }));
    }

    /**
     * Registers exception reporter
     */
    private static void registerUncaughtExceptionReporter() {
        UncaughtExceptionReporter.register();
        UncaughtExceptionReporter.addExceptionReportConsumer(exceptionReport -> {
            LOGGER.warn("Uncaught exception occurred! Sending to modules...", exceptionReport.getThrowable());
            moduleManager.processException(exceptionReport.getThrowable());
        });
    }

    /**
     * Loads {@link MayusJDAUtilities}
     */
    private static void loadMayusJDAUtilities() {
        baseMayusJDAUtilities = new MayusJDAUtilities();
        baseMayusJDAUtilities.setMessageInfoStyles(new ModularBotStyles(baseMayusJDAUtilities));
    }

    /**
     * Prepares module manager
     */
    private static void prepareModuleManager() {
        moduleManager = new DefaultModuleManager();
    }

    /**
     * Loads modules
     */
    private static void loadModules() {
        if (!moduleManager.loadModules()) {
            shutdown();
        }
    }

    /**
     * Loads DataManager
     */
    private static void loadDataManager() {
        modularBotDataManager = new ModularBotDataManager(config.getStorageSettings());

        LOGGER.mdebug("Preparing DataManager...");
        modularBotDataManager.prepareStorage();

        LOGGER.mdebug("Preparing GlobalDataHolder...");
        modularBotDataManager.getGlobalDataHolder();
    }

    /**
     * Enables modules
     */
    private static void enableModules() {
        moduleManager.enableModules();
    }

    /**
     * Initializes Discord stuff such as CommandClientBuilder, etc.
     */
    private static void initializeModules() {
        LOGGER.mdebug("Processing ConsoleParallax...");
        moduleManager.processConsoleParallax(consoleParallax);

        LOGGER.mdebug("Processing CommandClientBuilder...");
        moduleManager.processCommandClientBuilder(modularBotShardManager.getCommandClientBuilder());

        LOGGER.mdebug("Processing ShardManagerBuilder...");
        moduleManager.processShardBuilder(modularBotShardManager.getShardManagerBuilder());
    }

    /**
     * Creates ShardManager
     */
    private static void createModularBotShardManager() {
        modularBotShardManager = new ModularBotShardManager(config.getDiscord());

        LOGGER.mdebug("Initializing ModularBotShardManager...");
        if (!modularBotShardManager.init()) {
            shutdown();
        }
    }

    /**
     * Builds shard manager
     */
    private static void finishModularBotShardManager() {
        if (!modularBotShardManager.finish()) {
            shutdown();
        }
    }

    /**
     * Connects to Discord
     */
    private static void connectToDiscord() {
        if (!modularBotShardManager.connect()) {
            shutdown();
        }
    }

    /**
     * Initializes Presence Activity Cycle
     */
    private static void initializePresenceActivityCycle() {
        modularBotShardManager.initPresenceActivityCycle();
    }

    /**
     * Shutdowns ModularDiscordBot
     */
    public static void shutdown() {
        stopping = true;

        LOGGER.info("Shutting down ModularDiscordBot @ {}", ModularBotConstants.getVersion());

        LOGGER.info("Shutting down ConsoleParallax...");
        consoleParallax.interrupt();

        LOGGER.info("Unloading modules...");
        moduleManager.unloadModules();

        LOGGER.info("Disconnecting from Discord...");
        if (modularBotShardManager != null) {
            modularBotShardManager.shutdown();
        }

        LOGGER.success("Shutdown completed");
        Runtime.getRuntime().halt(0);
    }
}