private void connect() throws UnknownHostException, IOException {
IrcServer = new Socket(serverName, 6667);
BufferedReader br =
new BufferedReader(
new InputStreamReader(IrcServer.getInputStream(), BotStats.getInstance().getCharset()));
PrintWriter pw =
new PrintWriter(
new OutputStreamWriter(
IrcServer.getOutputStream(), BotStats.getInstance().getCharset()),
true);
ih = new InputHandler(br);
oh = new OutputHandler(pw);
ih.start();
oh.start();
new Message("", "NICK", BotStats.getInstance().getBotname(), "").send();
new Message(
"", "USER", "goat" + " nowhere.com " + serverName, BotStats.getInstance().getVersion())
.send();
// we sleep until we are connected, don't want to send these next messages too soon
while (!connected) {
try {
sleep(100);
} catch (InterruptedException e) {
e.printStackTrace();
}
}
joinChannels();
}
public Module getLoaded(String modName) {
List<Module> mods = bot.getModules();
for (Module mod : mods) {
if (mod.moduleName.equals(modName)) {
return mod;
}
}
return null;
}
public Module getLoaded(Class<?> modClass) {
Iterator<Module> it = bot.getModules().listIterator();
Module ret = null;
if (!modClass.equals(goat.core.Module.class) && !modClass.equals(Object.class))
while (it.hasNext()) {
Module mod = it.next();
if (modClass.isInstance(mod)) ret = mod;
}
return ret;
}
public boolean unload(String moduleName) {
Module mod = getLoaded(moduleName);
if (null == mod) return false;
else {
bot.removeModule(mod);
mod.stopDispatcher();
}
return true;
}
private void bootModule(Module module) {
if (null == module) return;
pool.execute(module);
while (!module
.isRunning()) { // wait to make sure module is running before adding it to loaded module
// list
try {
Thread.sleep(
5); // short wait, it shouldn't take too long for the thread pool to start the module
} catch (InterruptedException ie) {
}
}
System.out.print("running ... ");
bot.addModule(module);
bot.addCommands(module.getCommands());
System.out.println("loaded.");
}
/**
* Build the list of all public classes in package goat.module.
*
* <p>This gets sort of ugly. It sure would be nice if the java reflection API could give us a
* list of all classes in a package, wouldn't it?
*/
private void buildAllModulesList() {
if (bot.isTesting()) return;
Reflections reflections = new Reflections("goat");
Set<Class<? extends Module>> allModules = reflections.getSubTypesOf(Module.class);
ArrayList<String> tempList = new ArrayList<String>();
for (Class<? extends Module> modClass : allModules) {
tempList.add(modClass.getName());
}
Collections.sort(tempList);
System.out.println("Available Modules: ");
for (String modName : tempList) {
System.out.println(" " + modName);
}
System.out.println();
}
private void joinChannels() {
String[] channels = BotStats.getInstance().getChannels();
for (String channel : channels) new Message("", "JOIN", channel, "").send();
}
public void run() {
String messageString;
long lastActivity = System.currentTimeMillis();
while (keeprunning) {
try {
if (in.ready()) {
messageString = in.readLine();
if (messageString == null) continue;
Message m = new Message(messageString);
if (m.getCommand().matches("\\d+"))
try {
int intcommand = Integer.parseInt(m.getCommand());
if (intcommand == Constants.RPL_ENDOFMOTD) {
System.err.println("MOTD SEEN, must be connected.");
if (connected) joinChannels();
connected = true;
} else if (intcommand == Constants.ERR_NICKNAMEINUSE) {
System.err.println("NICKNAMEINUSE");
namecount++;
BotStats.getInstance().setBotname(botdefaultname + namecount);
new Message("", "NICK", BotStats.getInstance().getBotname(), "").send();
new Message(
"",
"USER",
BotStats.getInstance().getBotname()
+ " nowhere.com "
+ BotStats.getInstance().getServername(),
BotStats.getInstance().getClientName()
+ " v."
+ BotStats.getInstance().getVersion())
.send();
System.err.println("Setting nick to:" + botdefaultname + namecount);
}
} catch (NumberFormatException nfe) {
// we ignore this
System.err.println("Unknown command:" + m.getCommand());
}
if (m.getCommand().equals("PING")) {
outqueue.add(new Message("", "PONG", "", m.getTrailing()));
if (debug) System.out.println("PUNG at " + new Date());
} else if (BotStats.getInstance().containsIgnoreName(m.getSender())) {
if (debug) System.out.println("Ignored: " + m);
} else {
inqueue.add(m); // add to inqueue
if (debug) System.out.println(m.toString());
// System.out.println("Inbuffer: prefix: " + m.prefix + " params: " + m.params + "
// trailing:"
// + m.trailing + " command:" + m.command + " sender: " + m.sender + "\n "
// + "isCTCP:" + m.isCTCP + " isPrivate:" + m.isPrivate + " CTCPCommand:" +
// m.CTCPCommand
// + " CTCPMessage:" + m.CTCPMessage);
}
lastActivity = System.currentTimeMillis();
} else {
if (System.currentTimeMillis() - lastActivity > 400000) {
System.err.println("400 seconds since last activity! Attempting reconnect");
in.close();
oh.disconnect();
keeprunning = false;
reconnect();
return;
}
sleep(100);
}
} catch (IOException ioe) {
System.out.println("EOF on connection: " + ioe.getMessage());
} catch (InterruptedException ie) {
System.out.println("Interrupted: " + ie.getMessage());
} catch (Exception e) {
System.err.println("Unexpected exception in InputHandler :");
e.printStackTrace();
}
}
}
/**
* Loads and unloads Modules. Provides methods allowing you to find out what Modules are loaded, to
* control the Modules, and to return them.
*
* @version Date: 17-Dec-2003
* @author
*/
public class ModuleController {
private ExecutorService pool = Executors.newCachedThreadPool();
public ExecutorService getPool() {
return pool;
}
// private ArrayList<Module> loadedModules = new ArrayList<Module>();
private ArrayList<Class<? extends Module>> allModules = new ArrayList<Class<? extends Module>>();
// private ArrayList<String> allCommands = new ArrayList<String>() ;
private ArrayList<File> allScriptModules = new ArrayList<File>();
private BotStats bot = BotStats.getInstance();
public ModuleController() {
buildAllModulesList();
buildScriptModulesList();
}
/**
* Loads a module.
*
* @param moduleName
* @return
* @throws IllegalAccessException
* @throws InstantiationException
* @throws ClassNotFoundException
* @throws IOException
* @throws ScriptException
* @throws NoSuchMethodException
* @throws NoClassDefFoundError
* @throws ClassCastException
*/
public Module load(String moduleName)
throws ClassNotFoundException, IllegalAccessException, InstantiationException, IOException,
NoSuchMethodException, ScriptException {
if (moduleName.contains(".")) {
// if the name has a dot, assume a script
System.out.print("Loading Module " + moduleName + " ... ");
File file = new File(scriptDir(moduleName) + File.separatorChar + moduleName);
BufferedReader bis = new BufferedReader(new FileReader(file));
String line;
String script = "";
while ((line = bis.readLine()) != null) {
script += line + "\n";
}
String extension = moduleName.replaceAll(".*\\.", "");
ScriptEngine engine = new ScriptEngineManager().getEngineByExtension(extension);
engine.eval(script);
Invocable inv = (Invocable) engine;
Module mod;
Object moduleCandidate = inv.invokeFunction("getInstance");
if (moduleCandidate instanceof Module) {
mod = (Module) moduleCandidate;
} else {
mod = new JSR223Module(extension, script);
}
mod.moduleName = moduleName;
bootModule(mod);
return mod;
} else {
moduleName = "goat.module." + moduleName;
return load(Class.forName(moduleName));
}
}
public Module loadInAllChannels(String moduleName)
throws ClassNotFoundException, IllegalAccessException, InstantiationException, IOException,
NoSuchMethodException, ScriptException {
Module ret = load(moduleName);
if (null == ret) // wasn't loaded
ret = getLoaded(moduleName);
ret.inAllChannels = true;
return ret;
}
public Module load(Class<?> modClass) throws IllegalAccessException, InstantiationException {
if (null == modClass) return null;
System.out.print("Loading Module " + modClass.getName() + " ... ");
Module module = null;
// return null if module is already loaded
if (null != getLoaded(modClass)) return null;
module = (Module) modClass.newInstance();
bootModule(module);
module.moduleName = modClass.getName().replace("goat.module.", "");
return module;
}
private void bootModule(Module module) {
if (null == module) return;
pool.execute(module);
while (!module
.isRunning()) { // wait to make sure module is running before adding it to loaded module
// list
try {
Thread.sleep(
5); // short wait, it shouldn't take too long for the thread pool to start the module
} catch (InterruptedException ie) {
}
}
System.out.print("running ... ");
bot.addModule(module);
bot.addCommands(module.getCommands());
System.out.println("loaded.");
}
public Module loadInAllChannels(Class<?> modClass)
throws IllegalAccessException, InstantiationException {
Module ret = load(modClass);
if (null != ret) ret.inAllChannels = true;
return ret;
}
public boolean unload(String moduleName) {
Module mod = getLoaded(moduleName);
if (null == mod) return false;
else {
bot.removeModule(mod);
mod.stopDispatcher();
}
return true;
}
public Module getLoaded(int i) {
return bot.getModules().get(i);
}
public Module getLoaded(Class<?> modClass) {
Iterator<Module> it = bot.getModules().listIterator();
Module ret = null;
if (!modClass.equals(goat.core.Module.class) && !modClass.equals(Object.class))
while (it.hasNext()) {
Module mod = it.next();
if (modClass.isInstance(mod)) ret = mod;
}
return ret;
}
public Module getLoaded(String modName) {
List<Module> mods = bot.getModules();
for (Module mod : mods) {
if (mod.moduleName.equals(modName)) {
return mod;
}
}
return null;
}
/**
* Build the list of all public classes in package goat.module.
*
* <p>This gets sort of ugly. It sure would be nice if the java reflection API could give us a
* list of all classes in a package, wouldn't it?
*/
private void buildAllModulesList() {
if (bot.isTesting()) return;
Reflections reflections = new Reflections("goat");
Set<Class<? extends Module>> allModules = reflections.getSubTypesOf(Module.class);
ArrayList<String> tempList = new ArrayList<String>();
for (Class<? extends Module> modClass : allModules) {
tempList.add(modClass.getName());
}
Collections.sort(tempList);
System.out.println("Available Modules: ");
for (String modName : tempList) {
System.out.println(" " + modName);
}
System.out.println();
}
public final String pyModDir =
"src" + File.separatorChar + "main" + File.separatorChar + "python";
/** Looks in script dir(s) and assumes any files in there are modules. */
private void buildScriptModulesList() {
// FIXME: only python mods dir atm, should include anything that
// scriptDir() might return
String path = pyModDir;
File folder = new File(path);
File[] listOfFiles = folder.listFiles();
System.out.println("Available script modules:");
for (File scriptModuleFile : listOfFiles) {
System.out.println(" " + scriptModuleFile.getName());
allScriptModules.add(scriptModuleFile);
}
}
public List<Class<? extends Module>> getAllModules() {
return allModules;
}
public String scriptDir(String name) {
String ret = "scripts";
if (name.endsWith(".py")) ret = pyModDir;
return ret;
}
}
public Module getLoaded(int i) {
return bot.getModules().get(i);
}
