#!/usr/bin/env php <?php /* * StatusNet - the distributed open-source microblogging tool * Copyright (C) 2008, 2009, StatusNet, Inc. * * This program is free software: you can redistribute it and/or modify * it under the terms of the GNU Affero General Public License as published by * the Free Software Foundation, either version 3 of the License, or * (at your option) any later version. * * This program 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 Affero General Public License for more details. * * You should have received a copy of the GNU Affero General Public License * along with this program. If not, see <http://www.gnu.org/licenses/>. */ define('INSTALLDIR', realpath(dirname(__FILE__) . '/..')); $shortoptions = 'fi::a'; $longoptions = array('id::', 'foreground', 'all'); $helptext = <<<END_OF_IM_HELP Daemon script for receiving new notices from IM users. -i --id Identity (default none) -a --all Handle XMPP for all local sites (requires Stomp queue handler, status_network setup) -f --foreground Stay in the foreground (default background) END_OF_IM_HELP; require_once INSTALLDIR.'/scripts/commandline.inc'; class ImDaemon extends SpawningDaemon { protected $allsites = false; function __construct($id=null, $daemonize=true, $threads=1, $allsites=false) { if ($threads != 1) { // This should never happen. :) throw new Exception("IMDaemon can must run single-threaded"); } parent::__construct($id, $daemonize, $threads); $this->allsites = $allsites; } function runThread() { common_log(LOG_INFO, 'Waiting to listen to IM connections and queues'); $master = new ImMaster($this->get_id(), $this->processManager()); $master->init($this->allsites); $master->service(); common_log(LOG_INFO, 'terminating normally'); return $master->respawn ? self::EXIT_RESTART : self::EXIT_SHUTDOWN; } } class ImMaster extends IoMaster { protected $processManager; function __construct($id, $processManager) { parent::__construct($id); $this->processManager = $processManager; } /** * Initialize IoManagers for the currently configured site * which are appropriate to this instance. */ function initManagers() { $classes = array(); if (Event::handle('StartImDaemonIoManagers', array(&$classes))) { $qm = QueueManager::get(); $qm->setActiveGroup('im'); $classes[] = $qm; $classes[] = $this->processManager; } Event::handle('EndImDaemonIoManagers', array(&$classes)); foreach ($classes as $class) { $this->instantiate($class); } } } if (have_option('i', 'id')) { $id = get_option_value('i', 'id'); } else if (count($args) > 0) { $id = $args[0]; } else { $id = null; } $foreground = have_option('f', 'foreground'); $all = have_option('a') || have_option('--all'); $daemon = new ImDaemon($id, !$foreground, 1, $all); $daemon->runOnce();