me.gloriouseggroll.quorrabot.script.ScriptEventManager.java Source code

Introduction

Here is the source code for me.gloriouseggroll.quorrabot.script.ScriptEventManager.java

Source

/* 
 * Copyright (C) 2017 www.quorrabot.com
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
package me.gloriouseggroll.quorrabot.script;

import com.google.common.collect.Lists;
import com.google.common.eventbus.Subscribe;
import java.util.Iterator;
import java.util.List;
import java.util.concurrent.ConcurrentHashMap;
import me.gloriouseggroll.quorrabot.Quorrabot;
import me.gloriouseggroll.quorrabot.event.Event;
import me.gloriouseggroll.quorrabot.event.Listener;
import org.apache.commons.lang3.text.WordUtils;

public class ScriptEventManager implements Listener {

    private static final ScriptEventManager instance = new ScriptEventManager();

    public static ScriptEventManager instance() {
        return instance;
    }

    private static final String[] eventPackages = new String[] { "me.gloriouseggroll.quorrabot.event.command",
            "me.gloriouseggroll.quorrabot.event.console", "me.gloriouseggroll.quorrabot.event.discord",
            "me.gloriouseggroll.quorrabot.event.gamewisp", "me.gloriouseggroll.quorrabot.event.irc",
            "me.gloriouseggroll.quorrabot.event.irc.channel", "me.gloriouseggroll.quorrabot.event.irc.clearchat",
            "me.gloriouseggroll.quorrabot.event.irc.complete", "me.gloriouseggroll.quorrabot.event.irc.message",
            "me.gloriouseggroll.quorrabot.event.musicplayer", "me.gloriouseggroll.quorrabot.event.twitch",
            "me.gloriouseggroll.quorrabot.event.twitch.bits", "me.gloriouseggroll.quorrabot.event.twitch.follower",
            "me.gloriouseggroll.quorrabot.event.twitch.host",
            "me.gloriouseggroll.quorrabot.event.twitch.subscriber" };

    private ScriptEventManager() {
        Thread.setDefaultUncaughtExceptionHandler(com.gmt2001.UncaughtExceptionHandler.instance());
    }

    private static class EventHandlerEntry {

        Class<? extends Event> eventClass;
        ScriptEventHandler handler;

        private EventHandlerEntry(Class<? extends Event> eventClass, ScriptEventHandler handler) {
            this.eventClass = eventClass;
            this.handler = handler;
        }
    }

    private final List<EventHandlerEntry> entries = Lists.newCopyOnWriteArrayList();
    private final ConcurrentHashMap<String, EventHandlerEntry> hashEntries = new ConcurrentHashMap<String, EventHandlerEntry>();

    public void runDirect(Event event) {
        if (Quorrabot.instance().isExiting()) {
            return;
        }

        try {
            EventHandlerEntry entry = hashEntries.get(event.getClass().getName());
            if (entry != null) {
                entry.handler.handle(event);
                com.gmt2001.Console.out.println("Dispatched runDirect event " + entry.eventClass.getName());
            }
        } catch (Exception e) {
            com.gmt2001.Console.err.println("Failed to dispatch runDirect event " + event.getClass().getName());
            com.gmt2001.Console.err.printStackTrace(e);
        }
    }

    @Subscribe
    public void onEvent(Event event) {
        if (Quorrabot.instance().isExiting()) {
            return;
        }

        try {
            for (EventHandlerEntry entry : entries) {
                if (event.getClass().isAssignableFrom(entry.eventClass)) {
                    if (Quorrabot.enableDebugging) {
                        com.gmt2001.Console.out
                                .println(">>>[DEBUG] Dispatching event " + entry.eventClass.getName());
                    }

                    entry.handler.handle(event);
                }
            }
        } catch (Exception e) {
            com.gmt2001.Console.out.println(">>>[DEBUG] Failed to dispatch event " + event.getClass().getName());
            com.gmt2001.Console.err.printStackTrace(e);
        }
    }

    public void register(String eventName, ScriptEventHandler handler) {
        Class<? extends Event> eventClass = null;
        for (String eventPackage : eventPackages) {
            try {
                eventClass = Class.forName(eventPackage + "." + WordUtils.capitalize(eventName) + "Event")
                        .asSubclass(Event.class);
                break;
            } catch (ClassNotFoundException e) {
            }
        }

        if (eventClass == null) {
            throw new RuntimeException("Event class not found: " + eventName);
        }

        entries.add(new EventHandlerEntry(eventClass, handler));
    }

    public void unregister(ScriptEventHandler handler) {
        EventHandlerEntry entry;
        Iterator<EventHandlerEntry> iterator = entries.iterator();
        while (iterator.hasNext()) {
            entry = iterator.next();

            if (entry.handler == handler) {
                entries.remove(entry);
            }
        }
    }
}