Prompt.php

<?php
/*
    This file is part of Erebot, a modular IRC bot written in PHP.

    Copyright © 2010 François Poirotte

    Erebot is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    Erebot is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with Erebot.  If not, see <http://www.gnu.org/licenses/>.
*/

namespace Erebot;

class Prompt implements \Erebot\Interfaces\ReceivingConnection
{
    protected $bot;

    protected $socket;

    protected $io;

    public function __construct(
        \Erebot\Interfaces\Core $bot,
        $connector = null,
        $group = null,
        $perms = 0660
    ) {
        $this->bot = $bot;

        if ($connector === null) {
            $connector = sys_get_temp_dir() .
                        DIRECTORY_SEPARATOR .
                        'Erebot.sock';
        }
        $this->socket = stream_socket_server(
            "udg://".$connector,
            $errno,
            $errstr,
            STREAM_SERVER_BIND
        );
        if (!$this->socket) {
            throw new \Exception("Could not create prompt (".$errstr.")");
        }

        // Cleanup on shutdown.
        register_shutdown_function(
            function ($socket) {
                @unlink($socket);
            },
            $connector
        );

        // Change group.
        if ($group !== null) {
            if (!@chgrp($connector, $group)) {
                throw new \Exception(
                    "Could not change group to '$group' for '$connector'"
                );
            }
        }

        // Change permissions.
        if (!chmod($connector, $perms)) {
            throw new \Exception(
                "Could not set permissions to $perms on '$connector'"
            );
        }

        // Flush any received data on the socket, because
        // any data sent before the group and permissions
        // were set may have come from an untrusted source.
        $flush = array($this->socket);
        $dummy = null;
        while (stream_select($flush, $dummy, $dummy, 0) == 1) {
            if (fread($this->socket, 8192) === false) {
                throw new \Exception("Error while flushing the socket");
            }
        }

        $this->io  = new \Erebot\LineIO(\Erebot\LineIO::EOL_ANY, $this->socket);
        $logger = \Plop\Plop::getInstance();
        $logger->info(
            $bot->gettext('Prompt started in "%(path)s"'),
            array('path' => $connector)
        );
    }

    public function __destruct()
    {
        $this->disconnect();
    }

    public function connect()
    {
        $this->bot->addConnection($this);
    }

    public function disconnect($quitMessage = null)
    {
        $this->bot->removeConnection($this);
        if ($this->socket !== null) {
            stream_socket_shutdown($this->socket, STREAM_SHUT_RDWR);
        }
        $this->socket = null;
    }

    public function isConnected()
    {
        return true;
    }

    public function getSocket()
    {
        return $this->socket;
    }

    public function getIO()
    {
        return $this->io;
    }

    public function read()
    {
        $res = $this->io->read();
        if ($res === false) {
            throw new \Erebot\ConnectionFailureException('Disconnected');
        }
        return $res;
    }

    public function process()
    {
        for ($i = $this->io->inReadQueue(); $i > 0; $i--) {
            $this->handleMessage($this->io->pop());
        }
    }

    protected function handleMessage($line)
    {
        $pos = strpos($line, ' ');
        if ($pos === false) {
            return;
        }

        $pattern    = preg_quote(substr($line, 0, $pos), '@');
        $pattern    = strtr($pattern, array('\\?' => '.?', '\\*' => '.*'));
        $line       = substr($line, $pos + 1);
        if ($line === false) {
            return;
        }

        foreach ($this->bot->getConnections() as $connection) {
            if (!($connection instanceof \Erebot\Interfaces\SendingConnection) || $connection == $this) {
                continue;
            }

            $config = $connection->getConfig(null);
            $netConfig = $config->getNetworkCfg();
            if (preg_match('@^'.$pattern.'$@Di', $netConfig->getName())) {
                $connection->getIO()->push($line);
            }
        }
    }

    public function getBot()
    {
        return $this->bot;
    }

    public function getConfig($chan)
    {
        return null;
    }
}