Main.java

package tterrag.tppibot;

import static tterrag.tppibot.util.Logging.*;

import java.io.File;
import java.io.FileOutputStream;
import java.io.PrintStream;
import java.nio.charset.Charset;
import java.util.Arrays;
import java.util.Timer;
import java.util.TimerTask;

import lombok.SneakyThrows;

import org.apache.commons.cli.BasicParser;
import org.apache.commons.cli.CommandLine;
import org.apache.commons.cli.CommandLineParser;
import org.apache.commons.cli.Option;
import org.apache.commons.cli.OptionBuilder;
import org.apache.commons.cli.Options;
import org.apache.commons.cli.ParseException;
import org.apache.commons.io.output.TeeOutputStream;
import org.pircbotx.Configuration;
import org.pircbotx.PircBotX;
import org.pircbotx.hooks.events.DisconnectEvent;

import tterrag.tppibot.commands.*;
import tterrag.tppibot.listeners.EventProxy;
import tterrag.tppibot.listeners.JoinListener;
import tterrag.tppibot.listeners.MessageListener;
import tterrag.tppibot.listeners.PrivateMessageListener;
import tterrag.tppibot.reactions.BannedWords;
import tterrag.tppibot.reactions.CharacterSpam;
import tterrag.tppibot.reactions.FloodSpam;
import tterrag.tppibot.registry.EventHandler;
import tterrag.tppibot.registry.PermRegistry;
import tterrag.tppibot.registry.ReactionRegistry;
import tterrag.tppibot.registry.WhoisCache;
import tterrag.tppibot.runnables.ConsoleCommands;
import tterrag.tppibot.runnables.MessageSender;
import tterrag.tppibot.runnables.ReminderProcess;
import tterrag.tppibot.runnables.TimeoutChecker;
import tterrag.tppibot.util.IRCUtils;
import tterrag.tppibot.util.Logging;

import com.google.gson.Gson;
import com.google.gson.GsonBuilder;

public class Main {

    public static Reminders reminderCommand;

    public static ReminderProcess reminders;
    public static TimeoutChecker timeouts;

    public static CharacterSpam spamFilter;
    public static FloodSpam floodFilter;
    public static BannedWords bannedWords;

    public static PircBotX bot;

    public static String overrideFile;

    public static int autoSaveRateSeconds;

    public static final Gson gson = new GsonBuilder().setPrettyPrinting().create();

    @SneakyThrows
    public static void main(String[] args) throws ParseException {
        Options options = new Options();

        options.addOption("n", "name", true, "The nick/login of the bot");
        options.addOption("d", "dataDir", true, "The directory in which to create the .tppibot folder");
        options.addOption("p", "password", true, "Nickserv password for the bot");
        options.addOption("a", "autoSaveInterval", true, "Interval at which to save data automatically");

        OptionBuilder.withArgName("channel");
        OptionBuilder.hasArgs();
        OptionBuilder.withDescription("The channels to join on startup");
        Option channels = OptionBuilder.create("channels");
        options.addOption(channels);

        CommandLineParser parser = new BasicParser();
        CommandLine cmd = parser.parse(options, args);

        overrideFile = cmd.getOptionValue("dataDir");

        System.setErr(new PrintStream(new TeeOutputStream(new FileOutputStream(new File(Logging.logsDir, "pircbotx-latest.log")), System.err)));

        log("Starting...");

        // create base commands
        log("Creating commands...");
        new Help();
        new Kill();
        new Join();
        new EditCommand();
        new AddReminder();
        reminderCommand = new Reminders();
        new Topic();
        new Commands();
        new AddCommand();
        new RemoveCommand();
        new Perms();
        new Leave();
        new ToggleSpamFilters();
        new Forgive();
        new Recover();
        new Mode();
        new HTML();
        new Shortener();
        new Say();
        new Drama();
        new BanWord();
        new Nick();
        new UUID();
        new Strikes();
        new Rekt();
        new Tell();

        Timeout timeout = new Timeout();
        log("Commands created.");

        log("Creating reactions...");
        spamFilter = new CharacterSpam();
        ReactionRegistry.INSTANCE.registerReaction(spamFilter);
        floodFilter = new FloodSpam();
        ReactionRegistry.INSTANCE.registerReaction(floodFilter);
        bannedWords = new BannedWords();
        ReactionRegistry.INSTANCE.registerReaction(bannedWords);
        log("Reactions created.");

        log("Configuring bot...");
        Configuration.Builder<PircBotX> builder = new Configuration.Builder<PircBotX>();
        builder.setName(cmd.getOptionValue("name"));
        builder.setLogin(cmd.getOptionValue("name"));
        builder.setNickservPassword(cmd.getOptionValue("password"));
        builder.setEncoding(Charset.isSupported("UTF-8") ? Charset.forName("UTF-8") : Charset.defaultCharset());
        builder.setServer("irc.esper.net", 6667);
        builder.setAutoReconnect(true);

        Arrays.stream(cmd.getOptionValues("channels")).forEach(s -> builder.addAutoJoinChannel(IRCUtils.fmtChan(s)));

        builder.getListenerManager().addListener(MessageListener.instance);
        builder.getListenerManager().addListener(new JoinListener());
        builder.getListenerManager().addListener(new EventProxy());
        builder.getListenerManager().addListener(new PrivateMessageListener());

        bot = new PircBotX(builder.buildConfiguration());
        log("Configured.");

        // create and start threads
        log("Creating threads...");
        reminders = new ReminderProcess(bot,

        "[Reminder] You can open the chat and press tab to talk with us!", "[Reminder] Rules: Avoid swearing - No ETA requests - No modlist requests - Don't advertise - Use common sense.");

        Thread reminderThread = new Thread(reminders);
        reminderThread.start();

        timeouts = new TimeoutChecker(timeout);
        Thread timeoutThread = new Thread(timeouts);
        timeoutThread.start();

        Thread consoleThread = new Thread(new ConsoleCommands());
        consoleThread.start();

        Thread messageSenderThread = new Thread(MessageSender.INSTANCE);
        messageSenderThread.start();

        Timer timer = new Timer();
        String opt = cmd.getOptionValue("autoSaveInterval");
        long saveRate = 1000 * (opt == null ? 60 * 5 : IRCUtils.getSecondsFromString(cmd.getOptionValue("autoSaveInterval")));
        timer.scheduleAtFixedRate(new TimerTask() {

            @Override
            public void run() {
                Logging.log("Sending dummy DisconnectEvent for autosave");
                EventHandler.INSTANCE.post(new DisconnectEvent<PircBotX>(bot, bot.getUserChannelDao().createSnapshot(), null));
            }
        }, saveRate, saveRate);

        log("Threads created...");

        log("Registering extra event receivers...");
        EventHandler.INSTANCE.registerReceiver(reminders);
        EventHandler.INSTANCE.registerReceiver(PermRegistry.INSTANCE);
        EventHandler.INSTANCE.registerReceiver(spamFilter);
        EventHandler.INSTANCE.registerReceiver(bannedWords);
        EventHandler.INSTANCE.registerReceiver(WhoisCache.INSTANCE);
        log("Registered extra event reveivers.");

        // start 'er up
        try {
            log("Connecting bot...");
            bot.startBot();
        } catch (Exception e) {
            throw new RuntimeException(e);
        }
    }
}