X-Git-Url: http://git.pk910.de/?p=PHP-P10.git;a=blobdiff_plain;f=BotLoader%2FBotLoader.class.php;h=f7903d40e3762765a569d40351604afc9f5d23b0;hp=df06eeaed4d1f137aeb63ff3b6cecf3f5f45c5c6;hb=7f51193ddac01b0540000c756c50167ff015e02a;hpb=412e324075de1312fbc5f49b735180b46cdff82c diff --git a/BotLoader/BotLoader.class.php b/BotLoader/BotLoader.class.php index df06eea..f7903d4 100644 --- a/BotLoader/BotLoader.class.php +++ b/BotLoader/BotLoader.class.php @@ -1,12 +1,10 @@ . * * * ************************************************************************ - * + * * BotLoader/BotLoader.class.php * * This class loades / rehashs or unloads all Bots. @@ -29,18 +26,24 @@ require_once("Bot.class.php"); require_once("saxdb.class.php"); class BotLoader { + const BOT_DIR = "Bots"; + const TMP_DIR = "tmp"; private $uplink; private $saxdb; private $botdb; private $loadedBots = array(); - + private $botDatabases = array(); + private static $botloader; + public function __construct($uplink) { $this->uplink = $uplink; $this->saxdb = new saxdb(); $this->saxdb->loadDB("php_p10.db"); $this->botdb = $this->saxdb->getSection("BotLoader"); + self::$botloader = $this; + timer(60*10, array($this, "autosave"), array()); } - + public function loadBots() { if(array_key_exists("bots", $this->botdb) && is_array($this->botdb["bots"])) { foreach($this->botdb["bots"] as $name => $botfile) { @@ -50,15 +53,141 @@ class BotLoader { $this->botdb["bots"] = array(); } } - + + public function unloadBots() { + foreach($this->loadedBots as $name => $bot) { + $this->unloadBot($name, false); + } + } + + public function save() { + $this->saxdb->setSection("BotLoader", $this->botdb); + $this->saxdb->writeDB("php_p10.db"); + } + + public function autosave() { + foreach($this->loadedBots as $name => $bot) { + if($bot->getDBSection()) { + $db = $bot->writeDB(); + $this->saxdb->setSection($bot->getDBSection(),$db); + } + } + $this->save(); + timer(60*10, array($this, "autosave"), array()); + } + + public function loop() { + foreach($this->loadedBots as $name => $bot) { + $bot->loop(); + } + } + private function loadBot($name, $botfile) { - if(array_key_exists($name, $this->loadedBots)) return; + if(array_key_exists(strtolower($name), $this->loadedBots)) return false; //load bot + $bot = $this->loadClass($botfile, $name); + if(!$bot) return false; + $bot->load($this->uplink); + $this->loadedBots[strtolower($name)] = $bot; if(!array_key_exists($name, $this->botdb["bots"])) { $this->botdb["bots"][$name] = $botfile; } + return true; + } + + private function unloadBot($name, $delete = true) { + if(!(array_key_exists(strtolower($name), $this->loadedBots))) return false; + //unload bot + $bot = $this->loadedBots[strtolower($name)]; + if($bot->getDBSection()) { + $db = $bot->writeDB(); + $this->saxdb->setSection($bot->getDBSection(),$db); + } + ModCMD::unbindBot($bot); + $bot->unload(false); + unset($this->loadedBots[strtolower($name)]); + if(array_key_exists($name, $this->botdb["bots"]) && $delete) { + unset($this->botdb["bots"][$name]); + } + return true; + } + + private function rehashBot($name) { + if(!(array_key_exists(strtolower($name), $this->loadedBots))) return false; + $botfile = null; + foreach($this->botdb["bots"] as $botname => $file) { + if(strtolower($botname) == strtolower($name)) { + $botfile = $file; + break; + } + } + if(!$botfile) break; + //rehash bot + $bot = $this->loadedBots[strtolower($name)]; + if($bot->getDBSection()) { + $db = $bot->writeDB(); + $this->saxdb->setSection($bot->getDBSection(),$db); + } + ModCMD::unbindBot($bot); + $data = $bot->unload(true); + unset($this->loadedBots[strtolower($name)]); + //ok bot is unloaded... load it again... + $bot = $this->loadClass($botfile, $name); + if(!$bot) return false; + $bot->load($this->uplink, $data); + $this->loadedBots[strtolower($name)] = $bot; + return true; + } + + private function listLoadedBots() { + return $this->loadedBots; + } + + private function addDBsection($bot, $section) { + if(!is_a($bot, "Bot")) return false; + $bot->setDBSection($section); + $bot->readDB($this->saxdb->getSection($section)); + } + + private function loadClass($file, $classprefix) { + $dir = self::BOT_DIR; + $tmp = self::TMP_DIR; + $maincode=file_get_contents($dir."/".$file); + if(!$maincode) return; + $class = rand(10000,99999); + while(class_exists("bot_".$classprefix."_".$class)) { + $class = rand(10000,99999); + } + $maincode = str_replace('{$_NAME}', "module_".$classprefix."_".$class, $maincode); + $fp = fopen($tmp."/modules_".$classprefix."_".$class.".tmp.php", 'w'); + fwrite($fp, $maincode); + fclose($fp); + include($tmp."/modules_".$classprefix."_".$class.".tmp.php"); + $classname = "module_".$classprefix."_".$class; + $newclass = new $classname(); + unlink($tmp."/modules_".$classprefix."_".$class.".tmp.php"); + return $newclass; + } + + public static function load($name, $botfile) { + return self::$botloader->loadBot($name, $botfile); + } + + public static function unload($name) { + return self::$botloader->unloadBot($name); + } + + public static function rehash($name) { + return self::$botloader->rehashBot($name); + } + + public static function listBots() { + return self::$botloader->listLoadedBots(); + } + + public static function registerDB($bot, $name) { + return self::$botloader->addDBsection($bot, $name); } - } ?> \ No newline at end of file