com.kantenkugel.discordbot.Main.java Source code

Introduction

Here is the source code for com.kantenkugel.discordbot.Main.java

Source

/*
 * Copyright 2016 Michael Ritter (Kantenkugel)
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.kantenkugel.discordbot;

import com.kantenkugel.discordbot.commands.CommandRegistry;
import com.kantenkugel.discordbot.config.BotConfig;
import com.kantenkugel.discordbot.listener.DbListener;
import com.kantenkugel.discordbot.listener.InviteListener;
import com.kantenkugel.discordbot.listener.MessageListener;
import com.kantenkugel.discordbot.listener.StatusListener;
import com.kantenkugel.discordbot.modules.Module;
import com.kantenkugel.discordbot.moduleutils.DocParser;
import com.kantenkugel.discordbot.util.UpdateValidator;
import com.kantenkugel.discordbot.util.UpdateWatcher;
import net.dv8tion.jda.JDABuilder;
import net.dv8tion.jda.events.Event;
import net.dv8tion.jda.events.ReadyEvent;
import net.dv8tion.jda.hooks.EventListener;
import net.dv8tion.jda.utils.SimpleLog;
import org.apache.commons.lang3.StringUtils;

import javax.security.auth.login.LoginException;
import java.io.File;
import java.io.IOException;

/**
 * Created by Michael Ritter on 05.12.2015.
 */
public class Main {
    public static UpdateValidator checker = null;

    /*
    Args:
    0  bot-token
    1  system-time of wrapper-start (for uptime),
       OR:
       "true" -> 2nd bot for only db (rest of params ignored)
    2  success-indicator (true/false/"-")
    3  version-number
    4+ MULTIPLE/NONE strings describing the changelog of this version
    */
    public static void main(String[] args) {
        boolean isDbBot = false;
        if (args.length == 2 && Boolean.parseBoolean(args[1])) {
            isDbBot = true;
        } else if (args.length < 4) {
            System.out.println("Missing arguments!");
            return;
        }

        if (!BotConfig.load()) {
            System.out.println("Bot config created/updated. Please populate/check it before restarting the Bot");
            System.exit(Statics.NORMAL_EXIT_CODE);
        }

        if (BotConfig.get("logToFiles", true)) {
            try {
                SimpleLog.addFileLogs(new File("logs/main.txt"), new File("logs/err.txt"));
            } catch (IOException e) {
                e.printStackTrace();
            }
        }

        if (!isDbBot) {
            Statics.START_TIME = Long.parseLong(args[1]);
            Statics.VERSION = Integer.parseInt(args[3]);
        }

        if (!isDbBot && args.length > 4) {
            Statics.CHANGES = StringUtils.join(args, '\n', 4, args.length);
        } else {
            Statics.CHANGES = null;
        }

        if (isDbBot) {
            if ("".equals(BotConfig.get("historyBase"))) {
                Statics.LOG.fatal("Please specify a history-base in the config");
                System.exit(Statics.NORMAL_EXIT_CODE);
            }
            if (!DbEngine.init()) {
                Statics.LOG.fatal("Could not connect to db! shutting down");
                System.exit(Statics.NORMAL_EXIT_CODE);
            }
        } else {
            DocParser.init();
            Module.init();
        }
        try {
            JDABuilder jdaBuilder = new JDABuilder().setBotToken(args[0]).setAudioEnabled(false);
            if (isDbBot)
                jdaBuilder.addListener(new DbListener());
            else
                jdaBuilder.addListener(new StatusListener()).addListener(new InviteListener())
                        .addListener(new MessageListener());
            if (!isDbBot && !args[2].equals("-")) {
                boolean success = Boolean.parseBoolean(args[2]);
                if (success) {
                    checker = UpdateValidator.getInstance();
                    checker.start();
                }
                jdaBuilder.addListener(new UpdatePrintListener(success));
            }
            Statics.jdaInstance = jdaBuilder.buildAsync();
            if (!isDbBot)
                CommandRegistry.loadCommands(Statics.jdaInstance);
            new UpdateWatcher(Statics.jdaInstance);
        } catch (LoginException e) {
            Statics.LOG.fatal("Login informations were incorrect!");
            System.err.flush();
        }
    }

    private static class UpdatePrintListener implements EventListener {
        private final boolean success;

        private UpdatePrintListener(boolean success) {
            this.success = success;
        }

        @Override
        public void onEvent(Event event) {
            if (event instanceof ReadyEvent) {
                if (checker != null) {
                    checker.interrupt();
                }
                UpdateWatcher.getChannel(event.getJDA()).sendMessage("Update was " + (success ? "" : "**NOT**")
                        + "successful!\nCurrent revision: " + Statics.VERSION);
                if (Statics.CHANGES != null) {
                    UpdateWatcher.getChannel(event.getJDA())
                            .sendMessage("Changes for this revision:\n" + Statics.CHANGES);
                }
                event.getJDA().removeEventListener(this);
            }
        }
    }

}