Gazelle/classes/class_irc.php

<?
class IRC_DB extends DB_MYSQL {
	function halt($Msg) {
		global $Bot;
		global $DB;
		$Bot->send_to($Bot->get_channel(),'The database is currently unavailable try again later');
	}
}

abstract class IRC_BOT {
	abstract protected function connect_events();
	abstract protected function channel_events();
	abstract protected function query_events();
	abstract protected function listener_events();

	protected $Debug = false;
	protected $Socket = false;
	protected $Data = false;
	protected $Whois = false;
	protected $Identified = array();
	protected $Channels = array();
	protected $Messages = array();
	protected $LastChan = false;
	protected $ListenSocket =false;
	protected $Listened = false;
	protected $Connecting = false;
	protected $Bound = false; //Did we successfully bind to the socket?
	protected $State = 1; //Drones live
	public $Restart = 0; //Die by default

	public function __construct() {
		//ini_set('memory_limit', '12M');
		restore_error_handler(); //Avoid PHP error logging
		set_time_limit(0);
	}

	public function connect() {
		$this->connect_irc();
		$this->connect_listener();
		$this->post_connect();
	}

	public function connect_irc($Reconnect = false) {
		$this->Connecting = true;
		//Open a socket to the IRC server
		while (!$this->Socket = fsockopen('tls://'.BOT_SERVER, BOT_PORT_SSL)) {
			sleep(15);
		}
		stream_set_blocking($this->Socket, 0);
		$this->Connecting = false;
		if ($Reconnect) {
			$this->post_connect();
		}
	}

	public function connect_listener() {
		//create a socket to listen on
		$this->ListenSocket = socket_create(AF_INET, SOCK_STREAM, SOL_TCP);
		//socket_set_option($this->ListenSocket, SOL_TCP, SO_REUSEADDR, 1);
		socket_set_option($this->ListenSocket, SOL_SOCKET, SO_REUSEADDR, 1);
		$this->Bound = socket_bind($this->ListenSocket, SOCKET_LISTEN_ADDRESS, SOCKET_LISTEN_PORT);
		socket_listen($this->ListenSocket);
		socket_set_nonblock($this->ListenSocket);
	}

	public function post_connect() {
		fwrite($this->Socket, "NICK ".BOT_NICK."Init\n");
		fwrite($this->Socket, "USER ".BOT_NICK." * * :IRC Bot\n");
		$this->listen();
	}

 	public function disconnect() {
		socket_close($this->ListenSocket);
		$this->State = 0; //Drones dead
	}

	public function get_channel() {
		preg_match('/.+ PRIVMSG ([^:]+) :.+/', $this->Data, $Channel);
		if(preg_match('/#.+/',$Channel[1])) {
			return $Channel[1];
		} else {
			return false;
		}
	}

	public function get_nick() {
		preg_match('/:([^!:]+)!.+@[^\s]+ PRIVMSG [^:]+ :.+/', $this->Data, $Nick);
		return $Nick[1];
	}

	protected function get_message() {
		preg_match('/:.+ PRIVMSG [^:]+ :(.+)/', $this->Data, $Msg);
		return trim($Msg[1]);
	}

	protected function get_irc_host() {
		preg_match('/:[^!:]+!.+@([^\s]+) PRIVMSG [^:]+ :.+/', $this->Data, $Host);
		return trim($Host[1]);
	}

	protected function get_word($Select=1) {
		preg_match('/:.+ PRIVMSG [^:]+ :(.+)/', $this->Data, $Word);
		$Word = split(' ',$Word[1]);
		return trim($Word[$Select]);
	}

	protected function get_action() {
		preg_match('/:.+ PRIVMSG [^:]+ :!(\S+)/', $this->Data, $Action);
		return strtoupper($Action[1]);
	}

	protected function send_raw($Text) {
		if (!feof($this->Socket)) {
			fwrite($this->Socket, $Text."\n");
		} elseif (!$this->Connecting) {
			$this->Connecting = true;
			sleep(120);
			$this->connect_irc(true);
		}
	}

