9 * This source file is subject to the new BSD license that is bundled
10 * with this package in the file LICENSE.
11 * It is also available through the world-wide-web at this URL:
12 * http://phergie.org/license
16 * @author Phergie Development Team <team@phergie.org>
17 * @copyright 2008-2010 Phergie Development Team (http://phergie.org)
18 * @license http://phergie.org/license New BSD License
19 * @link http://pear.phergie.org/package/Phergie
23 * Driver that uses the sockets wrapper of the streams extension for
24 * communicating with the server and handles formatting and parsing of
29 * @author Phergie Development Team <team@phergie.org>
30 * @license http://phergie.org/license New BSD License
31 * @link http://pear.phergie.org/package/Phergie
33 class Phergie_Driver_Streams extends Phergie_Driver_Abstract
40 protected $sockets = array();
43 * Reference to the currently active socket handler
50 * Amount of time in seconds to wait to receive an event each time the
55 protected $timeout = 0.1;
58 * Handles construction of command strings and their transmission to the
61 * @param string $command Command to send
62 * @param string|array $args Optional string or array of sequential
65 * @return string Command string that was sent
66 * @throws Phergie_Driver_Exception
68 protected function send($command, $args = '')
70 $connection = $this->getConnection();
71 $encoding = $connection->getEncoding();
73 // Require an open socket connection to continue
74 if (empty($this->socket)) {
75 throw new Phergie_Driver_Exception(
76 'doConnect() must be called first',
77 Phergie_Driver_Exception::ERR_NO_INITIATED_CONNECTION
82 $buffer = strtoupper($command);
87 // Apply formatting if arguments are passed in as an array
88 if (is_array($args)) {
89 $end = count($args) - 1;
90 $args[$end] = ':' . $args[$end];
91 $args = implode(' ', $args);
96 $buffer .= ' ' . $args;
99 // Transmit the command over the socket connection
100 $attempts = $written = 0;
101 $temp = $buffer . "\r\n";
102 $is_multibyte = !substr($encoding, 0, 8) === 'ISO-8859' && $encoding !== 'ASCII' && $encoding !== 'CP1252';
103 $length = ($is_multibyte) ? mb_strlen($buffer, '8bit') : strlen($buffer);
105 $written += (int) fwrite($this->socket, $temp);
106 if ($written < $length) {
107 $temp = substr($temp, $written);
109 if ($attempts == 3) {
110 throw new Phergie_Driver_Exception(
111 'Unable to write to socket',
112 Phergie_Driver_Exception::ERR_CONNECTION_WRITE_FAILED
120 // Return the command string that was transmitted
125 * Overrides the parent class to set the currently active socket handler
126 * when the active connection is changed.
128 * @param Phergie_Connection $connection Active connection
130 * @return Phergie_Driver_Streams Provides a fluent interface
132 public function setConnection(Phergie_Connection $connection)
134 // Set the active socket handler
135 $hostmask = (string) $connection->getHostmask();
136 if (!empty($this->sockets[$hostmask])) {
137 $this->socket = $this->sockets[$hostmask];
140 // Set the active connection
141 return parent::setConnection($connection);
145 * Returns a list of hostmasks corresponding to sockets with data to read.
147 * @param int $sec Length of time to wait for new data (seconds)
148 * @param int $usec Length of time to wait for new data (microseconds)
150 * @return array List of hostmasks or an empty array if none were found
151 * to have data to read
153 public function getActiveReadSockets($sec = 0, $usec = 200000)
155 $read = $this->sockets;
160 if (count($this->sockets) > 0) {
161 $number = stream_select($read, $write, $error, $sec, $usec);
163 foreach ($read as $item) {
164 $active[] = array_search($item, $this->sockets);
173 * Sets the amount of time to wait for a new event each time the socket
176 * @param float $timeout Amount of time in seconds
178 * @return Phergie_Driver_Streams Provides a fluent interface
180 public function setTimeout($timeout)
182 $timeout = (float) $timeout;
184 $this->timeout = $timeout;
190 * Returns the amount of time to wait for a new event each time the
193 * @return float Amount of time in seconds
195 public function getTimeout()
197 return $this->timeout;
201 * Supporting method to parse event argument strings where the last
202 * argument may contain a colon.
204 * @param string $args Argument string to parse
205 * @param int $count Optional maximum number of arguments
207 * @return array Array of argument values
209 protected function parseArguments($args, $count = -1)
211 return preg_split('/ :?/S', $args, $count);
215 * Listens for an event on the current connection.
217 * @return Phergie_Event_Interface|null Event instance if an event was
218 * received, NULL otherwise
220 public function getEvent()
222 // Check the socket is still active
223 if (feof($this->socket)) {
224 throw new Phergie_Driver_Exception(
225 'EOF detected on socket',
226 Phergie_Driver_Exception::ERR_CONNECTION_READ_FAILED
230 // Check for a new event on the current connection
231 $buffer = fgets($this->socket, 512);
233 // If no new event was found, return NULL
234 if (empty($buffer)) {
238 // Strip the trailing newline from the buffer
239 $buffer = rtrim($buffer);
241 // If the event is from the server...
242 if (substr($buffer, 0, 1) != ':') {
244 // Parse the command and arguments
245 list($cmd, $args) = array_pad(explode(' ', $buffer, 2), 2, null);
246 $hostmask = new Phergie_Hostmask(null, null, $this->connection->getHost());
249 // If the event could be from the server or a user...
251 // Parse the server hostname or user hostmask, command, and arguments
252 list($prefix, $cmd, $args)
253 = array_pad(explode(' ', ltrim($buffer, ':'), 3), 3, null);
254 if (strpos($prefix, '@') !== false) {
255 $hostmask = Phergie_Hostmask::fromString($prefix);
257 $hostmask = new Phergie_Hostmask(null, null, $prefix);
261 // Parse the event arguments depending on the event type
262 $cmd = strtolower($cmd);
270 $args = array(ltrim($args, ':'));
275 $args = $this->parseArguments($args, 2);
276 list($source, $ctcp) = $args;
277 if (substr($ctcp, 0, 1) === "\001" && substr($ctcp, -1) === "\001") {
278 $ctcp = substr($ctcp, 1, -1);
279 $reply = ($cmd == 'notice');
280 list($cmd, $args) = array_pad(explode(' ', $ctcp, 2), 2, null);
281 $cmd = strtolower($cmd);
298 $args = array($source, $args);
306 $args = array($source, $args);
315 $args = $this->parseArguments($args);
321 $args = $this->parseArguments($args, 2);
325 $args = $this->parseArguments($args, 3);
328 // Remove the target from responses
330 $args = substr($args, strpos($args, ' ') + 1);
334 // Create, populate, and return an event object
335 if (ctype_digit($cmd)) {
336 $event = new Phergie_Event_Response;
339 ->setDescription($args);
341 $event = new Phergie_Event_Request;
344 ->setArguments($args);
345 if (isset($hostmask)) {
346 $event->setHostmask($hostmask);
349 $event->setRawData($buffer);
354 * Initiates a connection with the server.
358 public function doConnect()
360 // Listen for input indefinitely
363 // Get connection information
364 $connection = $this->getConnection();
365 $hostname = $connection->getHost();
366 $port = $connection->getPort();
367 $password = $connection->getPassword();
368 $username = $connection->getUsername();
369 $nick = $connection->getNick();
370 $realname = $connection->getRealname();
371 $transport = $connection->getTransport();
373 // Establish and configure the socket connection
374 $remote = $transport . '://' . $hostname . ':' . $port;
375 $this->socket = @stream_socket_client($remote, $errno, $errstr);
376 if (!$this->socket) {
377 throw new Phergie_Driver_Exception(
378 'Unable to connect: socket error ' . $errno . ' ' . $errstr,
379 Phergie_Driver_Exception::ERR_CONNECTION_ATTEMPT_FAILED
383 $seconds = (int) $this->timeout;
384 $microseconds = ($this->timeout - $seconds) * 1000000;
385 stream_set_timeout($this->socket, $seconds, $microseconds);
387 // Send the password if one is specified
388 if (!empty($password)) {
389 $this->send('PASS', $password);
392 // Send user information
403 $this->send('NICK', $nick);
405 // Add the socket handler to the internal array for socket handlers
406 $this->sockets[(string) $connection->getHostmask()] = $this->socket;
410 * Terminates the connection with the server.
412 * @param string $reason Reason for connection termination (optional)
416 public function doQuit($reason = null)
418 // Send a QUIT command to the server
419 $this->send('QUIT', $reason);
421 // Terminate the socket connection
422 fclose($this->socket);
424 // Remove the socket from the internal socket list
425 unset($this->sockets[(string) $this->getConnection()->getHostmask()]);
431 * @param string $channels Comma-delimited list of channels to join
432 * @param string $keys Optional comma-delimited list of channel keys
436 public function doJoin($channels, $keys = null)
438 $args = array($channels);
444 $this->send('JOIN', $args);
450 * @param string $channels Comma-delimited list of channels to leave
454 public function doPart($channels)
456 $this->send('PART', $channels);
460 * Invites a user to an invite-only channel.
462 * @param string $nick Nick of the user to invite
463 * @param string $channel Name of the channel
467 public function doInvite($nick, $channel)
469 $this->send('INVITE', array($nick, $channel));
473 * Obtains a list of nicks of usrs in currently joined channels.
475 * @param string $channels Comma-delimited list of one or more channels
479 public function doNames($channels)
481 $this->send('NAMES', $channels);
485 * Obtains a list of channel names and topics.
487 * @param string $channels Comma-delimited list of one or more channels
488 * to which the response should be restricted
493 public function doList($channels = null)
495 $this->send('LIST', $channels);
499 * Retrieves or changes a channel topic.
501 * @param string $channel Name of the channel
502 * @param string $topic New topic to assign (optional)
506 public function doTopic($channel, $topic = null)
508 $args = array($channel);
510 if (!empty($topic)) {
514 $this->send('TOPIC', $args);
518 * Retrieves or changes a channel or user mode.
520 * @param string $target Channel name or user nick
521 * @param string $mode New mode to assign (optional)
525 public function doMode($target, $mode = null)
527 $args = array($target);
533 $this->send('MODE', $args);
537 * Changes the client nick.
539 * @param string $nick New nick to assign
543 public function doNick($nick)
545 $this->send('NICK', $nick);
549 * Retrieves information about a nick.
551 * @param string $nick Nick
555 public function doWhois($nick)
557 $this->send('WHOIS', $nick);
561 * Sends a message to a nick or channel.
563 * @param string $target Channel name or user nick
564 * @param string $text Text of the message to send
568 public function doPrivmsg($target, $text)
570 $this->send('PRIVMSG', array($target, $text));
574 * Sends a notice to a nick or channel.
576 * @param string $target Channel name or user nick
577 * @param string $text Text of the notice to send
581 public function doNotice($target, $text)
583 $this->send('NOTICE', array($target, $text));
587 * Kicks a user from a channel.
589 * @param string $nick Nick of the user
590 * @param string $channel Channel name
591 * @param string $reason Reason for the kick (optional)
595 public function doKick($nick, $channel, $reason = null)
597 $args = array($nick, $channel);
599 if (!empty($reason)) {
603 $this->send('KICK', $args);
607 * Responds to a server test of client responsiveness.
609 * @param string $daemon Daemon from which the original request originates
613 public function doPong($daemon)
615 $this->send('PONG', $daemon);
619 * Sends a CTCP ACTION (/me) command to a nick or channel.
621 * @param string $target Channel name or user nick
622 * @param string $text Text of the action to perform
626 public function doAction($target, $text)
628 $buffer = rtrim('ACTION ' . $text);
630 $this->doPrivmsg($target, chr(1) . $buffer . chr(1));
634 * Sends a CTCP response to a user.
636 * @param string $nick User nick
637 * @param string $command Command to send
638 * @param string|array $args String or array of sequential arguments
643 protected function doCtcp($nick, $command, $args = null)
645 if (is_array($args)) {
646 $args = implode(' ', $args);
649 $buffer = rtrim(strtoupper($command) . ' ' . $args);
651 $this->doNotice($nick, chr(1) . $buffer . chr(1));
655 * Sends a CTCP PING request or response (they are identical) to a user.
657 * @param string $nick User nick
658 * @param string $hash Hash to use in the handshake
662 public function doPing($nick, $hash)
664 $this->doCtcp($nick, 'PING', $hash);
668 * Sends a CTCP VERSION request or response to a user.
670 * @param string $nick User nick
671 * @param string $version Version string to send for a response
675 public function doVersion($nick, $version = null)
678 $this->doCtcp($nick, 'VERSION', $version);
680 $this->doCtcp($nick, 'VERSION');
685 * Sends a CTCP TIME request to a user.
687 * @param string $nick User nick
688 * @param string $time Time string to send for a response
692 public function doTime($nick, $time = null)
695 $this->doCtcp($nick, 'TIME', $time);
697 $this->doCtcp($nick, 'TIME');
702 * Sends a CTCP FINGER request to a user.
704 * @param string $nick User nick
705 * @param string $finger Finger string to send for a response
709 public function doFinger($nick, $finger = null)
712 $this->doCtcp($nick, 'FINGER', $finger);
714 $this->doCtcp($nick, 'FINGER');
719 * Sends a raw command to the server.
721 * @param string $command Command string to send
725 public function doRaw($command)
727 $this->send('RAW', $command);