App.java

/*
 * The MIT License
 *
 * Copyright 2012 Jason Unger <entityreborn@gmail.com>.
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 */
package com.entityreborn.socpuppet;

import com.entityreborn.socpuppet.extensions.builtins.BuiltinListener;
import com.entityreborn.socbot.UserFactory;
import com.entityreborn.socbot.eventsystem.EventManager;
import com.entityreborn.socpuppet.config.BotConfig;
import com.entityreborn.socpuppet.config.ConnectionConfig;
import com.entityreborn.socpuppet.console.ConsoleManager;
import com.entityreborn.socpuppet.extensions.ExtensionManager;
import com.entityreborn.socpuppet.users.SocPuppetUser;
import com.entityreborn.socpuppet.util.ShutdownMonitorThread;
import com.laytonsmith.PureUtilities.ClassLoading.ClassDiscovery;
import com.laytonsmith.PureUtilities.ClassLoading.ClassDiscoveryCache;
import java.io.File;
import java.io.IOException;
import java.net.URL;
import java.util.HashMap;
import java.util.Map;

/**
 *
 * @author Jason Unger <entityreborn@gmail.com>
 */
public class App {
    private static App instance;
    
    public synchronized static App getInstance() {
        if (instance == null) {
            instance = new App();
        }
        
        return instance;
    }
    
    static String VERSION = "";
    private static final ConsoleManager consoleManager = ConsoleManager.getInstance();
    private static final Map<String, SocPuppet> bots = new HashMap<>();
    
    static {
        Package p = App.class.getPackage();

        if (p == null) {
            p = Package.getPackage("com.entityreborn.socpuppet");
        }

        if (p == null) {
            VERSION = "(unknown)";
        } else {
            String v = p.getImplementationVersion();

            if (v == null) {
                VERSION = "(unknown)";
            } else {
                VERSION = v;
            }
        }
    }
    
    private App() {
    }
    
    public static Map<String, SocPuppet> getBots() {
        return bots;
    }
    
    public static void shutdown() {
        System.out.println("Shutting down bots...");
        for (SocPuppet bot : bots.values()) {
            bot.quit();
        }
        
        bots.clear();
        
        ShutdownMonitorThread shutdown = new ShutdownMonitorThread(new Runnable() {
            @Override
            public void run() {
                System.out.println("Bye!");
            }
        });
        
        shutdown.start();
    }

    public static void main(String[] args) throws Exception {
        consoleManager.startConsole(true);
        
        BotConfig c = BotConfig.get();
        c.load();
        
        File cachedir = new File(".cache");
        cachedir.mkdirs();
        File extcache = new File(cachedir, "extensions");
        extcache.mkdirs();
        
        URL thisurl = ClassDiscovery.GetClassContainer(App.class);
        ExtensionManager em = ExtensionManager.Get();
        
        for (File dir : c.getDirectories("plugins")) {
            em.addDiscoveryLocation(dir);
        }
        
        // Cache extensions!
        
        ClassDiscoveryCache cache = new ClassDiscoveryCache(cachedir);
        ClassDiscovery cd = new ClassDiscovery();
        cd.setClassDiscoveryCache(cache);
        cd.addDiscoveryLocation(thisurl);
        
        em.cache(extcache, cachedir);
        
        // Load the cached extensions!
        
        cache = new ClassDiscoveryCache(cachedir);
        cd = ClassDiscovery.getDefaultInstance();
        cd.setClassDiscoveryCache(cache);
        cd.addDiscoveryLocation(thisurl);
        
        em.initialize(extcache, cd);
        em.startup();
        
        UserFactory factory = new SocPuppetUser.Factory();
        
        for (String connname : c.getConnectionNames()) {
            ConnectionConfig conn = c.getConnection(connname);
            
            if (!conn.isActive()) {
                continue;
            }
            
            final SocPuppet bot = new SocPuppet(conn, connname);
            bots.put(connname, bot);
            
            bot.setUserFactory(factory);
            EventManager.registerEvents(new BuiltinListener(c), bot);
            
            String nickname = conn.getNickname();
            bot.setNickname(nickname);
            
            final String server = conn.getServer();
            final int port = conn.getPort();
            final String password = conn.getPassword();
            
            Thread t = new Thread() {
                @Override
                public void run() {
                    boolean success = false;
                    
                    while (!success) {
                        try {
                            bot.connect(server, port, password);
                            success = true;
                        } catch (IOException ex) {
                            System.out.println("Retrying connection. " + ex.getLocalizedMessage());
                        }
                    }
                }
            };
            
            t.setName(connname + "-connector");
            t.start();
        }
    }
}