	public function send_to($Channel, $Text) {
		$this->send_raw("PRIVMSG $Channel :$Text");
	}

	protected function whois($Nick) {
		$this->Whois = $Nick;
		$this->send_raw("WHOIS $Nick");
	}

	/*
	This function uses blacklisted_ip, which is no longer in RC2.
	You can probably find it in old RC1 code kicking aronud if you need it.
	protected function ip_check($IP,$Gline=false,$Channel=BOT_REPORT_CHAN) {
		global $Cache, $DB;
		if(blacklisted_ip($IP)) {
			$this->send_to($Channel, 'TOR IP Detected: '.$IP);
			if ($Gline) {
				$this->send_raw('GLINE *@'.$IP.' 90d :DNSBL Proxy');
			}
		}
		if(Tools::site_ban_ip($IP)) {
			$this->send_to($Channel, 'Site IP Ban Detected: '.$IP);
			if ($Gline) {
				$this->send_raw('GLINE *@'.$IP.' 90d :IP Ban');
			}
		}
	}*/

	protected function listen() {
		global $Cache,$DB;
		$Cache->InternalCache = false;
		stream_set_timeout($this->Socket, 10000000000);
		while($this->State == 1){
			if($this->Data = fgets($this->Socket, 256)) {
				//IP checks
				//if(preg_match('/:\*\*\* (?:REMOTE)?CONNECT: Client connecting (?:.*) \[(\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3})\] \[(.+)\]/', $this->Data, $IP)) {
				//	$this->ip_check($IP[1],true);
				//}

				if($this->Debug === true) {
					$this->send_to(BOT_DEBUG_CHAN, $this->Data);
				}

				if($this->Whois !== false) {
					$Exp = explode(' ',$this->Data);
					if($Exp[1] == '307') {
						$this->Identified[$this->Whois] = 1;
						$this->send_to($this->LastChan, "$this->Whois correctly identified as a real person!");
						$this->Whois = false;
						$this->LastChan = false;
					} elseif($Exp[6] == '/WHOIS') {
						$this->Whois = false;
					}
				}

				if(preg_match("/:([^!]+)![^\s]* QUIT.* /", $this->Data, $Nick)) {
					if(isset($this->Identified[$Nick[1]])) {
						unset($this->Identified[$Nick[1]]);
					}
					if (isset($this->DisabledUsers[$Nick[1]])) {
						$DB->query("DELETE FROM disable_list WHERE Nick = '" . $Nick[1] . "'");
						unset($this->DisabledUsers[$Nick[1]]);
					}
				}

				if (preg_match("/:([^!]+)![^\s]* PART #what.cd-disabled/", $this->Data, $Nick)) {
					if (isset($this->DisabledUsers[$Nick[1]])) {
						$DB->query("DELETE FROM disable_list WHERE Nick = '" . $Nick[1] . "'");
						unset($this->DisabledUsers[$Nick[1]]);
					}
				}

				if (preg_match("/:([^!]+)![^\s]* KICK #what.cd-disabled.* /", $this->Data, $Nick)) {
					if (isset($this->DisabledUsers[$Nick[1]])) {
						$Nick = explode(" ", $Nick[0]);
						$DB->query("DELETE FROM disable_list WHERE Nick = '" . $Nick[3] . "'");
						unset($this->DisabledUsers[$Nick[1]]);
					}
				}

				if(preg_match("/End of message of the day./", $this->Data)) {
					$this->connect_events();
				}

				if(preg_match('/PING :(.+)/', $this->Data, $Ping)) {
					$this->send_raw('PONG :'.$Ping[1]);
				}

				if(preg_match('/.*PRIVMSG #.*/',$this->Data)) {
					$this->channel_events();
				}

				if(preg_match("/.* PRIVMSG ".BOT_NICK." .*/",$this->Data)) {
					$this->query_events();
				}
			}

			if($this->Listened = @socket_accept($this->ListenSocket)) {
				$this->listener_events();
			}

			$DB->LinkID = false;
			$DB->Queries = array();
			usleep(5000);
		}
	}
}
?>