/quit now handled properly.

[KiwiIRC.git] / class_irc.php

<?php\r
\r
class SocketStream {\r
    var $stream;\r
    function __construct($host = null, $port = null, $blocking = true) {\r
        if(!$host && !$port)\r
            return;\r
        $scheme = 'tcp';\r
        if($host{0}=='!')\r
        {\r
                $scheme = 'ssl';\r
                $port = 6697;\r
                $host = substr($host, 1);\r
        }    \r
        $this->stream = @stream_socket_client("{$scheme}://$host:$port", $errno, $errstr, 4);\r
        if(!$this->stream)\r
        {\r
            debug("Error creating socket: $host $errstr\n");\r
            return;\r
        }\r
//        socket_set_timeout($this->stream, 30);\r
        stream_set_blocking($this->stream, $blocking);\r
    }\r
    function __destruct()\r
    {\r
        if($this->stream)\r
        {\r
            fclose($this->stream);\r
            $this->stream = null;\r
        }\r
    }\r
    function SetBlocking($blocking = true)\r
    {\r
        if(!$this->stream)\r
            return false;\r
        stream_set_blocking($this->stream, $blocking);\r
    }\r
    function SetTimeout($timeout, $ms=0)\r
    {\r
        if(!$this->stream)\r
            return false;\r
        stream_set_timeout($this->stream, $timeout, $ms);\r
    }\r
    function Eof()\r
    {\r
        if(!$this->stream)\r
            return true;\r
        $x = stream_get_meta_data($this->stream);\r
        if($x['eof'] && $x['unread_bytes']==0 && feof($this->stream))\r
            return true;\r
        return false;\r
    }\r
    function Read($length = 512, $type = PHP_BINARY_READ)\r
    {\r
        if(!$this->stream) return false;\r
        if($type == PHP_NORMAL_READ){\r
            $ret = fgets($this->stream, $length);\r
        } elseif($type == PHP_BINARY_READ) {\r
            $ret = fread($this->stream, $length);\r
        }\r
		//file_put_contents('/home/wok/public_html/kiwi/rawlog', '> '.$ret, FILE_APPEND);\r
		return $ret;\r
    }\r
    function Write($data)\r
    {\r
        global $totalup;\r
        if(!$this->stream) return false;\r
		\r
		//file_put_contents('/home/wok/public_html/kiwi/rawlog', '< '.$data, FILE_APPEND);\r
        $x = @fwrite($this->stream, $data);\r
        return $x;\r
    }\r
    function GetInfo()\r
    {\r
        if(!$this->stream)\r
            return false;\r
        return array('local'=>stream_socket_get_name($this->stream,false),'remote'=>stream_socket_get_name($this->stream, true));\r
    }\r
}\r
class SocketStreamSSL extends SocketStream\r
{\r
    function __construct($host = null, $port = null, $blocking = true)\r
    {\r
        if(!$host && !$port)\r
            return;\r
        $this->stream = @stream_socket_client("ssl://$host:$port", $errno, $errstr, 4);\r
        if(!$this->stream)\r
        {\r
            debug("Error creating socket: $host $errstr\n");\r
            return;\r
        }\r
//        socket_set_timeout($this->stream, 30);\r
        stream_set_blocking($this->stream, $blocking);\r
    }\r
}\r
class IRC\r
{\r
    var $stream;\r
    var $url, $urlinfo;\r
    var $nick;\r
    protected $msgqueue = array();\r
    protected $dccs = array();\r
    private $reconnect=0;\r
    private $nextping=0;\r
    public $modes;\r
    public $chanlist;\r
	public $connected;\r
    function __construct($url, $kiwi=null, $opts=array()){		\r
        $this->url = $url;\r
        $urlinfo = parse_url($url);\r
        $this->urlinfo = $urlinfo;\r
        if(!ereg("^ircs?$", $urlinfo['scheme']))\r
            return false;\r
        $ssl = ($urlinfo['scheme']=='ircs');\r
        $host = $urlinfo['host'];\r
        $port = isset($urlinfo['port'])?$urlinfo['port']:6667;\r
        $this->nick = $nick = isset($urlinfo['user'])?$urlinfo['user']:'kiwi_user|'.(string)rand(0,9999);\r
        //$ident = isset($urlinfo['pass'])?$urlinfo['pass']:$nick;\r
		$ident = (isset($opts['ident'])) ? "{$opts['ident']}_kiwi" : "{$nick}_kiwi";\r
        $chans = false;\r
        if(isset($urlinfo['path'])){\r
            $path = trim($urlinfo['path'], "/");\r
            $chans = explode(",", $path); //"#".str_replace(array(":", ","), array(" ", ",#"), $path);\r
        }\r
		$this->connected = false;\r
        if($ssl)\r
            $this->stream = new SocketStreamSSL($host, $port, false);\r
        else\r
            $this->stream = new SocketStream($host, $port, false);\r
        if(!$this->stream || !$this->stream->stream){\r
            return false;\r
        }\r
		\r
		// Send the login data\r
		if($kiwi != null) $this->stream->Write($kiwi."\r\n");\r
        if(isset($urlinfo['fragment'])) $this->stream->Write("PASS {$urlinfo['fragment']}\r\n");\r
		\r
        $this->stream->Write("NICK $nick\r\n");\r
        $this->stream->Write("USER $ident 0 0 :$nick\r\n");\r
        //if($chans){\r
        //    foreach($chans as $chan)\r
        //        $this->stream->Write("JOIN ".str_replace(":", " ", $chan)."\r\n");\r
        //} else {\r
			$chans = array();\r
		//}\r
        $this->chans = $chans;\r
		$this->connected = true;\r
        return true;\r
    }\r
    function __destruct(){\r
        if($this->stream){\r
            $this->stream->Write("QUIT :kiwi\r\n");    \r
        }\r
    }\r
    function Process(){\r
        if((!$this->stream || !$this->stream->stream)){\r
            if(time() > $this->reconnect) {\r
            $this->__construct($this->url);\r
            $this->reconnect = time() + 10;\r
            }\r
            usleep(50000);\r
            return;\r
        }\r
        if($this->stream->Eof()){\r
            $this->stream = null;\r
            return;\r
        }\r
        $r=array($this->stream->stream);\r
		$w = $e = array();\r
        if(($num=@stream_select($r, $w,$e,0,50000))===false){\r
            \r
        } elseif($num>0){\r
            $data=$this->stream->Read(512, PHP_NORMAL_READ);\r
		//deb($data);\r
            if(preg_match("/^(?::(?:([a-z0-9\x5B-\x60\x7B-\x7D\.\-]+)|([a-z0-9\x5B-\x60\x7B-\x7D\.\-]+)!([a-z0-9~\.\-_|]+)@([a-z0-9\.\-:]+)) )?([a-z0-9]+)(?:(?: ([^:]+))?(?: :(.+))?)$/i", $data, $regs))\r
            {\r
                    unset($prefix, $nick, $ident, $hostname, $command, $params, $trailing, $flarp);\r
                    $prefix = $regs[1];\r
                    $nick = $regs[2];\r
                    $ident = $regs[3];\r
                    $hostname = $regs[4];\r
                    $command = $regs[5];\r
                    $params = isset($regs[6]) ? $regs[6] : '';\r
                    $trailing = trim(isset($regs[7]) ? $regs[7] : '');\r
                    $flarp = compact('prefix', 'nick', 'ident', 'hostname', 'command', 'params', 'trailing');\r
                    $this->ProcessMessage($flarp);\r
            }\r
        }\r
        else\r
        {\r
            if(sizeof($this->msgqueue) && (microtime(true)>($this->lastqueuewrite+0.1)) && ($msg=array_shift($this->msgqueue)))\r
            {    $this->lastqueuewrite = microtime(true); $this->stream->Write($msg); }\r
            foreach($this->dccs as $k=>&$dcc)\r
            {\r
                if(!$dcc->Process())\r
                {\r
                    if($dcc->errorinfo)\r
                        $this->SendNotice($dcc->errorinfo['nick'], $dcc->errorinfo['text']);\r
                    unset($this->dccs[$k]);\r
                }\r
            }\r
            if(($now=microtime(1))>$this->nextping)\r
            {\r
                $this->stream->Write("PONG x\r\n");\r
                $this->nextping = $now + 150;\r
            }\r
        }\r
    }\r
    function ProcessMessage($message)\r
    {\r
        //echo $message['command']."\n";\r
        switch($message['command'])\r
        {\r
            case "PING":\r
                $this->stream->Write("PONG {$message['trailing']}\r\n");\r
            break;\r
            case "PRIVMSG":\r
                $this->ProcessPrivMsg($message);\r
            break;\r
            case "001":\r
            if($this->chans)\r
                foreach($this->chans as $chan)\r
                    $this->stream->Write("JOIN ".str_replace(":", " ", $chan)."\r\n");\r
            break;\r
            case "443":\r
                $newnick = 'kiwi_user|'.rand(0,99999);\r
                $this->SendCmd("NICK ".$newnick);\r
                $this->nick = $newnick;\r
            break;\r
            case "MODE":\r
                $this->OnMODE($message);\r
            break;\r
            case "353":\r
                $this->On353($message);\r
				break;\r
			case "QUIT":\r
				$chan = $message['params'];\r
				foreach($this->chanlist as &$chan){\r
					if(isset($chan['userlist'][$message['nick']]))\r
						unset($chan['userlist'][$message['nick']]);\r
				}\r
				break;\r
			case "PART":\r
				$chan = $message['params'];\r
				//debug("Parting {$message['nick']} from $chan");\r
				if(isset($this->chanlist[ltrim($chan, "#")])){\r
					unset($this->chanlist[ltrim($chan, "#")]['userlist'][$message['nick']]);\r
				}\r
				break;\r
			case "JOIN":\r
				$chan = $message['trailing'];\r
				$nick = array($message['nick'] => '');\r
				$nicklist = is_array($this->chanlist[ltrim($chan, "#")]['userlist']) ? $this->chanlist[ltrim($chan, "#")]['userlist'] :\r
																						array();\r
				\r
				$this->chanlist[ltrim($chan, "#")]['userlist'] = array_merge($nicklist, $nick);\r
				break;\r
			case "NICK":\r
				if($message['nick'] == $this->nick){\r
					$this->nick = $message['trailing'];\r
				}\r
				foreach($this->chanlist as $chan_name => $chan){\r
					foreach($chan['userlist'] as $nick => $status){\r
						if($nick == $message['nick']){\r
							$this->chanlist[$chan_name]['userlist'][$message['trailing']] = $this->chanlist[$chan_name]['userlist'][$message['nick']];\r
							unset($this->chanlist[$chan_name]['userlist'][$message['nick']]);\r
							break;\r
						}\r
					}\r
				}\r
				break;\r
        }\r
    }\r
    function OnMODE($message){\r
		if($message['params'] == $this->nick) {\r
			$modes = $message['trailing'];\r
			$size = strlen($modes);\r
			$add = 1;\r
			for($i=0;$i<$size;$i++){\r
				if($modes{$i} == '+'){\r
					$add = 1;\r
					continue;\r
				} elseif($modes{$i} == '-'){\r
					$add = 0;\r
					continue;\r
				}\r
				if($add && strpos($this->modes, $modes{$i}) === false)\r
					$this->modes.=$modes{$i};\r
				if(!$add && strpos($this->modes, $modes{$i}))\r
					$this->modes = str_replace($modes{$i}, "", $this->modes);\r
			}\r
		}\r
		$params = $message['params'];\r
		$chan = trim(strtok($params, ' '));\r
		if(in_array(trim($chan,'#'), $this->chans)){\r
			$modes = strtok(' ');\r
			$therest = explode(" ", trim(strtok('')));\r
			$size = strlen($modes);\r
			$add = 1;\r
			$offset = 0;\r
			for($i=0;$i<$size;$i++){\r
				if($modes{$i} == '+'){\r
					$add = 1;\r
					$offset--;\r
					continue;\r
				} elseif($modes{$i} == '-'){\r
					$add = 0;\r
					$offset--;\r
					continue;\r
				}\r
					if($modes[$i]=='v')\r
					{\r
						$user = $therest[$i+$offset];\r
						if($add)\r
						{\r
							if(stripos($this->chanlist[trim($chan,'#')]['userlist'][$user], '+')===false)\r
								$this->chanlist[trim($chan,'#')]['userlist'][$user] .= '+';\r
						}\r
						else\r
							$this->chanlist[trim($chan,'#')]['userlist'][$user] = str_replace('+','',$this->chanlist[trim($chan,'#')]['userlist'][$user]);\r
						continue;\r
					}\r
					if($modes[$i]=='o')\r
					{\r
						$user = $therest[$i+$offset];\r
						if($add)\r
						{\r
							if(stripos($this->chanlist[trim($chan,'#')]['userlist'][$user], '@')===false)\r
								$this->chanlist[trim($chan,'#')]['userlist'][$user] .= '@';\r
						}\r
						else\r
							$this->chanlist[trim($chan,'#')]['userlist'][$user] = str_replace('@','',$this->chanlist[trim($chan,'#')]['userlist'][$user]);\r
						continue;\r
					}\r
					if($modes[$i]=='h')\r
					{\r
						$user = $therest[$i+$offset];\r
						if($add)\r
						{\r
							if(stripos($this->chanlist[trim($chan,'#')]['userlist'][$user], '%')===false)\r
								$this->chanlist[trim($chan,'#')]['userlist'][$user] .= '%';\r
						}\r
						else\r
							$this->chanlist[trim($chan,'#')]['userlist'][$user] = str_replace('%','',$this->chanlist[trim($chan,'#')]['userlist'][$user]);\r
						continue;\r
					}\r
					if($modes[$i]=='q')\r
					{\r
						$user = $therest[$i+$offset];\r
						if($add)\r
						{\r
							if(stripos($this->chanlist[trim($chan,'#')]['userlist'][$user], '~')===false)\r
								$this->chanlist[trim($chan,'#')]['userlist'][$user] .= '~';\r
						}\r
						else\r
							$this->chanlist[trim($chan,'#')]['userlist'][$user] = str_replace('~','',$this->chanlist[trim($chan,'#')]['userlist'][$user]);\r
						continue;\r
					}\r
			}\r
		}\r
		\r
	}\r
	function On353($message){\r
		// Response to NAMES command\r
		list($nick,,$chan) = explode(" ", $message['params']);\r
		$nicks = explode(" ", $message['trailing']);\r
		$prefixes = array('~', '&', '@','%','+');\r
		$nicklist = array();\r
		foreach($nicks as $nick){\r
			if(in_array($nick{0}, $prefixes)){\r
				$prefix = $nick{0};\r
				$nick = substr($nick,1);\r
			}\r
			else\r
				$prefix = '';\r
			$nicklist[$nick] = $prefix;\r
		}\r
		if(sizeof($nicklist)){\r
			// If we havn't got a list of nicks for this channel yet, create it\r
			if(!isset($this->chanlist[ltrim($chan, "#")]['userlist'])){\r
				$this->chanlist[ltrim($chan, "#")]['userlist'] = array();\r
			}\r
			$this->chanlist[ltrim($chan, "#")]['userlist'] = array_merge($this->chanlist[ltrim($chan, "#")]['userlist'], $nicklist);\r
		}\r
	}\r
    function ProcessPrivMsg($message){\r
                //$cmd = strtok($message['trailing'], ' ');\r
                //switch($cmd){\r
                //}\r
    }\r
    function SendMessage($dest, $text){\r
        if(!$this->stream)\r
            return false;\r
        $this->stream->Write("PRIVMSG $dest :$text\r\n");\r
    }\r
	function Join($chan){\r
		$this->stream->Write("JOIN ".str_replace(":", " ", $chan)."\r\n");\r
	}\r
    function SendNotice($dest, $text)\r
    {\r
        $this->stream->Write("NOTICE $dest :$text\r\n");\r
    }\r
    function QueueMessage($dest, $text)\r
    {\r
        $this->msgqueue[] = "PRIVMSG $dest :$text\r\n";\r
    }\r
    function QueueNotice($dest, $text)\r
    {\r
        $this->msgqueue[] = "NOTICE $dest :$text\r\n";        \r
    }\r
    function GetMainChan()\r
    {\r
        return '#'.$this->chans[0];    \r
    }\r
}